AK-21/Graphite-Industrial-Intelligence

1import { connect } from "@tidbcloud/serverless";2import { entityKind } from "../entity.js";3import { DefaultLogger } from "../logger.js";4import { MySqlDatabase } from "../mysql-core/db.js";5import { MySqlDialect } from "../mysql-core/dialect.js";6import {7  createTableRelationsHelpers,8  extractTablesRelationalConfig9} from "../relations.js";10import { isConfig } from "../utils.js";11import { TiDBServerlessSession } from "./session.js";12class TiDBServerlessDatabase extends MySqlDatabase {13  static [entityKind] = "TiDBServerlessDatabase";14}15function construct(client, config = {}) {16  const dialect = new MySqlDialect({ casing: config.casing });17  let logger;18  if (config.logger === true) {19    logger = new DefaultLogger();20  } else if (config.logger !== false) {21    logger = config.logger;22  }23  let schema;24  if (config.schema) {25    const tablesConfig = extractTablesRelationalConfig(26      config.schema,27      createTableRelationsHelpers28    );29    schema = {30      fullSchema: config.schema,31      schema: tablesConfig.tables,32      tableNamesMap: tablesConfig.tableNamesMap33    };34  }35  const session = new TiDBServerlessSession(client, dialect, void 0, schema, { logger, cache: config.cache });36  const db = new TiDBServerlessDatabase(dialect, session, schema, "default");37  db.$client = client;38  db.$cache = config.cache;39  if (db.$cache) {40    db.$cache["invalidate"] = config.cache?.onMutate;41  }42  return db;43}44function drizzle(...params) {45  if (typeof params[0] === "string") {46    const instance = connect({47      url: params[0]48    });49    return construct(instance, params[1]);50  }51  if (isConfig(params[0])) {52    const { connection, client, ...drizzleConfig } = params[0];53    if (client) return construct(client, drizzleConfig);54    const instance = typeof connection === "string" ? connect({55      url: connection56    }) : connect(connection);57    return construct(instance, drizzleConfig);58  }59  return construct(params[0], params[1]);60}61((drizzle2) => {62  function mock(config) {63    return construct({}, config);64  }65  drizzle2.mock = mock;66})(drizzle || (drizzle = {}));67export {68  TiDBServerlessDatabase,69  drizzle70};71//# sourceMappingURL=driver.js.map