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

update: return when meeting non-0 video status code in snapshotVideo job

Browse Source
This commit is contained in:
alikia2x (寒寒) 2025-03-26 23:25:07 +08:00
parent 5450c17e13
commit afee7f58bf
Signed by: alikia2x
GPG Key ID: 56209E0CCD8420C6
2 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
lib/db/allData.ts
View File

@ -75,3 +75,13 @@ export async function setBiliVideoStatus(client: Client, aid: number, status: nu
[status, aid],
);
}
export async function getBiliVideoStatus(client: Client, aid: number) {
const queryResult = await client.queryObject<{ status: number }>(
`SELECT status FROM bilibili_metadata WHERE aid = $1`,
[aid],
);
const rows = queryResult.rows;
if (rows.length === 0) return 0;
return rows[0].status;
}

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

@ -19,7 +19,7 @@ import logger from "lib/log/logger.ts";
import { SnapshotQueue } from "lib/mq/index.ts";
import { insertVideoSnapshot } from "lib/mq/task/getVideoStats.ts";
import { NetSchedulerError } from "lib/mq/scheduler.ts";
import { setBiliVideoStatus } from "lib/db/allData.ts";
import { getBiliVideoStatus, setBiliVideoStatus } from "lib/db/allData.ts";
import { truncate } from "lib/utils/truncate.ts";
import { lockManager } from "lib/mq/lockManager.ts";
import { getSongsPublihsedAt } from "lib/db/songs.ts";
@ -197,13 +197,15 @@ export const takeSnapshotForVideoWorker = async (job: Job) => {
if (!exists) {
return;
}
const status = await getBiliVideoStatus(client, aid);
if (status !== 0) return `REFUSE_WORKING_BILI_STATUS_${status}`
try {
await setSnapshotStatus(client, id, "processing");
const stat = await insertVideoSnapshot(client, aid, task);
if (typeof stat === "number") {
await setBiliVideoStatus(client, aid, stat);
await setSnapshotStatus(client, id, "completed");
return `BILI_STATUS_${stat}`;
return `GET_BILI_STATUS_${stat}`;
}
await setSnapshotStatus(client, id, "completed");
if (type === "normal") {