ZY2025010601/app/Console/Commands/SwooleTask.php

<?php

namespace App\Console\Commands;

use App\Services\Api\FinanceService;
use App\Services\Api\MemberService;
use App\Services\Api\PledgeOrderService;
use App\Services\Api\PriceLogService;
use App\Services\Common\CurrencyService;
use App\Services\Common\TickerService;
use App\Services\RedisService;
use App\Services\WalletService;
use Illuminate\Console\Command;
use Illuminate\Support\Facades\DB;

class SwooleTask extends Command
{

    protected $serv;
    protected $host = '127.0.0.1';
    protected $port = 6660;
    // 进程名称
    protected $taskName = 'swooleTask';
    // PID路径
    protected $pidPath = '/storage/swoole.pid';
    // task
    protected $onlyReloadTaskWorker = false;
    // 设置运行时参数
    protected $options = [
        'worker_num' => 8, //worker进程数,一般设置为CPU数的1-4倍
        'daemonize' => true, //启用守护进程
        'log_file' => '/storage/logs/swoole-task.log', //指定swoole错误日志文件
        'log_level' => 0, //日志级别 范围是0-5，0-DEBUG，1-TRACE，2-INFO，3-NOTICE，4-WARNING，5-ERROR
        'dispatch_mode' => 1, //数据包分发策略,1-轮询模式
        'task_worker_num' => 6, //task进程的数量
        'task_ipc_mode' => 3, //使用消息队列通信，并设置为争抢模式
    ];


    /**
     * The name and signature of the console command.
     *
     * @var string
     */
    protected $signature = 'swoole:task {op}';

    /**
     * The console command description.
     *
     * @var string
     */
    protected $description = 'Swoole task server description';

    /**
     * Create a new command instance.
     *
     * @return void
     */
    public function __construct()
    {
        parent::__construct();
    }

    /**
     * 入口
     * Execute the console command.
     *
     * @return mixed
     */
    public function handle()
    {
        ini_set("default_socket_timeout", -1);
        // 项目根目录
        defined('ROOT_PATH') or define('ROOT_PATH', base_path());

        // 文件上传目录
        defined('ATTACHMENT_PATH') or define('ATTACHMENT_PATH', base_path('public/uploads'));

        // 图片上传目录
        defined('IMG_PATH') or define('IMG_PATH', base_path('public/uploads/images'));

        // 临时存放目录
        defined('UPLOAD_TEMP_PATH') or define('UPLOAD_TEMP_PATH', ATTACHMENT_PATH . "/temp");

        // 定义普通图片域名
        defined('IMG_URL') or define('IMG_URL', env('IMG_URL'));

        // 数据表前缀
        defined('DB_PREFIX') or define('DB_PREFIX', DB::connection()->getTablePrefix());

        $this->options['log_file'] = base_path() . $this->options['log_file'];
        $this->pidPath = base_path() . $this->pidPath;
        $op = $this->argument('op');
        switch ($op) {
            case 'status': // 状态
                $res = $this->status();
                echo $res ? $res : 0;
                break;
            case 'start': // 运行
                return $this->start();
                break;
            case 'reload': // 平滑重启
                return $this->reload();
                break;
            case 'stop': // 停止运行
                return $this->stop();
                break;
            default:
                exit("{$op} command does not exist");
                break;
        }
    }

    /**
     * 启动
     */
    public function start()
    {
        date_default_timezone_set('PRC');
        // 构建Server对象，监听对应地址
        $this->serv = new \Swoole\Server($this->host, $this->port);
        $this->serv->set($this->options);

        // 注册事件
        $this->serv->on('start', [$this, 'onStart']);
        $this->serv->on('receive', [$this, 'onReceive']);
        $this->serv->on('task', [$this, 'onTask']);
        $this->serv->on('finish', [$this, 'onFinish']);

        // Run worker
        echo "swoole start...\n";
        $this->serv->start();

    }

    // 安全重启
    public function reload()
    {

        $pids = file_exists($this->pidPath) ? file_get_contents($this->pidPath) : '';
        $pids = $pids ? explode("\n", $pids) : [];
        $masterPid = isset($pids[0]) ? $pids[0] : '';
        $managePid = isset($pids[1]) ? $pids[1] : '';
        if (empty($masterPid)) {
            return false;
        }

        if (!$this->status($masterPid)) {
            return false;
        }

        \Swoole\Process::kill($managePid, SIGUSR1);

        echo "swoole reload...\n";
    }

