Mirror/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2025-12-12 06:27:10 +03:00
Code Issues Packages Projects Releases Wiki Activity

Count active postgres and redis requests

Browse Source
This commit is contained in:
Ajay
2022-10-26 01:31:29 -04:00
parent 0ced78d798
commit 94eb37cb1c
3 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/utils/redis.ts
View File

@@ -27,6 +27,7 @@ let exportClient: RedisSB = {
let lastClientFail = 0;
let lastReadFail = 0;
let activeRequests = 0;
if (config.redis?.enabled) {
Logger.info("Connected to redis");
@@ -40,10 +41,13 @@ if (config.redis?.enabled) {
const get = client.get.bind(client);
const getRead = readClient?.get?.bind(readClient);
exportClient.get = (key) => new Promise((resolve, reject) => {
activeRequests++;
const timeout = config.redis.getTimeout ? setTimeout(() => reject(), config.redis.getTimeout) : null;
const chosenGet = pickChoice(get, getRead);
chosenGet(key).then((reply) => {
if (timeout !== null) clearTimeout(timeout);
activeRequests--;
resolve(reply);
}).catch((err) => {
if (chosenGet === get) {
@@ -52,6 +56,7 @@ if (config.redis?.enabled) {
lastReadFail = Date.now();
}
activeRequests--;
reject(err);
});
});
@@ -91,4 +96,8 @@ function pickChoice<T>(client: T, readClient: T): T {
}
}
export function getRedisActiveRequests(): number {
return activeRequests;
}
export default exportClient;