mirror of
https://gitee.com/ShopeX/OMS
synced 2026-03-23 19:05:34 +08:00
122 lines
3.0 KiB
PHP
122 lines
3.0 KiB
PHP
<?php
|
|
/**
|
|
* Copyright 2012-2026 ShopeX (https://www.shopex.cn)
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
if (!defined('__CONNECTER_MODE')){
|
|
require_once dirname(__FILE__) . '/../../shell/shell.php';
|
|
}
|
|
|
|
class taskmgr_swprocess_queue
|
|
{
|
|
// private static $driver;
|
|
|
|
/**
|
|
*
|
|
*
|
|
* @return mixed
|
|
* @author
|
|
**/
|
|
public static function getDriver($taskType)
|
|
{
|
|
$connecterClass = sprintf('taskmgr_connecter_%s', __CONNECTER_MODE);
|
|
$driver = new $connecterClass();
|
|
|
|
$config = strtoupper(sprintf('__%s_CONFIG', __CONNECTER_MODE));
|
|
|
|
$isConnect = $driver->load($taskType, $GLOBALS[$config]);
|
|
if (!$isConnect) {
|
|
return null;
|
|
}
|
|
|
|
return $driver;
|
|
}
|
|
|
|
/**
|
|
* 获取所有队列名
|
|
*
|
|
* @return array
|
|
* @author
|
|
**/
|
|
public static function getKeys()
|
|
{
|
|
$config = strtoupper(sprintf('__%s_CONFIG', __CONNECTER_MODE));
|
|
|
|
$queueNames = [];
|
|
|
|
$tasks = taskmgr_whitelist::get_all_task_list();
|
|
foreach ($tasks as $key => $value) {
|
|
if (false !== strpos($key, 'domainqueue')) {
|
|
continue;
|
|
}
|
|
|
|
$prefix = $GLOBALS[$config]['QUEUE_PREFIX'] ?: 'ERP';
|
|
|
|
$queueNames[] = sprintf('%s_TASK_%s_QUEUE', $prefix, strtoupper($key));
|
|
}
|
|
|
|
return $queueNames;
|
|
}
|
|
|
|
/**
|
|
* 获取队列名
|
|
*
|
|
* @return string
|
|
* @author
|
|
**/
|
|
public static function getKey($key)
|
|
{
|
|
$config = strtoupper(sprintf('__%s_CONFIG', __CONNECTER_MODE));
|
|
|
|
$prefix = $GLOBALS[$config]['QUEUE_PREFIX'] ?: 'ERP';
|
|
|
|
$name = sprintf('%s_TASK_%s_QUEUE', $prefix, strtoupper($key));
|
|
|
|
return $name;
|
|
}
|
|
|
|
/**
|
|
* 获取队列长度
|
|
*
|
|
* @return void
|
|
* @author
|
|
**/
|
|
public function count($key)
|
|
{
|
|
$split = explode('_', $key);
|
|
// 弹出第一，第二个
|
|
array_shift($split); array_shift($split); array_pop($split);
|
|
|
|
$taskType = strtolower(implode('_', $split));
|
|
|
|
$tasks = taskmgr_whitelist::get_all_task_list();
|
|
|
|
if (!$tasks[$taskType]) {
|
|
taskmgr_swconsole_output::error(sprintf("队列#%s#不存在", $key) . PHP_EOL);
|
|
|
|
exit();
|
|
}
|
|
|
|
$driver = self::getDriver($taskType);
|
|
if (!is_object($driver)) {
|
|
taskmgr_swconsole_output::error('队列服务未启用' . PHP_EOL);
|
|
|
|
exit();
|
|
}
|
|
|
|
return $driver->length();
|
|
}
|
|
}
|