]> git.mxchange.org Git - friendica.git/blobdiff - bin/daemon.php
Convert to tabs
[friendica.git] / bin / daemon.php
index acb26d8199bb68e94d2d5b4442caee6c170d7cdc..1682d366ede0ca16fd05c3ab8f0df7bd04b13a0a 100755 (executable)
@@ -1,45 +1,91 @@
 #!/usr/bin/env php
 <?php
 /**
- * @file bin/daemon.php
- * @brief Run the worker from a daemon.
+ * @copyright Copyright (C) 2010-2022, the Friendica project
+ *
+ * @license GNU AGPL version 3 or any later version
+ *
+ * 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 <https://www.gnu.org/licenses/>.
+ *
+ */
+
+/**
+ * Run the worker from a daemon.
  *
  * This script was taken from http://php.net/manual/en/function.pcntl-fork.php
  */
+if (php_sapi_name() !== 'cli') {
+       header($_SERVER["SERVER_PROTOCOL"] . ' 403 Forbidden');
+       exit();
+}
 
-use Friendica\App;
-use Friendica\BaseObject;
-use Friendica\Core\Config;
+use Dice\Dice;
+use Friendica\App\Mode;
+use Friendica\Core\Logger;
 use Friendica\Core\Worker;
+use Friendica\Database\DBA;
+use Friendica\DI;
+use Friendica\Util\DateTimeFormat;
+use Psr\Log\LoggerInterface;
+
+// Get options
+$shortopts = 'f';
+$longopts = ['foreground'];
+$options = getopt($shortopts, $longopts);
 
 // Ensure that daemon.php is executed from the base path of the installation
 if (!file_exists("boot.php") && (sizeof($_SERVER["argv"]) != 0)) {
        $directory = dirname($_SERVER["argv"][0]);
 
        if (substr($directory, 0, 1) != "/") {
-               $directory = $_SERVER["PWD"]."/".$directory;
+               $directory = $_SERVER["PWD"] . "/" . $directory;
        }
-       $directory = realpath($directory."/..");
+       $directory = realpath($directory . "/..");
 
        chdir($directory);
 }
 
-require_once "boot.php";
-require_once "include/dba.php";
+require dirname(__DIR__) . '/vendor/autoload.php';
 
-$a = new App(dirname(__DIR__));
-BaseObject::setApp($a);
+$dice = (new Dice())->addRules(include __DIR__ . '/../static/dependencies.config.php');
+$dice = $dice->addRule(LoggerInterface::class,['constructParams' => ['daemon']]);
 
-require_once ".htconfig.php";
-dba::connect($db_host, $db_user, $db_pass, $db_data);
+DI::init($dice);
+\Friendica\Core\Logger\Handler\ErrorHandler::register($dice->create(\Psr\Log\LoggerInterface::class));
+$a = DI::app();
+
+if (DI::mode()->isInstall()) {
+       die("Friendica isn't properly installed yet.\n");
+}
 
-Config::load();
+DI::mode()->setExecutor(Mode::DAEMON);
 
