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

Switch postgres instances if there is a failure

Browse Source
This commit is contained in:
Ajay
2022-06-25 12:02:01 -04:00
parent 1a232600a1
commit 08003bc2f2
1 changed files with 15 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/databases/Postgres.ts
View File

@@ -27,20 +27,26 @@ export interface DatabaseConfig {
export class Postgres implements IDatabase {
private pool: Pool;
private lastPoolFail: number = 0;
private poolRead: Pool;
private lastPoolReadFail: number = 0;
constructor(private config: DatabaseConfig) {}
async init(): Promise<void> {
this.pool = new Pool(this.config.postgres);
const errorHandler = (err: Error) => {
this.pool.on("error", (err) => {
Logger.error(err.stack);
};
this.pool.on("error", errorHandler);
this.lastPoolFail = Date.now();
});
if (this.config.postgresReadOnly) {
this.poolRead = new Pool(this.config.postgresReadOnly);
this.poolRead.on("error", errorHandler);
this.poolRead.on("error", (err) => {
Logger.error(err.stack);
this.lastPoolReadFail = Date.now();
});
}
if (!this.config.readOnly) {
@@ -104,8 +110,11 @@ export class Postgres implements IDatabase {
}
private getClient(type: string): Promise<PoolClient> {
if (this.poolRead && (type === "get" || type === "all")
&& Math.random() > 1 / (this.config.postgresReadOnly.weight + 1)) {
const readAvailable = this.poolRead && (type === "get" || type === "all");
const ignroreReadDueToFailure = this.lastPoolReadFail < Date.now() - 1000 * 30;
const readDueToFailure = this.lastPoolFail < Date.now() - 1000 * 30;
if (readAvailable && !ignroreReadDueToFailure && (readDueToFailure ||
Math.random() > 1 / (this.config.postgresReadOnly.weight + 1))) {
return this.poolRead.connect();
} else {
return this.pool.connect();