mirror of
https://github.com/kuaifan/dootask.git
synced 2026-01-12 08:58:11 +00:00
281 lines
8.5 KiB
PHP
281 lines
8.5 KiB
PHP
<?php
|
||
|
||
namespace App\Tasks;
|
||
|
||
use App\Models\File;
|
||
use App\Models\User;
|
||
use App\Models\Project;
|
||
use App\Models\ProjectTask;
|
||
use App\Models\WebSocketDialogMsg;
|
||
use App\Module\Apps;
|
||
use App\Module\Manticore\ManticoreBase;
|
||
use App\Module\Manticore\ManticoreFile;
|
||
use App\Module\Manticore\ManticoreUser;
|
||
use App\Module\Manticore\ManticoreProject;
|
||
use App\Module\Manticore\ManticoreTask;
|
||
use App\Module\Manticore\ManticoreMsg;
|
||
use Carbon\Carbon;
|
||
use Illuminate\Support\Facades\Cache;
|
||
|
||
/**
|
||
* 通用 Manticore Search 同步任务（MVA 权限方案）
|
||
*
|
||
* 支持文件、用户、项目、任务的同步操作
|
||
* 使用 MVA (Multi-Value Attribute) 内联权限过滤
|
||
*/
|
||
class ManticoreSyncTask extends AbstractTask
|
||
{
|
||
private $action;
|
||
|
||
private $data;
|
||
|
||
public function __construct($action = null, $data = null)
|
||
{
|
||
parent::__construct(...func_get_args());
|
||
$this->action = $action;
|
||
$this->data = $data;
|
||
}
|
||
|
||
/**
|
||
* 获取任务动作类型（用于去重）
|
||
*
|
||
* @return string|null
|
||
*/
|
||
public function getAction(): ?string
|
||
{
|
||
return $this->action;
|
||
}
|
||
|
||
/**
|
||
* 获取数据ID（用于去重）
|
||
*
|
||
* @return int|null
|
||
*/
|
||
public function getDataId(): ?int
|
||
{
|
||
if (!is_array($this->data)) {
|
||
return null;
|
||
}
|
||
|
||
// 根据不同的 action 类型提取对应的 ID
|
||
return $this->data['id']
|
||
?? $this->data['userid']
|
||
?? $this->data['file_id']
|
||
?? $this->data['project_id']
|
||
?? $this->data['task_id']
|
||
?? $this->data['msg_id']
|
||
?? $this->data['dialog_id']
|
||
?? null;
|
||
}
|
||
|
||
public function start()
|
||
{
|
||
if (!Apps::isInstalled("manticore")) {
|
||
return;
|
||
}
|
||
|
||
switch ($this->action) {
|
||
// ==============================
|
||
// 文件同步动作
|
||
// ==============================
|
||
case 'file_sync':
|
||
$file = File::find($this->data['id'] ?? 0);
|
||
if ($file) {
|
||
ManticoreFile::sync($file);
|
||
}
|
||
break;
|
||
|
||
case 'file_delete':
|
||
$fileId = $this->data['id'] ?? 0;
|
||
if ($fileId > 0) {
|
||
ManticoreFile::delete($fileId);
|
||
}
|
||
break;
|
||
|
||
case 'file_pshare_update':
|
||
$fileIds = $this->data['file_ids'] ?? [];
|
||
$pshare = $this->data['pshare'] ?? 0;
|
||
if (!empty($fileIds)) {
|
||
ManticoreBase::batchUpdatePshare($fileIds, $pshare);
|
||
}
|
||
break;
|
||
|
||
case 'update_file_allowed_users':
|
||
// 更新文件权限
|
||
$fileId = $this->data['file_id'] ?? 0;
|
||
if ($fileId > 0) {
|
||
ManticoreFile::updateAllowedUsers($fileId);
|
||
}
|
||
break;
|
||
|
||
// ==============================
|
||
// 用户同步动作
|
||
// ==============================
|
||
case 'user_sync':
|
||
$user = User::find($this->data['userid'] ?? 0);
|
||
if ($user) {
|
||
ManticoreUser::sync($user);
|
||
}
|
||
break;
|
||
|
||
case 'user_delete':
|
||
$userid = $this->data['userid'] ?? 0;
|
||
if ($userid > 0) {
|
||
ManticoreUser::delete($userid);
|
||
}
|
||
break;
|
||
|
||
// ==============================
|
||
// 项目同步动作
|
||
// ==============================
|
||
case 'project_sync':
|
||
$project = Project::find($this->data['id'] ?? 0);
|
||
if ($project) {
|
||
ManticoreProject::sync($project);
|
||
}
|
||
break;
|
||
|
||
case 'project_delete':
|
||
$projectId = $this->data['project_id'] ?? 0;
|
||
if ($projectId > 0) {
|
||
ManticoreProject::delete($projectId);
|
||
}
|
||
break;
|
||
|
||
case 'update_project_allowed_users':
|
||
// 更新项目权限
|
||
$projectId = $this->data['project_id'] ?? 0;
|
||
if ($projectId > 0) {
|
||
ManticoreProject::updateAllowedUsers($projectId);
|
||
}
|
||
break;
|
||
|
||
case 'cascade_project_users':
|
||
// 项目成员变更时，级联更新该项目下所有 visibility=1 的任务
|
||
// 异步执行，避免阻塞
|
||
$projectId = $this->data['project_id'] ?? 0;
|
||
if ($projectId > 0) {
|
||
ManticoreTask::cascadeUpdateByProject($projectId);
|
||
}
|
||
break;
|
||
|
||
// ==============================
|
||
// 任务同步动作
|
||
// ==============================
|
||
case 'task_sync':
|
||
$task = ProjectTask::find($this->data['id'] ?? 0);
|
||
if ($task) {
|
||
ManticoreTask::sync($task);
|
||
}
|
||
break;
|
||
|
||
case 'task_delete':
|
||
$taskId = $this->data['task_id'] ?? 0;
|
||
if ($taskId > 0) {
|
||
ManticoreTask::delete($taskId);
|
||
}
|
||
break;
|
||
|
||
case 'update_task_allowed_users':
|
||
// 更新任务权限
|
||
$taskId = $this->data['task_id'] ?? 0;
|
||
if ($taskId > 0) {
|
||
ManticoreTask::updateAllowedUsers($taskId);
|
||
// 级联更新子任务
|
||
ManticoreTask::cascadeToChildren($taskId);
|
||
}
|
||
break;
|
||
|
||
// ==============================
|
||
// 消息同步动作
|
||
// ==============================
|
||
case 'msg_sync':
|
||
$msg = WebSocketDialogMsg::find($this->data['msg_id'] ?? 0);
|
||
if ($msg) {
|
||
ManticoreMsg::sync($msg);
|
||
}
|
||
break;
|
||
|
||
case 'msg_delete':
|
||
$msgId = $this->data['msg_id'] ?? 0;
|
||
if ($msgId > 0) {
|
||
ManticoreMsg::delete($msgId);
|
||
}
|
||
break;
|
||
|
||
case 'update_dialog_allowed_users':
|
||
// 更新对话消息权限
|
||
$dialogId = $this->data['dialog_id'] ?? 0;
|
||
if ($dialogId > 0) {
|
||
ManticoreMsg::updateDialogAllowedUsers($dialogId);
|
||
}
|
||
break;
|
||
|
||
default:
|
||
// 增量更新
|
||
$this->incrementalUpdate();
|
||
break;
|
||
}
|
||
}
|
||
|
||
/**
|
||
* 增量更新（定时执行 - 兜底机制）
|
||
*
|
||
* 命令本身会持续处理直到完成，定时器只是确保命令在运行
|
||
* 如果命令正在运行（有锁），则跳过本次触发
|
||
*
|
||
* @return void
|
||
*/
|
||
private function incrementalUpdate()
|
||
{
|
||
// 兜底触发：每 2 分钟检查一次，如果命令没在运行则启动
|
||
$time = intval(Cache::get("ManticoreSyncTask:CheckTime"));
|
||
if (time() - $time < 2 * 60) {
|
||
return;
|
||
}
|
||
Cache::put("ManticoreSyncTask:CheckTime", time(), Carbon::now()->addMinutes(5));
|
||
|
||
// 执行增量全文索引同步
|
||
$this->runIncrementalSync();
|
||
|
||
// 执行向量生成
|
||
$this->runVectorGeneration();
|
||
}
|
||
|
||
/**
|
||
* 执行增量全文索引同步（兜底触发）
|
||
*
|
||
* 命令内部有锁机制，如果已在运行会自动跳过
|
||
* 命令会持续处理直到无新数据，然后自动退出
|
||
*/
|
||
private function runIncrementalSync(): void
|
||
{
|
||
// 启动各类型的增量同步命令
|
||
@shell_exec("php /var/www/artisan manticore:sync-files --i 2>&1 &");
|
||
@shell_exec("php /var/www/artisan manticore:sync-users --i 2>&1 &");
|
||
@shell_exec("php /var/www/artisan manticore:sync-projects --i 2>&1 &");
|
||
@shell_exec("php /var/www/artisan manticore:sync-tasks --i 2>&1 &");
|
||
@shell_exec("php /var/www/artisan manticore:sync-msgs --i 2>&1 &");
|
||
}
|
||
|
||
/**
|
||
* 执行向量生成（兜底触发）
|
||
*
|
||
* 命令内部有锁机制，如果已在运行会自动跳过
|
||
* 命令会持续处理直到无待处理数据，然后自动退出
|
||
*/
|
||
private function runVectorGeneration(): void
|
||
{
|
||
if (!Apps::isInstalled("ai")) {
|
||
return;
|
||
}
|
||
|
||
// 启动向量生成命令
|
||
@shell_exec("php /var/www/artisan manticore:generate-vectors --type=all --batch=50 2>&1 &");
|
||
}
|
||
|
||
public function end()
|
||
{
|
||
}
|
||
}
|