77 lines
2.1 KiB
TypeScript
77 lines
2.1 KiB
TypeScript
|
import { ConsoleHelper } from "../../ConsoleHelper";
|
||
|
|
import { createPool, Pool } from "mysql2";
|
||
|
|
|
||
|
|
export class Database {
|
||
|
|
private connectionPool:Pool;
|
||
|
|
private static readonly CONNECTION_LIMIT = 128;
|
||
|
|
|
||
|
|
public constructor(databaseAddress:string, databasePort:number = 3306, databaseUsername:string, databasePassword:string, databaseName:string, connectedCallback:Function) {
|
||
|
|
this.connectionPool = createPool({
|
||
|
|
connectionLimit: Database.CONNECTION_LIMIT,
|
||
|
|
host: databaseAddress,
|
||
|
|
port: databasePort,
|
||
|
|
user: databaseUsername,
|
||
|
|
password: databasePassword,
|
||
|
|
database: databaseName
|
||
|
|
});
|
||
|
|
|
||
|
|
const classCreationTime:number = Date.now();
|
||
|
|
const connectionCheckInterval = setInterval(() => {
|
||
|
|
this.query("SELECT name FROM osu_info LIMIT 1")
|
||
|
|
.then(data => {
|
||
|
|
ConsoleHelper.printBancho(`Connected to database. Took ${Date.now() - classCreationTime}ms`);
|
||
|
|
clearInterval(connectionCheckInterval);
|
||
|
|
|
||
|
|
connectedCallback();
|
||
|
|
})
|
||
|
|
.catch(err => {});
|
||
|
|
}, 17); // Roughly 6 times per sec
|
||
|
|
}
|
||
|
|
|
||
|
|
public query(query = "", data?:Array<any>) {
|
||
|
|
const limited = query.includes("LIMIT 1");
|
||
|
|
|
||
|
|
return new Promise((resolve, reject) => {
|
||
|
|
this.connectionPool.getConnection((err, connection) => {
|
||
|
|
if (err) {
|
||
|
|
reject(err);
|
||
|
|
try {
|
||
|
|
connection.release();
|
||
|
|
} catch (e) {
|
||
|
|
ConsoleHelper.printError("Failed to release mysql connection\n" + err);
|
||
|
|
}
|
||
|
|
} else {
|
||
|
|
// Use old query
|
||
|
|
if (data == null) {
|
||
|
|
connection.query(query, (err, data) => {
|
||
|
|
if (err) {
|
||
|
|
reject(err);
|
||
|
|
connection.release();
|
||
|
|
} else {
|
||
|
|
dataReceived(resolve, data, limited);
|
||
|
|
connection.release();
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
// Use new prepared statements w/ placeholders
|
||
|
|
else {
|
||
|
|
connection.execute(query, data, (err, data) => {
|
||
|
|
if (err) {
|
||
|
|
reject(err);
|
||
|
|
connection.release();
|
||
|
|
} else {
|
||
|
|
dataReceived(resolve, data, limited);
|
||
|
|
connection.release();
|
||
|
|
}
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
});
|
||
|
|
});
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
function dataReceived(resolveCallback:(value:unknown) => void, data:any, limited:boolean = false) : void {
|
||
|
|
if (limited) resolveCallback(data[0]);
|
||
|
|
else resolveCallback(data);
|
||
|
|
}
|