Repository navigation
Expand file tree
/
Copy pathquota.ts
More file actions
47 lines (47 loc) · 1.75 KB
/
Copy pathquota.ts
File metadata and controls
47 lines (47 loc) · 1.75 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
/** 단일 중계 프로세스에서 사용하는 재시작 보존 호출 상한. */
import { readFile, rename, writeFile } from 'node:fs/promises';
interface Ledger {
day: number;
minute: number;
dayCount: number;
minuteCount: number;
}
export async function createFileQuota(
path: string,
now: () => number = Date.now,
): Promise<() => Promise<boolean>> {
let ledger: Ledger = { day: 0, minute: 0, dayCount: 0, minuteCount: 0 };
try {
const value = JSON.parse(await readFile(path, 'utf8')) as Ledger;
if (
!value ||
![value.day, value.minute, value.dayCount, value.minuteCount].every(
(n) => Number.isSafeInteger(n) && n >= 0,
)
) {
throw new Error('Invalid quota ledger');
}
ledger = value;
} catch (error) {
if ((error as NodeJS.ErrnoException).code !== 'ENOENT') throw error;
}
let tail: Promise<unknown> = Promise.resolve();
return () => {
const task = tail.then(async () => {
const time = now();
const day = Math.max(ledger.day, Math.floor(time / 86400000));
const minute = Math.max(ledger.minute, Math.floor(time / 60000));
const dayCount = day === ledger.day ? ledger.dayCount : 0;
const minuteCount = minute === ledger.minute ? ledger.minuteCount : 0;
if (dayCount >= 3000 || minuteCount >= 30) return false;
const next = { day, minute, dayCount: dayCount + 1, minuteCount: minuteCount + 1 };
// 기록 실패를 재시도해도 프로세스 내에서 이미 소비한 예산은 되돌리지 않습니다.
ledger = next;
await writeFile(`${path}.tmp`, JSON.stringify(next), { mode: 0o600, flush: true });
await rename(`${path}.tmp`, path);
return true;
});
tail = task.catch(() => undefined);
return task;
};
}