123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145
import { Connection, ExecuteResult } from "./connection.ts";import { DeferredStack } from "./deferred.ts";import { config as logConfig, log } from "./logger.ts";
/** * Clinet Config */export interface ClientConfig {  /** Database hostname */  hostname?: string;  /** Database username */  username?: string;  /** Database password */  password?: string;  /** Database port */  port?: number;  /** Database name */  db?: string;  /** Whether to Display Packet Debugging Information */  debug?: boolean;  /** Connect timeout */  timeout?: number;  /** TODO: auto reconnect */  reconnect?: boolean;  /** Number of retries that failed in the link process */  retry?: number;  /** Connection pool size default 1 */  pool?: number;}
/** Transaction processor */export interface TransactionProcessor<T> {  (connection: Connection): Promise<T>;}
/** * MySQL client */export class Client {  config: ClientConfig;  private _pool: DeferredStack<Connection>;  private _connections: Connection[] = [];
  private async createConnection(): Promise<Connection> {    let connection: Connection = new Connection(this);    await connection.connect();    return connection;  }
  /** get size of the pool, Number of connections created */  get poolSize() {    return this._pool.size;  }
  /** get length of the pool, Number of connections available */  get poolLength() {    return this._pool.length;  }
  /**   * connect to database   * @param config config for client   * @returns Clinet instance   */  async connect(config: ClientConfig): Promise<Client> {    await logConfig({      debug: config.debug,      logFile: "mysql.log"    });    this.config = {      hostname: "127.0.0.1",      username: "root",      port: 3306,      pool: 1,      ...config    };    Object.freeze(this.config);    this._connections = [];    this._pool = new DeferredStack<Connection>(      this.config.pool,      this._connections,      this.createConnection.bind(this)    );    return this;  }
  /**   * excute query sql   * @param sql query sql string   * @param params query params   */  async query(sql: string, params?: any[]): Promise<any> {    const connection = await this._pool.pop();    try {      const result = await connection.query(sql, params);      return result;    } finally {      this._pool.push(connection);    }  }
  /**   * excute sql   * @param sql sql string   * @param params query params   */  async execute(sql: string, params?: any[]): Promise<ExecuteResult> {    const connection = await this._pool.pop();    try {      const result = await connection.execute(sql, params);      return result;    } finally {      this._pool.push(connection);    }  }
  /**   * Execute a transaction process, and the transaction successfully   * returns the return value of the transaction process   * @param processor transation processor   */  async transaction<T = any>(processor: TransactionProcessor<T>): Promise<T> {    const connection = await this._pool.pop();    try {      await connection.execute("BEGIN");      const result = await processor(connection);      await connection.execute("COMMIT");      return result;    } catch (error) {      log.info(`ROLLBACK: ${error.message}`);      await connection.execute("ROLLBACK");      throw error;    } finally {      this._pool.push(connection);    }  }
  /**   * close connection   */  async close() {    await Promise.all(this._connections.map(conn => conn.close()));  }}