Commit 1a11de42 by haojie

1

parent e1ac83ab
......@@ -14,6 +14,8 @@ class TaskService
public const TYPE_IMG_TO_IMG = 1; // 图片转图片
public const TYPE_TEXT_TO_IMG = 2; // 文字转图片
public const GPT_prompt_num = 1; // 指令数量
public function submit($user_id, array $data = [])
{
# task表新增一条记录
......@@ -45,15 +47,30 @@ public function submit($user_id, array $data = [])
'prompt' => $prompt,
'prompt_img' => json_encode($prompt_img),
'user_id' => $user_id,
'prompt_num' => 2,
'prompt_num' => self::GPT_prompt_num,
];
# 保存任务
$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
return [
'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