|
|
@@ -0,0 +1,338 @@
|
|
|
+<?php
|
|
|
+
|
|
|
+namespace App\Console\Commands;
|
|
|
+
|
|
|
+use App\Services\RedisService;
|
|
|
+use App\Services\WalletService;
|
|
|
+use Illuminate\Console\Command;
|
|
|
+use Illuminate\Support\Facades\DB;
|
|
|
+
|
|
|
+class WalletTask extends Command
|
|
|
+{
|
|
|
+
|
|
|
+ protected $serv;
|
|
|
+ protected $host = '127.0.0.1';
|
|
|
+ protected $port = 6626;
|
|
|
+ // 进程名称
|
|
|
+ protected $taskName = 'walletTask';
|
|
|
+ // PID路径
|
|
|
+ protected $pidPath = '/storage/wallet.pid';
|
|
|
+ // task
|
|
|
+ protected $onlyReloadTaskWorker = false;
|
|
|
+ // 设置运行时参数
|
|
|
+ protected $options = [
|
|
|
+ 'worker_num' => 8, //worker进程数,一般设置为CPU数的1-4倍
|
|
|
+ 'daemonize' => true, //启用守护进程
|
|
|
+ 'log_file' => '/storage/logs/wallet-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 = 'wallet:task {op}';
|
|
|
+
|
|
|
+ /**
|
|
|
+ * The console command description.
|
|
|
+ *
|
|
|
+ * @var string
|
|
|
+ */
|
|
|
+ protected $description = 'Wallet 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 "wallet 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 "wallet 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] : '';
|
|
|
+ $managePid = isset($pids[1]) ? $pids[1] : '';
|
|
|
+ if (empty($masterPid)) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ if (!$this->status($masterPid)) {
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ if ($smooth) {
|
|
|
+ \Swoole\Process::kill($masterPid, SIGTERM);
|
|
|
+ try {
|
|
|
+ while (true) {
|
|
|
+ \Swoole\Process::kill($masterPid, 0);
|
|
|
+ }
|
|
|
+ } catch (\Exception $exception) {
|
|
|
+
|
|
|
+ }
|
|
|
+ } else {
|
|
|
+ \Swoole\Process::kill($masterPid, SIGKILL);
|
|
|
+ }
|
|
|
+
|
|
|
+ if($managePid){
|
|
|
+ \Swoole\Process::kill($managePid, SIGKILL);
|
|
|
+ }
|
|
|
+
|
|
|
+ // 直接杀
|
|
|
+ if(function_exists('exec')){
|
|
|
+ exec('pkill -9 php artisan wallet:task start');
|
|
|
+ }
|
|
|
+
|
|
|
+ @unlink($this->pidPath);
|
|
|
+ echo "wallet 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");
|
|
|
+ }
|
|
|
+
|
|
|
+ \swoole_timer_tick(20000, function ($timer) use ($serv, &$time) { // 启用定时器,每20秒执行一次
|
|
|
+ $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++;
|
|
|
+ $wallets = WalletService::make()->getWalletList();
|
|
|
+ if($wallets){
|
|
|
+ if(!RedisService::get('caches:task:lock:wallet_loaded')){
|
|
|
+ foreach ($wallets as $item){
|
|
|
+ $taskData = [
|
|
|
+ 'taskName' => 'listenWallet',
|
|
|
+ 'name' => "平台钱包地址监控",
|
|
|
+ 'params' => $item,
|
|
|
+ 'date' => date('Y-m-d'),
|
|
|
+ ];
|
|
|
+ $res = $serv->task($taskData);
|
|
|
+ }
|
|
|
+ RedisService::set('caches:task:lock:wallet_loaded', true, rand(3,5));
|
|
|
+ echo "[Task listenWallet {$date}] 平台钱包地址监控调用完成:{$res}\n";
|
|
|
+ }else{
|
|
|
+ echo "[Task listenWallet {$date}] 间隔时间调用\n";
|
|
|
+ }
|
|
|
+ }else{
|
|
|
+ echo "[Task listenWallet {$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'] : [];
|
|
|
+ switch ($taskName) {
|
|
|
+ case 'listenWallet': // 钱包监控
|
|
|
+ $address = isset($params['address'])? $params['address'] : '';
|
|
|
+ $type = isset($params['type'])? $params['type'] : 1;
|
|
|
+ if(empty($address)){
|
|
|
+ echo "[Task {$taskName} {$date}] address 地址参数错误\n";
|
|
|
+ return false;
|
|
|
+ }
|
|
|
+
|
|
|
+ // 调用处理
|
|
|
+ if($res = WalletService::make()->listenTrcWallet($address, $type)){
|
|
|
+ $success = isset($res['success'])? $res['success'] : 0;
|
|
|
+ $total = isset($res['total'])? $res['total'] : 0;
|
|
|
+ echo "[Task {$taskName} {$date}] 钱包地址监控处理结果:".($total? "监控{$total}条记录,成功处理{$success}条":'success')."\n";
|
|
|
+ }else{
|
|
|
+ $error = WalletService::make()->getError();
|
|
|
+ $error = $error? lang($error) : 'failed';
|
|
|
+ echo "[Task {$taskName} {$date}] 钱包地址监控处理结果:{$error}\n";
|
|
|
+ }
|
|
|
+ break;
|
|
|
+ }
|
|
|
+
|
|
|
+ return "[Task {$taskName} {$date}] 暂无任务处理\n";
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ /**
|
|
|
+ * @param $serv swoole_server swoole_server对象
|
|
|
+ * @param $task_id int 任务id
|
|
|
+ * @param $data string 任务返回的数据
|
|
|
+ */
|
|
|
+ public function onFinish(\Swoole\Server $serv, $task_id, $data)
|
|
|
+ {
|
|
|
+ //
|
|
|
+ echo "任务处理完成...\n";
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ // 监听连接关闭事件
|
|
|
+ public function onClose($serv, $fd, $from_id)
|
|
|
+ {
|
|
|
+ echo "Client {$fd} close connection\n";
|
|
|
+ $serv->close();
|
|
|
+ }
|
|
|
+}
|