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:
2026-01-26 15:14:51 +08:00
co-authored by Claude
parent 114ae76abb
commit 6c65cb7cc0
2 changed files with 91 additions and 0 deletions
@@ -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 }
)
}
}
+15
View File
@@ -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>