mirror of
https://github.com/ajayyy/SponsorBlockServer.git
synced 2025-12-16 08:26:59 +03:00
Add locks to different write operations
This commit is contained in:
37
src/utils/redisLock.ts
Normal file
37
src/utils/redisLock.ts
Normal file
@@ -0,0 +1,37 @@
|
||||
import redis from "../utils/redis";
|
||||
import { Logger } from "./logger";
|
||||
|
||||
const defaultTimeout = 5000;
|
||||
|
||||
export type AcquiredLock = {
|
||||
status: false
|
||||
} | {
|
||||
status: true;
|
||||
unlock: () => void;
|
||||
};
|
||||
|
||||
export async function acquireLock(key: string, timeout = defaultTimeout): Promise<AcquiredLock> {
|
||||
try {
|
||||
const result = await redis.set(key, "1", {
|
||||
PX: timeout,
|
||||
NX: true
|
||||
});
|
||||
|
||||
if (result) {
|
||||
return {
|
||||
status: true,
|
||||
unlock: () => void redis.del(key).catch((err) => Logger.error(err))
|
||||
};
|
||||
} else {
|
||||
return {
|
||||
status: false
|
||||
};
|
||||
}
|
||||
} catch (e) {
|
||||
Logger.error(e as string);
|
||||
}
|
||||
|
||||
return {
|
||||
status: false
|
||||
};
|
||||
}
|
||||
Reference in New Issue
Block a user