146 lines
3.5 KiB
PHP
146 lines
3.5 KiB
PHP
<?php
|
||
// +----------------------------------------------------------------------
|
||
// | CRMEB [ CRMEB赋能开发者，助力企业发展 ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2016~2026 https://www.crmeb.com All rights reserved.
|
||
// +----------------------------------------------------------------------
|
||
// | Licensed CRMEB并不是自由软件，未经许可不能去掉CRMEB相关版权
|
||
// +----------------------------------------------------------------------
|
||
// | Author: CRMEB Team <admin@crmeb.com>
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace crmeb\traits;
|
||
|
||
use think\facade\Queue;
|
||
|
||
/**
|
||
* 快捷加入消息队列
|
||
* Trait QueueTrait
|
||
* @package crmeb\traits
|
||
*/
|
||
trait QueueTrait
|
||
{
|
||
|
||
/**
|
||
* 错误次数
|
||
* @var int
|
||
*/
|
||
protected static int $errorCount = 3;
|
||
|
||
/**
|
||
* @var string
|
||
*/
|
||
protected static string $defaultDo = 'doJob';
|
||
|
||
/**
|
||
* 列名
|
||
* @return string
|
||
*/
|
||
protected static function queueName()
|
||
{
|
||
return null;
|
||
}
|
||
|
||
/**
|
||
* 加入队列
|
||
* @param $action
|
||
* @param array $data
|
||
* @param string|null $queueName
|
||
* @return mixed
|
||
*/
|
||
public static function dispatch($action = null, array $data = [], string $queueName = null)
|
||
{
|
||
$do = '';
|
||
if (is_array($action)) {
|
||
$data = $action;
|
||
} else if (is_string($action)) {
|
||
$do = $action;
|
||
}
|
||
|
||
if (self::queueName()) {
|
||
$queueName = self::queueName();
|
||
}
|
||
|
||
if (!$do) {
|
||
$do = self::$defaultDo;
|
||
}
|
||
|
||
return self::queuePush($do, $data, null, $queueName);
|
||
}
|
||
|
||
/**
|
||
* 延迟加入消息队列
|
||
* @param int $secs
|
||
* @param $action
|
||
* @param array $data
|
||
* @param string|null $queueName
|
||
* @return mixed
|
||
*/
|
||
public static function dispatchSece(int $secs, $action = null, array $data = [], string $queueName = null)
|
||
{
|
||
$do = '';
|
||
|
||
if (is_array($action)) {
|
||
$data = $action;
|
||
} else if (is_string($action)) {
|
||
$do = $action;
|
||
}
|
||
|
||
if (self::queueName()) {
|
||
$queueName = self::queueName();
|
||
}
|
||
|
||
if (!$do) {
|
||
$do = self::$defaultDo;
|
||
}
|
||
|
||
return self::queuePush($do, $data, $secs, $queueName);
|
||
}
|
||
|
||
/**
|
||
* 加入小队列
|
||
* @param string $do
|
||
* @param array $data
|
||
* @param int|null $secs
|
||
* @param string|null $queueName
|
||
* @return mixed
|
||
*/
|
||
public static function dispatchDo(string $do, array $data = [], int $secs = null, string $queueName = null)
|
||
{
|
||
if (self::queueName()) {
|
||
$queueName = self::queueName();
|
||
}
|
||
return self::queuePush($do, $data, $secs, $queueName);
|
||
}
|
||
|
||
/**
|
||
* 放入消息队列
|
||
* @param string $do
|
||
* @param array $data
|
||
* @param int|null $secs
|
||
* @param string|null $queueName
|
||
* @return mixed|null
|
||
* @author 等风来
|
||
* @email 136327134@qq.com
|
||
* @date 2023/9/28
|
||
*/
|
||
protected static function queuePush(string $do, array $data = [], int $secs = null, string $queueName = null)
|
||
{
|
||
$job = __CLASS__ . '@fire';
|
||
|
||
$jobData = [
|
||
'data' => $data,
|
||
'do' => $do,
|
||
'errorCount' => self::$errorCount,
|
||
];
|
||
|
||
if ($secs) {
|
||
$res = Queue::later($secs, $job, $jobData, $queueName);
|
||
} else {
|
||
$res = Queue::push($job, $jobData, $queueName);
|
||
}
|
||
|
||
return $res;
|
||
}
|
||
}
|