【触客宝】自动分配好友根群

2025-04-25 09:28:52 +08:00
parent 16a723bcaf
commit 3b12d6befd
5 changed files with 540 additions and 0 deletions
--- a/Server/application/command/AllotFriendCommand.php
+++ b/Server/application/command/AllotFriendCommand.php
@@ -0,0 +1,102 @@
+<?php
+
+namespace app\command;
+
+use think\console\Command;
+use think\console\Input;
+use think\console\Output;
+use think\console\input\Option;
+use think\facade\Log;
+use think\Queue;
+use app\job\AllotFriendJob;
+use think\facade\Cache;
+
+class AllotFriendCommand extends Command
+{
+    // 队列名称
+    protected $queueName = 'allot_friends';
+    
+    protected function configure()
+    {
+        $this->setName('allotFriends:run')
+            ->setDescription('自动分配微信好友')
+            ->addOption('toAccountId', null, Option::VALUE_REQUIRED, '目标账号ID')
+            ->addOption('wechatAccountKeyword', null, Option::VALUE_REQUIRED, '微信账号关键字')
+            ->addOption('isDeleted', null, Option::VALUE_OPTIONAL, '是否已删除状态: 0=未删除(false), 1=已删除(true)', 0)
+            ->addOption('jobId', null, Option::VALUE_OPTIONAL, '任务ID，用于区分不同实例', date('YmdHis') . rand(1000, 9999));
+    }
+
+    protected function execute(Input $input, Output $output)
+    {
+        $output->writeln('开始处理微信好友自动分配任务...');
+        
+        try {
+            // 获取命令参数
+            $toAccountId = $input->getOption('toAccountId');
+            $wechatAccountKeyword = $input->getOption('wechatAccountKeyword');
+            $isDeleted = $input->getOption('isDeleted');
+            $jobId = $input->getOption('jobId');
+            
+            // 验证必填参数
+            if (empty($toAccountId)) {
+                $output->writeln('错误: 目标账号ID不能为空');
+                return false;
+            }
+            
+            if (empty($wechatAccountKeyword)) {
+                $output->writeln('错误: 微信账号关键字不能为空');
+                return false;
+            }
+            
+            $output->writeln('目标账号ID: ' . $toAccountId);
+            $output->writeln('微信账号关键字: ' . $wechatAccountKeyword);
+            $output->writeln('删除状态: ' . ($isDeleted ? '已删除' : '未删除'));
+            $output->writeln('任务ID: ' . $jobId);
+            
+            // 检查队列是否已经在运行
+            $queueLockKey = "queue_lock:{$this->queueName}:{$wechatAccountKeyword}";
+            if (Cache::get($queueLockKey)) {
+                $output->writeln("队列 {$this->queueName} 已经在运行中，wechatAccountKeyword:{$wechatAccountKeyword}，跳过执行");
+                Log::warning("队列 {$this->queueName} 已经在运行中，wechatAccountKeyword:{$wechatAccountKeyword}，跳过执行");
+                return false;
+            }
+            
+            // 设置队列运行锁，有效期1小时
+            Cache::set($queueLockKey, $jobId, 3600);
+            $output->writeln("已设置队列运行锁，键名:{$queueLockKey}，值:{$jobId}，有效期:1小时");
+            
+            // 将任务添加到队列
+            $this->addToQueue($toAccountId, $wechatAccountKeyword, $isDeleted, $jobId, $queueLockKey);
+            
+            $output->writeln('微信好友自动分配任务已添加到队列');
+        } catch (\Exception $e) {
+            Log::error('微信好友自动分配任务添加失败：' . $e->getMessage());
+            $output->writeln('微信好友自动分配任务添加失败：' . $e->getMessage());
+            return false;
+        }
+        
+        return true;
+    }
+    
+    /**
+     * 添加任务到队列
+     * @param string $toAccountId 目标账号ID
+     * @param string $wechatAccountKeyword 微信账号关键字
+     * @param bool $isDeleted 是否已删除状态
+     * @param string $jobId 任务ID
+     * @param string $queueLockKey 队列锁键名
+     */
+    public function addToQueue($toAccountId, $wechatAccountKeyword, $isDeleted = false, $jobId = '', $queueLockKey = '')
+    {
+        $data = [
+            'toAccountId' => $toAccountId,
+            'wechatAccountKeyword' => $wechatAccountKeyword,
+            'isDeleted' => $isDeleted,
+            'jobId' => $jobId,
+            'queueLockKey' => $queueLockKey
+        ];
+        
+        // 添加到队列
+        Queue::push(AllotFriendJob::class, $data, $this->queueName);
+    }
+}