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
'tag' =>
array('dropoff' => 864000.0),
'daemon' =>
array('piddir' => '/var/run',
'user' => false,
'group' => false)
);
$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);
class QueueHandler {
if (!defined('LACONICA')) { exit(1); }
require_once(INSTALLDIR.'/lib/daemon.php');
class QueueHandler extends Daemon {
var $_id = 'generic';
@ -33,6 +37,10 @@ class QueueHandler {
return ucfirst($this->transport()) . 'Handler';
}
function name() {
return strtolower($this->class_name().'.'.$this->get_id());
}
function get_id() {
return $this->_id;
}
@ -55,7 +63,10 @@ class QueueHandler {
return true;
}
function handle_queue() {
function run() {
if (!$this->start()) {
return false;
}
$this->log(LOG_INFO, 'checking for queued notices');
$transport = $this->transport();
do {
@ -87,6 +98,10 @@ class QueueHandler {
$this->idle(5);
}
} while (true);
if (!$this->finish()) {
return false;
}
return true;
}
function idle($timeout=0) {

View File

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

View File

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

View File

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

View File

@ -63,7 +63,10 @@ class XmppConfirmHandler {
$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');
do {
$confirm = $this->next_confirm();
@ -99,6 +102,10 @@ class XmppConfirmHandler {
$this->idle(10);
}
} while (true);
if (!$this->finish()) {
return false;
}
return true;
}
function next_confirm() {
@ -181,8 +188,5 @@ $resource = ($argc > 1) ? $argv[1] : (common_config('xmpp', 'resource').'-confir
$handler = new XmppConfirmHandler($resource);
if ($handler->start()) {
$handler->handle_queue();
}
$handler->runOnce();
$handler->finish();

View File

@ -29,6 +29,7 @@ define('LACONICA', true);
require_once(INSTALLDIR . '/lib/common.php');
require_once(INSTALLDIR . '/lib/jabber.php');
require_once(INSTALLDIR . '/lib/daemon.php');
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
# might be to use make this a subclass of XMPP.
class XMPPDaemon {
class XMPPDaemon extends Daemon {
function XMPPDaemon($resource=NULL) {
static $attrs = array('server', 'port', 'user', 'password', 'host');
@ -74,13 +75,20 @@ class XMPPDaemon {
return !$this->conn->isDisconnected();
}
function handle() {
function name() {
return strtolower('xmppdaemon.'.$this->resource);
}
function run() {
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) {
$this->conn->processUntil('session_start');
@ -325,6 +333,4 @@ $resource = ($argc > 1) ? $argv[1] : (common_config('xmpp','resource') . '-liste
$daemon = new XMPPDaemon($resource);
if ($daemon->connect()) {
$daemon->handle();
}
$daemon->runOnce();

View File

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