    /**
     * 停止
     * @param bool $smooth
     * @return bool
     */
    public function stop($smooth = false)
    {
        $pids = file_exists($this->pidPath) ? file_get_contents($this->pidPath) : '';
        $pids = $pids ? explode("\n", $pids) : [];
        $masterPid = isset($pids[0]) ? $pids[0] : '';
        if (empty($masterPid)) {
            return false;
        }

        if (!$this->status($masterPid)) {
            return false;
        }

        // 直接杀
        $stoSh = base_path() . '/crontab/swooleTaskStop.sh';
        if (file_exists($stoSh) && function_exists('exec')) {
            exec("{$stoSh}");
        }

        @unlink($this->pidPath);
        echo "swoole stop...\n";
    }

    /**
     * 状态
     * @return mixed
     */
    public function status($masterPid = 0)
    {
        $res = false;
        if (empty($masterPid) && file_exists($this->pidPath)) {
            $pids = file_get_contents($this->pidPath);
            $pids = $pids ? explode("\n", $pids) : [];
            $masterPid = isset($pids[0]) ? $pids[0] : '';
        }

        if ($masterPid) {
            $res = \Swoole\Process::kill($masterPid, 0);
        }
        return $res;
    }

    public function onStart($serv)
    {
        if (!is_dir(dirname($this->pidPath))) {
            @mkdir(dirname($this->pidPath), true, 755);
        }

        //记录进程id,脚本实现自动重启
        $pid = "{$serv->master_pid}\n{$serv->manager_pid}";
        file_put_contents($this->pidPath, $pid);

        // 定时任务
        $time = 0;
        $date = date('Y-m-d H:i:s');
        if (file_exists($this->options['log_file'])) {
            $time = 0;
            file_put_contents($this->options['log_file'], "Task {$date}:清空日志\n");
        }

        // TODO 更新USDT价格
        \swoole_timer_tick(2000, function ($timer) use ($serv, &$time) { // 启用定时器，每2秒钟执行一次
            $date = date('Y-m-d H:i:s');
            if ($time > 3600 && file_exists($this->options['log_file'])) {
                $time = 0;
                file_put_contents($this->options['log_file'], "Task {$date}:清空日志\n");
            }
            $time++;
            if (!RedisService::get('caches:task:lock:usdt_loaded')) {
                $taskData = [
                    'taskName' => 'UpdateUsdtPrice',
                    'name' => "更新USDT价格",
                    'date' => date('Y-m-d'),
                ];
                $res = $serv->task($taskData);
                RedisService::set('caches:task:lock:usdt_loaded', true, rand(2, 3));
                echo "[Task UpdateUsdtPrice {$date}] 更新USDT价格：{$res}\n";
            } else {
                echo "[Task UpdateUsdtPrice {$date}] 间隔时间调用\n";
            }

        });

        // TODO 定期清理价格记录
        \swoole_timer_tick(20000, function ($timer) use ($serv, &$time) { // 启用定时器，每2分钟执行一次
//        \swoole_timer_tick(120000, function ($timer) use ($serv, &$time) { // 启用定时器，每2分钟执行一次
            $date = date('Y-m-d H:i:s');
            if ($time > 7200 && file_exists($this->options['log_file'])) {
                $time = 0;
                file_put_contents($this->options['log_file'], "Task {$date}:清空日志\n");
            }
            $time++;
            $ccyList = CurrencyService::make()->getSyncList();
            if ($ccyList) {
                foreach ($ccyList as $item) {
                    $ccyId = $item['id'];
                    $ccy = $item['name'];
                    $cacheKey = "caches:task:lock:currency_clear_loaded_{$ccy}";
                    if ($ccyId && !RedisService::get($cacheKey)) {
                        $taskData = [
                            'taskName' => 'ClearCurrencyPrice',
                            'name' => "定期清理价格记录",
                            'params' => $item,
                            'date' => date('Y-m-d'),
                        ];
                        $res = $serv->task($taskData);
                        RedisService::set($cacheKey, true, rand(2, 3));
                        echo "[Task ClearCurrencyPrice {$date}] 清理[{$ccy}-USDT]价格记录：{$res}\n";
                    }else {
                        echo "[Task ClearCurrencyPrice {$date}] 间隔时间调用\n";
                    }
                }
            } else {
                echo "[Task ClearCurrencyPrice {$date}] 暂无需清理价格记录币种\n";
            }

        });

        return false;

        // TODO 自动监控方案下单
        \swoole_timer_tick(3000, function ($timer) use ($serv, &$time) { // 启用定时器，每3秒执行一次

            $date = date('Y-m-d H:i:s');
            if ($time > 7200 && file_exists($this->options['log_file'])) {
                $time = 0;
                file_put_contents($this->options['log_file'], "Task {$date}:清空日志\n");
            }
            $time++;
            $userList = MemberService::make()->pledgeUserList();
            if ($userList) {
                if (!RedisService::get('caches:task:lock:pledge_loaded')) {
                    foreach ($userList as $item) {
                        $userId = $item['id'];
                        if ($userId) {
                            $taskData = [
                                'taskName' => 'PledgeAutoTrade',
                                'name' => "自动质押交易",
                                'params' => $item,
                                'date' => date('Y-m-d'),
                            ];
                            $res = $serv->task($taskData);
                            echo "[Task PledgeAutoTrade {$date}] 用户[{$userId}]自动质押交易结果：{$res}\n";
                        }
                    }
                    RedisService::set('caches:task:lock:pledge_loaded', true, rand(5, 10));
                } else {
                    echo "[Task PledgeAutoTrade {$date}] 间隔时间调用\n";
                }
            } else {
                echo "[Task PledgeAutoTrade-0 {$date}] 暂无可自动质押交易用户\n";
            }

        });

        // TODO 检测质押订单退本，到期退本后（USDT余额足够）自动质押
//        \swoole_timer_tick(180000, function ($timer) use ($serv, &$time) { // 启用定时器，每3分钟执行一次
//            $date = date('Y-m-d H:i:s');
//            if ($time > 3600 && file_exists($this->options['log_file'])) {
//                $time = 0;
//                file_put_contents($this->options['log_file'], "Task {$date}:清空日志\n");
//            }
//            $time++;
//            $userList = PledgeOrderService::make()->getRefundOrderList();
//            if ($userList) {
//                if (!RedisService::get('caches:task:lock:plende_refund_loaded')) {
//                    foreach ($userList as $item) {
//                        $orderId = $item['id'];
//                        $userId = $item['user_id'];
//                        $orderNo = $item['order_no'];
//                        if ($orderId) {
//                            $taskData = [
//                                'taskName' => 'PledgeRefund',
//                                'name' => "质押订单自动退本",
//                                'params'=> $item,
//                                'date' => date('Y-m-d'),
//                            ];
//                            $res = $serv->task($taskData);
//                            echo "[Task PledgeRefund {$date}] 用户[{$userId}]质押订单[{$orderNo}]到期退本：{$res}\n";
//                        }
//                    }
//
//                    RedisService::set('caches:task:lock:pledge_settle_loaded', true, rand(3, 5));
//                }else{
//                    echo "[Task PledgeRefund {$date}] 质押订单到期退本调用间隔\n";
//                }
//            } else {
//                echo "[Task PledgeRefund {$date}] 间隔时间调用\n";
//            }
//        });
//
//        // TODO 质押订单到期结算收益，同时发放奖励（质押收益、推荐奖、管理奖、平级奖-基于管理奖）
//        \swoole_timer_tick(120000, function ($timer) use ($serv, &$time) { // 启用定时器，每120秒执行一次
//            $date = date('Y-m-d H:i:s');
//            if ($time > 3600 && file_exists($this->options['log_file'])) {
//                $time = 0;
//                file_put_contents($this->options['log_file'], "Task {$date}:清空日志\n");
//            }
//            $time++;
//            $userList = PledgeOrderService::make()->getSettleOrderList();
//            if ($userList) {
//                if (!RedisService::get('caches:task:lock:pledge_settle_loaded')) {
//                    foreach ($userList as $item) {
//                        $orderId = $item['id'];
//                        $userId = $item['user_id'];
//                        $orderNo = $item['order_no'];
//                        if ($orderId) {
//                            $taskData = [
//                                'taskName' => 'PledgeSettle',
//                                'name' => "质押订单到期结算",
//                                'params'=> $item,
//                                'date' => date('Y-m-d'),
//                            ];
//                            $res = $serv->task($taskData);
//                            echo "[Task PledgeSettle {$date}] 用户[{$userId}]质押订单[{$orderNo}]到期结算：{$res}\n";
//                        }
//                    }
//
//                    RedisService::set('caches:task:lock:pledge_settle_loaded', true, rand(3, 5));
//                }else{
//                    echo "[Task PledgeSettle {$date}] 质押订单到期结算调用间隔\n";
//                }
//            } else {
//                echo "[Task PledgeSettle {$date}] 间隔时间调用\n";
//            }
//
//        });

    }

