scripts daemonize themselves

darcs-hash:20080904184031-84dde-eba2061f3aa898d0c791ffeb70837f759778c567.gz
This commit is contained in:
Evan Prodromou 2008-09-04 14:40:31 -04:00
parent 95d9920d78
commit aed0fff4ea
9 changed files with 167 additions and 37 deletions

View File

@ -94,6 +94,10 @@ $config =
'public' => array()), # JIDs of users who want to receive the public stream 'public' => array()), # JIDs of users who want to receive the public stream
'tag' => 'tag' =>
array('dropoff' => 864000.0), array('dropoff' => 864000.0),
'daemon' =>
array('piddir' => '/var/run',
'user' => false,
'group' => false)
); );
$config['db'] = &PEAR::getStaticProperty('DB_DataObject','options'); $config['db'] = &PEAR::getStaticProperty('DB_DataObject','options');

117
lib/daemon.php Normal file
View File

@ -0,0 +1,117 @@
<?php
/*
* Laconica - a distributed open-source microblogging tool
* Copyright (C) 2008, Controlez-Vous, Inc.
*
* This program is free software: you can redistribute it and/or modify
* it under the terms of the GNU Affero General Public License as published by
* the Free Software Foundation, either version 3 of the License, or
* (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU Affero General Public License for more details.
*
* You should have received a copy of the GNU Affero General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
if (!defined('LACONICA')) { exit(1); }
class Daemon {
function name() {
return NULL;
}
function background() {
$pid = pcntl_fork();
if ($pid < 0) { # error
return false;
} else if ($pid > 0) { # parent
common_log(LOG_INFO, "Successfully forked.");
exit(0);
} else { # child
return true;
}
}
function alreadyRunning() {
$pidfilename = $this->pidFilename();
if (!$pidfilename) {
return false;
}
if (!file_exists($pidfilename)) {
return false;
}
$contents = file_get_contents($pidfilename);
if (posix_kill(trim($contents),0)) {
return true;
} else {
return false;
}
}
function writePidFile() {
$pidfilename = $this->pidFilename();
if (!$pidfilename) {
return false;
}
file_put_contents($pidfilename, posix_getpid());
}
function clearPidFile() {
$pidfilename = $this->pidFilename();
unlink($pidfilename);
}
function pidFilename() {
$piddir = common_config('daemon', 'piddir');
if (!$piddir) {
return NULL;
}
$name = $this->name();
if (!$name) {
return NULL;
}
return $piddir . '/' . $name;
}
function changeUser() {
if (common_config('daemon', 'user')) {
$user_info = posix_getpwnam(common_config('daemon', 'user'));
common_log(LOG_INFO, "Setting user to " . common_config('daemon', 'user'));
posix_setuid($user_info['uid']);
}
if (common_config('daemon', 'group')) {
$group_info = posix_getgrnam(common_config('daemon', 'group'));
common_log(LOG_INFO, "Setting group to " . common_config('daemon', 'group'));
posix_setgid($group_info['gid']);
}
}
function runOnce() {
if ($this->alreadyRunning()) {
common_log(LOG_INFO, $this->name() . ' already running. Exiting.');
exit(0);
}
if ($this->background()) {
$this->writePidFile();
$this->changeUser();
$this->run();
$this->clearPidFile();
}
}
function run() {
return true;
}
}

View File

@ -19,7 +19,11 @@
define('CLAIM_TIMEOUT', 1200); define('CLAIM_TIMEOUT', 1200);
class QueueHandler { if (!defined('LACONICA')) { exit(1); }
require_once(INSTALLDIR.'/lib/daemon.php');
class QueueHandler extends Daemon {
var $_id = 'generic'; var $_id = 'generic';
@ -32,6 +36,10 @@ class QueueHandler {
function class_name() { function class_name() {
return ucfirst($this->transport()) . 'Handler'; return ucfirst($this->transport()) . 'Handler';
} }
function name() {
return strtolower($this->class_name().'.'.$this->get_id());
}
function get_id() { function get_id() {
return $this->_id; return $this->_id;
@ -55,7 +63,10 @@ class QueueHandler {
return true; return true;
} }
function handle_queue() { function run() {
if (!$this->start()) {
return false;
}
$this->log(LOG_INFO, 'checking for queued notices'); $this->log(LOG_INFO, 'checking for queued notices');
$transport = $this->transport(); $transport = $this->transport();
do { do {
@ -87,6 +98,10 @@ class QueueHandler {
$this->idle(5); $this->idle(5);
} }
} while (true); } while (true);
if (!$this->finish()) {
return false;
}
return true;
} }
function idle($timeout=0) { function idle($timeout=0) {

View File

@ -71,8 +71,4 @@ $id = ($argc > 1) ? $argv[1] : NULL;
$handler = new OmbQueueHandler($id); $handler = new OmbQueueHandler($id);
if ($handler->start()) { $handler->runOnce();
$handler->handle_queue();
}
$handler->finish();

View File

@ -105,8 +105,4 @@ $resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-publi
$handler = new PublicQueueHandler($resource); $handler = new PublicQueueHandler($resource);
if ($handler->start()) { $handler->runOnce();
$handler->handle_queue();
}
$handler->finish();

View File

@ -61,8 +61,4 @@ $id = ($argc > 1) ? $argv[1] : NULL;
$handler = new SmsQueueHandler($id); $handler = new SmsQueueHandler($id);
if ($handler->start()) { $handler->runOnce();
$handler->handle_queue();
}
$handler->finish();

View File

@ -63,7 +63,10 @@ class XmppConfirmHandler {
$this->conn->presence(NULL, 'available', NULL, 'available', -1); $this->conn->presence(NULL, 'available', NULL, 'available', -1);
} }
function handle_queue() { function run() {
if (!$this->start()) {
return false;
}
$this->log(LOG_INFO, 'checking for queued confirmations'); $this->log(LOG_INFO, 'checking for queued confirmations');
do { do {
$confirm = $this->next_confirm(); $confirm = $this->next_confirm();
@ -99,6 +102,10 @@ class XmppConfirmHandler {
$this->idle(10); $this->idle(10);
} }
} while (true); } while (true);
if (!$this->finish()) {
return false;
}
return true;
} }
function next_confirm() { function next_confirm() {
@ -181,8 +188,5 @@ $resource = ($argc > 1) ? $argv[1] : (common_config('xmpp', 'resource').'-confir
$handler = new XmppConfirmHandler($resource); $handler = new XmppConfirmHandler($resource);
if ($handler->start()) { $handler->runOnce();
$handler->handle_queue();
}
$handler->finish();

View File

@ -29,6 +29,7 @@ define('LACONICA', true);
require_once(INSTALLDIR . '/lib/common.php'); require_once(INSTALLDIR . '/lib/common.php');
require_once(INSTALLDIR . '/lib/jabber.php'); require_once(INSTALLDIR . '/lib/jabber.php');
require_once(INSTALLDIR . '/lib/daemon.php');
set_error_handler('common_error_handler'); set_error_handler('common_error_handler');
@ -36,7 +37,7 @@ set_error_handler('common_error_handler');
# in jabber.php, which create a new XMPP class. A more elegant (?) solution # in jabber.php, which create a new XMPP class. A more elegant (?) solution
# might be to use make this a subclass of XMPP. # might be to use make this a subclass of XMPP.
class XMPPDaemon { class XMPPDaemon extends Daemon {
function XMPPDaemon($resource=NULL) { function XMPPDaemon($resource=NULL) {
static $attrs = array('server', 'port', 'user', 'password', 'host'); static $attrs = array('server', 'port', 'user', 'password', 'host');
@ -74,12 +75,19 @@ class XMPPDaemon {
return !$this->conn->isDisconnected(); return !$this->conn->isDisconnected();
} }
function handle() { function name() {
$this->conn->addEventHandler('message', 'handle_message', $this); return strtolower('xmppdaemon.'.$this->resource);
$this->conn->addEventHandler('presence', 'handle_presence', $this); }
$this->conn->addEventHandler('reconnect', 'handle_reconnect', $this);
function run() {
$this->conn->process(); if ($this->connect()) {
$this->conn->addEventHandler('message', 'handle_message', $this);
$this->conn->addEventHandler('presence', 'handle_presence', $this);
$this->conn->addEventHandler('reconnect', 'handle_reconnect', $this);
$this->conn->process();
}
} }
function handle_reconnect(&$pl) { function handle_reconnect(&$pl) {
@ -325,6 +333,4 @@ $resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-liste
$daemon = new XMPPDaemon($resource); $daemon = new XMPPDaemon($resource);
if ($daemon->connect()) { $daemon->runOnce();
$daemon->handle();
}

View File

@ -110,8 +110,4 @@ $resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-queue
$handler = new XmppQueueHandler($resource); $handler = new XmppQueueHandler($resource);
if ($handler->start()) { $handler->runOnce();
$handler->handle_queue();
}
$handler->finish();