mirror of
https://gitee.com/zoujingli/ThinkAdmin.git
synced 2025-04-05 19:41:44 +08:00
223 lines
7.3 KiB
PHP
223 lines
7.3 KiB
PHP
<?php
|
||
|
||
// +----------------------------------------------------------------------
|
||
// | ThinkAdmin
|
||
// +----------------------------------------------------------------------
|
||
// | 版权所有 2014~2020 广州楚才信息科技有限公司 [ http://www.cuci.cc ]
|
||
// +----------------------------------------------------------------------
|
||
// | 官方网站: https://gitee.com/zoujingli/ThinkLibrary
|
||
// +----------------------------------------------------------------------
|
||
// | 开源协议 ( https://mit-license.org )
|
||
// +----------------------------------------------------------------------
|
||
// | gitee 代码仓库：https://gitee.com/zoujingli/ThinkLibrary
|
||
// | github 代码仓库：https://github.com/zoujingli/ThinkLibrary
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace think\admin\service;
|
||
|
||
use think\admin\Exception;
|
||
use think\admin\extend\CodeExtend;
|
||
use think\admin\Service;
|
||
use think\exception\InvalidArgumentException;
|
||
|
||
/**
|
||
* 任务基础服务
|
||
* Class QueueService
|
||
* @package think\admin\service
|
||
*/
|
||
class QueueService extends Service
|
||
{
|
||
|
||
/**
|
||
* 当前任务编号
|
||
* @var string
|
||
*/
|
||
protected $code = 0;
|
||
|
||
/**
|
||
* 当前任务标题
|
||
* @var string
|
||
*/
|
||
protected $title = '';
|
||
|
||
/**
|
||
* 当前任务参数
|
||
* @var array
|
||
*/
|
||
protected $data = [];
|
||
|
||
/**
|
||
* 当前任务数据
|
||
* @var array
|
||
*/
|
||
protected $queue = [];
|
||
|
||
/**
|
||
* 数据初始化
|
||
* @param integer $code
|
||
* @return static
|
||
* @throws \think\Exception
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
*/
|
||
public function initialize($code = 0)
|
||
{
|
||
if (!empty($code)) {
|
||
$this->code = $code;
|
||
$this->queue = $this->app->db->name('SystemQueue')->where(['code' => $this->code])->find();
|
||
if (empty($this->queue)) {
|
||
$this->app->log->error("Qeueu initialize failed, Queue {$code} not found.");
|
||
throw new \think\Exception("Qeueu initialize failed, Queue {$code} not found.");
|
||
}
|
||
$this->code = $this->queue['code'];
|
||
$this->title = $this->queue['title'];
|
||
$this->data = json_decode($this->queue['exec_data'], true) ?: [];
|
||
}
|
||
return $this;
|
||
}
|
||
|
||
/**
|
||
* 获取当前对象值
|
||
* @param string $name
|
||
* @return mixed
|
||
*/
|
||
public function __get($name)
|
||
{
|
||
if (isset($this->$name)) {
|
||
return $this->$name;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 判断是否WIN环境
|
||
* @return boolean
|
||
*/
|
||
protected function iswin()
|
||
{
|
||
return ProcessService::instance()->iswin();
|
||
}
|
||
|
||
/**
|
||
* 重发异步任务
|
||
* @param integer $wait 等待时间
|
||
* @return $this
|
||
* @throws \think\Exception
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
*/
|
||
public function reset($wait = 0)
|
||
{
|
||
if (empty($this->queue)) {
|
||
$this->app->log->error("Qeueu reset failed, Queue {$this->code} data cannot be empty!");
|
||
throw new \think\Exception("Qeueu reset failed, Queue {$this->code} data cannot be empty!");
|
||
}
|
||
$map = ['code' => $this->code];
|
||
$this->app->db->name('SystemQueue')->where($map)->strict(false)->failException(true)->update([
|
||
'exec_pid' => '0', 'exec_time' => time() + $wait, 'status' => '1',
|
||
]);
|
||
return $this->initialize($this->code);
|
||
}
|
||
|
||
/**
|
||
* 添加清理7天前的记录及超时任务
|
||
* @throws Exception
|
||
* @throws \think\Exception
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
*/
|
||
public function addCleanQueue()
|
||
{
|
||
$this->register('清理7天前记录及执行超时的任务', "xtask:clean", 0, [], 0, 3600);
|
||
}
|
||
|
||
/**
|
||
* 注册异步处理任务
|
||
* @param string $title 任务名称
|
||
* @param string $command 执行脚本
|
||
* @param integer $later 延时时间
|
||
* @param array $data 任务附加数据
|
||
* @param integer $rscript 任务类型(0单例,1多例)
|
||
* @param integer $loops 循环等待时间
|
||
* @return $this
|
||
* @throws Exception
|
||
* @throws \think\Exception
|
||
* @throws \think\db\exception\DataNotFoundException
|
||
* @throws \think\db\exception\DbException
|
||
* @throws \think\db\exception\ModelNotFoundException
|
||
*/
|
||
public function register($title, $command, $later = 0, $data = [], $rscript = 1, $loops = 0)
|
||
{
|
||
$map = [['title', '=', $title], ['status', 'in', ['1', '2']]];
|
||
if (empty($rscript) && ($queue = $this->app->db->name('SystemQueue')->where($map)->find())) {
|
||
throw new Exception(lang('think_library_queue_exist'), 0, $queue);
|
||
}
|
||
$this->app->db->name('SystemQueue')->strict(false)->failException(true)->insert([
|
||
'code' => $this->code = CodeExtend::uniqidDate(16, 'Q'),
|
||
'title' => $title,
|
||
'command' => $command,
|
||
'attempts' => '0',
|
||
'rscript' => intval(boolval($rscript)),
|
||
'exec_data' => json_encode($data, JSON_UNESCAPED_UNICODE),
|
||
'exec_time' => $later > 0 ? time() + $later : time(),
|
||
'enter_time' => '0',
|
||
'outer_time' => '0',
|
||
'loops_time' => $loops,
|
||
]);
|
||
$this->progress($this->code, 1, '>>> 任务创建成功！', 0.00);
|
||
return $this->initialize($this->code);
|
||
}
|
||
|
||
/**
|
||
* 更新任务进度信息
|
||
* @param string $code 任务编号
|
||
* @param null|integer $status 任务状态
|
||
* @param null|string $message 进度消息
|
||
* @param null|integer $progress 进度数值
|
||
* @return array
|
||
*/
|
||
public function progress($code, $status = null, $message = null, $progress = null)
|
||
{
|
||
$ckey = "queue_{$code}_progress";
|
||
$data = $this->app->cache->get($ckey, [
|
||
'code' => $code,
|
||
'status' => $status,
|
||
'message' => $message,
|
||
'progress' => $progress,
|
||
'history' => [],
|
||
]);
|
||
if (is_numeric($progress)) {
|
||
$progress = sprintf("%.2f", $progress);
|
||
}
|
||
if (is_string($message) && is_null($progress)) {
|
||
$data['message'] = $message;
|
||
$data['history'][] = ['message' => $message, 'progress' => $data['progress']];
|
||
} elseif (is_null($message) && is_numeric($progress)) {
|
||
$data['progress'] = $progress;
|
||
$data['history'][] = ['message' => $data['message'], 'progress' => $progress];
|
||
} elseif (is_string($message) && is_numeric($progress)) {
|
||
$data['message'] = $message;
|
||
$data['progress'] = $progress;
|
||
$data['history'][] = ['message' => $message, 'progress' => $progress];
|
||
}
|
||
if (is_numeric($status)) {
|
||
$data['status'] = intval($status);
|
||
}
|
||
if (is_string($message) || is_numeric($progress)) {
|
||
$this->app->cache->set($ckey, $data);
|
||
}
|
||
return $data;
|
||
}
|
||
|
||
/**
|
||
* 执行任务处理
|
||
* @param array $data 任务参数
|
||
* @return mixed
|
||
*/
|
||
public function execute(array $data = [])
|
||
{
|
||
}
|
||
|
||
} |