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', dirname(__DIR__));
  29 define('PUBLICDIR', INSTALLDIR . DIRECTORY_SEPARATOR . 'public');
  30
  31 $shortoptions = 'ur';
  32 $longoptions = array('update', 'restart', 'stop');
  33
  34 $helptext = <<<END_OF_QUEUECTL_HELP
  35 Send broadcast events to control any running queue handlers.
  36 (Currently for Stomp queues only.)
  37
  38 Events relating to current site (as selected with -s etc)
  39     -u --update       Announce new site or updated configuration. Running
  40                       daemons will start subscribing to any new queues needed
  41                       for this site.
  42
  43 Global events:
  44     -r --restart      Graceful restart of all threads
  45        --stop         Graceful shutdown of all threads
  46
  47 END_OF_QUEUECTL_HELP;
  48
  49 require_once INSTALLDIR.'/scripts/commandline.inc';
  50
  51 function doSendControl($message, $event, $param='')
  52 {
  53     print $message;
  54     $qm = QueueManager::get();
  55     if ($qm->sendControlSignal($event, $param)) {
  56         print " sent.\n";
  57     } else {
  58         print " FAILED.\n";
  59     }
  60 }
  61
  62 $actions = 0;
  63
  64 if (have_option('u') || have_option('--update')) {
  65     $nickname = common_config('site', 'nickname');
  66     doSendControl("Sending site update signal to queue daemons for $nickname",
  67                   "update", $nickname);
  68     $actions++;
  69 }
  70
  71 if (have_option('r') || have_option('--restart')) {
  72     doSendControl("Sending graceful restart signal to queue daemons...",
  73                   "restart");
  74     $actions++;
  75 }
  76
  77 if (have_option('--stop')) {
  78     doSendControl("Sending graceful shutdown signal to queue daemons...",
  79                   "shutdown");
  80     $actions++;
  81 }
  82
  83 if (!$actions) {
  84     show_help();
  85 }
  86