bin/opensrf-perl.pl

   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 = undef;
  31 my $opt_pid_dir = '/tmp';
  32 my $opt_no_daemon = 0;
  33 my $opt_help = 0;
  34 my $sclient;
  35 my $hostname = hostfqdn();
  36 my @hosted_services;
  37
  38 GetOptions(
  39     'action=s' => \$opt_action,
  40     'service=s' => \$opt_service,
  41     'config=s' => \$opt_config,
  42     'pid_dir=s' => \$opt_pid_dir,
  43     'no_daemon' => \$opt_no_daemon,
  44     'help' => \$opt_help,
  45 );
  46
  47
  48 sub haltme {
  49     kill('INT', -$$); #kill all in process group
  50     exit;
  51 };
  52 $SIG{INT} = \&haltme;
  53 $SIG{TERM} = \&haltme;
  54
  55 sub get_pid_file {
  56     my $service = shift;
  57     return "$opt_pid_dir/$service.pid";
  58 }
  59
  60 # stop a specific service
  61 sub do_stop {
  62     my $service = shift;
  63     my $pid_file = get_pid_file($service);
  64     if(-e $pid_file) {
  65         my $pid = `cat $pid_file`;
  66         kill('INT', $pid);
  67         waitpid($pid, 0);
  68         unlink $pid_file;
  69     } else {
  70         msg("$service not running");
  71     }
  72 }
  73
  74 sub do_init {
  75     OpenSRF::System->bootstrap_client(config_file => $opt_config);
  76     die "Unable to bootstrap client for requests\n"
  77         unless OpenSRF::Transport::PeerHandle->retrieve;
  78
  79     load_settings(); # load the settings config if we can
  80
  81     my $sclient = OpenSRF::Utils::SettingsClient->new;
  82     my $apps = $sclient->config_value("activeapps", "appname");
  83
  84     # disconnect the top-level network handle
  85     OpenSRF::Transport::PeerHandle->retrieve->disconnect;
  86
  87     if($apps) {
  88         $apps = [$apps] unless ref $apps;
  89         for my $app (@$apps) {
  90             if($app eq $service) {
  91                 push(@hosted_services, $app)
  92                     if $sclient->config_value('apps', $app, 'language') =~ /perl/i);
  93             }
  94         }
  95     }
  96 }
  97
  98 # start a specific service
  99 sub do_start {
 100     my $service = shift;
 101     load_settings() if $service eq 'opensrf.settings';
 102
 103     my $sclient = OpenSRF::Utils::SettingsClient->new;
 104     my $apps = $sclient->config_value("activeapps", "appname");
 105     OpenSRF::Transport::PeerHandle->retrieve->disconnect;
 106
 107     if(grep { $_ eq $service } @hosted_services) {
 108         do_daemon($service) unless $opt_no_daemon;
 109         launch_net_server($service);
 110         launch_listener($service);
 111         $0 = "OpenSRF controller [$service]";
 112         while(my $pid = waitpid(-1, 0)) {
 113             $logger->debug("Cleaning up Perl $service process $pid");
 114         }
 115     }
 116
 117     msg("$service is not configured to run on $hostname");
 118 }
 119
 120 sub do_start_all {
 121     do_start('opensrf.settings') if grep {$_ eq 'opensrf.settings'} @hosted_services);
 122     for my $service (@hosted_services) {
 123         do_start($service) unless $service eq 'opensrf.settings';
 124     }
 125 }
 126
 127 sub do_stop_all {
 128     do_stop($_) for @hosted_services;
 129 }
 130
 131 # daemonize us
 132 sub do_daemon {
 133     my $service = shift;
 134     my $pid_file = get_pid_file($service);
 135     exit if OpenSRF::Utils::safe_fork();
 136     chdir('/');
 137     setsid();
 138     close STDIN;
 139     close STDOUT;
 140     close STDERR;
 141     `echo $$ > $pid_file`;
 142 }
 143
 144 # parses the local settings file
 145 sub load_settings {
 146     my $conf = OpenSRF::Utils::Config->current;
 147     my $cfile = $conf->bootstrap->settings_config;
 148     return unless $cfile;
 149     my $parser = OpenSRF::Utils::SettingsParser->new();
 150     $parser->initialize( $cfile );
 151     $OpenSRF::Utils::SettingsClient::host_config =
 152         $parser->get_server_config($conf->env->hostname);
 153 }
 154
 155 # starts up the unix::server master process
 156 sub launch_net_server {
 157     my $service = shift;
 158     push @OpenSRF::UnixServer::ISA, 'Net::Server::PreFork';
 159     unless(OpenSRF::Utils::safe_fork()) {
 160         $0 = "OpenSRF Drone [$service]";
 161         OpenSRF::UnixServer->new($service)->serve;
 162         exit;
 163     }
 164     return 1;
 165 }
 166
 167 # starts up the inbound listener process
 168 sub launch_listener {
 169     my $service = shift;
 170     unless(OpenSRF::Utils::safe_fork()) {
 171         $0 = "OpenSRF listener [$service]";
 172         OpenSRF::Transport::Listener->new($service)->initialize->listen;
 173         exit;
 174     }
 175     return 1;
 176 }
 177
 178 sub msg {
 179     my $m = shift;
 180     print "* $m\n";
 181 }
 182
 183 sub do_help {
 184     print <<HELP;
 185
 186     Usage: perl $0 --pid_dir /var/run/opensrf --config /etc/opensrf/opensrf_core.xml --service opensrf.settings --action start
 187
 188     --action <action>
 189         Actions include start, stop, restart, and start_all, stop_all, and restart_all
 190
 191     --service <service>
 192         Specifies which OpenSRF service to control
 193
 194     --config <file>
 195         OpenSRF configuration file
 196
 197     --pid_dir <dir>
 198         Directory where process-specific PID files are kept
 199
 200     --no_daemon
 201         Do not detach and run as a daemon process.  Useful for debugging.
 202
 203     --help
 204         Print this help message
 205 HELP
 206 exit;
 207 }
 208
 209
 210 do_help() if $opt_help or not $opt_action;
 211 do_init() and do_start($opt_service) if $opt_action eq 'start';
 212 do_stop($opt_service) if $opt_action eq 'stop';
 213 do_init() and do_stop($opt_service) and do_start($opt_service) if $opt_action eq 'restart';
 214 do_init() and do_start_all() if $opt_action eq 'start_all';
 215 do_stop_all() if $opt_action eq 'stop_all';
 216 do_init() and do_stop_all() and do_start_all() if $opt_action eq 'restart_all';
 217
 218