smartclickhouse/ts/smartclickhouse.classes.smartclickhouse.ts

62 lines
2.2 KiB
TypeScript
Raw Permalink 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-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;
password?: string;
}
export class SmartClickHouseDb {
public options: IClickhouseConstructorOptions;
public clickhouseClient: plugins.clickhouse.ClickHouseClient;
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
const defaultOptions: {[keyArg: string]: string} = {};
// the protocol, url and host
const parsedUrl = plugins.smarturl.Smarturl.createFromUrl(this.options.url);
parsedUrl.protocol === 'https' ? defaultOptions.protocol = plugins.clickhouse.ClickHouseConnectionProtocol.HTTPS : null;
parsedUrl.protocol === 'http' ? defaultOptions.protocol = plugins.clickhouse.ClickHouseConnectionProtocol.HTTP : null;
defaultOptions.host = parsedUrl.hostname;
defaultOptions.port = parsedUrl.port;
// the database
defaultOptions.database = this.options.database;
// the password
this.options.password ? defaultOptions.password = this.options.password : null;
// lets connect
2022-03-02 15:35:20 +00:00
const defaultClient = new plugins.clickhouse.ClickHouseClient({
2022-07-27 06:29:59 +00:00
...defaultOptions,
2022-03-07 14:49:47 +00:00
database: 'default',
2022-03-02 15:35:20 +00:00
});
console.log(`Create database ${this.options.database}, if it does not exist...`);
2022-03-07 14:49:47 +00:00
if (dropOld) {
await defaultClient.queryPromise(`DROP DATABASE IF EXISTS ${this.options.database}`);
}
2022-03-02 15:35:20 +00:00
await defaultClient.queryPromise(`CREATE DATABASE IF NOT EXISTS ${this.options.database}`);
2022-03-07 14:49:47 +00:00
console.log(`Ensured database. Now connecting to wanted database: ${this.options.database}`);
2022-03-02 15:35:20 +00:00
this.clickhouseClient = new plugins.clickhouse.ClickHouseClient({
2022-07-27 06:29:59 +00:00
...defaultOptions
2022-03-02 15:35:20 +00:00
});
console.log(`trying to ping database...`);
const result = await this.clickhouseClient.ping();
console.log(`Ping successfull?: ${result}`);
}
/**
* 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
}