JSON 类型

This commit is contained in:
2023-02-18 20:47:21 +08:00
parent d5bae2d772
commit 1a925bc129
2 changed files with 5 additions and 17 deletions

View File

@@ -20,28 +20,16 @@ export default defineEventHandler(async event => {
return tasks
}
// 修改任务状态(仅限bot访问)
if (event.node.req.method === 'PATCH') {
const { ids } = await readBody(event)
// 取出ids对应的任务
const tasks = await Promise.all(ids.map(async (id: any) => await useStorage().getItem(`task:${id}`)))
// 修改任务状态
tasks.forEach(async (task: any) => {
task.status = 'finished'
await useStorage().setItem(`task:${task.id}`, task)
})
return tasks
}
// 添加任务到队列(生成一个32位随机字符串作为任务ID)
if (event.node.req.method === 'POST') {
const task = await readBody(event) // 读取body数据到
task.id = Math.random().toString(36).substring(2, 34) // 生成随机任务ID
task.id = Math.random().toString(36).substring(2, 34) // 生成随机任务ID
task.uid = 1234567890 // 用户ID
task.createdAt = new Date().getTime() // 设定创建时间
// 验证需要输入的参数
task.model = 'SD2' // 模型
task.ckpt = '768-v-ema' // 风格
task.image = '' // 图片数据(img2img)
task.prompt = '猫猫' // 提示词
task.remove = '狗狗' // 排除词
task.number = 1 // 生成数量
@@ -54,7 +42,6 @@ export default defineEventHandler(async event => {
// 输出参数
task.status = 'waiting' // 任务状态
task.progress = 0.00 // 任务进度
task.image = '' // 图片数据
task.data = null // 生成结果
console.log('task:', task)
await useStorage().setItem(`task:${task.id}`, task) // 保存任务到内存