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),
|
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
|
* 检查任务去重 Schema
|
||||||
* 用于在创建任务前检查 URL 是否已存在
|
* 用于在创建任务前检查 URL 是否已存在
|
||||||
@@ -125,3 +139,4 @@ export type TaskStatus = z.infer<typeof TaskStatusEnum>
|
|||||||
export type CreateDiscoveryTask = z.infer<typeof CreateDiscoveryTaskSchema>
|
export type CreateDiscoveryTask = z.infer<typeof CreateDiscoveryTaskSchema>
|
||||||
export type UpdateDiscoveryTask = z.infer<typeof UpdateDiscoveryTaskSchema>
|
export type UpdateDiscoveryTask = z.infer<typeof UpdateDiscoveryTaskSchema>
|
||||||
export type GetDiscoveryTasksQuery = z.infer<typeof GetDiscoveryTasksQuerySchema>
|
export type GetDiscoveryTasksQuery = z.infer<typeof GetDiscoveryTasksQuerySchema>
|
||||||
|
export type BatchResetTasks = z.infer<typeof BatchResetTasksSchema>
|
||||||
|
|||||||
Reference in New Issue
Block a user