smartclickhouse/ts/smartclickhouse.classes.smartclickhouse.ts

68 lines
2.1 KiB
TypeScript
Raw Normal View History

2022-03-14 13:29:23 +00:00
import * as plugins from './smartclickhouse.plugins.js';
import { TimeDataTable } from './smartclickhouse.classes.timedatatable.js';
2022-07-28 14:53:07 +00:00
import { ClickhouseHttpClient } from './smartclickhouse.classes.httpclient.js';
2022-03-02 15:35:20 +00:00
export interface IClickhouseConstructorOptions {
2022-07-27 06:29:59 +00:00
url: string;
2022-03-02 15:35:20 +00:00
database: string;
2022-07-28 15:36:08 +00:00
username?: string;
2022-03-02 15:35:20 +00:00
password?: string;
2022-07-30 16:03:17 +00:00
/**
* allow services to exit when waiting for clickhouse startup
* this allows to leave the lifecycle flow to other processes
* like a listening server.
*/
unref?: boolean;
2022-03-02 15:35:20 +00:00
}
export class SmartClickHouseDb {
public options: IClickhouseConstructorOptions;
2024-06-14 14:33:00 +00:00
public clickhouseHttpClient: ClickhouseHttpClient;
2022-03-02 15:35:20 +00:00
constructor(optionsArg: IClickhouseConstructorOptions) {
this.options = optionsArg;
}
/**
* starts the connection to the Clickhouse db
*/
2022-03-07 14:49:47 +00:00
public async start(dropOld = false) {
console.log(`Connecting to default database first.`);
2022-07-27 06:29:59 +00:00
// lets connect
2024-06-14 14:33:00 +00:00
this.clickhouseHttpClient = await ClickhouseHttpClient.createAndStart(this.options);
2022-07-27 20:42:08 +00:00
await this.pingDatabaseUntilAvailable();
2022-03-02 15:35:20 +00:00
console.log(`Create database ${this.options.database}, if it does not exist...`);
2022-07-27 20:42:08 +00:00
await this.createDatabase(dropOld);
}
public async createDatabase(dropOld: boolean = false) {
if (dropOld) {
2024-06-14 14:33:00 +00:00
await this.clickhouseHttpClient.queryPromise(`DROP DATABASE IF EXISTS ${this.options.database}`);
2022-07-27 20:42:08 +00:00
}
2024-06-14 14:33:00 +00:00
await this.clickhouseHttpClient.queryPromise(
2022-08-05 11:31:11 +00:00
`CREATE DATABASE IF NOT EXISTS ${this.options.database}`
);
2022-07-27 20:42:08 +00:00
}
public async pingDatabaseUntilAvailable() {
let available = false;
2022-08-05 11:31:11 +00:00
while (!available) {
2024-06-14 14:33:00 +00:00
available = await this.clickhouseHttpClient.ping().catch((err) => {
2022-07-27 20:42:08 +00:00
return false;
});
if (!available) {
console.log(`NOT OK: tried pinging ${this.options.url}... Trying again in 5 seconds.`);
2022-07-30 16:03:17 +00:00
await plugins.smartdelay.delayFor(5000, null, this.options.unref);
2022-07-27 20:42:08 +00:00
}
}
2022-03-02 15:35:20 +00:00
}
/**
* gets a table
*/
public async getTable(tableName: string) {
const newTable = TimeDataTable.getTable(this, tableName);
2022-03-07 14:49:47 +00:00
return newTable;
2022-03-02 15:35:20 +00:00
}
2022-03-07 14:49:47 +00:00
}