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

improve: cache result for adjustSnapshotTime

Browse Source
This commit is contained in:
alikia2x (寒寒) 2025-03-27 01:48:10 +08:00
parent c80b047e0c
commit d5c278ae06
Signed by: alikia2x
GPG Key ID: 56209E0CCD8420C6
1 changed files with 12 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
lib/db/snapshotSchedule.ts
View File

@ -9,6 +9,7 @@ import { Redis } from "ioredis";
const WINDOW_SIZE = 2880;
const REDIS_KEY = "cvsa:snapshot_window_counts";
let lastAvailableWindow: { offset: number; count: number } | null = null;
function getCurrentWindowIndex(): number {
const now = new Date();
@ -43,6 +44,8 @@ export async function refreshSnapshotWindowCounts(client: Client, redisClient: R
await redisClient.hset(REDIS_KEY, offset.toString(), Number(row.count));
}
}
lastAvailableWindow = null;
}
export async function initSnapshotWindowCounts(client: Client, redisClient: Redis) {
@ -57,10 +60,6 @@ async function getWindowCount(redisClient: Redis, offset: number): Promise<numbe
return count ? parseInt(count, 10) : 0;
}
async function updateWindowCount(redisClient: Redis, offset: number, increment: number): Promise<void> {
await redisClient.hincrby(REDIS_KEY, offset.toString(), increment);
}
export async function snapshotScheduleExists(client: Client, id: number) {
const res = await client.queryObject<{ id: number }>(
`SELECT id FROM snapshot_schedule WHERE id = $1`,
@ -203,17 +202,23 @@ export async function adjustSnapshotTime(
redisClient: Redis,
): Promise<Date> {
const currentWindow = getCurrentWindowIndex();
const targetOffset = Math.floor((expectedStartTime.getTime() - Date.now()) / (5 * MINUTE));
let initialOffset = 0;
if (lastAvailableWindow && lastAvailableWindow.count < allowedCounts) {
initialOffset = lastAvailableWindow.offset;
}
let timePerIteration = 0;
const t = performance.now();
for (let i = 0; i < WINDOW_SIZE; i++) {
for (let i = initialOffset; i < WINDOW_SIZE; i++) {
const offset = (currentWindow + targetOffset + i) % WINDOW_SIZE;
const count = await getWindowCount(redisClient, offset);
if (count < allowedCounts) {
await updateWindowCount(redisClient, offset, 1);
const newCount = await redisClient.hincrby(REDIS_KEY, offset.toString(), 1);
lastAvailableWindow = { offset, count: newCount };
const startPoint = new Date();
startPoint.setHours(0, 0, 0, 0);