scripts/queuectl.php

   1 #!/usr/bin/env php
   2 <?php
   3 /*
   4  * StatusNet - the distributed open-source microblogging tool
   5  * Copyright (C) 2010, StatusNet, Inc.
   6  *
   7  * This program is free software: you can redistribute it and/or modify
   8  * it under the terms of the GNU Affero General Public License as published by
   9  * the Free Software Foundation, either version 3 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15  * GNU Affero General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU Affero General Public License
  18  * along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19  */
  20
  21 /**
  22  * Sends control signals to running queue daemons.
  23  *
  24  * @author Brion Vibber <brion@status.net>
  25  * @package QueueHandler
  26  */
  27
  28 define('INSTALLDIR', realpath(dirname(__FILE__) . '/..'));
  29
  30 $shortoptions = 'ur';
  31 $longoptions = array('update', 'restart', 'stop');
  32
  33 $helptext = <<<END_OF_QUEUECTL_HELP
  34 Send broadcast events to control any running queue handlers.
  35 (Currently for Stomp queues only.)
  36
  37 Events relating to current site (as selected with -s etc)
  38     -u --update       Announce new site or updated configuration. Running
  39                       daemons will start subscribing to any new queues needed
  40                       for this site.
  41
  42 Global events:
  43     -r --restart      Graceful restart of all threads
  44        --stop         Graceful shutdown of all threads
  45
  46 END_OF_QUEUECTL_HELP;
  47
  48 require_once INSTALLDIR.'/scripts/commandline.inc';
  49
  50 function doSendControl($message, $event, $param='')
  51 {
  52     print $message;
  53     $qm = QueueManager::get();
  54     if ($qm->sendControlSignal($event, $param)) {
  55         print " sent.\n";
  56     } else {
  57         print " FAILED.\n";
  58     }
  59 }
  60
  61 $actions = 0;
  62
  63 if (have_option('u') || have_option('--update')) {
  64     $nickname = common_config('site', 'nickname');
  65     doSendControl("Sending site update signal to queue daemons for $nickname",
  66                   "update", $nickname);
  67     $actions++;
  68 }
  69
  70 if (have_option('r') || have_option('--restart')) {
  71     doSendControl("Sending graceful restart signal to queue daemons...",
  72                   "restart");
  73     $actions++;
  74 }
  75
  76 if (have_option('--stop')) {
  77     doSendControl("Sending graceful shutdown signal to queue daemons...",
  78                   "shutdown");
  79     $actions++;
  80 }
  81
  82 if (!$actions) {
  83     show_help();
  84 }
  85