Initial commit

node_modules/drizzle-orm/tidb-serverless/driver.cjs

@@ -0,0 +1,90 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var driver_exports = {};
+__export(driver_exports, {
+  TiDBServerlessDatabase: () => TiDBServerlessDatabase,
+  drizzle: () => drizzle
+});
+module.exports = __toCommonJS(driver_exports);
+var import_serverless = require("@tidbcloud/serverless");
+var import_entity = require("../entity.cjs");
+var import_logger = require("../logger.cjs");
+var import_db = require("../mysql-core/db.cjs");
+var import_dialect = require("../mysql-core/dialect.cjs");
+var import_relations = require("../relations.cjs");
+var import_utils = require("../utils.cjs");
+var import_session = require("./session.cjs");
+class TiDBServerlessDatabase extends import_db.MySqlDatabase {
+  static [import_entity.entityKind] = "TiDBServerlessDatabase";
+}
+function construct(client, config = {}) {
+  const dialect = new import_dialect.MySqlDialect({ casing: config.casing });
+  let logger;
+  if (config.logger === true) {
+    logger = new import_logger.DefaultLogger();
+  } else if (config.logger !== false) {
+    logger = config.logger;
+  }
+  let schema;
+  if (config.schema) {
+    const tablesConfig = (0, import_relations.extractTablesRelationalConfig)(
+      config.schema,
+      import_relations.createTableRelationsHelpers
+    );
+    schema = {
+      fullSchema: config.schema,
+      schema: tablesConfig.tables,
+      tableNamesMap: tablesConfig.tableNamesMap
+    };
+  }
+  const session = new import_session.TiDBServerlessSession(client, dialect, void 0, schema, { logger });
+  const db = new TiDBServerlessDatabase(dialect, session, schema, "default");
+  db.$client = client;
+  return db;
+}
+function drizzle(...params) {
+  if (typeof params[0] === "string") {
+    const instance = (0, import_serverless.connect)({
+      url: params[0]
+    });
+    return construct(instance, params[1]);
+  }
+  if ((0, import_utils.isConfig)(params[0])) {
+    const { connection, client, ...drizzleConfig } = params[0];
+    if (client)
+      return construct(client, drizzleConfig);
+    const instance = typeof connection === "string" ? (0, import_serverless.connect)({
+      url: connection
+    }) : (0, import_serverless.connect)(connection);
+    return construct(instance, drizzleConfig);
+  }
+  return construct(params[0], params[1]);
+}
+((drizzle2) => {
+  function mock(config) {
+    return construct({}, config);
+  }
+  drizzle2.mock = mock;
+})(drizzle || (drizzle = {}));
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TiDBServerlessDatabase,
+  drizzle
+});
+//# sourceMappingURL=driver.cjs.map

node_modules/drizzle-orm/tidb-serverless/driver.d.cts

@@ -0,0 +1,31 @@
+import { type Config, type Connection } from '@tidbcloud/serverless';
+import { entityKind } from "../entity.cjs";
+import type { Logger } from "../logger.cjs";
+import { MySqlDatabase } from "../mysql-core/db.cjs";
+import { type DrizzleConfig } from "../utils.cjs";
+import type { TiDBServerlessPreparedQueryHKT, TiDBServerlessQueryResultHKT } from "./session.cjs";
+export interface TiDBServerlessSDriverOptions {
+    logger?: Logger;
+}
+export declare class TiDBServerlessDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends MySqlDatabase<TiDBServerlessQueryResultHKT, TiDBServerlessPreparedQueryHKT, TSchema> {
+    static readonly [entityKind]: string;
+}
+export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>, TClient extends Connection = Connection>(...params: [
+    TClient | string
+] | [
+    TClient | string,
+    DrizzleConfig<TSchema>
+] | [
+    ({
+        connection: string | Config;
+    } | {
+        client: TClient;
+    }) & DrizzleConfig<TSchema>
+]): TiDBServerlessDatabase<TSchema> & {
+    $client: TClient;
+};
+export declare namespace drizzle {
+    function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DrizzleConfig<TSchema>): TiDBServerlessDatabase<TSchema> & {
+        $client: '$client is not available on drizzle.mock()';
+    };
+}

