1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
import { PoolClient } from "./client.ts";import { Connection } from "./connection.ts";import { ConnectionParams, IConnectionParams } from "./connection_params.ts";import { Query, QueryConfig, QueryResult } from "./query.ts";import { DeferredStack } from "./deferred.ts";
export class Pool {  private _connectionParams: ConnectionParams;  private _connections: Array<Connection>;  private _availableConnections: DeferredStack<Connection>;  private _size: number;  private _ready: Promise<void>;
  constructor(connectionParams: IConnectionParams, size: number) {    this._connectionParams = new ConnectionParams(connectionParams);    this._size = size;    this._ready = this._startup();  }
  private async _createConnection(): Promise<Connection> {    const connection = new Connection(this._connectionParams);    await connection.startup();    await connection.initSQL();    return connection;  }
  get size(): number {    return this._size;  }
  get available(): number {    return this._availableConnections.size;  }
  private async _startup(): Promise<void> {    const connecting = [...Array(this.size)].map(      async () => await this._createConnection()    );    this._connections = await Promise.all(connecting);    this._availableConnections = new DeferredStack(this._connections);  }
  private async _execute(query: Query): Promise<QueryResult> {    await this._ready;    const connection = await this._availableConnections.pop();    const result = await connection.query(query);    this._availableConnections.push(connection);    return result;  }
  async connect(): Promise<PoolClient> {    await this._ready;    const connection = await this._availableConnections.pop();    const release = () => this._availableConnections.push(connection);    return new PoolClient(connection, release);  }
  // TODO: can we use more specific type for args?  async query(    text: string | QueryConfig,    ...args: any[]  ): Promise<QueryResult> {    const query = new Query(text, ...args);    return await this._execute(query);  }
  async end(): Promise<void> {    await this._ready;    const ending = this._connections.map(c => c.end());    await Promise.all(ending);  }
  // Support `using` module  _aenter = () => {};  _aexit = this.end;}