58 lines
2.2 KiB
TypeScript
58 lines
2.2 KiB
TypeScript
import { Worker } from "bullmq"
|
|
|
|
import { config } from "./config"
|
|
import { judgeQueueName, type JudgeJobData } from "./judge/job"
|
|
import { failAbandonedSubmission, judgeSubmission } from "./judge/run"
|
|
import { flowchartQueueName, type FlowchartJobData } from "./flowchart/job"
|
|
import { evaluateFlowchart } from "./flowchart/run"
|
|
import { createBlockingRedis } from "./redis"
|
|
|
|
const worker = new Worker<JudgeJobData>(judgeQueueName, async (job) => judgeSubmission(job.data), {
|
|
connection: createBlockingRedis(),
|
|
concurrency: config.judgeConcurrency,
|
|
})
|
|
|
|
const flowchartWorker = new Worker<FlowchartJobData>(
|
|
flowchartQueueName,
|
|
// attemptsMade 是「此前已经失败过几次」，当前这次还没计进去，
|
|
// 所以最后一次尝试的判据是 attemptsMade + 1 >= attempts
|
|
async (job) =>
|
|
evaluateFlowchart(job.data, {
|
|
isFinalAttempt: job.attemptsMade + 1 >= (job.opts.attempts ?? 1),
|
|
}),
|
|
{ connection: createBlockingRedis(), concurrency: 2 },
|
|
)
|
|
|
|
worker.on("ready", () => {
|
|
console.log(`Judge worker ready (concurrency=${config.judgeConcurrency})`)
|
|
})
|
|
worker.on("failed", async (job, error) => {
|
|
console.error(`Judge job ${job?.id ?? "unknown"} failed`, error)
|
|
// 队列没配 attempts，失败即终局；worker 被杀掉那种 BullMQ 走完 stalled 重试也会
|
|
// 落到这里。不在这里写一个终态，提交就永远停在「等待评分」，没有任何人会再管它。
|
|
const submissionId = job?.data.submissionId
|
|
if (!submissionId) return
|
|
try {
|
|
await failAbandonedSubmission(submissionId, error)
|
|
} catch (markError) {
|
|
console.error(`Failed to mark submission ${submissionId} as system error`, markError)
|
|
}
|
|
})
|
|
worker.on("error", (error) => {
|
|
console.error("Judge worker error", error)
|
|
})
|
|
flowchartWorker.on("ready", () => console.log("Flowchart worker ready (concurrency=2)"))
|
|
flowchartWorker.on("failed", (job, error) =>
|
|
console.error(`Flowchart job ${job?.id ?? "unknown"} failed`, error),
|
|
)
|
|
flowchartWorker.on("error", (error) => console.error("Flowchart worker error", error))
|
|
|
|
async function shutdown() {
|
|
await worker.close()
|
|
await flowchartWorker.close()
|
|
process.exit(0)
|
|
}
|
|
|
|
process.on("SIGINT", shutdown)
|
|
process.on("SIGTERM", shutdown)
|