Commit 1a11de42 by haojie

1

parent e1ac83ab
...@@ -14,6 +14,8 @@ class TaskService ...@@ -14,6 +14,8 @@ class TaskService
public const TYPE_IMG_TO_IMG = 1; // 图片转图片 public const TYPE_IMG_TO_IMG = 1; // 图片转图片
public const TYPE_TEXT_TO_IMG = 2; // 文字转图片 public const TYPE_TEXT_TO_IMG = 2; // 文字转图片
public const GPT_prompt_num = 1; // 指令数量
public function submit($user_id, array $data = []) public function submit($user_id, array $data = [])
{ {
# task表新增一条记录 # task表新增一条记录
...@@ -45,15 +47,30 @@ public function submit($user_id, array $data = []) ...@@ -45,15 +47,30 @@ public function submit($user_id, array $data = [])
'prompt' => $prompt, 'prompt' => $prompt,
'prompt_img' => json_encode($prompt_img), 'prompt_img' => json_encode($prompt_img),
'user_id' => $user_id, 'user_id' => $user_id,
'prompt_num' => 2, 'prompt_num' => self::GPT_prompt_num,
]; ];
# 保存任务 # 保存任务
$task_id = Task::query()->create($list)->id; $task_id = Task::query()->create($list)->id;
self::setRedis($task_id, $list); # old
# self::setRedis($task_id, $list);
# 改成直接提交到自动化任务队列
$list = [
'prompt' => $prompt,
'user_id' => $user_id,
'task_id' => $task_id,
'prompt_img' => json_encode($prompt_img),
'type' => $type,
'callback' => config('common.prompt_callback'),
'policy' => config('common.policy_callback'),
];
$prompt_id = promptTask::query()->create($list)->id;
$list['prompt_id'] = $prompt_id;
# 插入redis,这是自动化的任务
Redis::rpush('midjourney_prompt', json_encode($list));
# 返回任务id # 返回任务id
return [ return [
'task_id' => $task_id, 'task_id' => $task_id,
'prompt_num' => 2 'prompt_num' => self::GPT_prompt_num,
]; ];
} }
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment