97 lines
2.1 KiB
PHP
97 lines
2.1 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake\Crontab;
|
|
|
|
|
|
use Exception;
|
|
use Snowflake\Abstracts\Component;
|
|
use Snowflake\Snowflake;
|
|
|
|
|
|
/**
|
|
* Class Producer
|
|
* @package Snowflake\Abstracts
|
|
*/
|
|
class Producer extends Component
|
|
{
|
|
|
|
const CRONTAB_KEY = '_application:{crontab}:system:crontab';
|
|
|
|
|
|
/**
|
|
* @param Crontab $crontab
|
|
* @throws Exception
|
|
*/
|
|
public function dispatch(Crontab $crontab)
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
|
|
$name = $crontab->getName();
|
|
if ($redis->exists(self::CRONTAB_KEY) && $redis->type(self::CRONTAB_KEY) !== \Redis::REDIS_ZSET) {
|
|
throw new Exception('Cache key ' . self::CRONTAB_KEY . ' types error.');
|
|
}
|
|
|
|
$redis->del('stop:crontab:' . $name);
|
|
|
|
$redis->del('crontab:' . $name);
|
|
$redis->zRem(static::CRONTAB_KEY, $name);
|
|
|
|
$redis->zAdd(self::CRONTAB_KEY, time() + $crontab->getTickTime(), $name);
|
|
$redis->set('crontab:' . $name, swoole_serialize($crontab));
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @throws Exception
|
|
*/
|
|
public function clear(string $name)
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
|
|
$redis->del('crontab:' . md5($name));
|
|
$redis->zRem(static::CRONTAB_KEY, md5($name));
|
|
|
|
$redis->setex('stop:crontab:' . md5($name), 120, 1);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @return bool
|
|
* @throws Exception
|
|
*/
|
|
public function exists(string $name): bool
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
if ($redis->exists('crontab:' . md5($name))) {
|
|
return true;
|
|
}
|
|
if ($redis->zRank(static::CRONTAB_KEY, md5($name))) {
|
|
return true;
|
|
}
|
|
if ($redis->hExists(Crontab::WAIT_END, md5($name))) {
|
|
return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function clearAll()
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
$data = $redis->zRange(self::CRONTAB_KEY, 0, -1);
|
|
foreach ($data as $datum) {
|
|
$redis->setex('stop:crontab:' . $datum, 120, 1);
|
|
$redis->del('crontab:' . $datum);
|
|
}
|
|
$redis->release();
|
|
}
|
|
|
|
|
|
}
|