Files
OMS/app/taskmgr/lib/func.php
2026-01-04 19:08:31 +08:00

100 lines
3.1 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.
*/
/**
* 公共方法
*/
class taskmgr_func
{
/**
* 分RDS多队列
*
* @return void
* @author
**/
public static function multiQueue($queue_config, $queue_exchange, $queue_name, $routerKey, $data, $type = 'task', $uniqid = '0')
{
//获取http或者https
$http_prefix = kernel::request()->get_schema();
$http_prefix = strtolower($http_prefix);
$http_prefix = ($http_prefix ? $http_prefix : 'http');
if ($task == 'api') {
$url = sprintf("%s://%s%s", $http_prefix, $_SERVER['SERVER_NAME'], $_SERVER['REQUEST_URI']);
// $data['_FROM_MQ_QUEUE'] = true;
} else {
$url = kernel::openapi_url('openapi.autotask', 'service');
$data['taskmgr_sign'] = taskmgr_rpc_sign::gen_sign($data);
}
$mq = kernel::single('base_queue_mq');
$mq->connect($queue_config, $queue_exchange, $queue_name);
$postAttr = array();
foreach ($data as $key => $val) {
$postAttr[] = $key . '=' . urlencode($val);
}
$message = array();
$message['spider_data']['url'] = $url;
$message['spider_data']['params'] = empty($postAttr) ? '' : join('&', $postAttr);
$message['relation']['to_node_id'] = base_shopnode::node_id('ome');
$message['relation']['from_node_id'] = '0';
$message['relation']['tid'] = $uniqid ? $uniqid : $data['uniqid'];
$message['relation']['to_url'] = $message['spider_data']['url'];
$message['relation']['time'] = time();
$mq->publish(json_encode($message), $routerKey);
$mq->disconnect();
}
public static function singleQueue($data)
{
$message = array(
'data' => $data,
'url' => kernel::openapi_url('openapi.autotask', 'service'),
);
kernel::single('taskmgr_interface_connecter')->push($message);
}
/**
*
*
* @return void
* @author
**/
public static function publishMsg($data, $type = 'task', $uniqid = '0', $config_args = [])
{
if ($config_args['SAAS_MQ'] == 'true') {
self::multiQueue($config_args['queue_config'],
$config_args['queue_exchange'],
$config_args['queue_name'],
$config_args['routerKey'],
$data,
$type,
$uniqid);
} else {
self::singleQueue($data);
}
}
}