69 lines
4.0 KiB
TypeScript
69 lines
4.0 KiB
TypeScript
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:')))
|
|
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) // 筛选出指定用户的任务
|
|
if (bot) {
|
|
tasks = tasks.filter((task: any) => task.model === tasks[0].model) // 筛选出首位model的任务
|
|
tasks.forEach(async (task: any) => {
|
|
task.status = 'processing'
|
|
await useStorage().setItem(`task:${task.tid}`, task)
|
|
})
|
|
}
|
|
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.tid}`, task)
|
|
})
|
|
return tasks
|
|
}
|
|
|
|
// 添加任务到队列(生成一个32位随机字符串作为任务ID)
|
|
if (event.node.req.method === 'POST') {
|
|
const body = await readBody(event) // 读取body数据到
|
|
body.tid = Math.random().toString(36).substring(2, 34) // 生成随机任务ID
|
|
body.uid = 1234567890 // 用户ID
|
|
body.createdAt = new Date().getTime() // 设定创建时间
|
|
// 验证需要输入的参数
|
|
body.model = 'SD2' // 模型
|
|
body.ckpt = '768-v-ema' // 风格
|
|
body.prompt = '猫猫' // 提示词
|
|
body.remove = '狗狗' // 排除词
|
|
body.number = 1 // 生成数量
|
|
body.w = 512 // 图片宽度
|
|
body.h = 512 // 图片高度
|
|
body.seed = 0 // 随机种子
|
|
body.sampler = 'pndm' // 扩散采样器
|
|
body.prompt_guidance = 0.5 // 提示词权重
|
|
body.quality_details = 25 // 质量和细节(步数)
|
|
// 输出参数
|
|
body.status = 'waiting' // 任务状态
|
|
body.progress = 0.00 // 任务进度
|
|
body.image = '' // 图片数据
|
|
body.data = null // 生成结果
|
|
console.log('task:', body)
|
|
await useStorage().setItem(`task:${body.tid}`, body) // 保存任务到内存
|
|
// 筛选出所有的任务(并打印)
|
|
let tasks = await useStorage().getKeys().then((keys: any[]) => keys.filter(key => key.startsWith('task:')))
|
|
console.log('tasks:', tasks)
|
|
// 如果无需排队，直接返回结果, 否则返回任务排队状态(预估等待时间)
|
|
return body
|
|
}
|
|
|
|
})
|