瀏覽代碼

重启任务时ai worker自动启动

visuddhinanda 1 年之前
父節點
當前提交
1d4a490a38
共有 1 個文件被更改,包括 6 次插入2 次删除
  1. 6 2
      api-v8/app/Http/Controllers/TaskStatusController.php

+ 6 - 2
api-v8/app/Http/Controllers/TaskStatusController.php

@@ -110,6 +110,9 @@ class TaskStatusController extends Controller
                 $task->executor_id = $user['user_uid'];
                 $task->executor_id = $user['user_uid'];
                 $this->pushChange('running', $task->id);
                 $this->pushChange('running', $task->id);
                 break;
                 break;
+            case 'restarted':
+                $this->pushChange('restarted', $task->id);
+                break;
             case 'done':
             case 'done':
                 $this->pushChange('done', $task->id);
                 $this->pushChange('done', $task->id);
                 $task->finished_at = now();
                 $task->finished_at = now();
@@ -160,7 +163,7 @@ class TaskStatusController extends Controller
                     ->select('task_id')->get();
                     ->select('task_id')->get();
                 foreach ($preTasks as $key => $value) {
                 foreach ($preTasks as $key => $value) {
                     //$restartTask[] = $value->task_id;
                     //$restartTask[] = $value->task_id;
-                    $this->pushChange('restart', $value->task_id);
+                    $this->pushChange('restarted', $value->task_id);
                 }
                 }
                 break;
                 break;
         }
         }
@@ -168,7 +171,8 @@ class TaskStatusController extends Controller
         $task->editor_id = $user['user_uid'];
         $task->editor_id = $user['user_uid'];
         $task->save();
         $task->save();
         # auto start with ai assistant
         # auto start with ai assistant
-        foreach ($this->getChange('published') as $taskId) {
+        $autoStart = array_merge($this->getChange('published'), $this->getChange('restarted'));
+        foreach ($autoStart as $taskId) {
             $taskAssignee = TaskAssignee::where('task_id', $taskId)
             $taskAssignee = TaskAssignee::where('task_id', $taskId)
                 ->select('assignee_id')->get();
                 ->select('assignee_id')->get();
             $aiAssistant = AiModel::whereIn('uid', $taskAssignee)->first();
             $aiAssistant = AiModel::whereIn('uid', $taskAssignee)->first();