This commit is contained in:
2023-02-14 02:05:11 +08:00
parent 93562752b5
commit 8e581e3a56

View File

@@ -4,9 +4,9 @@ export default defineEventHandler(async event => {
if (event.node.req.method === 'GET') {
let {status, model, uid, bot} = getQuery(event)
let tasks = await useStorage().getKeys().then((keys: any[]) => keys.filter(key => key.startsWith('task:')))
if (tasks.length === 0) return [] // 如果没有任务,直接返回空数组
tasks = await Promise.all(tasks.map(async (key: any) => await useStorage().getItem(key))) // 读取所有任务的数据
tasks.sort((a: any, b: any) => a.createdAt - b.createdAt) // 按照创建时间排序
if (tasks.length === 0) return [] // 如果没有任务,直接返回空数组
if (status) tasks = tasks.filter((task: any) => task.status === status) // 筛选出指定状态的任务
if (model) tasks = tasks.filter((task: any) => task.model === model) // 筛选出指定model的任务
if (uid) tasks = tasks.filter((task: any) => task.uid === uid) // 筛选出指定用户的任务
@@ -37,6 +37,7 @@ export default defineEventHandler(async event => {
if (event.node.req.method === 'POST') {
event.context.body = await readBody(event) // 读取body数据到
event.context.body.tid = Math.random().toString(36).substring(2, 34)
event.context.body.uid = 1234567890 // 默认用户ID
event.context.body.status = 'waiting' // 默认当前状态
event.context.body.createdAt = new Date().getTime() // 设定创建时间
await useStorage().setItem(`task:${event.context.body.tid}`, event.context.body)