MadelineProto/src/danog/MadelineProto/Shutdown.php

93 lines
2.6 KiB
PHP

<?php
/**
* Shutdown module.
*
* This file is part of MadelineProto.
* MadelineProto 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.
* MadelineProto 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 General Public License along with MadelineProto.
* If not, see <http://www.gnu.org/licenses/>.
*
* @author Daniil Gentili <daniil@daniil.it>
* @copyright 2016-2020 Daniil Gentili <daniil@daniil.it>
* @license https://opensource.org/licenses/AGPL-3.0 AGPLv3
*
* @link https://docs.madelineproto.xyz MadelineProto documentation
*/
namespace danog\MadelineProto;
/**
* Class that controls script shutdown.
*/
class Shutdown
{
/**
* Callbacks to call on shutdown.
*
* @var array<callable>
*/
private static $callbacks = [];
/**
* Whether the main shutdown was registered.
*
* @var boolean
*/
private static $registered = false;
/**
* Incremental ID for new callback.
*
* @var integer
*/
private static $id = 0;
/**
* Function to be called on shutdown.
*
* @return void
*/
public static function shutdown()
{
foreach (self::$callbacks as $callback) {
$callback();
}
Magic::shutdown(0);
}
/**
* Add a callback for script shutdown.
*
* @param callable $callback The callback to set
* @param null|string $id The optional callback ID
*
* @return int|string The callback ID
*/
public static function addCallback($callback, $id = null)
{
if (!$id) {
$id = self::$id++;
}
self::$callbacks[$id] = $callback;
if (!self::$registered) {
\register_shutdown_function([__CLASS__, 'shutdown']);
self::$registered = true;
}
return $id;
}
/**
* Remove a callback from the script shutdown callable list.
*
* @param null|string|int $id The optional callback ID
*
* @return bool true if the callback was removed correctly, false otherwise
*/
public static function removeCallback($id)
{
if (isset(self::$callbacks[$id])) {
unset(self::$callbacks[$id]);
return true;
}
return false;
}
}