-if (!isset($pidfile)) {
-       die('Please specify a pid file in the variable $pidfile in the .htconfig.php. For example:'."\n".
-               '$pidfile = "/path/to/daemon.pid";'."\n");
+DI::config()->load();
+
+if (empty(DI::config()->get('system', 'pidfile'))) {
+       die(<<<TXT
+Please set system.pidfile in config/local.config.php. For example:
+    
+    'system' => [ 
+        'pidfile' => '/path/to/daemon.pid',
+    ],
+TXT
+    );
 }
 
+$pidfile = DI::config()->get('system', 'pidfile');
+
 if (in_array("start", $_SERVER["argv"])) {
        $mode = "start";
 }
@@ -52,6 +98,8 @@ if (in_array("status", $_SERVER["argv"])) {
        $mode = "status";
 }
 
+$foreground = array_key_exists('f', $options) || array_key_exists('foreground', $options);
+
 if (!isset($mode)) {
        die("Please use either 'start', 'stop' or 'status'.\n");
 }
@@ -60,10 +108,14 @@ if (empty($_SERVER["argv"][0])) {
        die("Unexpected script behaviour. This message should never occur.\n");
 }
 
-$pid = @file_get_contents($pidfile);
+$pid = null;
+
+if (is_readable($pidfile)) {
+       $pid = intval(file_get_contents($pidfile));
+}
 
 if (empty($pid) && in_array($mode, ["stop", "status"])) {
-       Config::set('system', 'worker_daemon_mode', false);
+       DI::config()->set('system', 'worker_daemon_mode', false);
        die("Pidfile wasn't found. Is the daemon running?\n");
 }
 
@@ -74,7 +126,7 @@ if ($mode == "status") {
 
        unlink($pidfile);
 
-       Config::set('system', 'worker_daemon_mode', false);
+       DI::config()->set('system', 'worker_daemon_mode', false);
        die("Daemon process $pid isn't running.\n");
 }
 
@@ -83,9 +135,9 @@ if ($mode == "stop") {
 
        unlink($pidfile);
 
-       logger("Worker daemon process $pid was killed.", LOGGER_DEBUG);
+       Logger::notice("Worker daemon process was killed", ["pid" => $pid]);
 
-       Config::set('system', 'worker_daemon_mode', false);
+       DI::config()->set('system', 'worker_daemon_mode', false);
        die("Worker daemon process $pid was killed.\n");
 }
 
@@ -93,43 +145,48 @@ if (!empty($pid) && posix_kill($pid, 0)) {
        die("Daemon process $pid is already running.\n");
 }
 
-logger('Starting worker daemon.', LOGGER_DEBUG);
-echo "Starting worker daemon.\n";
+Logger::notice('Starting worker daemon.', ["pid" => $pid]);
 
-// Switch over to daemon mode.
-if ($pid = pcntl_fork()) {
-       return;     // Parent
-}
+if (!$foreground) {
+       echo "Starting worker daemon.\n";
+
+       DBA::disconnect();
 
-fclose(STDIN);  // Close all of the standard
-fclose(STDOUT); // file descriptors as we
-fclose(STDERR); // are running as a daemon.
+       // Fork a daemon process
+       $pid = pcntl_fork();
+       if ($pid == -1) {
+               echo "Daemon couldn't be forked.\n";
+               Logger::warning('Could not fork daemon');
+               exit(1);
+       } elseif ($pid) {
+               // The parent process continues here
+               echo 'Child process started with pid ' . $pid . ".\n";
+               Logger::notice('Child process started', ['pid' => $pid]);
+               file_put_contents($pidfile, $pid);
+               exit(0);
+       }
 
-dba::disconnect();
+       // We now are in the child process
+       register_shutdown_function('shutdown');
 
-register_shutdown_function('shutdown');
+       // Make the child the main process, detach it from the terminal
+       if (posix_setsid() < 0) {
+               return;
+       }
 
-if (posix_setsid() < 0) {
-       return;
-}
+       // Closing all existing connections with the outside
+       fclose(STDIN);
 
-if ($pid = pcntl_fork()) {
-       return;     // Parent
+       // And now connect the database again
+       DBA::connect();
 }
 
-// We lose the database connection upon forking
-dba::connect($db_host, $db_user, $db_pass, $db_data);
-unset($db_host, $db_user, $db_pass, $db_data);
-
-Config::set('system', 'worker_daemon_mode', true);
+DI::config()->set('system', 'worker_daemon_mode', true);
 
 // Just to be sure that this script really runs endlessly
 set_time_limit(0);
 
-$pid = getmypid();
-file_put_contents($pidfile, $pid);
-
-$wait_interval = intval(Config::get('system', 'cron_interval', 5)) * 60;
+$wait_interval = intval(DI::config()->get('system', 'cron_interval', 5)) * 60;
 
 $do_cron = true;
 $last_cron = 0;
@@ -137,32 +194,52 @@ $last_cron = 0;
 // Now running as a daemon.
 while (true) {
        if (!$do_cron && ($last_cron + $wait_interval) < time()) {
-               logger('Forcing cron worker call.', LOGGER_DEBUG);
+               Logger::info('Forcing cron worker call.', ["pid" => $pid]);
                $do_cron = true;
        }
 
-       Worker::spawnWorker($do_cron);
+       if ($do_cron || (!DI::system()->isMaxLoadReached() && Worker::entriesExists() && Worker::isReady())) {
+               Worker::spawnWorker($do_cron);
+       } else {
+               Logger::info('Cool down for 5 seconds', ['pid' => $pid]);
+               sleep(5);
+       }
 
        if ($do_cron) {
                // We force a reconnect of the database connection.
                // This is done to ensure that the connection don't get lost over time.
-               dba::reconnect();
+               DBA::reconnect();
 
                $last_cron = time();
        }
 
-       logger("Sleeping", LOGGER_DEBUG);
-       $i = 0;
+       $start = time();
+       Logger::info("Sleeping", ["pid" => $pid, 'until' => gmdate(DateTimeFormat::MYSQL, $start + $wait_interval)]);
+
        do {
-               sleep(1);
-       } while (($i++ < $wait_interval) && !Worker::IPCJobsExists());
+               $seconds = (time() - $start);
+
+               // logarithmic wait time calculation.
+               // Background: After jobs had been started, they often fork many workers.
+               // To not waste too much time, the sleep period increases.
+               $arg = (($seconds + 1) / ($wait_interval / 9)) + 1;
+               $sleep = min(1000000, round(log10($arg) * 1000000, 0));
+               usleep($sleep);
+
+               $pid = pcntl_waitpid(-1, $status, WNOHANG);
+               if ($pid > 0) {
+                       Logger::info('Children quit via pcntl_waitpid', ['pid' => $pid, 'status' => $status]);
+               }
+
+               $timeout = ($seconds >= $wait_interval);
+       } while (!$timeout && !Worker\IPC::JobsExists());
 
-       if ($i >= $wait_interval) {
+       if ($timeout) {
                $do_cron = true;
-               logger("Woke up after $wait_interval seconds.", LOGGER_DEBUG);
+               Logger::info("Woke up after $wait_interval seconds.", ["pid" => $pid, 'sleep' => $wait_interval]);
        } else {
                $do_cron = false;
-               logger("Worker jobs are calling to be forked.", LOGGER_DEBUG);
+               Logger::info("Worker jobs are calling to be forked.", ["pid" => $pid]);
        }
 }