node_modules/drizzle-orm/tidb-serverless/driver.d.ts

@@ -0,0 +1,31 @@
+import { type Config, type Connection } from '@tidbcloud/serverless';
+import { entityKind } from "../entity.js";
+import type { Logger } from "../logger.js";
+import { MySqlDatabase } from "../mysql-core/db.js";
+import { type DrizzleConfig } from "../utils.js";
+import type { TiDBServerlessPreparedQueryHKT, TiDBServerlessQueryResultHKT } from "./session.js";
+export interface TiDBServerlessSDriverOptions {
+    logger?: Logger;
+}
+export declare class TiDBServerlessDatabase<TSchema extends Record<string, unknown> = Record<string, never>> extends MySqlDatabase<TiDBServerlessQueryResultHKT, TiDBServerlessPreparedQueryHKT, TSchema> {
+    static readonly [entityKind]: string;
+}
+export declare function drizzle<TSchema extends Record<string, unknown> = Record<string, never>, TClient extends Connection = Connection>(...params: [
+    TClient | string
+] | [
+    TClient | string,
+    DrizzleConfig<TSchema>
+] | [
+    ({
+        connection: string | Config;
+    } | {
+        client: TClient;
+    }) & DrizzleConfig<TSchema>
+]): TiDBServerlessDatabase<TSchema> & {
+    $client: TClient;
+};
+export declare namespace drizzle {
+    function mock<TSchema extends Record<string, unknown> = Record<string, never>>(config?: DrizzleConfig<TSchema>): TiDBServerlessDatabase<TSchema> & {
+        $client: '$client is not available on drizzle.mock()';
+    };
+}

node_modules/drizzle-orm/tidb-serverless/driver.js

@@ -0,0 +1,68 @@
+import { connect } from "@tidbcloud/serverless";
+import { entityKind } from "../entity.js";
+import { DefaultLogger } from "../logger.js";
+import { MySqlDatabase } from "../mysql-core/db.js";
+import { MySqlDialect } from "../mysql-core/dialect.js";
+import {
+  createTableRelationsHelpers,
+  extractTablesRelationalConfig
+} from "../relations.js";
+import { isConfig } from "../utils.js";
+import { TiDBServerlessSession } from "./session.js";
+class TiDBServerlessDatabase extends MySqlDatabase {
+  static [entityKind] = "TiDBServerlessDatabase";
+}
+function construct(client, config = {}) {
+  const dialect = new MySqlDialect({ casing: config.casing });
+  let logger;
+  if (config.logger === true) {
+    logger = new DefaultLogger();
+  } else if (config.logger !== false) {
+    logger = config.logger;
+  }
+  let schema;
+  if (config.schema) {
+    const tablesConfig = extractTablesRelationalConfig(
+      config.schema,
+      createTableRelationsHelpers
+    );
+    schema = {
+      fullSchema: config.schema,
+      schema: tablesConfig.tables,
+      tableNamesMap: tablesConfig.tableNamesMap
+    };
+  }
+  const session = new TiDBServerlessSession(client, dialect, void 0, schema, { logger });
+  const db = new TiDBServerlessDatabase(dialect, session, schema, "default");
+  db.$client = client;
+  return db;
+}
+function drizzle(...params) {
+  if (typeof params[0] === "string") {
+    const instance = connect({
+      url: params[0]
+    });
+    return construct(instance, params[1]);
+  }
+  if (isConfig(params[0])) {
+    const { connection, client, ...drizzleConfig } = params[0];
+    if (client)
+      return construct(client, drizzleConfig);
+    const instance = typeof connection === "string" ? connect({
+      url: connection
+    }) : connect(connection);
+    return construct(instance, drizzleConfig);
+  }
+  return construct(params[0], params[1]);
+}
+((drizzle2) => {
+  function mock(config) {
+    return construct({}, config);
+  }
+  drizzle2.mock = mock;
+})(drizzle || (drizzle = {}));
+export {
+  TiDBServerlessDatabase,
+  drizzle
+};
+//# sourceMappingURL=driver.js.map

