123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101
import { configMySQLLogger, MySQLClient, MySQLConnection } from "../../deps.ts";import type { LoggerConfig } from "../../deps.ts";import type { Connector, ConnectorOptions } from "./connector.ts";import { SQLTranslator } from "../translators/sql-translator.ts";import type { SupportedSQLDatabaseDialect } from "../translators/sql-translator.ts";import type { QueryDescription } from "../query-builder.ts";
export interface MySQLOptions extends ConnectorOptions {  database: string;  host: string;  username: string;  password: string;  port?: number;  charset?: string;  logger?: LoggerConfig;}
export class MySQLConnector implements Connector {  _dialect: SupportedSQLDatabaseDialect = "mysql";
  _client: MySQLClient;  _options: MySQLOptions;  _translator: SQLTranslator;  _connected = false;
  /** Create a MySQL connection. */  constructor(options: MySQLOptions) {    this._options = options;    this._client = new MySQLClient();    this._translator = new SQLTranslator(this._dialect);  }
  async _makeConnection() {    if (this._connected) {      return;    }
    if (this._options.logger !== undefined) {      await configMySQLLogger(this._options.logger);    }
    await this._client.connect({      hostname: this._options.host,      username: this._options.username,      db: this._options.database,      password: this._options.password,      port: this._options.port ?? 3306,      charset: this._options.charset ?? "utf8",    });
    this._connected = true;  }
  async ping() {    await this._makeConnection();
    try {      const [{ result }] = await this._client.query("SELECT 1 + 1 as result");      return result === 2;    } catch {      return false;    }  }
  async query(    queryDescription: QueryDescription,    client?: MySQLClient | MySQLConnection,  // deno-lint-ignore no-explicit-any  ): Promise<any | any[]> {    await this._makeConnection();
    const queryClient = client ?? this._client;    const query = this._translator.translateToQuery(queryDescription);    const subqueries = query.split(/;(?=(?:[^'"]|'[^']*'|"[^"]*")*$)/);    const queryMethod = query.toLowerCase().startsWith("select")      ? "query"      : "execute";
    for (let i = 0; i < subqueries.length; i++) {      const result = await queryClient[queryMethod](subqueries[i]);
      if (i === subqueries.length - 1) {        return result;      }    }  }
  transaction(queries: () => Promise<void>) {    return this._client.transaction(queries);  }
  async close() {    if (!this._connected) {      return;    }
    await this._client.close();    this._connected = false;  }}