2011-03-13 22:27:03 +01:00
|
|
|
#!/usr/bin/env php
|
|
|
|
<?php
|
|
|
|
|
|
|
|
$root = dirname(dirname(dirname(__FILE__)));
|
|
|
|
require_once $root.'/scripts/__init_script__.php';
|
|
|
|
|
2011-03-14 23:43:56 +01:00
|
|
|
$control = new PhabricatorDaemonControl();
|
|
|
|
|
2011-08-17 21:54:52 +02:00
|
|
|
must_have_extension('pcntl');
|
|
|
|
must_have_extension('posix');
|
|
|
|
|
|
|
|
function must_have_extension($ext) {
|
|
|
|
if (!extension_loaded($ext)) {
|
|
|
|
echo "ERROR: The PHP extension '{$ext}' is not installed. You must ".
|
|
|
|
"install it to run daemons on this machine.\n";
|
|
|
|
exit(1);
|
|
|
|
}
|
2013-04-07 17:42:59 +02:00
|
|
|
|
|
|
|
$extension = new ReflectionExtension($ext);
|
|
|
|
foreach ($extension->getFunctions() as $function) {
|
|
|
|
$function = $function->name;
|
|
|
|
if (!function_exists($function)) {
|
|
|
|
echo "ERROR: The PHP function {$function}() is disabled. You must ".
|
|
|
|
"enable it to run daemons on this machine.\n";
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
}
|
2011-08-17 21:54:52 +02:00
|
|
|
}
|
|
|
|
|
2012-05-09 19:29:37 +02:00
|
|
|
$command = isset($argv[1]) ? $argv[1] : 'help';
|
|
|
|
switch ($command) {
|
2011-03-14 20:33:20 +01:00
|
|
|
case 'list':
|
2011-03-14 23:43:56 +01:00
|
|
|
$err = $control->executeListCommand();
|
|
|
|
exit($err);
|
2011-03-14 20:33:20 +01:00
|
|
|
|
|
|
|
case 'status':
|
2011-03-14 23:43:56 +01:00
|
|
|
$err = $control->executeStatusCommand();
|
|
|
|
exit($err);
|
2011-03-14 20:33:20 +01:00
|
|
|
|
2011-03-14 23:43:56 +01:00
|
|
|
case 'stop':
|
2011-10-01 08:35:53 +02:00
|
|
|
$pass_argv = array_slice($argv, 2);
|
|
|
|
$err = $control->executeStopCommand($pass_argv);
|
2011-03-14 23:43:56 +01:00
|
|
|
exit($err);
|
2011-03-14 20:33:20 +01:00
|
|
|
|
2012-05-09 19:29:37 +02:00
|
|
|
case 'restart':
|
|
|
|
$err = $control->executeStopCommand(array());
|
|
|
|
if ($err) {
|
|
|
|
exit($err);
|
|
|
|
}
|
|
|
|
/* Fall Through */
|
|
|
|
case 'start':
|
|
|
|
$running = $control->loadRunningDaemons();
|
2012-06-20 21:36:31 +02:00
|
|
|
// "running" might not mean actually running so much as was running at
|
|
|
|
// some point. ergo, do a quick grouping and only barf if daemons are
|
|
|
|
// *actually* running.
|
|
|
|
$running_dict = mgroup($running, 'isRunning');
|
|
|
|
if (!empty($running_dict[true])) {
|
2012-05-09 19:29:37 +02:00
|
|
|
echo phutil_console_wrap(
|
|
|
|
"phd start: Unable to start daemons because daemons are already ".
|
|
|
|
"running.\n".
|
2012-05-09 20:33:10 +02:00
|
|
|
"You can view running daemons with 'phd status'.\n".
|
2012-05-09 19:29:37 +02:00
|
|
|
"You can stop running daemons with 'phd stop'.\n".
|
|
|
|
"You can use 'phd restart' to stop all daemons before starting new ".
|
|
|
|
"daemons.\n");
|
|
|
|
exit(1);
|
|
|
|
}
|
|
|
|
|
|
|
|
$daemons = array(
|
|
|
|
array('PhabricatorRepositoryPullLocalDaemon', array()),
|
|
|
|
array('PhabricatorGarbageCollectorDaemon', array()),
|
|
|
|
);
|
|
|
|
|
|
|
|
$taskmasters = PhabricatorEnv::getEnvConfig('phd.start-taskmasters');
|
|
|
|
for ($ii = 0; $ii < $taskmasters; $ii++) {
|
|
|
|
$daemons[] = array('PhabricatorTaskmasterDaemon', array());
|
|
|
|
}
|
|
|
|
|
|
|
|
will_launch($control);
|
|
|
|
foreach ($daemons as $spec) {
|
|
|
|
list($name, $argv) = $spec;
|
|
|
|
echo "Launching '{$name}'...\n";
|
|
|
|
$control->launchDaemon($name, $argv);
|
|
|
|
}
|
|
|
|
|
|
|
|
echo "Done.\n";
|
|
|
|
break;
|
|
|
|
|
2011-03-15 21:38:14 +01:00
|
|
|
case 'launch':
|
2011-06-13 19:01:06 +02:00
|
|
|
case 'debug':
|
|
|
|
$is_debug = ($argv[1] == 'debug');
|
|
|
|
|
2011-03-14 20:33:20 +01:00
|
|
|
$daemon = idx($argv, 2);
|
|
|
|
if (!$daemon) {
|
|
|
|
throw new Exception("Daemon name required!");
|
|
|
|
}
|
|
|
|
|
2011-03-15 21:38:14 +01:00
|
|
|
$pass_argv = array_slice($argv, 3);
|
2011-03-14 23:43:56 +01:00
|
|
|
|
2011-03-14 20:33:20 +01:00
|
|
|
$n = 1;
|
2011-06-13 19:01:06 +02:00
|
|
|
if (!$is_debug) {
|
|
|
|
if (is_numeric($daemon)) {
|
|
|
|
$n = $daemon;
|
|
|
|
if ($n < 1) {
|
|
|
|
throw new Exception("Count must be at least 1!");
|
|
|
|
}
|
|
|
|
$daemon = idx($argv, 3);
|
|
|
|
if (!$daemon) {
|
|
|
|
throw new Exception("Daemon name required!");
|
|
|
|
}
|
|
|
|
$pass_argv = array_slice($argv, 4);
|
2011-03-14 20:33:20 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
$loader = new PhutilSymbolLoader();
|
|
|
|
$symbols = $loader
|
|
|
|
->setAncestorClass('PhutilDaemon')
|
|
|
|
->selectSymbolsWithoutLoading();
|
|
|
|
|
|
|
|
$symbols = ipull($symbols, 'name');
|
|
|
|
$match = array();
|
|
|
|
foreach ($symbols as $symbol) {
|
|
|
|
if (stripos($symbol, $daemon) !== false) {
|
|
|
|
if (strtolower($symbol) == strtolower($daemon)) {
|
|
|
|
$match = array($symbol);
|
|
|
|
break;
|
|
|
|
} else {
|
|
|
|
$match[] = $symbol;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
if (count($match) == 0) {
|
|
|
|
throw new Exception(
|
|
|
|
"No daemons match! Use 'phd list' for a list of daemons.");
|
|
|
|
} else if (count($match) > 1) {
|
|
|
|
throw new Exception(
|
|
|
|
"Which of these daemons did you mean?\n".
|
|
|
|
" ".implode("\n ", $match));
|
|
|
|
} else {
|
|
|
|
$daemon = reset($match);
|
|
|
|
}
|
|
|
|
|
2011-06-27 05:14:11 +02:00
|
|
|
$with_logs = true;
|
|
|
|
if ($is_debug) {
|
|
|
|
// In debug mode, we emit errors straight to stdout, so nothing useful
|
|
|
|
// will show up in the logs. Don't echo the message about stuff showing
|
|
|
|
// up in them, since it would be confusing.
|
|
|
|
$with_logs = false;
|
|
|
|
}
|
|
|
|
|
|
|
|
will_launch($control, $with_logs);
|
2011-06-14 03:39:23 +02:00
|
|
|
|
2011-06-13 19:01:06 +02:00
|
|
|
if ($is_debug) {
|
|
|
|
echo "Launching {$daemon} in debug mode (nondaemonized)...\n";
|
|
|
|
} else {
|
|
|
|
echo "Launching {$n} x {$daemon}";
|
|
|
|
}
|
2011-03-14 23:43:56 +01:00
|
|
|
|
2011-03-14 20:33:20 +01:00
|
|
|
for ($ii = 0; $ii < $n; $ii++) {
|
2011-06-13 19:01:06 +02:00
|
|
|
$control->launchDaemon($daemon, $pass_argv, $is_debug);
|
|
|
|
if (!$is_debug) {
|
|
|
|
echo ".";
|
|
|
|
}
|
2011-03-14 20:33:20 +01:00
|
|
|
}
|
2011-03-15 21:38:14 +01:00
|
|
|
|
2011-03-14 20:33:20 +01:00
|
|
|
echo "\n";
|
|
|
|
echo "Done.\n";
|
|
|
|
|
|
|
|
break;
|
|
|
|
|
2011-03-13 22:27:03 +01:00
|
|
|
case '--help':
|
|
|
|
case 'help':
|
|
|
|
default:
|
2011-03-14 23:43:56 +01:00
|
|
|
$err = $control->executeHelpCommand();
|
|
|
|
exit($err);
|
2011-03-13 22:27:03 +01:00
|
|
|
}
|
2011-03-15 21:38:14 +01:00
|
|
|
|
2011-06-27 05:14:11 +02:00
|
|
|
function will_launch($control, $with_logs = true) {
|
2011-06-14 03:39:23 +02:00
|
|
|
echo "Staging launch...\n";
|
|
|
|
$control->pingConduit();
|
2011-06-27 05:14:11 +02:00
|
|
|
if ($with_logs) {
|
2012-08-04 10:27:57 +02:00
|
|
|
$log_dir = $control->getLogDirectory().'/daemons.log';
|
2011-06-27 05:14:11 +02:00
|
|
|
echo "NOTE: Logs will appear in '{$log_dir}'.\n\n";
|
|
|
|
}
|
2011-06-14 03:39:23 +02:00
|
|
|
}
|
|
|
|
|