node_modules/drizzle-orm/tidb-serverless/index.cjs

@@ -0,0 +1,25 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __reExport = (target, mod, secondTarget) => (__copyProps(target, mod, "default"), secondTarget && __copyProps(secondTarget, mod, "default"));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var tidb_serverless_exports = {};
+module.exports = __toCommonJS(tidb_serverless_exports);
+__reExport(tidb_serverless_exports, require("./driver.cjs"), module.exports);
+__reExport(tidb_serverless_exports, require("./session.cjs"), module.exports);
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  ...require("./driver.cjs"),
+  ...require("./session.cjs")
+});
+//# sourceMappingURL=index.cjs.map

node_modules/drizzle-orm/tidb-serverless/index.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/tidb-serverless/index.ts"],"sourcesContent":["export * from './driver.ts';\nexport * from './session.ts';\n"],"mappings":";;;;;;;;;;;;;;;AAAA;AAAA;AAAA,oCAAc,wBAAd;AACA,oCAAc,yBADd;","names":[]}

node_modules/drizzle-orm/tidb-serverless/index.d.cts

@@ -0,0 +1,2 @@
+export * from "./driver.cjs";
+export * from "./session.cjs";

node_modules/drizzle-orm/tidb-serverless/index.d.ts

@@ -0,0 +1,2 @@
+export * from "./driver.js";
+export * from "./session.js";

node_modules/drizzle-orm/tidb-serverless/index.js

@@ -0,0 +1,3 @@
+export * from "./driver.js";
+export * from "./session.js";
+//# sourceMappingURL=index.js.map

node_modules/drizzle-orm/tidb-serverless/index.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/tidb-serverless/index.ts"],"sourcesContent":["export * from './driver.ts';\nexport * from './session.ts';\n"],"mappings":"AAAA,cAAc;AACd,cAAc;","names":[]}

node_modules/drizzle-orm/tidb-serverless/migrator.cjs

@@ -0,0 +1,33 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var migrator_exports = {};
+__export(migrator_exports, {
+  migrate: () => migrate
+});
+module.exports = __toCommonJS(migrator_exports);
+var import_migrator = require("../migrator.cjs");
+async function migrate(db, config) {
+  const migrations = (0, import_migrator.readMigrationFiles)(config);
+  await db.dialect.migrate(migrations, db.session, config);
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  migrate
+});
+//# sourceMappingURL=migrator.cjs.map

