smartclickhouse/ts/smartclickhouse.classes.smartclickhouse.ts

import * as plugins from './smartclickhouse.plugins.js';
import { TimeDataTable } from './smartclickhouse.classes.timedatatable.js';

export interface IClickhouseConstructorOptions {
  url: string;
  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
   */
  public async start(dropOld = false) {
    console.log(`Connecting to default database first.`);
    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
    const defaultClient = new plugins.clickhouse.ClickHouseClient({
      ...defaultOptions,
      database: 'default',
    });
    console.log(`Create database ${this.options.database}, if it does not exist...`);
    if (dropOld) {
      await defaultClient.queryPromise(`DROP DATABASE IF EXISTS ${this.options.database}`);
    }
    await defaultClient.queryPromise(`CREATE DATABASE IF NOT EXISTS ${this.options.database}`);

    console.log(`Ensured database. Now connecting to wanted database: ${this.options.database}`);
    this.clickhouseClient = new plugins.clickhouse.ClickHouseClient({
      ...defaultOptions
    });
    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);
    return newTable;
  }
}