264 lines
9.1 KiB
PHP
264 lines
9.1 KiB
PHP
<?php
|
||
// +----------------------------------------------------------------------
|
||
// | CRMEB [ CRMEB赋能开发者，助力企业发展 ]
|
||
// +----------------------------------------------------------------------
|
||
// | Copyright (c) 2016~2026 https://www.crmeb.com All rights reserved.
|
||
// +----------------------------------------------------------------------
|
||
// | Licensed CRMEB并不是自由软件，未经许可不能去掉CRMEB相关版权
|
||
// +----------------------------------------------------------------------
|
||
// | Author: CRMEB Team <admin@crmeb.com>
|
||
// +----------------------------------------------------------------------
|
||
|
||
namespace app\services\user\label;
|
||
|
||
use app\services\BaseServices;
|
||
use app\services\user\UserServices;
|
||
use app\jobs\user\UserAutoLabelJob;
|
||
use think\facade\Log;
|
||
use crmeb\traits\ServicesTrait;
|
||
|
||
/**
|
||
* 用户标签批量处理服务
|
||
* Class UserLabelBatchProcessServices
|
||
* @package app\services\user\label
|
||
*/
|
||
class UserLabelBatchProcessServices extends BaseServices
|
||
{
|
||
use ServicesTrait;
|
||
|
||
/**
|
||
* 批量大小
|
||
*/
|
||
const BATCH_SIZE = 1000;
|
||
|
||
/**
|
||
* 处理标签变更后的用户批量检查
|
||
* @param array $labelIds 标签ID数组
|
||
* @param string $action 操作类型：create(新建) 或 update(修改)
|
||
* @return bool
|
||
*/
|
||
public function processLabelChange(array $labelIds, string $action = 'update')
|
||
{
|
||
try {
|
||
Log::info('开始处理标签变更批量用户检查:' . var_export([
|
||
'label_ids' => $labelIds,
|
||
'action' => $action
|
||
], true));
|
||
|
||
// 获取所有用户ID
|
||
$userIds = $this->getAllUserIds();
|
||
|
||
if (empty($userIds)) {
|
||
Log::info('没有找到用户，跳过处理');
|
||
return true;
|
||
}
|
||
|
||
// 初始化处理状态缓存
|
||
$cacheKey = "label_batch_process_" . implode('_', $labelIds);
|
||
$statusData = [
|
||
'label_ids' => $labelIds,
|
||
'status' => 'processing',
|
||
'processed_count' => 0,
|
||
'total_count' => count($userIds),
|
||
'start_time' => date('Y-m-d H:i:s'),
|
||
'end_time' => null,
|
||
'message' => '正在处理中...'
|
||
];
|
||
cache($cacheKey, $statusData, 3600); // 缓存1小时
|
||
|
||
// 分批处理用户
|
||
$batches = array_chunk($userIds, self::BATCH_SIZE);
|
||
$totalBatches = count($batches);
|
||
|
||
Log::info('开始分批处理用户:' . var_export([
|
||
'total_users' => count($userIds),
|
||
'batch_size' => self::BATCH_SIZE,
|
||
'total_batches' => $totalBatches
|
||
], true));
|
||
|
||
$processedCount = 0;
|
||
foreach ($batches as $index => $batchUserIds) {
|
||
$this->processBatchUsers($batchUserIds, $labelIds, $action, $index + 1, $totalBatches);
|
||
|
||
$processedCount += count($batchUserIds);
|
||
|
||
// 更新处理状态
|
||
$statusData['processed_count'] = $processedCount;
|
||
$statusData['message'] = "已处理 {$processedCount}/{$statusData['total_count']} 个用户";
|
||
cache($cacheKey, $statusData, 3600);
|
||
}
|
||
|
||
// 更新最终状态
|
||
$statusData['status'] = 'completed';
|
||
$statusData['end_time'] = date('Y-m-d H:i:s');
|
||
$statusData['message'] = '处理完成';
|
||
cache($cacheKey, $statusData, 3600);
|
||
|
||
Log::info('标签变更批量用户检查处理完成:' . var_export([
|
||
'label_ids' => $labelIds,
|
||
'total_users' => count($userIds)
|
||
], true));
|
||
return true;
|
||
} catch (\Throwable $e) {
|
||
// 更新错误状态
|
||
$cacheKey = "label_batch_process_" . implode('_', $labelIds);
|
||
$errorStatus = [
|
||
'label_ids' => $labelIds,
|
||
'status' => 'error',
|
||
'processed_count' => 0,
|
||
'total_count' => 0,
|
||
'start_time' => date('Y-m-d H:i:s'),
|
||
'end_time' => date('Y-m-d H:i:s'),
|
||
'message' => '处理失败：' . $e->getMessage()
|
||
];
|
||
cache($cacheKey, $errorStatus, 3600);
|
||
|
||
Log::error('标签变更批量用户检查处理失败:' . var_export([
|
||
'label_ids' => $labelIds,
|
||
'action' => $action,
|
||
'error' => $e->getMessage(),
|
||
'file' => $e->getFile(),
|
||
'line' => $e->getLine()
|
||
], true));
|
||
return false;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 获取所有用户ID
|
||
* @return array
|
||
*/
|
||
protected function getAllUserIds(): array
|
||
{
|
||
/** @var UserServices $userServices */
|
||
$userServices = app()->make(UserServices::class);
|
||
|
||
// 获取所有启用的用户ID
|
||
$where = [
|
||
'status' => 1, // 启用状态
|
||
'is_del' => 0 // 未删除
|
||
];
|
||
|
||
// 分批获取用户ID，避免内存溢出
|
||
$userIds = [];
|
||
$page = 1;
|
||
$limit = 5000; // 每次获取5000个用户ID
|
||
|
||
do {
|
||
$users = $userServices->getList($where, 'uid', $page, $limit);
|
||
$batchIds = array_column($users, 'uid');
|
||
$userIds = array_merge($userIds, $batchIds);
|
||
$page++;
|
||
} while (count($batchIds) === $limit);
|
||
|
||
return array_unique($userIds);
|
||
}
|
||
|
||
/**
|
||
* 处理批次用户
|
||
* @param array $userIds 用户ID数组
|
||
* @param array $labelIds 标签ID数组
|
||
* @param string $action 操作类型
|
||
* @param int $batchIndex 当前批次索引
|
||
* @param int $totalBatches 总批次数
|
||
*/
|
||
protected function processBatchUsers(array $userIds, array $labelIds, string $action, int $batchIndex, int $totalBatches)
|
||
{
|
||
try {
|
||
Log::info('处理用户批次:' . var_export([
|
||
'batch_index' => $batchIndex,
|
||
'total_batches' => $totalBatches,
|
||
'user_ids' => $userIds,
|
||
'user_count' => count($userIds),
|
||
'label_ids' => $labelIds
|
||
], true));
|
||
|
||
// 批量分发自动标签检查任务，每100个用户一个任务
|
||
$batchSize = 100;
|
||
$userBatches = array_chunk($userIds, $batchSize);
|
||
|
||
foreach ($userBatches as $userBatch) {
|
||
// 分发批量处理任务，传入用户数组而不是单个用户
|
||
|
||
UserAutoLabelJob::dispatch([
|
||
$userBatch,
|
||
'label_change', // 触发类型：标签变更
|
||
[
|
||
'action' => $action,
|
||
'changed_labels' => $labelIds,
|
||
'timestamp' => time()
|
||
],
|
||
$labelIds // 只检查变更的标签
|
||
]);
|
||
}
|
||
|
||
Log::info('批次用户处理完成:' . var_export([
|
||
'batch_index' => $batchIndex,
|
||
'user_batches' => count($userBatches),
|
||
'dispatched_jobs' => count($userBatches),
|
||
'total_users' => count($userIds)
|
||
], true));
|
||
} catch (\Throwable $e) {
|
||
Log::error('处理用户批次失败:' . var_export([
|
||
'batch_index' => $batchIndex,
|
||
'user_count' => count($userIds),
|
||
'error' => $e->getMessage()
|
||
], true));
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 处理单个标签的用户检查
|
||
* @param int $labelId 标签ID
|
||
* @param string $action 操作类型
|
||
* @return bool
|
||
*/
|
||
public function processSingleLabel(int $labelId, string $action = 'update')
|
||
{
|
||
return $this->processLabelChange([$labelId], $action);
|
||
}
|
||
|
||
/**
|
||
* 获取批量处理状态
|
||
* @param int $labelId
|
||
* @return array
|
||
*/
|
||
public function getBatchProcessStatus(int $labelId): array
|
||
{
|
||
try {
|
||
// 从缓存中获取处理状态
|
||
$cacheKey = "label_batch_process_{$labelId}";
|
||
$status = cache($cacheKey);
|
||
|
||
if (!$status) {
|
||
return [
|
||
'label_id' => $labelId,
|
||
'status' => 'not_started',
|
||
'processed_count' => 0,
|
||
'total_count' => 0,
|
||
'start_time' => null,
|
||
'end_time' => null,
|
||
'message' => '未找到处理记录'
|
||
];
|
||
}
|
||
|
||
return $status;
|
||
} catch (\Exception $e) {
|
||
Log::error('获取批量处理状态失败:' . var_export([
|
||
'label_id' => $labelId,
|
||
'error' => $e->getMessage()
|
||
], true));
|
||
|
||
return [
|
||
'label_id' => $labelId,
|
||
'status' => 'error',
|
||
'processed_count' => 0,
|
||
'total_count' => 0,
|
||
'start_time' => null,
|
||
'end_time' => null,
|
||
'message' => '获取状态失败：' . $e->getMessage()
|
||
];
|
||
}
|
||
}
|
||
}
|