node_modules/drizzle-orm/tidb-serverless/migrator.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/tidb-serverless/migrator.ts"],"sourcesContent":["import type { MigrationConfig } from '~/migrator.ts';\nimport { readMigrationFiles } from '~/migrator.ts';\nimport type { TiDBServerlessDatabase } from './driver.ts';\n\nexport async function migrate<TSchema extends Record<string, unknown>>(\n\tdb: TiDBServerlessDatabase<TSchema>,\n\tconfig: MigrationConfig,\n) {\n\tconst migrations = readMigrationFiles(config);\n\tawait db.dialect.migrate(migrations, db.session, config);\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,sBAAmC;AAGnC,eAAsB,QACrB,IACA,QACC;AACD,QAAM,iBAAa,oCAAmB,MAAM;AAC5C,QAAM,GAAG,QAAQ,QAAQ,YAAY,GAAG,SAAS,MAAM;AACxD;","names":[]}

node_modules/drizzle-orm/tidb-serverless/migrator.d.cts

@@ -0,0 +1,3 @@
+import type { MigrationConfig } from "../migrator.cjs";
+import type { TiDBServerlessDatabase } from "./driver.cjs";
+export declare function migrate<TSchema extends Record<string, unknown>>(db: TiDBServerlessDatabase<TSchema>, config: MigrationConfig): Promise<void>;

node_modules/drizzle-orm/tidb-serverless/migrator.d.ts

@@ -0,0 +1,3 @@
+import type { MigrationConfig } from "../migrator.js";
+import type { TiDBServerlessDatabase } from "./driver.js";
+export declare function migrate<TSchema extends Record<string, unknown>>(db: TiDBServerlessDatabase<TSchema>, config: MigrationConfig): Promise<void>;

node_modules/drizzle-orm/tidb-serverless/migrator.js

@@ -0,0 +1,9 @@
+import { readMigrationFiles } from "../migrator.js";
+async function migrate(db, config) {
+  const migrations = readMigrationFiles(config);
+  await db.dialect.migrate(migrations, db.session, config);
+}
+export {
+  migrate
+};
+//# sourceMappingURL=migrator.js.map

node_modules/drizzle-orm/tidb-serverless/migrator.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/tidb-serverless/migrator.ts"],"sourcesContent":["import type { MigrationConfig } from '~/migrator.ts';\nimport { readMigrationFiles } from '~/migrator.ts';\nimport type { TiDBServerlessDatabase } from './driver.ts';\n\nexport async function migrate<TSchema extends Record<string, unknown>>(\n\tdb: TiDBServerlessDatabase<TSchema>,\n\tconfig: MigrationConfig,\n) {\n\tconst migrations = readMigrationFiles(config);\n\tawait db.dialect.migrate(migrations, db.session, config);\n}\n"],"mappings":"AACA,SAAS,0BAA0B;AAGnC,eAAsB,QACrB,IACA,QACC;AACD,QAAM,aAAa,mBAAmB,MAAM;AAC5C,QAAM,GAAG,QAAQ,QAAQ,YAAY,GAAG,SAAS,MAAM;AACxD;","names":[]}

node_modules/drizzle-orm/tidb-serverless/session.cjs

@@ -0,0 +1,169 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var session_exports = {};
+__export(session_exports, {
+  TiDBServerlessPreparedQuery: () => TiDBServerlessPreparedQuery,
+  TiDBServerlessSession: () => TiDBServerlessSession,
+  TiDBServerlessTransaction: () => TiDBServerlessTransaction
+});
+module.exports = __toCommonJS(session_exports);
+var import_column = require("../column.cjs");
+var import_entity = require("../entity.cjs");
+var import_logger = require("../logger.cjs");
+var import_session = require("../mysql-core/session.cjs");
+var import_sql = require("../sql/sql.cjs");
+var import_utils = require("../utils.cjs");
+const executeRawConfig = { fullResult: true };
+const queryConfig = { arrayMode: true };
+class TiDBServerlessPreparedQuery extends import_session.MySqlPreparedQuery {
+  constructor(client, queryString, params, logger, fields, customResultMapper, generatedIds, returningIds) {
+    super();
+    this.client = client;
+    this.queryString = queryString;
+    this.params = params;
+    this.logger = logger;
+    this.fields = fields;
+    this.customResultMapper = customResultMapper;
+    this.generatedIds = generatedIds;
+    this.returningIds = returningIds;
+  }
+  static [import_entity.entityKind] = "TiDBPreparedQuery";
+  async execute(placeholderValues = {}) {
+    const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
+    this.logger.logQuery(this.queryString, params);
+    const { fields, client, queryString, joinsNotNullableMap, customResultMapper, returningIds, generatedIds } = this;
+    if (!fields && !customResultMapper) {
+      const res = await client.execute(queryString, params, executeRawConfig);
+      const insertId = res.lastInsertId ?? 0;
+      const affectedRows = res.rowsAffected ?? 0;
+      if (returningIds) {
+        const returningResponse = [];
+        let j = 0;
+        for (let i = insertId; i < insertId + affectedRows; i++) {
+          for (const column of returningIds) {
+            const key = returningIds[0].path[0];
+            if ((0, import_entity.is)(column.field, import_column.Column)) {
+              if (column.field.primary && column.field.autoIncrement) {
+                returningResponse.push({ [key]: i });
+              }
+              if (column.field.defaultFn && generatedIds) {
+                returningResponse.push({ [key]: generatedIds[j][key] });
+              }
+            }
+          }
+          j++;
+        }
+        return returningResponse;
+      }
+      return res;
+    }
+    const rows = await client.execute(queryString, params, queryConfig);
+    if (customResultMapper) {
+      return customResultMapper(rows);
+    }
+    return rows.map((row) => (0, import_utils.mapResultRow)(fields, row, joinsNotNullableMap));
+  }
+  iterator(_placeholderValues) {
+    throw new Error("Streaming is not supported by the TiDB Cloud Serverless driver");
+  }
+}
+class TiDBServerlessSession extends import_session.MySqlSession {
+  constructor(baseClient, dialect, tx, schema, options = {}) {
+    super(dialect);
+    this.baseClient = baseClient;
+    this.schema = schema;
+    this.options = options;
+    this.client = tx ?? baseClient;
+    this.logger = options.logger ?? new import_logger.NoopLogger();
+  }
+  static [import_entity.entityKind] = "TiDBServerlessSession";
+  logger;
+  client;
+  prepareQuery(query, fields, customResultMapper, generatedIds, returningIds) {
+    return new TiDBServerlessPreparedQuery(
+      this.client,
+      query.sql,
+      query.params,
+      this.logger,
+      fields,
+      customResultMapper,
+      generatedIds,
+      returningIds
+    );
+  }
+  all(query) {
+    const querySql = this.dialect.sqlToQuery(query);
+    this.logger.logQuery(querySql.sql, querySql.params);
+    return this.client.execute(querySql.sql, querySql.params);
+  }
+  async count(sql2) {
+    const res = await this.execute(sql2);
+    return Number(
+      res["rows"][0]["count"]
+    );
+  }
+  async transaction(transaction) {
+    const nativeTx = await this.baseClient.begin();
+    try {
+      const session = new TiDBServerlessSession(this.baseClient, this.dialect, nativeTx, this.schema, this.options);
+      const tx = new TiDBServerlessTransaction(
+        this.dialect,
+        session,
+        this.schema
+      );
+      const result = await transaction(tx);
+      await nativeTx.commit();
+      return result;
+    } catch (err) {
+      await nativeTx.rollback();
+      throw err;
+    }
+  }
+}
+class TiDBServerlessTransaction extends import_session.MySqlTransaction {
+  static [import_entity.entityKind] = "TiDBServerlessTransaction";
+  constructor(dialect, session, schema, nestedIndex = 0) {
+    super(dialect, session, schema, nestedIndex, "default");
+  }
+  async transaction(transaction) {
+    const savepointName = `sp${this.nestedIndex + 1}`;
+    const tx = new TiDBServerlessTransaction(
+      this.dialect,
+      this.session,
+      this.schema,
+      this.nestedIndex + 1
+    );
+    await tx.execute(import_sql.sql.raw(`savepoint ${savepointName}`));
+    try {
+      const result = await transaction(tx);
+      await tx.execute(import_sql.sql.raw(`release savepoint ${savepointName}`));
+      return result;
+    } catch (err) {
+      await tx.execute(import_sql.sql.raw(`rollback to savepoint ${savepointName}`));
+      throw err;
+    }
+  }
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  TiDBServerlessPreparedQuery,
+  TiDBServerlessSession,
+  TiDBServerlessTransaction
+});
+//# sourceMappingURL=session.cjs.map

