external/SponsorBlockServer
1
0
Fork 0
mirror of https://github.com/ajayyy/SponsorBlockServer.git synced 2024-11-10 09:07:47 +01:00
Code Issues Projects Releases Packages Wiki Activity

Add timeout for all postgres requests

Browse source
This commit is contained in:
Ajay 2022-11-15 14:19:15 -05:00
parent 4139bf8f8c
commit 9e2e1343da
3 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
src/config.ts
View file

@ -77,7 +77,8 @@ addDefaults(config, {
max: 10,
idleTimeoutMillis: 10000,
maxTries: 3,
maxActiveRequests: 0
maxActiveRequests: 0,
timeout: 60000
},
postgresReadOnly: {
enabled: false,

1
src/databases/Postgres.ts
View file

@ -125,6 +125,7 @@ export class Postgres implements IDatabase {
pendingQueries.push(savePromiseState(lastPool.query({ text: query, values: params })));
const currentPromises = [...pendingQueries];
if (options.useReplica && maxTries() - tries > 1) currentPromises.push(savePromiseState(timeoutPomise(this.config.postgresReadOnly.readTimeout)));
else if (this.config.postgres.timeout) currentPromises.push(savePromiseState(timeoutPomise(this.config.postgres.timeout)));
const queryResult = await nextFulfilment(currentPromises);
this.activePostgresRequests--;

1
src/types/config.model.ts
View file

@ -21,6 +21,7 @@ export interface CustomPostgresConfig extends PoolConfig {
export interface CustomWritePostgresConfig extends CustomPostgresConfig {
maxActiveRequests: number;
timeout: number;
}
export interface CustomPostgresReadOnlyConfig extends CustomPostgresConfig {