500 lines
17 KiB
PHP
500 lines
17 KiB
PHP
<?php
|
||
|
||
namespace app\job;
|
||
|
||
use think\queue\Job;
|
||
use think\facade\Log;
|
||
use think\Queue;
|
||
use think\facade\Config;
|
||
use think\facade\Cache;
|
||
use think\facade\Env;
|
||
use app\cunkebao\model\Workbench;
|
||
use app\cunkebao\model\WorkbenchAutoLike;
|
||
use think\Db;
|
||
use app\api\controller\WebSocketController;
|
||
use app\api\controller\AutomaticAssign;
|
||
use app\api\controller\WechatFriendController;
|
||
|
||
class WorkbenchAutoLikeJob
|
||
{
|
||
/**
|
||
* 最大重试次数
|
||
*/
|
||
const MAX_RETRY_ATTEMPTS = 3;
|
||
|
||
/**
|
||
* 队列任务处理
|
||
* @param Job $job 队列任务
|
||
* @param array $data 任务数据
|
||
* @return bool
|
||
*/
|
||
public function fire(Job $job, $data)
|
||
{
|
||
$jobId = $data['jobId'] ?? '';
|
||
$queueLockKey = $data['queueLockKey'] ?? '';
|
||
try {
|
||
$this->logJobStart($jobId, $queueLockKey);
|
||
$workbenches = $this->getActiveWorkbenches();
|
||
if (empty($workbenches)) {
|
||
$this->handleEmptyWorkbenches($job, $queueLockKey);
|
||
return true;
|
||
}
|
||
$this->processWorkbenches($workbenches);
|
||
$this->handleJobSuccess($job, $queueLockKey);
|
||
return true;
|
||
|
||
} catch (\Exception $e) {
|
||
return $this->handleJobError($e, $job, $queueLockKey);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取活跃的工作台
|
||
* @return \think\Collection
|
||
*/
|
||
protected function getActiveWorkbenches()
|
||
{
|
||
return Workbench::where([
|
||
['status', '=', 1],
|
||
['isDel', '=', 0],
|
||
['type', '=', 1] // 只获取自动点赞类型的工作台
|
||
])->order('id DESC')->select();
|
||
}
|
||
|
||
/**
|
||
* 处理工作台列表
|
||
* @param \think\Collection $workbenches
|
||
*/
|
||
protected function processWorkbenches($workbenches)
|
||
{
|
||
foreach ($workbenches as $workbench) {
|
||
try {
|
||
$this->processSingleWorkbench($workbench);
|
||
} catch (\Exception $e) {
|
||
Log::error("处理工作台 {$workbench->id} 失败: " . $e->getMessage());
|
||
}
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 处理单个工作台
|
||
* @param Workbench $workbench
|
||
*/
|
||
protected function processSingleWorkbench($workbench)
|
||
{
|
||
$config = WorkbenchAutoLike::where('workbenchId', $workbench->id)->find();
|
||
if (!$config) {
|
||
Log::error("工作台 {$workbench->id} 配置获取失败");
|
||
return;
|
||
}
|
||
|
||
$this->handleAutoLike($workbench, $config);
|
||
}
|
||
|
||
/**
|
||
* 处理自动点赞任务
|
||
* @param Workbench $workbench
|
||
* @param WorkbenchAutoLike $config
|
||
*/
|
||
protected function handleAutoLike($workbench, $config)
|
||
{
|
||
if (!$this->validateAutoLikeConfig($workbench, $config)) {
|
||
return;
|
||
}
|
||
|
||
// 验证是否在点赞时间范围内
|
||
if (!$this->isWithinLikeTimeRange($config)) {
|
||
return;
|
||
}
|
||
|
||
// 处理分页获取好友列表
|
||
$this->processAllFriends($workbench, $config);
|
||
}
|
||
|
||
/**
|
||
* 处理所有好友分页
|
||
* @param Workbench $workbench
|
||
* @param WorkbenchAutoLike $config
|
||
* @param int $page 当前页码
|
||
* @param int $pageSize 每页大小
|
||
*/
|
||
protected function processAllFriends($workbench, $config, $page = 1, $pageSize = 100)
|
||
{
|
||
$friendList = $this->getFriendList($config, $page, $pageSize);
|
||
|
||
if (empty($friendList)) {
|
||
return;
|
||
}
|
||
|
||
// 直接顺序处理所有好友
|
||
foreach ($friendList as $friend) {
|
||
// 验证是否达到点赞次数上限
|
||
$likeCount = $this->getTodayLikeCount($workbench, $config, $friend['deviceId']);
|
||
if ($likeCount >= $config['maxLikes']) {
|
||
Log::info("工作台 {$workbench->id} 点赞次数已达上限");
|
||
continue;
|
||
}
|
||
|
||
// 验证是否达到好友点赞次数上限
|
||
$friendMaxLikes = Db::name('workbench_auto_like_item')
|
||
->where('workbenchId', $workbench->id)
|
||
->where('wechatFriendId', $friend['friendId'])
|
||
->count();
|
||
|
||
if ($friendMaxLikes < $config['friendMaxLikes']) {
|
||
$this->processFriendMoments($workbench, $config, $friend);
|
||
}
|
||
}
|
||
|
||
// 如果当前页数据量等于页大小，说明可能还有更多数据，继续处理下一页
|
||
if (count($friendList) == $pageSize) {
|
||
$this->processAllFriends($workbench, $config, $page + 1, $pageSize);
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取好友列表
|
||
* @param WorkbenchAutoLike $config 配置
|
||
* @param int $page 页码
|
||
* @param int $pageSize 每页大小
|
||
* @return array
|
||
*/
|
||
protected function getFriendList($config, $page = 1, $pageSize = 100)
|
||
{
|
||
$friends = json_decode($config['friends'], true);
|
||
$devices = json_decode($config['devices'], true);
|
||
|
||
$list = Db::table('s2_company_account')
|
||
->alias('ca')
|
||
->join(['s2_wechat_account' => 'wa'], 'ca.id = wa.deviceAccountId')
|
||
->join(['s2_wechat_friend' => 'wf'], 'wf.wechatAccountId = wa.id')
|
||
->join('workbench_auto_like_item wali', 'wali.wechatFriendId = wf.id AND wali.workbenchId = ' . $config['workbenchId'], 'left')
|
||
->where([
|
||
'ca.status' => 0,
|
||
'wf.isDeleted' => 0,
|
||
'wa.deviceAlive' => 1,
|
||
'wa.wechatAlive' => 1
|
||
])
|
||
->whereIn('wa.currentDeviceId', $devices)
|
||
->field([
|
||
'ca.id as accountId',
|
||
'ca.userName',
|
||
'wf.id as friendId',
|
||
'wf.wechatId',
|
||
'wf.wechatAccountId',
|
||
'wa.wechatId as wechatAccountWechatId',
|
||
'wa.currentDeviceId as deviceId',
|
||
'COUNT(wali.id) as like_count'
|
||
]);
|
||
|
||
if (!empty($friends) && is_array($friends) && count($friends) > 0) {
|
||
$list = $list->whereIn('wf.id', $friends);
|
||
}
|
||
|
||
$list = $list->group('wf.wechatId')
|
||
->having('like_count < ' . $config['friendMaxLikes'])
|
||
->order('wf.id DESC')
|
||
->page($page, $pageSize)
|
||
->select();
|
||
|
||
return $list;
|
||
}
|
||
|
||
/**
|
||
* 处理好友朋友圈
|
||
* @param Workbench $workbench
|
||
* @param WorkbenchAutoLike $config
|
||
* @param array $friend
|
||
*/
|
||
protected function processFriendMoments($workbench, $config, $friend)
|
||
{
|
||
$toAccountId = '';
|
||
$username = Env::get('api.username', '');
|
||
$password = Env::get('api.password', '');
|
||
if (!empty($username) || !empty($password)) {
|
||
$toAccountId = Db::name('users')->where('account',$username)->value('s2_accountId');
|
||
}
|
||
|
||
try {
|
||
// 执行切换好友命令
|
||
$automaticAssign = new AutomaticAssign();
|
||
$automaticAssign->allotWechatFriend(['wechatFriendId' => $friend['friendId'], 'toAccountId' => $toAccountId], true);
|
||
|
||
//记录切换好友
|
||
$cacheKey = 'allotWechatFriend';
|
||
$cacheFriend = $friend;
|
||
$cacheFriend['time'] = time() + 120;
|
||
$maxRetry = 5;
|
||
$retry = 0;
|
||
do {
|
||
$cacheFriendData = Cache::get($cacheKey, []);
|
||
// 去重：移除同friendId的旧数据
|
||
$cacheFriendData = array_filter($cacheFriendData, function($item) use ($cacheFriend) {
|
||
return $item['friendId'] !== $cacheFriend['friendId'];
|
||
});
|
||
$cacheFriendData[] = $cacheFriend;
|
||
$success = Cache::set($cacheKey, $cacheFriendData);
|
||
$retry++;
|
||
} while (!$success && $retry < $maxRetry);
|
||
|
||
|
||
// 创建WebSocket链接
|
||
$webSocket = new WebSocketController(['userName' => $username, 'password' => $password, 'accountId' => $toAccountId]);
|
||
|
||
// 查询未点赞的朋友圈
|
||
$moments = $this->getUnlikedMoments($friend['wechatId']);
|
||
if (empty($moments) || count($moments) == 0) {
|
||
//采集最新朋友圈
|
||
$webSocket->getMoments(['wechatFriendId' => $friend['friendId'], 'wechatAccountId' => $friend['wechatAccountId']]);
|
||
$moments = $this->getUnlikedMoments($friend['wechatId']);
|
||
}
|
||
|
||
|
||
if (empty($moments) || count($moments) == 0) {
|
||
// 处理完毕切换回原账号
|
||
$automaticAssign->allotWechatFriend(['wechatFriendId' => $friend['friendId'], 'toAccountId' => $friend['accountId']], true);
|
||
Log::info("好友 {$friend['friendId']} 没有需要点赞的朋友圈");
|
||
return;
|
||
}
|
||
|
||
|
||
foreach ($moments as $moment) {
|
||
// 点赞朋友圈
|
||
$this->likeMoment($workbench, $config, $friend, $moment, $webSocket);
|
||
|
||
if(!empty($config['enableFriendTags']) && !empty($config['friendTags'])){
|
||
// 修改好友标签
|
||
$labels = $this->getFriendLabels($friend);
|
||
$labels[] = $config['friendTags'];
|
||
$webSocket->modifyFriendLabel(['wechatFriendId' => $friend['friendId'], 'wechatAccountId' => $friend['wechatAccountId'], 'labels' => $labels]);
|
||
|
||
//更新用户标签
|
||
$friendData = Db::table('s2_wechat_friend')->where('id', $friend['friendId'])->find();
|
||
Db::table('s2_wechat_friend')->where('id', $friend['friendId'])->update(['labels' => json_encode($labels,256)]);
|
||
}
|
||
break;
|
||
}
|
||
|
||
// 处理完毕切换回原账号
|
||
$automaticAssign->allotWechatFriend(['wechatFriendId' => $friend['friendId'], 'toAccountId' => $friend['accountId']], true);
|
||
} catch (\Exception $e) {
|
||
// 异常情况下也要确保切换回原账号
|
||
$automaticAssign = new AutomaticAssign();
|
||
$automaticAssign->allotWechatFriend(['wechatFriendId' => $friend['friendId'], 'toAccountId' => $friend['accountId']], true);
|
||
Log::error("处理好友 {$friend['friendId']} 朋友圈失败异常: " . $e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取未点赞的朋友圈
|
||
* @param int $wechatId
|
||
* @return \think\Collection
|
||
*/
|
||
protected function getUnlikedMoments($wechatId)
|
||
{
|
||
return Db::table('s2_wechat_moments')
|
||
->alias('wm')
|
||
->join('workbench_auto_like_item wali', 'wali.momentsId = wm.id', 'left')
|
||
->where([
|
||
['wm.userName', '=', $wechatId],
|
||
['wali.id', 'null', null]
|
||
])
|
||
->where('wm.update_time', '>=', time() - 86400)
|
||
->field('wm.id, wm.snsId')
|
||
->group('wali.wechatFriendId')
|
||
->order('wm.createTime DESC')
|
||
->page(1,1)
|
||
->select();
|
||
}
|
||
|
||
/**
|
||
* 点赞朋友圈
|
||
* @param Workbench $workbench
|
||
* @param WorkbenchAutoLike $config
|
||
* @param array $friend
|
||
* @param array $moment
|
||
* @param WebSocketController $webSocket
|
||
*/
|
||
protected function likeMoment($workbench, $config, $friend, $moment, $webSocket)
|
||
{
|
||
try {
|
||
$result = $webSocket->momentInteract([
|
||
'snsId' => $moment['snsId'],
|
||
'wechatAccountId' => $friend['wechatAccountId'],
|
||
]);
|
||
|
||
$result = json_decode($result, true);
|
||
|
||
if ($result['code'] == 200) {
|
||
$this->recordLike($workbench, $moment, $friend);
|
||
|
||
// 添加间隔时间
|
||
if (!empty($config['interval'])) {
|
||
sleep($config['interval']);
|
||
}
|
||
} else {
|
||
Log::error("工作台 {$workbench->id} 点赞失败: " . ($result['msg'] ?? '未知错误'));
|
||
}
|
||
} catch (\Exception $e) {
|
||
Log::error("工作台 {$workbench->id} 点赞异常: " . $e->getMessage());
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 记录点赞
|
||
* @param Workbench $workbench
|
||
* @param array $moment
|
||
* @param array $friend
|
||
*/
|
||
protected function recordLike($workbench, $moment, $friend)
|
||
{
|
||
Db::name('workbench_auto_like_item')->insert([
|
||
'workbenchId' => $workbench->id,
|
||
'deviceId' => $friend['deviceId'],
|
||
'momentsId' => $moment['id'],
|
||
'snsId' => $moment['snsId'],
|
||
'wechatAccountId' => $friend['wechatAccountId'],
|
||
'wechatFriendId' => $friend['friendId'],
|
||
'createTime' => time()
|
||
]);
|
||
Log::info("工作台 {$workbench->id} 点赞成功: {$moment['snsId']}");
|
||
}
|
||
|
||
/**
|
||
* 获取好友标签
|
||
* @param array $friend
|
||
* @return array
|
||
*/
|
||
protected function getFriendLabels($friend)
|
||
{
|
||
$wechatFriendController = new WechatFriendController();
|
||
$result = $wechatFriendController->getlist([
|
||
'friendKeyword' => $friend['wechatId'],
|
||
'wechatAccountKeyword' => $friend['wechatAccountWechatId']
|
||
], true);
|
||
|
||
$result = json_decode($result, true);
|
||
$labels = [];
|
||
|
||
if(!empty($result['data'])){
|
||
foreach($result['data'] as $item){
|
||
$labels = array_merge($labels, $item['labels']);
|
||
}
|
||
}
|
||
|
||
return $labels;
|
||
}
|
||
|
||
/**
|
||
* 验证自动点赞配置
|
||
* @param Workbench $workbench
|
||
* @param WorkbenchAutoLike $config
|
||
* @return bool
|
||
*/
|
||
protected function validateAutoLikeConfig($workbench, $config)
|
||
{
|
||
$requiredFields = ['contentTypes', 'interval', 'maxLikes', 'startTime', 'endTime'];
|
||
foreach ($requiredFields as $field) {
|
||
if (empty($config[$field])) {
|
||
Log::error("工作台 {$workbench->id} 配置字段 {$field} 为空");
|
||
return false;
|
||
}
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 获取今日点赞次数
|
||
* @param Workbench $workbench
|
||
* @param WorkbenchAutoLike $config
|
||
* @return int
|
||
*/
|
||
protected function getTodayLikeCount($workbench, $config, $deviceId)
|
||
{
|
||
return Db::name('workbench_auto_like_item')
|
||
->where('workbenchId', $workbench->id)
|
||
->where('deviceId', $deviceId)
|
||
->whereTime('createTime', 'between', [
|
||
strtotime(date('Y-m-d') . ' ' . $config['startTime'] . ':00'),
|
||
strtotime(date('Y-m-d') . ' ' . $config['endTime'] . ':00')
|
||
])
|
||
->count();
|
||
}
|
||
|
||
/**
|
||
* 检查是否在点赞时间范围内
|
||
* @param WorkbenchAutoLike $config
|
||
* @return bool
|
||
*/
|
||
protected function isWithinLikeTimeRange($config)
|
||
{
|
||
$currentTime = date('H:i');
|
||
if ($currentTime < $config['startTime'] || $currentTime > $config['endTime']) {
|
||
Log::info("当前时间 {$currentTime} 不在点赞时间范围内 ({$config['startTime']} - {$config['endTime']})");
|
||
return false;
|
||
}
|
||
return true;
|
||
}
|
||
|
||
/**
|
||
* 记录任务开始
|
||
* @param string $jobId
|
||
* @param string $queueLockKey
|
||
*/
|
||
protected function logJobStart($jobId, $queueLockKey)
|
||
{
|
||
Log::info('开始处理工作台自动点赞任务: ' . json_encode([
|
||
'jobId' => $jobId,
|
||
'queueLockKey' => $queueLockKey
|
||
]));
|
||
}
|
||
|
||
/**
|
||
* 处理任务成功
|
||
* @param Job $job
|
||
* @param string $queueLockKey
|
||
*/
|
||
protected function handleJobSuccess($job, $queueLockKey)
|
||
{
|
||
$job->delete();
|
||
Cache::rm($queueLockKey);
|
||
Log::info('工作台自动点赞任务执行成功');
|
||
}
|
||
|
||
/**
|
||
* 处理任务错误
|
||
* @param \Exception $e
|
||
* @param Job $job
|
||
* @param string $queueLockKey
|
||
* @return bool
|
||
*/
|
||
protected function handleJobError(\Exception $e, $job, $queueLockKey)
|
||
{
|
||
Log::error('工作台自动点赞任务异常：' . $e->getMessage());
|
||
|
||
if (!empty($queueLockKey)) {
|
||
Cache::rm($queueLockKey);
|
||
Log::info("由于异常释放队列锁: {$queueLockKey}");
|
||
}
|
||
|
||
if ($job->attempts() > self::MAX_RETRY_ATTEMPTS) {
|
||
$job->delete();
|
||
} else {
|
||
$job->release(Config::get('queue.failed_delay', 10));
|
||
}
|
||
|
||
return false;
|
||
}
|
||
|
||
/**
|
||
* 处理空工作台情况
|
||
* @param Job $job
|
||
* @param string $queueLockKey
|
||
*/
|
||
protected function handleEmptyWorkbenches(Job $job, $queueLockKey)
|
||
{
|
||
Log::info('没有需要处理的工作台自动点赞任务');
|
||
$job->delete();
|
||
Cache::rm($queueLockKey);
|
||
}
|
||
}
|