alikia2x/cvsa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: prevent duplicate jobs added to queue

Browse Source
This commit is contained in:
alikia2x (寒寒) 2025-03-25 21:33:58 +08:00
parent b33fd790d1
commit 2d8e990bc9
Signed by: alikia2x
GPG Key ID: 56209E0CCD8420C6
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/mq/exec/snapshotTick.ts
View File

@ -10,6 +10,7 @@ import {
scheduleSnapshot, scheduleSnapshot,
setSnapshotStatus, setSnapshotStatus,
snapshotScheduleExists, snapshotScheduleExists,
videoHasProcessingSchedule,
} from "lib/db/snapshotSchedule.ts"; } from "lib/db/snapshotSchedule.ts";
import { Client } from "https://deno.land/x/postgres@v0.19.3/mod.ts"; import { Client } from "https://deno.land/x/postgres@v0.19.3/mod.ts";
import { HOUR, MINUTE, SECOND, WEEK } from "$std/datetime/constants.ts"; import { HOUR, MINUTE, SECOND, WEEK } from "$std/datetime/constants.ts";
@ -38,6 +39,9 @@ export const snapshotTickWorker = async (_job: Job) => {
try { try {
const schedules = await getSnapshotsInNextSecond(client); const schedules = await getSnapshotsInNextSecond(client);
for (const schedule of schedules) { for (const schedule of schedules) {
if (await videoHasProcessingSchedule(client, schedule.aid)) {
return `ALREADY_PROCESSING`;
}
let priority = 3; let priority = 3;
if (schedule.type && priorityMap[schedule.type]) { if (schedule.type && priorityMap[schedule.type]) {
priority = priorityMap[schedule.type]; priority = priorityMap[schedule.type];
@ -192,6 +196,7 @@ export const takeSnapshotForVideoWorker = async (job: Job) => {
return; return;
} }
try { try {
await setSnapshotStatus(client, id, "processing");
const stat = await insertVideoSnapshot(client, aid, task); const stat = await insertVideoSnapshot(client, aid, task);
if (typeof stat === "number") { if (typeof stat === "number") {
await setBiliVideoStatus(client, aid, stat); await setBiliVideoStatus(client, aid, stat);