[DAEMONS] Properly disconnect daemons from the database

This commit is contained in:
Alexei Sorokin 2019-09-10 20:25:28 +03:00 committed by Diogo Peralta Cordeiro
parent cb412d228c
commit 72f5576795
2 changed files with 81 additions and 82 deletions

View File

@ -1,68 +1,56 @@
<?php <?php
/** // This file is part of GNU social - https://www.gnu.org/software/social
* StatusNet - the distributed open-source microblogging tool //
* Copyright (C) 2008, 2009, StatusNet, Inc. // GNU social is free software: you can redistribute it and/or modify
* // it under the terms of the GNU Affero General Public License as published by
* This program is free software: you can redistribute it and/or modify // the Free Software Foundation, either version 3 of the License, or
* it under the terms of the GNU Affero General Public License as published by // (at your option) any later version.
* the Free Software Foundation, either version 3 of the License, or //
* (at your option) any later version. // GNU social is distributed in the hope that it will be useful,
* // but WITHOUT ANY WARRANTY; without even the implied warranty of
* This program is distributed in the hope that it will be useful, // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* but WITHOUT ANY WARRANTY; without even the implied warranty of // GNU Affero General Public License for more details.
* 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 GNU social. If not, see <http://www.gnu.org/licenses/>.
* 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('STATUSNET') && !defined('LACONICA')) { defined('GNUSOCIAL') || die();
exit(1);
}
class Daemon class Daemon
{ {
var $daemonize = true; public $daemonize = true;
var $_id = 'generic'; public $_id = 'generic';
function __construct($daemonize = true) public function __construct($daemonize = true)
{ {
$this->daemonize = $daemonize; $this->daemonize = $daemonize;
} }
function name() public function name()
{ {
return null; return null;
} }
function get_id() public function get_id()
{ {
return $this->_id; return $this->_id;
} }
function set_id($id) public function set_id($id)
{ {
$this->_id = $id; $this->_id = $id;
} }
function background() public function background()
{ {
/* // Database connection will likely get lost after forking
* This prefers to Starting PHP 5.4 (dotdeb), maybe earlier for some version/distrib $this->resetDb();
* seems MySQL connection using mysqli driver get lost when fork.
* Need to unset it so that child process recreate it.
*
* @todo FIXME cleaner way to do it ?
*/
global $_DB_DATAOBJECT;
unset($_DB_DATAOBJECT['CONNECTIONS']);
$pid = pcntl_fork(); $pid = pcntl_fork();
if ($pid < 0) { // error if ($pid < 0) { // error
common_log(LOG_ERR, "Could not fork."); common_log(LOG_ERR, "Could not fork.");
return false; return false;
} else if ($pid > 0) { // parent } elseif ($pid > 0) { // parent
common_log(LOG_INFO, "Successfully forked."); common_log(LOG_INFO, "Successfully forked.");
exit(0); exit(0);
} else { // child } else { // child
@ -70,7 +58,7 @@ class Daemon
} }
} }
function alreadyRunning() public function alreadyRunning()
{ {
$pidfilename = $this->pidFilename(); $pidfilename = $this->pidFilename();
@ -89,7 +77,7 @@ class Daemon
} }
} }
function writePidFile() public function writePidFile()
{ {
$pidfilename = $this->pidFilename(); $pidfilename = $this->pidFilename();
@ -100,7 +88,7 @@ class Daemon
return file_put_contents($pidfilename, posix_getpid() . "\n"); return file_put_contents($pidfilename, posix_getpid() . "\n");
} }
function clearPidFile() public function clearPidFile()
{ {
$pidfilename = $this->pidFilename(); $pidfilename = $this->pidFilename();
if (!$pidfilename) { if (!$pidfilename) {
@ -109,7 +97,7 @@ class Daemon
return unlink($pidfilename); return unlink($pidfilename);
} }
function pidFilename() public function pidFilename()
{ {
$piddir = common_config('daemon', 'piddir'); $piddir = common_config('daemon', 'piddir');
if (!$piddir) { if (!$piddir) {
@ -122,15 +110,17 @@ class Daemon
return $piddir . '/' . $name . '.pid'; return $piddir . '/' . $name . '.pid';
} }
function changeUser() public function changeUser()
{ {
$groupname = common_config('daemon', 'group'); $groupname = common_config('daemon', 'group');
if ($groupname) { if ($groupname) {
$group_info = posix_getgrnam($groupname); $group_info = posix_getgrnam($groupname);
if (!$group_info) { if (!$group_info) {
common_log(LOG_WARNING, common_log(
'Ignoring unknown group for daemon: ' . $groupname); LOG_WARNING,
'Ignoring unknown group for daemon: ' . $groupname
);
} else { } else {
common_log(LOG_INFO, "Setting group to " . $groupname); common_log(LOG_INFO, "Setting group to " . $groupname);
posix_setgid($group_info['gid']); posix_setgid($group_info['gid']);
@ -142,8 +132,10 @@ class Daemon
if ($username) { if ($username) {
$user_info = posix_getpwnam($username); $user_info = posix_getpwnam($username);
if (!$user_info) { if (!$user_info) {
common_log(LOG_WARNING, common_log(
'Ignoring unknown user for daemon: ' . $username); LOG_WARNING,
'Ignoring unknown user for daemon: ' . $username
);
} else { } else {
common_log(LOG_INFO, "Setting user to " . $username); common_log(LOG_INFO, "Setting user to " . $username);
posix_setuid($user_info['uid']); posix_setuid($user_info['uid']);
@ -151,7 +143,7 @@ class Daemon
} }
} }
function runOnce() public function runOnce()
{ {
if ($this->alreadyRunning()) { if ($this->alreadyRunning()) {
common_log(LOG_INFO, $this->name() . ' already running. Exiting.'); common_log(LOG_INFO, $this->name() . ' already running. Exiting.');
@ -169,7 +161,7 @@ class Daemon
$this->clearPidFile(); $this->clearPidFile();
} }
function run() public function run()
{ {
return true; return true;
} }

View File

@ -1,21 +1,20 @@
<?php <?php
/* // This file is part of GNU social - https://www.gnu.org/software/social
* StatusNet - the distributed open-source microblogging tool //
* Copyright (C) 2010, StatusNet, Inc. // GNU social is free software: you can redistribute it and/or modify
* // it under the terms of the GNU Affero General Public License as published by
* This program is free software: you can redistribute it and/or modify // the Free Software Foundation, either version 3 of the License, or
* it under the terms of the GNU Affero General Public License as published by // (at your option) any later version.
* the Free Software Foundation, either version 3 of the License, or //
* (at your option) any later version. // GNU social is distributed in the hope that it will be useful,
* // but WITHOUT ANY WARRANTY; without even the implied warranty of
* This program is distributed in the hope that it will be useful, // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* but WITHOUT ANY WARRANTY; without even the implied warranty of // GNU Affero General Public License for more details.
* 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 GNU social. If not, see <http://www.gnu.org/licenses/>.
* 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/>. defined('GNUSOCIAL') || die();
*/
/** /**
* Base class for daemon that can launch one or more processing threads, * Base class for daemon that can launch one or more processing threads,
@ -34,14 +33,14 @@
*/ */
abstract class SpawningDaemon extends Daemon abstract class SpawningDaemon extends Daemon
{ {
protected $threads=1; protected $threads = 1;
const EXIT_OK = 0; const EXIT_OK = 0;
const EXIT_ERR = 1; const EXIT_ERR = 1;
const EXIT_SHUTDOWN = 100; const EXIT_SHUTDOWN = 100;
const EXIT_RESTART = 101; const EXIT_RESTART = 101;
function __construct($id=null, $daemonize=true, $threads=1) public function __construct($id = null, $daemonize = true, $threads = 1)
{ {
parent::__construct($daemonize); parent::__construct($daemonize);
@ -56,7 +55,7 @@ abstract class SpawningDaemon extends Daemon
* *
* @return int exit code; use self::EXIT_SHUTDOWN to request not to respawn. * @return int exit code; use self::EXIT_SHUTDOWN to request not to respawn.
*/ */
public abstract function runThread(); abstract public function runThread();
/** /**
* Spawn one or more background processes and let them start running. * Spawn one or more background processes and let them start running.
@ -69,17 +68,17 @@ abstract class SpawningDaemon extends Daemon
* though ParallelizingDaemon is probably better for workloads * though ParallelizingDaemon is probably better for workloads
* that have forseeable endpoints. * that have forseeable endpoints.
*/ */
function run() public function run()
{ {
$this->initPipes(); $this->initPipes();
$children = array(); $children = [];
for ($i = 1; $i <= $this->threads; $i++) { for ($i = 1; $i <= $this->threads; $i++) {
$pid = pcntl_fork(); $pid = pcntl_fork();
if ($pid < 0) { if ($pid < 0) {
$this->log(LOG_ERR, "Couldn't fork for thread $i; aborting\n"); $this->log(LOG_ERR, "Couldn't fork for thread $i; aborting\n");
exit(1); exit(1);
} else if ($pid == 0) { } elseif ($pid === 0) {
$this->initAndRunChild($i); $this->initAndRunChild($i);
} else { } else {
$this->log(LOG_INFO, "Spawned thread $i as pid $pid"); $this->log(LOG_INFO, "Spawned thread $i as pid $pid");
@ -101,7 +100,7 @@ abstract class SpawningDaemon extends Daemon
if (pcntl_wifexited($status)) { if (pcntl_wifexited($status)) {
$exitCode = pcntl_wexitstatus($status); $exitCode = pcntl_wexitstatus($status);
$info = "status $exitCode"; $info = "status $exitCode";
} else if (pcntl_wifsignaled($status)) { } elseif (pcntl_wifsignaled($status)) {
$exitCode = self::EXIT_ERR; $exitCode = self::EXIT_ERR;
$signal = pcntl_wtermsig($status); $signal = pcntl_wtermsig($status);
$info = "signal $signal"; $info = "signal $signal";
@ -114,7 +113,7 @@ abstract class SpawningDaemon extends Daemon
$pid = pcntl_fork(); $pid = pcntl_fork();
if ($pid < 0) { if ($pid < 0) {
$this->log(LOG_ERR, "Couldn't fork to respawn thread $i; aborting thread.\n"); $this->log(LOG_ERR, "Couldn't fork to respawn thread $i; aborting thread.\n");
} else if ($pid == 0) { } elseif ($pid === 0) {
$this->initAndRunChild($i); $this->initAndRunChild($i);
} else { } else {
$this->log(LOG_INFO, "Respawned thread $i as pid $pid"); $this->log(LOG_INFO, "Respawned thread $i as pid $pid");
@ -134,7 +133,7 @@ abstract class SpawningDaemon extends Daemon
* Create an IPC socket pair which child processes can use to detect * Create an IPC socket pair which child processes can use to detect
* if the parent process has been killed. * if the parent process has been killed.
*/ */
function initPipes() public function initPipes()
{ {
$sockets = stream_socket_pair(STREAM_PF_UNIX, STREAM_SOCK_STREAM, 0); $sockets = stream_socket_pair(STREAM_PF_UNIX, STREAM_SOCK_STREAM, 0);
if ($sockets) { if ($sockets) {
@ -197,10 +196,19 @@ abstract class SpawningDaemon extends Daemon
*/ */
protected function resetDb() protected function resetDb()
{ {
// @fixme do we need to explicitly open the db too
// or is this implied?
global $_DB_DATAOBJECT; global $_DB_DATAOBJECT;
unset($_DB_DATAOBJECT['CONNECTIONS']);
// Can't be called statically
$user = new User();
$conn = $user->getDatabaseConnection();
$conn->disconnect();
// Remove the disconnected connection from the list
foreach ($_DB_DATAOBJECT['CONNECTIONS'] as $k => $v) {
if ($v === $conn) {
unset($_DB_DATAOBJECT['CONNECTIONS'][$k]);
}
}
// Reconnect main memcached, or threads will stomp on // Reconnect main memcached, or threads will stomp on
// each other and corrupt their requests. // each other and corrupt their requests.
@ -216,14 +224,13 @@ abstract class SpawningDaemon extends Daemon
} }
} }
function log($level, $msg) public function log($level, $msg)
{ {
common_log($level, get_class($this) . ' ('. $this->get_id() .'): '.$msg); common_log($level, get_class($this) . ' ('. $this->get_id() .'): '.$msg);
} }
function name() public function name()
{ {
return strtolower(get_class($this).'.'.$this->get_id()); return strtolower(get_class($this).'.'.$this->get_id());
} }
} }