es-entity/lib/handlers/Handler.ts

ORM framework for Node js based on Ecmascript 6 and Typescript.

import { Readable } from 'stream';
import * as bean from '../bean/index.js';
import * as sql from '../sql/index.js';

export default abstract class Handler {
  config: bean.IConnectionConfig;

  constructor(config: bean.IConnectionConfig) {
    this.config = config;
  }

  abstract init(): Promise<void>;

  prepareQuery(queryStmt: string | sql.Statement | sql.Statement[]) {
    let query: string;
    let dataArgs: any[] = [];
    if (Array.isArray(queryStmt)) {
      let tempQueries: string[] = [];
      queryStmt.forEach(a => {
        if (!(a instanceof sql.Statement)) throw new Error('Invalid Statement');
        tempQueries.push(a.eval(this));
        dataArgs.push(...a.args);
      });
      query = tempQueries.join('; ').concat(';');
    } else if (queryStmt instanceof sql.Statement) {
      query = queryStmt.eval(this);
      dataArgs.push(...queryStmt.args);
    } else {
      query = queryStmt;
    }
    return { query, dataArgs };
  }

  abstract run(query: string | sql.Statement | sql.Statement[], connection?: any): Promise<bean.ResultSet>;
  abstract stream(query: string | sql.Statement | sql.Statement[], connection?: any): Promise<Readable>;

  // Connetion manage functions
  abstract getConnection(): Promise<any>;
  abstract initTransaction(conn: any): Promise<void>;
  abstract commit(conn: any): Promise<void>;
  abstract rollback(conn: any): Promise<void>;
  abstract close(conn: any): Promise<void>;
  abstract end(): Promise<void>;

  // Comparison Operators
  eq(val0: string, val1: string): string {
    return `${val0} = ${val1}`;
  }
  neq(val0: string, val1: string): string {
    return `${val0} != ${val1}`;
  }
  lt(val0: string, val1: string): string {
    return `${val0} < ${val1}`;
  }
  gt(val0: string, val1: string): string {
    return `${val0} > ${val1}`;
  }
  lteq(val0: string, val1: string): string {
    return `${val0} <= ${val1}`;
  }
  gteq(val0: string, val1: string): string {
    return `${val0} >= ${val1}`;
  }

  // Logical Operators
  and(values: string[]): string {
    return values
      .filter(x => x)
      .map(val => {
        return `(${val})`;
      })
      .join(' and ');
  }

  or(values: string[]): string {
    return values
      .filter(x => x)
      .map(val => {
        return `(${val})`;
      })
      .join(' or ');
  }

  not(val0: string): string {
    return ` not ${val0}`;
  }

  // Inclusion Funtions
  in(values: string[]): string {
    let lhs = values[0];
    let rhs = values.slice(1).join(', ');
    return `${lhs} in (${rhs})`;
  }
  between(val0: string, val1: string, val2: string): string {
    return `${val0} between ${val1} and ${val2}`;
  }
  like(val0: string, val1: string): string {
    return `${val0} like ${val1}`;
  }
  isNull(val0: string): string {
    return `${val0} is null`;
  }
  isNotNull(val0: string): string {
    return `${val0} is not null`;
  }
  exists(val0: string): string {
    return ` exists (${val0})`;
  }
  limit(size: string, index?: string): string {
    let indexStr = index ? `${index}, ` : '';
    return ` limit ${indexStr}${size}`;
  }

  // Arithmatic Operators
  plus(val0: string, val1: string): string {
    return `${val0} + ${val1}`;
  }
  minus(val0: string, val1: string): string {
    return `${val0} - ${val1}`;
  }
  multiply(val0: string, val1: string): string {
    return `${val0} * ${val1}`;
  }
  devide(val0: string, val1: string): string {
    return `${val0} / ${val1}`;
  }

  // Sorting Operators
  asc(val0: string): string {
    return `${val0} asc`;
  }
  desc(val0: string): string {
    return `${val0} desc`;
  }

  // Group Functions
  sum(val0: string): string {
    return `sum(${val0})`;
  }
  min(val0: string): string {
    return `min(${val0})`;
  }
  max(val0: string): string {
    return `max(${val0})`;
  }
  count(val0: string): string {
    return `count(${val0})`;
  }
  average(val0: string): string {
    return `avg(${val0})`;
  }
}