This commit is contained in:
2021-07-23 17:38:09 +08:00
parent b12ebb9c7e
commit e57e600945
4 changed files with 82 additions and 72 deletions
+16 -1
View File
@@ -11,7 +11,22 @@ namespace Server\SInterface;
interface PipeMessage interface PipeMessage
{ {
public function process(); /**
*
*/
public function process(): void;
/**
*
*/
public function max_execute(): void;
/**
* @return bool
*/
public function isStop(): bool;
} }
+50 -60
View File
@@ -6,16 +6,17 @@ namespace Snowflake\Crontab;
use Exception; use Exception;
use JetBrains\PhpStorm\Pure; use JetBrains\PhpStorm\Pure;
use Snowflake\Abstracts\BaseObject; use Server\SInterface\PipeMessage;
use Snowflake\Core\Json; use Snowflake\Application;
use Snowflake\Snowflake; use Snowflake\Snowflake;
use Swoole\Timer; use Swoole\Timer;
/** /**
* Class Async * Class Async
* @package Snowflake * @package Snowflake
* @property Application $application
*/ */
abstract class Crontab extends BaseObject abstract class Crontab implements PipeMessage
{ {
const WAIT_END = 'crontab:wait:execute'; const WAIT_END = 'crontab:wait:execute';
@@ -42,6 +43,29 @@ abstract class Crontab extends BaseObject
private int $execute_number = 0; private int $execute_number = 0;
/**
* Crontab constructor.
* @param mixed $params
* @param false $isLoop
* @param int $tickTime
*/
public function __construct(mixed $params, bool $isLoop = false, int $tickTime = 1)
{
$this->params = $params;
$this->isLoop = $isLoop;
$this->tickTime = $tickTime;
}
/**
* @return Application
*/
#[Pure] private function getApplication(): Application
{
return Snowflake::app();
}
/** /**
* @return $this * @return $this
*/ */
@@ -110,27 +134,6 @@ abstract class Crontab extends BaseObject
} }
public function setParams(): void
{
$this->params = func_get_args();
}
/**
* @param int $tickTime
*/
public function setTickTime(int $tickTime): void
{
$this->tickTime = $tickTime;
}
/**
* @param bool $isLoop
*/
public function setIsLoop(bool $isLoop): void
{
$this->isLoop = $isLoop;
}
/** /**
* @param int $max_execute_number * @param int $max_execute_number
*/ */
@@ -155,14 +158,6 @@ abstract class Crontab extends BaseObject
return $this->timerId; return $this->timerId;
} }
/**
* @param int $timerId
*/
public function setTimerId(int $timerId): void
{
$this->timerId = $timerId;
}
/** /**
* *
@@ -170,26 +165,22 @@ abstract class Crontab extends BaseObject
*/ */
public function clearTimer() public function clearTimer()
{ {
$this->warning('crontab timer clear.'); $this->application->warning('crontab timer clear.');
if (Timer::exists($this->timerId)) { if (Timer::exists($this->timerId)) {
Timer::clear($this->timerId); Timer::clear($this->timerId);
} }
} }
abstract public function process(): mixed;
abstract public function max_execute(): mixed;
abstract public function isStop(): bool;
/** /**
* @param $name * @param $name
* @return mixed * @return mixed
*/ */
public function __get($name): mixed #[Pure] public function __get($name): mixed
{ {
if ($name === 'application') {
return $this->getApplication();
}
if (!isset($this->params[$name])) { if (!isset($this->params[$name])) {
return null; return null;
} }
@@ -202,32 +193,32 @@ abstract class Crontab extends BaseObject
*/ */
public function execute(): void public function execute(): void
{ {
defer(fn() => $this->afterExecute());
try { try {
$redis = Snowflake::app()->getRedis(); $redis = $this->application->getRedis();
$name_md5 = $this->getName(); $name_md5 = $this->getName();
$redis->hSet(self::WAIT_END, $name_md5, static::getSerialize($this)); $redis->hSet(self::WAIT_END, $name_md5, static::getSerialize($this));
$params = call_user_func([$this, 'process'], ...$this->params); call_user_func([$this, 'process'], ...$this->params);
$redis->hDel(self::WAIT_END, $name_md5); $redis->hDel(self::WAIT_END, $name_md5);
if ($params === null) {
return;
}
write(Json::encode(['name' => $this->name, 'response' => serialize($params)]), 'crontab');
} catch (\Throwable $throwable) { } catch (\Throwable $throwable) {
logger()->addError($throwable, 'throwable'); $this->application->addError($throwable, 'throwable');
} finally {
$this->afterExecute();
} }
} }
/**
* @throws Exception
*/
public function afterExecute() public function afterExecute()
{ {
if ($this->isRecover() !== 999) { if ($this->isRecover() !== 999) {
return; return;
} }
$redis = Snowflake::app()->getRedis(); $redis = $this->application->getRedis();
$name = $this->getName(); $name = $this->getName();
if (!$redis->exists('stop:crontab:' . $name)) { if (!$redis->exists('stop:crontab:' . $name)) {
@@ -248,22 +239,21 @@ abstract class Crontab extends BaseObject
public function isRecover(): bool|int public function isRecover(): bool|int
{ {
try { try {
$redis = Snowflake::app()->getRedis(); $redis = $this->application->getRedis();
if ($redis->exists('stop:crontab:' . $this->getName())) { $crontab_name = $this->getName();
$redis->del('stop:crontab:' . $this->getName()); if ($redis->exists('stop:crontab:' . $crontab_name)) {
return true; return $redis->del('stop:crontab:' . $crontab_name);
} }
if ($this->isExit()) { if ($this->isExit()) {
return $redis->del('crontab:' . $this->getName()); return $redis->del('crontab:' . $crontab_name);
} }
if ($this->isMaxExecute()) { if (!$this->isMaxExecute()) {
call_user_func([$this, 'max_execute'], ...$this->getParams());
return $redis->del('crontab:' . $this->getName());
} else {
return 999; return 999;
} }
call_user_func([$this, 'max_execute'], ...$this->getParams());
return $redis->del('crontab:' . $crontab_name);
} catch (\Throwable $throwable) { } catch (\Throwable $throwable) {
return logger()->addError($throwable, 'throwable'); return $this->application->addError($throwable, 'throwable');
} }
} }
+6 -4
View File
@@ -17,21 +17,23 @@ class DefaultCrontab extends Crontab
*/ */
public function isStop(): bool public function isStop(): bool
{ {
// TODO: Implement isStop() method. return true;
} }
/** /**
* *
*/ */
public function process(): mixed public function process(): void
{ {
// TODO: Implement process() method. // TODO: Implement process() method.
} }
/**
public function max_execute(): mixed *
*/
public function max_execute(): void
{ {
// TODO: Implement max_execute() method. // TODO: Implement max_execute() method.
} }
+10 -7
View File
@@ -5,6 +5,7 @@ namespace Snowflake\Crontab;
use Exception; use Exception;
use Server\ServerManager;
use Server\SInterface\CustomProcess; use Server\SInterface\CustomProcess;
use Snowflake\Abstracts\Config; use Snowflake\Abstracts\Config;
use Snowflake\Cache\Redis; use Snowflake\Cache\Redis;
@@ -25,9 +26,6 @@ class Zookeeper implements CustomProcess
private int $workerNum = 0; private int $workerNum = 0;
private mixed $server;
/** /**
* @param Process $process * @param Process $process
* @return string * @return string
@@ -49,7 +47,7 @@ class Zookeeper implements CustomProcess
*/ */
public function onHandler(Process $process): void public function onHandler(Process $process): void
{ {
Timer::tick(100, [$this, 'loop']); Timer::tick(300, [$this, 'loop']);
} }
@@ -81,8 +79,8 @@ class Zookeeper implements CustomProcess
return; return;
} }
$params['handler'] = swoole_unserialize($handler); $params['handler'] = swoole_unserialize($handler);
$server = ServerManager::getContext()->getServer();
$this->server->sendMessage($params, $this->getWorker()); $server->sendMessage($params, $this->getWorker());
} catch (Throwable $exception) { } catch (Throwable $exception) {
logger()->addError($exception); logger()->addError($exception);
} }
@@ -91,10 +89,15 @@ class Zookeeper implements CustomProcess
/** /**
* @return int * @return int
* @throws \Exception * @throws Exception
*/ */
private function getWorker(): int private function getWorker(): int
{ {
if ($this->workerNum == 0) {
$server = ServerManager::getContext()->getServer();
$this->workerNum = $server->setting['worker_num'] + ($server->setting['task_worker_num'] ?? 0);
}
return random_int(0, $this->workerNum - 1); return random_int(0, $this->workerNum - 1);
} }