node_modules/drizzle-orm/tidb-serverless/session.d.cts

@@ -0,0 +1,50 @@
+import type { Connection, FullResult, Tx } from '@tidbcloud/serverless';
+import { entityKind } from "../entity.cjs";
+import type { Logger } from "../logger.cjs";
+import type { MySqlDialect } from "../mysql-core/dialect.cjs";
+import type { SelectedFieldsOrdered } from "../mysql-core/query-builders/select.types.cjs";
+import { MySqlPreparedQuery, type MySqlPreparedQueryConfig, type MySqlPreparedQueryHKT, type MySqlQueryResultHKT, MySqlSession, MySqlTransaction } from "../mysql-core/session.cjs";
+import type { RelationalSchemaConfig, TablesRelationalConfig } from "../relations.cjs";
+import { type Query, type SQL } from "../sql/sql.cjs";
+import { type Assume } from "../utils.cjs";
+export declare class TiDBServerlessPreparedQuery<T extends MySqlPreparedQueryConfig> extends MySqlPreparedQuery<T> {
+    private client;
+    private queryString;
+    private params;
+    private logger;
+    private fields;
+    private customResultMapper?;
+    private generatedIds?;
+    private returningIds?;
+    static readonly [entityKind]: string;
+    constructor(client: Tx | Connection, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined, generatedIds?: Record<string, unknown>[] | undefined, returningIds?: SelectedFieldsOrdered | undefined);
+    execute(placeholderValues?: Record<string, unknown> | undefined): Promise<T['execute']>;
+    iterator(_placeholderValues?: Record<string, unknown>): AsyncGenerator<T['iterator']>;
+}
+export interface TiDBServerlessSessionOptions {
+    logger?: Logger;
+}
+export declare class TiDBServerlessSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MySqlSession<TiDBServerlessQueryResultHKT, TiDBServerlessPreparedQueryHKT, TFullSchema, TSchema> {
+    private baseClient;
+    private schema;
+    private options;
+    static readonly [entityKind]: string;
+    private logger;
+    private client;
+    constructor(baseClient: Connection, dialect: MySqlDialect, tx: Tx | undefined, schema: RelationalSchemaConfig<TSchema> | undefined, options?: TiDBServerlessSessionOptions);
+    prepareQuery<T extends MySqlPreparedQueryConfig = MySqlPreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute'], generatedIds?: Record<string, unknown>[], returningIds?: SelectedFieldsOrdered): MySqlPreparedQuery<T>;
+    all<T = unknown>(query: SQL): Promise<T[]>;
+    count(sql: SQL): Promise<number>;
+    transaction<T>(transaction: (tx: TiDBServerlessTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}
+export declare class TiDBServerlessTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MySqlTransaction<TiDBServerlessQueryResultHKT, TiDBServerlessPreparedQueryHKT, TFullSchema, TSchema> {
+    static readonly [entityKind]: string;
+    constructor(dialect: MySqlDialect, session: MySqlSession, schema: RelationalSchemaConfig<TSchema> | undefined, nestedIndex?: number);
+    transaction<T>(transaction: (tx: TiDBServerlessTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}
+export interface TiDBServerlessQueryResultHKT extends MySqlQueryResultHKT {
+    type: FullResult;
+}
+export interface TiDBServerlessPreparedQueryHKT extends MySqlPreparedQueryHKT {
+    type: TiDBServerlessPreparedQuery<Assume<this['config'], MySqlPreparedQueryConfig>>;
+}

node_modules/drizzle-orm/tidb-serverless/session.d.ts

@@ -0,0 +1,50 @@
+import type { Connection, FullResult, Tx } from '@tidbcloud/serverless';
+import { entityKind } from "../entity.js";
+import type { Logger } from "../logger.js";
+import type { MySqlDialect } from "../mysql-core/dialect.js";
+import type { SelectedFieldsOrdered } from "../mysql-core/query-builders/select.types.js";
+import { MySqlPreparedQuery, type MySqlPreparedQueryConfig, type MySqlPreparedQueryHKT, type MySqlQueryResultHKT, MySqlSession, MySqlTransaction } from "../mysql-core/session.js";
+import type { RelationalSchemaConfig, TablesRelationalConfig } from "../relations.js";
+import { type Query, type SQL } from "../sql/sql.js";
+import { type Assume } from "../utils.js";
+export declare class TiDBServerlessPreparedQuery<T extends MySqlPreparedQueryConfig> extends MySqlPreparedQuery<T> {
+    private client;
+    private queryString;
+    private params;
+    private logger;
+    private fields;
+    private customResultMapper?;
+    private generatedIds?;
+    private returningIds?;
+    static readonly [entityKind]: string;
+    constructor(client: Tx | Connection, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined, generatedIds?: Record<string, unknown>[] | undefined, returningIds?: SelectedFieldsOrdered | undefined);
+    execute(placeholderValues?: Record<string, unknown> | undefined): Promise<T['execute']>;
+    iterator(_placeholderValues?: Record<string, unknown>): AsyncGenerator<T['iterator']>;
+}
+export interface TiDBServerlessSessionOptions {
+    logger?: Logger;
+}
+export declare class TiDBServerlessSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MySqlSession<TiDBServerlessQueryResultHKT, TiDBServerlessPreparedQueryHKT, TFullSchema, TSchema> {
+    private baseClient;
+    private schema;
+    private options;
+    static readonly [entityKind]: string;
+    private logger;
+    private client;
+    constructor(baseClient: Connection, dialect: MySqlDialect, tx: Tx | undefined, schema: RelationalSchemaConfig<TSchema> | undefined, options?: TiDBServerlessSessionOptions);
+    prepareQuery<T extends MySqlPreparedQueryConfig = MySqlPreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute'], generatedIds?: Record<string, unknown>[], returningIds?: SelectedFieldsOrdered): MySqlPreparedQuery<T>;
+    all<T = unknown>(query: SQL): Promise<T[]>;
+    count(sql: SQL): Promise<number>;
+    transaction<T>(transaction: (tx: TiDBServerlessTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}
+export declare class TiDBServerlessTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MySqlTransaction<TiDBServerlessQueryResultHKT, TiDBServerlessPreparedQueryHKT, TFullSchema, TSchema> {
+    static readonly [entityKind]: string;
+    constructor(dialect: MySqlDialect, session: MySqlSession, schema: RelationalSchemaConfig<TSchema> | undefined, nestedIndex?: number);
+    transaction<T>(transaction: (tx: TiDBServerlessTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}
+export interface TiDBServerlessQueryResultHKT extends MySqlQueryResultHKT {
+    type: FullResult;
+}
+export interface TiDBServerlessPreparedQueryHKT extends MySqlPreparedQueryHKT {
+    type: TiDBServerlessPreparedQuery<Assume<this['config'], MySqlPreparedQueryConfig>>;
+}

node_modules/drizzle-orm/tidb-serverless/session.js

@@ -0,0 +1,147 @@
+import { Column } from "../column.js";
+import { entityKind, is } from "../entity.js";
+import { NoopLogger } from "../logger.js";
+import {
+  MySqlPreparedQuery,
+  MySqlSession,
+  MySqlTransaction
+} from "../mysql-core/session.js";
+import { fillPlaceholders, sql } from "../sql/sql.js";
+import { mapResultRow } from "../utils.js";
+const executeRawConfig = { fullResult: true };
+const queryConfig = { arrayMode: true };
+class TiDBServerlessPreparedQuery extends MySqlPreparedQuery {
+  constructor(client, queryString, params, logger, fields, customResultMapper, generatedIds, returningIds) {
+    super();
+    this.client = client;
+    this.queryString = queryString;
+    this.params = params;
+    this.logger = logger;
+    this.fields = fields;
+    this.customResultMapper = customResultMapper;
+    this.generatedIds = generatedIds;
+    this.returningIds = returningIds;
+  }
+  static [entityKind] = "TiDBPreparedQuery";
+  async execute(placeholderValues = {}) {
+    const params = fillPlaceholders(this.params, placeholderValues);
+    this.logger.logQuery(this.queryString, params);
+    const { fields, client, queryString, joinsNotNullableMap, customResultMapper, returningIds, generatedIds } = this;
+    if (!fields && !customResultMapper) {
+      const res = await client.execute(queryString, params, executeRawConfig);
+      const insertId = res.lastInsertId ?? 0;
+      const affectedRows = res.rowsAffected ?? 0;
+      if (returningIds) {
+        const returningResponse = [];
+        let j = 0;
+        for (let i = insertId; i < insertId + affectedRows; i++) {
+          for (const column of returningIds) {
+            const key = returningIds[0].path[0];
+            if (is(column.field, Column)) {
+              if (column.field.primary && column.field.autoIncrement) {
+                returningResponse.push({ [key]: i });
+              }
+              if (column.field.defaultFn && generatedIds) {
+                returningResponse.push({ [key]: generatedIds[j][key] });
+              }
+            }
+          }
+          j++;
+        }
+        return returningResponse;
+      }
+      return res;
+    }
+    const rows = await client.execute(queryString, params, queryConfig);
+    if (customResultMapper) {
+      return customResultMapper(rows);
+    }
+    return rows.map((row) => mapResultRow(fields, row, joinsNotNullableMap));
+  }
+  iterator(_placeholderValues) {
+    throw new Error("Streaming is not supported by the TiDB Cloud Serverless driver");
+  }
+}
+class TiDBServerlessSession extends MySqlSession {
+  constructor(baseClient, dialect, tx, schema, options = {}) {
+    super(dialect);
+    this.baseClient = baseClient;
+    this.schema = schema;
+    this.options = options;
+    this.client = tx ?? baseClient;
+    this.logger = options.logger ?? new NoopLogger();
+  }
+  static [entityKind] = "TiDBServerlessSession";
+  logger;
+  client;
+  prepareQuery(query, fields, customResultMapper, generatedIds, returningIds) {
+    return new TiDBServerlessPreparedQuery(
+      this.client,
+      query.sql,
+      query.params,
+      this.logger,
+      fields,
+      customResultMapper,
+      generatedIds,
+      returningIds
+    );
+  }
+  all(query) {
+    const querySql = this.dialect.sqlToQuery(query);
+    this.logger.logQuery(querySql.sql, querySql.params);
+    return this.client.execute(querySql.sql, querySql.params);
+  }
+  async count(sql2) {
+    const res = await this.execute(sql2);
+    return Number(
+      res["rows"][0]["count"]
+    );
+  }
+  async transaction(transaction) {
+    const nativeTx = await this.baseClient.begin();
+    try {
+      const session = new TiDBServerlessSession(this.baseClient, this.dialect, nativeTx, this.schema, this.options);
+      const tx = new TiDBServerlessTransaction(
+        this.dialect,
+        session,
+        this.schema
+      );
+      const result = await transaction(tx);
+      await nativeTx.commit();
+      return result;
+    } catch (err) {
+      await nativeTx.rollback();
+      throw err;
+    }
+  }
+}
+class TiDBServerlessTransaction extends MySqlTransaction {
+  static [entityKind] = "TiDBServerlessTransaction";
+  constructor(dialect, session, schema, nestedIndex = 0) {
+    super(dialect, session, schema, nestedIndex, "default");
+  }
+  async transaction(transaction) {
+    const savepointName = `sp${this.nestedIndex + 1}`;
+    const tx = new TiDBServerlessTransaction(
+      this.dialect,
+      this.session,
+      this.schema,
+      this.nestedIndex + 1
+    );
+    await tx.execute(sql.raw(`savepoint ${savepointName}`));
+    try {
+      const result = await transaction(tx);
+      await tx.execute(sql.raw(`release savepoint ${savepointName}`));
+      return result;
+    } catch (err) {
+      await tx.execute(sql.raw(`rollback to savepoint ${savepointName}`));
+      throw err;
+    }
+  }
+}
+export {
+  TiDBServerlessPreparedQuery,
+  TiDBServerlessSession,
+  TiDBServerlessTransaction
+};
+//# sourceMappingURL=session.js.map