    //监听连接进入事件
    public function onConnect($serv, $fd, $from_id)
    {
        $serv->send($fd, "Success {$fd}!");
    }

    // 监听数据接收事件
    public function onReceive(\Swoole\Server $serv, $fd, $from_id, $data)
    {
        echo "Get Message From Client {$fd}:{$data}\n";
        $res['result'] = 'success';
        $serv->send($fd, json_encode($res)); // 同步返回消息给客户端
        $serv->task($data);  // 执行异步任务
    }

    /**
     * @param \Swoole\Server $serv
     * @param $task_id
     * @param $from_id
     * @param $data
     * @return false|string
     */
    public function onTask(\Swoole\Server $serv, $task_id, $from_id, $data)
    {

        $date = date('Y-m-d H:i:s');
        $taskName = isset($data['taskName']) ? $data['taskName'] : '';
        $params = isset($data['params']) ? $data['params'] : [];
        try {
            switch ($taskName) {
                case 'UpdateUsdtPrice': // 更新USDT价格
                    // 调用处理
                    $res = WalletService::make()->getUsdtPrice('sell');
                    if ($res && $res1 = WalletService::make()->getUsdtPrice('buy')) {
                        echo "[Task {$taskName}-{$task_id} {$date}] 更新USDT价格：sell-{$res} buy-{$res1}\n";
                    } else {
                        $error = WalletService::make()->getError();
                        $error = $error ? lang($error) : '处理失败';
                        echo "[Task {$taskName}-{$task_id} {$date}] 更新USDT价格：{$error}\n";
                    }
                    break;
                case 'ClearCurrencyPrice': // 定期清理价格记录
                    // 调用处理
                    $ccy = isset($params['name'])? $params['name'] : 0;
                    if(empty($ccy)){
                        echo "[Task {$taskName}-{$task_id} {$date}] 币种参数错误\n";
                        return false;
                    }

                    if(date('H:i') >= '04:00'){
                        echo "[Task {$taskName}-{$task_id} {$date}] 不在运行时间内\n";
                        return false;
                    }

                    if ($res = TickerService::make()->clearPriceLog($ccy)) {
                        $res = is_array($res) && $res ? json_encode($res, 256) : '价格记录清理成功';
                        echo "[Task {$taskName}-{$task_id} {$date}] 币种[{$ccy}-USDT]定期清理价格记录结果：{$res}\n";
                    } else {
                        $error = TickerService::make()->getError();
                        $error = $error ? lang($error) : '价格记录清理失败';
                        echo "[Task {$taskName}-{$task_id} {$date}] 币种[{$ccy}-USDT]定期清理价格记录结果：{$error}\n";
                    }

                    break;
            }
        } catch (\Exception $exception) {
            return $exception->getMessage();
        }

        return '暂无任务处理';
    }


    /**
     * @param $serv swoole_server swoole_server对象
     * @param $task_id int 任务id
     * @param $data string 任务返回的数据
     */
    public function onFinish(\Swoole\Server $serv, $task_id, $data)
    {
        //
        echo "任务[{$task_id}]处理完成...\n";
    }


    // 监听连接关闭事件
    public function onClose($serv, $fd, $from_id)
    {
        echo "Client {$fd} close connection\n";
        $serv->close();
    }
}