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

Support schema upgrading with postgres

Browse Source
This commit is contained in:
Ajay Ramachandran
2021-03-01 22:20:44 -05:00
parent 46b42da5bd
commit 88855ab695
3 changed files with 50 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
databases/_sponsorTimes.db.sql
View File

@@ -33,4 +33,7 @@ CREATE TABLE IF NOT EXISTS "config" (
CREATE INDEX IF NOT EXISTS sponsorTimes_videoID on sponsorTimes(videoID);
CREATE INDEX IF NOT EXISTS sponsorTimes_UUID on sponsorTimes(UUID);
CREATE EXTENSION IF NOT EXISTS pgcrypto; --!sqlite-ignore
COMMIT;

40
src/databases/Postgres.ts
View File

@@ -2,6 +2,8 @@ import { Logger } from '../utils/logger';
import { IDatabase, QueryType } from './IDatabase';
import { Pool } from 'pg';
import fs from "fs";
export class Mysql implements IDatabase {
private pool: Pool;
@@ -9,10 +11,25 @@ export class Mysql implements IDatabase {
init(): void {
this.pool = new Pool();
if (!this.config.readOnly) {
// Upgrade database if required
this.upgradeDB(this.config.fileNamePrefix, this.config.dbSchemaFolder);
}
}
async prepare(type: QueryType, query: string, params?: any[]) {
Logger.debug(`prepare (postgres): type: ${type}, query: ${query}, params: ${params}`);
// Convert query to use numbered parameters
let count = 1;
for (let char = 0; char < query.length; char++) {
if (query.charAt(char) === '?') {
query = query.slice(0, char) + "$" + count + query.slice(char + 1);
count++;
}
}
const queryResult = await this.pool.query(query, params);
switch (type) {
@@ -27,5 +44,28 @@ export class Mysql implements IDatabase {
}
}
}
private async upgradeDB(fileNamePrefix: string, schemaFolder: string) {
const versionCodeInfo = await this.pool.query("SELECT value FROM config WHERE key = 'version'");
let versionCode = versionCodeInfo ? versionCodeInfo.rows[0].value : 0;
let path = schemaFolder + "/_upgrade_" + fileNamePrefix + "_" + (parseInt(versionCode) + 1) + ".sql";
Logger.debug('db update: trying ' + path);
while (fs.existsSync(path)) {
Logger.debug('db update: updating ' + path);
await this.pool.query(this.processUpgradeQuery(fs.readFileSync(path).toString()));
versionCode = (await this.pool.query("SELECT value FROM config WHERE key = 'version'"))?.rows[0]?.value;
path = schemaFolder + "/_upgrade_" + fileNamePrefix + "_" + (parseInt(versionCode) + 1) + ".sql";
Logger.debug('db update: trying ' + path);
}
Logger.debug('db update: no file ' + path);
}
private processUpgradeQuery(query: string): string {
let result = query.replace(/sha256\((.*?)\)/gm, "digest($1, 'sha256')");
return result;
}
}

8
src/databases/Sqlite.ts
View File

@@ -84,7 +84,7 @@ export class Sqlite implements IDatabase {
Logger.debug('db update: trying ' + path);
while (fs.existsSync(path)) {
Logger.debug('db update: updating ' + path);
db.exec(fs.readFileSync(path).toString());
db.exec(this.processUpgradeQuery(fs.readFileSync(path).toString()));
versionCode = db.prepare("SELECT value FROM config WHERE key = ?").get("version").value;
path = schemaFolder + "/_upgrade_" + fileNamePrefix + "_" + (parseInt(versionCode) + 1) + ".sql";
@@ -92,6 +92,12 @@ export class Sqlite implements IDatabase {
}
Logger.debug('db update: no file ' + path);
}
private static processUpgradeQuery(query: string): string {
let result = query.replace(/^.*--!sqlite-ignore/gm, "");
return result;
}
}
export interface SqliteConfig {