bin/opensrf-perl.pl.in

   1 #!/usr/bin/perl
   2 # ---------------------------------------------------------------
   3 # Copyright (C) 2008  Georgia Public Library Service
   4 # Bill Erickson <erickson@esilibrary.com>
   5 #
   6 # This program is free software; you can redistribute it and/or
   7 # modify it under the terms of the GNU General Public License
   8 # as published by the Free Software Foundation; either version 2
   9 # of the License, or (at your option) any later version.
  10 #
  11 # This program is distributed in the hope that it will be useful,
  12 # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14 # GNU General Public License for more details.
  15 # ---------------------------------------------------------------
  16 use strict; use warnings;
  17 use Getopt::Long;
  18 use Net::Domain qw/hostfqdn/;
  19 use POSIX qw/setsid :sys_wait_h/;
  20 use OpenSRF::Utils::Logger q/$logger/;
  21 use OpenSRF::System;
  22 use OpenSRF::Transport::PeerHandle;
  23 use OpenSRF::Utils::SettingsClient;
  24 use OpenSRF::Transport::Listener;
  25 use OpenSRF::Utils;
  26 use OpenSRF::Utils::Config;
  27
  28 my $opt_action = undef;
  29 my $opt_service = undef;
  30 my $opt_config = "@CONF_DIR@/opensrf_core.xml";
  31 my $opt_pid_dir = "@PID_DIR@/run/opensrf";
  32 my $opt_no_daemon = 0;
  33 my $opt_settings_pause = 0;
  34 my $opt_localhost = 0;
  35 my $opt_help = 0;
  36 my $verbose = 0;
  37 my $sclient;
  38 my $hostname = $ENV{OSRF_HOSTNAME} || hostfqdn();
  39 my @hosted_services;
  40
  41 GetOptions(
  42     'action=s' => \$opt_action,
  43     'service=s' => \$opt_service,
  44     'config=s' => \$opt_config,
  45     'pid-dir=s' => \$opt_pid_dir,
  46     'no-daemon' => \$opt_no_daemon,
  47     'settings-startup-pause=i' => \$opt_settings_pause,
  48     'localhost' => \$opt_localhost,
  49     'help' => \$opt_help,
  50     'verbose' => \$verbose,
  51 );
  52
  53 if ($opt_localhost) {
  54     $hostname = 'localhost';
  55     $ENV{OSRF_HOSTNAME} = $hostname;
  56 }
  57
  58 sub haltme {
  59     kill('INT', -$$); #kill all in process group
  60     exit;
  61 };
  62 $SIG{INT} = \&haltme;
  63 $SIG{TERM} = \&haltme;
  64
  65 sub get_pid_file {
  66     my $service = shift;
  67     return "$opt_pid_dir/$service.pid";
  68 }
  69
  70 # stop a specific service
  71 sub do_stop {
  72     my $service = shift;
  73     my $pid_file = get_pid_file($service);
  74     if(-e $pid_file) {
  75         my $pid = `cat $pid_file`;
  76         chomp $pid;
  77         msg("stopping service pid=$pid $service", 1);
  78         kill('INT', $pid);
  79         waitpid($pid, 0);
  80         unlink $pid_file;
  81     } else {
  82         msg("$service not running");
  83     }
  84     return 1;
  85 }
  86
  87 sub do_init {
  88     OpenSRF::System->bootstrap_client(config_file => $opt_config);
  89     die "Unable to bootstrap client for requests\n"
  90         unless OpenSRF::Transport::PeerHandle->retrieve;
  91
  92     load_settings(); # load the settings config if we can
  93
  94     my $sclient = OpenSRF::Utils::SettingsClient->new;
  95     my $apps = $sclient->config_value("activeapps", "appname");
  96
  97     # disconnect the top-level network handle
  98     OpenSRF::Transport::PeerHandle->retrieve->disconnect;
  99
 100     if($apps) {
 101         $apps = [$apps] unless ref $apps;
 102         for my $app (@$apps) {
 103             if (!$sclient->config_value('apps', $app)) {
 104                 msg("Service '$app' is listed for this host, but there is no configuration for it in $opt_config");
 105                 next;
 106             }
 107             if ($sclient->config_value('apps', $app, 'language') =~ /perl/i) {
 108                 push(@hosted_services, $app);
 109             }
 110         }
 111     }
 112     return 1;
 113 }
 114
 115 # start a specific service
 116 sub do_start {
 117     my $service = shift;
 118
 119     if(-e get_pid_file($service)) {
 120         msg("$service is already running");
 121         return;
 122     }
 123
 124     load_settings() if $service eq 'opensrf.settings';
 125
 126     if(grep { $_ eq $service } @hosted_services) {
 127         return unless do_daemon($service);
 128         OpenSRF::System->run_service($service, $opt_pid_dir);
 129     }
 130
 131     msg("$service is not configured to run on $hostname");
 132     return 1;
 133 }
 134
 135 sub do_start_all {
 136     msg("starting all services for $hostname", 1);
 137     if(grep {$_ eq 'opensrf.settings'} @hosted_services) {
 138         do_start('opensrf.settings');
 139         # in batch mode, give opensrf.settings plenty of time to start
 140         # before any non-Perl services try to connect
 141         sleep $opt_settings_pause if $opt_settings_pause;
 142     }
 143     for my $service (@hosted_services) {
 144         do_start($service) unless $service eq 'opensrf.settings';
 145     }
 146     return 1;
 147 }
 148
 149 sub do_stop_all {
 150     msg("stopping all services for $hostname", 1);
 151     do_stop($_) for @hosted_services;
 152     return 1;
 153 }
 154
 155 # daemonize us.  return true if we're the child, false if parent
 156 sub do_daemon {
 157     return 1 if $opt_no_daemon;
 158     my $service = shift;
 159     my $pid_file = get_pid_file($service);
 160     #exit if OpenSRF::Utils::safe_fork();
 161     return 0 if OpenSRF::Utils::safe_fork();
 162     msg("starting service pid=$$ $service", 1);
 163     chdir('/');
 164     setsid();
 165     close STDIN;
 166     close STDOUT;
 167     close STDERR;
 168     open STDIN, '</dev/null';
 169     open STDOUT, '>/dev/null';
 170     open STDERR, '>/dev/null';
 171     `echo $$ > $pid_file`;
 172     return 1;
 173 }
 174
 175 # parses the local settings file
 176 sub load_settings {
 177     my $conf = OpenSRF::Utils::Config->current;
 178     my $cfile = $conf->bootstrap->settings_config;
 179     return unless $cfile;
 180     my $parser = OpenSRF::Utils::SettingsParser->new();
 181     $parser->initialize( $cfile );
 182     $OpenSRF::Utils::SettingsClient::host_config =
 183         $parser->get_server_config($conf->env->hostname);
 184 }
 185
 186 sub msg {
 187     my $m = shift;
 188     my $v = shift;
 189     print "* $m\n" unless $v and not $verbose;
 190 }
 191
 192 sub do_help {
 193     print <<HELP;
 194
 195     Usage: perl $0 --pid-dir @TMP@ --config @CONF_DIR@/opensrf_core.xml --service opensrf.settings --action start
 196
 197     --action <action>
 198         Actions include start, stop, restart, and start_all, stop_all, and restart_all
 199
 200     --service <service>
 201         Specifies which OpenSRF service to control
 202
 203     --config <file>
 204         OpenSRF configuration file
 205
 206     --pid-dir <dir>
 207         Directory where process-specific PID files are kept
 208
 209     --no-daemon
 210         Do not detach and run as a daemon process.  Useful for debugging.
 211
 212     --settings-startup-pause
 213         How long to give the opensrf.settings server to start up when running
 214         in batch mode (start_all).  The purpose is to give plenty of time for
 215         the settings server to be up and active before any non-Perl services
 216         attempt to connect.
 217
 218     --localhost
 219         Force the hostname to be 'localhost', instead of the fully qualified
 220         domain name for the machine.
 221
 222     --help
 223         Print this help message
 224 HELP
 225 exit;
 226 }
 227
 228
 229 do_help() if $opt_help or not $opt_action;
 230 do_init() and do_start($opt_service) if $opt_action eq 'start';
 231 do_stop($opt_service) if $opt_action eq 'stop';
 232 do_init() and do_stop($opt_service) and do_start($opt_service) if $opt_action eq 'restart';
 233 do_init() and do_start_all() if $opt_action eq 'start_all';
 234 do_init() and do_stop_all() if $opt_action eq 'stop_all';
 235 do_init() and do_stop_all() and do_start_all() if $opt_action eq 'restart_all';
 236
 237