feat: 添加批量重置探索任务 API 端点
新增 POST /api/discovery/tasks/batch-reset 接口,用于 n8n 探索流程失败后批量重置任务状态。 功能: - 支持按任务 ID 列表重置 - 支持按状态筛选重置(默认 IN_PROGRESS 和 FAILED) - 重置时清除 startedAt、completedAt、errorMessage Co-Authored-By: Claude <noreply@anthropic.com>
This commit is contained in:
@@ -0,0 +1,76 @@
|
||||
import { NextRequest, NextResponse } from 'next/server'
|
||||
import crypto from 'crypto'
|
||||
import { prisma } from '@/lib/prisma'
|
||||
import { BatchResetTasksSchema } from '@/lib/validations'
|
||||
|
||||
/**
|
||||
* POST /api/discovery/tasks/batch-reset
|
||||
* 批量重置任务状态为 PENDING
|
||||
*
|
||||
* 用于 n8n 探索流程失败后重置任务,支持两种模式:
|
||||
* 1. 按任务 ID 列表重置:{ taskIds: ["id1", "id2"] }
|
||||
* 2. 按状态筛选重置:{ statuses: ["IN_PROGRESS", "FAILED"] }
|
||||
* 不传 statuses 则默认重置 IN_PROGRESS 和 FAILED
|
||||
*/
|
||||
export async function POST(request: NextRequest) {
|
||||
try {
|
||||
const body = await request.json()
|
||||
const validation = BatchResetTasksSchema.safeParse(body)
|
||||
|
||||
if (!validation.success) {
|
||||
return NextResponse.json(
|
||||
{
|
||||
success: false,
|
||||
error: 'Validation error',
|
||||
details: validation.error.errors.map((e) => e.message),
|
||||
},
|
||||
{ status: 400 }
|
||||
)
|
||||
}
|
||||
|
||||
const { apiKey, taskIds, statuses } = validation.data
|
||||
|
||||
// 验证API密钥
|
||||
const validApiKey = process.env.WEBHOOK_API_KEY
|
||||
if (
|
||||
!validApiKey ||
|
||||
!crypto.timingSafeEqual(Buffer.from(apiKey), Buffer.from(validApiKey))
|
||||
) {
|
||||
return NextResponse.json(
|
||||
{ success: false, error: 'Unauthorized' },
|
||||
{ status: 401 }
|
||||
)
|
||||
}
|
||||
|
||||
// 构建查询条件
|
||||
const where = taskIds
|
||||
? { id: { in: taskIds } }
|
||||
: { status: { in: statuses || ['IN_PROGRESS', 'FAILED'] } }
|
||||
|
||||
// 批量更新任务状态
|
||||
const result = await prisma.projectDiscoveryTask.updateMany({
|
||||
where,
|
||||
data: {
|
||||
status: 'PENDING',
|
||||
startedAt: null,
|
||||
completedAt: null,
|
||||
errorMessage: null,
|
||||
},
|
||||
})
|
||||
|
||||
console.warn(
|
||||
`[Discovery] Batch reset ${result.count} tasks to PENDING. Condition: ${JSON.stringify(where)}`
|
||||
)
|
||||
|
||||
return NextResponse.json({
|
||||
success: true,
|
||||
reset: result.count,
|
||||
})
|
||||
} catch (error) {
|
||||
console.error('[Discovery] Error batch resetting tasks:', error)
|
||||
return NextResponse.json(
|
||||
{ success: false, error: 'Internal server error' },
|
||||
{ status: 500 }
|
||||
)
|
||||
}
|
||||
}
|
||||
@@ -91,6 +91,20 @@ export const GetDiscoveryTasksQuerySchema = z.object({
|
||||
offset: z.coerce.number().int().nonnegative().default(0),
|
||||
})
|
||||
|
||||
/**
|
||||
* 批量重置任务 Schema
|
||||
* 支持两种模式:按 ID 列表重置 或 按状态筛选重置
|
||||
*/
|
||||
export const BatchResetTasksSchema = WebhookAuthSchema.extend({
|
||||
// 模式1: 指定任务 ID 列表
|
||||
taskIds: z.array(z.string()).optional(),
|
||||
// 模式2: 按状态筛选(不传则默认重置 IN_PROGRESS 和 FAILED)
|
||||
statuses: z.array(TaskStatusEnum).optional(),
|
||||
}).refine(
|
||||
(data) => data.taskIds || data.statuses,
|
||||
{ message: '必须提供 taskIds 或 statuses 之一' }
|
||||
)
|
||||
|
||||
/**
|
||||
* 检查任务去重 Schema
|
||||
* 用于在创建任务前检查 URL 是否已存在
|
||||
@@ -125,3 +139,4 @@ export type TaskStatus = z.infer<typeof TaskStatusEnum>
|
||||
export type CreateDiscoveryTask = z.infer<typeof CreateDiscoveryTaskSchema>
|
||||
export type UpdateDiscoveryTask = z.infer<typeof UpdateDiscoveryTaskSchema>
|
||||
export type GetDiscoveryTasksQuery = z.infer<typeof GetDiscoveryTasksQuerySchema>
|
||||
export type BatchResetTasks = z.infer<typeof BatchResetTasksSchema>
|
||||
|
||||
Reference in New Issue
Block a user