drizzle-orm 0.44.2 → 0.44.3-aca84cf

package/node-mssql/pool.d.ts

@@ -0,0 +1,8 @@
+import mssql from 'mssql';
+import { entityKind } from "../entity.js";
+export declare class AutoPool {
+    static readonly [entityKind]: string;
+    private pool;
+    constructor(config: string | mssql.config);
+    $instance(): Promise<mssql.ConnectionPool>;
+}

package/node-mssql/pool.js

@@ -0,0 +1,20 @@
+import mssql from "mssql";
+import { entityKind } from "../entity.js";
+class AutoPool {
+  static [entityKind] = "AutoPool";
+  pool;
+  constructor(config) {
+    this.pool = new mssql.ConnectionPool(config);
+  }
+  async $instance() {
+    await this.pool.connect().catch((err) => {
+      console.error("\u274C AutoPool failed to connect:", err);
+      throw err;
+    });
+    return this.pool;
+  }
+}
+export {
+  AutoPool
+};
+//# sourceMappingURL=pool.js.map

package/node-mssql/pool.js.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/node-mssql/pool.ts"],"sourcesContent":["import mssql from 'mssql';\nimport { entityKind } from '~/entity.ts';\n\nexport class AutoPool {\n\tstatic readonly [entityKind]: string = 'AutoPool';\n\n\tprivate pool: mssql.ConnectionPool;\n\n\tconstructor(config: string | mssql.config) {\n\t\tthis.pool = new mssql.ConnectionPool(config as any);\n\t}\n\n\tasync $instance() {\n\t\tawait this.pool.connect().catch((err) => {\n\t\t\tconsole.error('❌ AutoPool failed to connect:', err);\n\t\t\tthrow err;\n\t\t});\n\t\treturn this.pool;\n\t}\n}\n"],"mappings":"AAAA,OAAO,WAAW;AAClB,SAAS,kBAAkB;AAEpB,MAAM,SAAS;AAAA,EACrB,QAAiB,UAAU,IAAY;AAAA,EAE/B;AAAA,EAER,YAAY,QAA+B;AAC1C,SAAK,OAAO,IAAI,MAAM,eAAe,MAAa;AAAA,EACnD;AAAA,EAEA,MAAM,YAAY;AACjB,UAAM,KAAK,KAAK,QAAQ,EAAE,MAAM,CAAC,QAAQ;AACxC,cAAQ,MAAM,sCAAiC,GAAG;AAClD,YAAM;AAAA,IACP,CAAC;AACD,WAAO,KAAK;AAAA,EACb;AACD;","names":[]}

package/node-mssql/session.cjs

@@ -0,0 +1,256 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+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 __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var session_exports = {};
+__export(session_exports, {
+  NodeMsSqlPreparedQuery: () => NodeMsSqlPreparedQuery,
+  NodeMsSqlSession: () => NodeMsSqlSession,
+  NodeMsSqlTransaction: () => NodeMsSqlTransaction
+});
+module.exports = __toCommonJS(session_exports);
+var import_mssql = __toESM(require("mssql"), 1);
+var import_node_events = require("node:events");
+var import_entity = require("../entity.cjs");
+var import_logger = require("../logger.cjs");
+var import_session = require("../mssql-core/session.cjs");
+var import_sql = require("../sql/sql.cjs");
+var import_utils = require("../utils.cjs");
+var import_pool = require("./pool.cjs");
+class NodeMsSqlPreparedQuery extends import_session.PreparedQuery {
+  constructor(client, queryString, params, logger, fields, customResultMapper) {
+    super();
+    this.client = client;
+    this.params = params;
+    this.logger = logger;
+    this.fields = fields;
+    this.customResultMapper = customResultMapper;
+    this.rawQuery = {
+      sql: queryString,
+      parameters: params
+    };
+  }
+  static [import_entity.entityKind] = "NodeMsSqlPreparedQuery";
+  rawQuery;
+  async execute(placeholderValues = {}) {
+    const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
+    this.logger.logQuery(this.rawQuery.sql, params);
+    const {
+      fields,
+      client,
+      rawQuery,
+      joinsNotNullableMap,
+      customResultMapper
+    } = this;
+    let queryClient = client;
+    if ((0, import_entity.is)(client, import_pool.AutoPool)) {
+      queryClient = await client.$instance();
+    }
+    const request = queryClient.request();
+    for (const [index, param] of params.entries()) {
+      request.input(`par${index}`, param);
+    }
+    if (!fields && !customResultMapper) {
+      return request.query(rawQuery.sql);
+    }
+    request.arrayRowMode = true;
+    const rows = await request.query(rawQuery.sql);
+    if (customResultMapper) {
+      return customResultMapper(rows.recordset);
+    }
+    return rows.recordset.map((row) => (0, import_utils.mapResultRow)(fields, row, joinsNotNullableMap));
+  }
+  async *iterator(placeholderValues = {}) {
+    const params = (0, import_sql.fillPlaceholders)(this.params, placeholderValues);
+    const {
+      fields,
+      rawQuery,
+      joinsNotNullableMap,
+      client,
+      customResultMapper
+    } = this;
+    let queryClient = client;
+    if ((0, import_entity.is)(client, import_pool.AutoPool)) {
+      queryClient = await client.$instance();
+    }
+    const request = queryClient.request();
+    request.stream = true;
+    const hasRowsMapper = Boolean(fields || customResultMapper);
+    if (hasRowsMapper) {
+      request.arrayRowMode = true;
+    }
+    for (const [index, param] of params.entries()) {
+      request.input(`par${index}`, param);
+    }
+    const stream = request.toReadableStream();
+    request.query(rawQuery.sql);
+    function dataListener() {
+      stream.pause();
+    }
+    stream.on("data", dataListener);
+    try {
+      const onEnd = (0, import_node_events.once)(stream, "end");
+      const onError = (0, import_node_events.once)(stream, "error");
+      while (true) {
+        stream.resume();
+        const row = await Promise.race([
+          onEnd,
+          onError,
+          new Promise((resolve) => stream.once("data", resolve))
+        ]);
+        if (row === void 0 || Array.isArray(row) && row.length === 0) {
+          break;
+        } else if (row instanceof Error) {
+          throw row;
+        } else {
+          if (hasRowsMapper) {
+            if (customResultMapper) {
+              const mappedRow = customResultMapper([row]);
+              yield Array.isArray(mappedRow) ? mappedRow[0] : mappedRow;
+            } else {
+              yield (0, import_utils.mapResultRow)(
+                fields,
+                row,
+                joinsNotNullableMap
+              );
+            }
+          } else {
+            yield row;
+          }
+        }
+      }
+    } finally {
+      stream.off("data", dataListener);
+      request.cancel();
+    }
+  }
+}
+class NodeMsSqlSession extends import_session.MsSqlSession {
+  constructor(client, dialect, schema, options) {
+    super(dialect);
+    this.client = client;
+    this.schema = schema;
+    this.options = options;
+    this.logger = options.logger ?? new import_logger.NoopLogger();
+  }
+  static [import_entity.entityKind] = "NodeMsSqlSession";
+  logger;
+  prepareQuery(query, fields, customResultMapper) {
+    return new NodeMsSqlPreparedQuery(
+      this.client,
+      query.sql,
+      query.params,
+      this.logger,
+      fields,
+      customResultMapper
+    );
+  }
+  /**
+   * @internal
+   * What is its purpose?
+   */
+  async query(query, params) {
+    this.logger.logQuery(query, params);
+    let queryClient = this.client;
+    if ((0, import_entity.is)(this.client, import_pool.AutoPool)) {
+      queryClient = await this.client.$instance();
+    }
+    const request = queryClient.request();
+    request.arrayRowMode = true;
+    for (const [index, param] of params.entries()) {
+      request.input(`par${index}`, param);
+    }
+    return request.query(query);
+  }
+  async all(query) {
+    const querySql = this.dialect.sqlToQuery(query);
+    this.logger.logQuery(querySql.sql, querySql.params);
+    return await this.query(querySql.sql, querySql.params).then(
+      (result) => result.recordset
+    );
+  }
+  async transaction(transaction, config) {
+    const mssqlTransaction = this.client.transaction();
+    const session = new NodeMsSqlSession(
+      mssqlTransaction,
+      this.dialect,
+      this.schema,
+      this.options
+    );
+    const tx = new NodeMsSqlTransaction(
+      this.dialect,
+      session,
+      this.schema,
+      0
+    );
+    await mssqlTransaction.begin(
+      config?.isolationLevel ? isolationLevelMap[config.isolationLevel] : void 0
+    );
+    try {
+      const result = await transaction(tx);
+      await mssqlTransaction.commit();
+      return result;
+    } catch (err) {
+      await mssqlTransaction.rollback();
+      throw err;
+    }
+  }
+}
+class NodeMsSqlTransaction extends import_session.MsSqlTransaction {
+  static [import_entity.entityKind] = "NodeMsSqlTransaction";
+  async transaction(transaction) {
+    const savepointName = `sp${this.nestedIndex + 1}`;
+    const tx = new NodeMsSqlTransaction(
+      this.dialect,
+      this.session,
+      this.schema,
+      this.nestedIndex + 1
+    );
+    await tx.execute(import_sql.sql.raw(`save transaction ${savepointName}`));
+    try {
+      const result = await transaction(tx);
+      return result;
+    } catch (err) {
+      await tx.execute(import_sql.sql.raw(`rollback transaction ${savepointName}`));
+      throw err;
+    }
+  }
+}
+const isolationLevelMap = {
+  "read uncommitted": import_mssql.default.ISOLATION_LEVEL.READ_UNCOMMITTED,
+  "read committed": import_mssql.default.ISOLATION_LEVEL.READ_COMMITTED,
+  "repeatable read": import_mssql.default.ISOLATION_LEVEL.REPEATABLE_READ,
+  serializable: import_mssql.default.ISOLATION_LEVEL.SERIALIZABLE,
+  snapshot: import_mssql.default.ISOLATION_LEVEL.SNAPSHOT
+};
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  NodeMsSqlPreparedQuery,
+  NodeMsSqlSession,
+  NodeMsSqlTransaction
+});
+//# sourceMappingURL=session.cjs.map

package/node-mssql/session.cjs.map

@@ -0,0 +1 @@
+{"version":3,"sources":["../../src/node-mssql/session.ts"],"sourcesContent":["import type { ConnectionPool, IResult, Request } from 'mssql';\nimport mssql from 'mssql';\nimport { once } from 'node:events';\nimport { entityKind, is } from '~/entity.ts';\nimport type { Logger } from '~/logger.ts';\nimport { NoopLogger } from '~/logger.ts';\nimport type { MsSqlDialect } from '~/mssql-core/dialect.ts';\nimport type { SelectedFieldsOrdered } from '~/mssql-core/query-builders/select.types.ts';\nimport {\n\tMsSqlSession,\n\tMsSqlTransaction,\n\ttype MsSqlTransactionConfig,\n\tPreparedQuery,\n\ttype PreparedQueryConfig,\n\ttype PreparedQueryHKT,\n\ttype PreparedQueryKind,\n\ttype QueryResultHKT,\n} from '~/mssql-core/session.ts';\nimport type { RelationalSchemaConfig, TablesRelationalConfig } from '~/relations.ts';\nimport { fillPlaceholders, type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { type Assume, mapResultRow } from '~/utils.ts';\nimport { AutoPool } from './pool.ts';\n\nexport type NodeMsSqlClient = Pick<ConnectionPool, 'request'> | AutoPool;\n\nexport type MsSqlQueryResult<T extends unknown | unknown[] = any> = IResult<T>;\n\nexport class NodeMsSqlPreparedQuery<\n\tT extends PreparedQueryConfig,\n> extends PreparedQuery<T> {\n\tstatic override readonly [entityKind]: string = 'NodeMsSqlPreparedQuery';\n\n\tprivate rawQuery: {\n\t\tsql: string;\n\t\tparameters: unknown[];\n\t};\n\n\tconstructor(\n\t\tprivate client: NodeMsSqlClient,\n\t\tqueryString: string,\n\t\tprivate params: unknown[],\n\t\tprivate logger: Logger,\n\t\tprivate fields: SelectedFieldsOrdered | undefined,\n\t\tprivate customResultMapper?: (rows: unknown[][]) => T['execute'],\n\t) {\n\t\tsuper();\n\t\tthis.rawQuery = {\n\t\t\tsql: queryString,\n\t\t\tparameters: params,\n\t\t};\n\t}\n\n\tasync execute(\n\t\tplaceholderValues: Record<string, unknown> = {},\n\t): Promise<T['execute']> {\n\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\n\t\tthis.logger.logQuery(this.rawQuery.sql, params);\n\n\t\tconst {\n\t\t\tfields,\n\t\t\tclient,\n\t\t\trawQuery,\n\t\t\tjoinsNotNullableMap,\n\t\t\tcustomResultMapper,\n\t\t} = this;\n\t\tlet queryClient = client as ConnectionPool;\n\t\tif (is(client, AutoPool)) {\n\t\t\tqueryClient = await client.$instance();\n\t\t}\n\t\tconst request = queryClient.request() as Request & { arrayRowMode: boolean };\n\t\tfor (const [index, param] of params.entries()) {\n\t\t\trequest.input(`par${index}`, param);\n\t\t}\n\n\t\tif (!fields && !customResultMapper) {\n\t\t\treturn request.query(rawQuery.sql) as Promise<T['execute']>;\n\t\t}\n\n\t\trequest.arrayRowMode = true;\n\t\tconst rows = await request.query<any[]>(rawQuery.sql);\n\n\t\tif (customResultMapper) {\n\t\t\treturn customResultMapper(rows.recordset);\n\t\t}\n\n\t\treturn rows.recordset.map((row) => mapResultRow<T['execute']>(fields!, row, joinsNotNullableMap));\n\t}\n\n\tasync *iterator(\n\t\tplaceholderValues: Record<string, unknown> = {},\n\t): AsyncGenerator<\n\t\tT['execute'] extends any[] ? T['execute'][number] : T['execute']\n\t> {\n\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\n\t\tconst {\n\t\t\tfields,\n\t\t\trawQuery,\n\t\t\tjoinsNotNullableMap,\n\t\t\tclient,\n\t\t\tcustomResultMapper,\n\t\t} = this;\n\t\tlet queryClient = client as ConnectionPool;\n\t\tif (is(client, AutoPool)) {\n\t\t\tqueryClient = await client.$instance();\n\t\t}\n\t\tconst request = queryClient.request() as Request & { arrayRowMode: boolean };\n\t\trequest.stream = true;\n\t\tconst hasRowsMapper = Boolean(fields || customResultMapper);\n\n\t\tif (hasRowsMapper) {\n\t\t\trequest.arrayRowMode = true;\n\t\t}\n\n\t\tfor (const [index, param] of params.entries()) {\n\t\t\trequest.input(`par${index}`, param);\n\t\t}\n\n\t\tconst stream = request.toReadableStream();\n\n\t\trequest.query(rawQuery.sql);\n\n\t\tfunction dataListener() {\n\t\t\tstream.pause();\n\t\t}\n\n\t\tstream.on('data', dataListener);\n\n\t\ttry {\n\t\t\tconst onEnd = once(stream, 'end');\n\t\t\tconst onError = once(stream, 'error');\n\n\t\t\twhile (true) {\n\t\t\t\tstream.resume();\n\t\t\t\tconst row = await Promise.race([\n\t\t\t\t\tonEnd,\n\t\t\t\t\tonError,\n\t\t\t\t\tnew Promise((resolve) => stream.once('data', resolve)),\n\t\t\t\t]);\n\t\t\t\tif (row === undefined || (Array.isArray(row) && row.length === 0)) {\n\t\t\t\t\tbreak;\n\t\t\t\t\t// eslint-disable-next-line no-instanceof/no-instanceof\n\t\t\t\t} else if (row instanceof Error) {\n\t\t\t\t\tthrow row;\n\t\t\t\t} else {\n\t\t\t\t\tif (hasRowsMapper) {\n\t\t\t\t\t\tif (customResultMapper) {\n\t\t\t\t\t\t\tconst mappedRow = customResultMapper([row as unknown[]]);\n\t\t\t\t\t\t\tyield Array.isArray(mappedRow) ? mappedRow[0] : mappedRow;\n\t\t\t\t\t\t} else {\n\t\t\t\t\t\t\tyield mapResultRow(\n\t\t\t\t\t\t\t\tfields!,\n\t\t\t\t\t\t\t\trow as unknown[],\n\t\t\t\t\t\t\t\tjoinsNotNullableMap,\n\t\t\t\t\t\t\t);\n\t\t\t\t\t\t}\n\t\t\t\t\t} else {\n\t\t\t\t\t\tyield row as T['execute'];\n\t\t\t\t\t}\n\t\t\t\t}\n\t\t\t}\n\t\t} finally {\n\t\t\tstream.off('data', dataListener);\n\t\t\trequest.cancel();\n\t\t}\n\t}\n}\n\nexport interface NodeMsSqlSessionOptions {\n\tlogger?: Logger;\n}\n\nexport class NodeMsSqlSession<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends TablesRelationalConfig,\n> extends MsSqlSession<\n\tNodeMsSqlQueryResultHKT,\n\tNodeMsSqlPreparedQueryHKT,\n\tTFullSchema,\n\tTSchema\n> {\n\tstatic override readonly [entityKind]: string = 'NodeMsSqlSession';\n\n\tprivate logger: Logger;\n\n\tconstructor(\n\t\tprivate client: NodeMsSqlClient,\n\t\tdialect: MsSqlDialect,\n\t\tprivate schema: RelationalSchemaConfig<TSchema> | undefined,\n\t\tprivate options: NodeMsSqlSessionOptions,\n\t) {\n\t\tsuper(dialect);\n\t\tthis.logger = options.logger ?? new NoopLogger();\n\t}\n\n\tprepareQuery<T extends PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tcustomResultMapper?: (rows: unknown[][]) => T['execute'],\n\t): PreparedQueryKind<NodeMsSqlPreparedQueryHKT, T> {\n\t\treturn new NodeMsSqlPreparedQuery(\n\t\t\tthis.client,\n\t\t\tquery.sql,\n\t\t\tquery.params,\n\t\t\tthis.logger,\n\t\t\tfields,\n\t\t\tcustomResultMapper,\n\t\t) as PreparedQueryKind<NodeMsSqlPreparedQueryHKT, T>;\n\t}\n\n\t/**\n\t * @internal\n\t * What is its purpose?\n\t */\n\tasync query(query: string, params: unknown[]): Promise<MsSqlQueryResult> {\n\t\tthis.logger.logQuery(query, params);\n\n\t\tlet queryClient = this.client as ConnectionPool;\n\t\tif (is(this.client, AutoPool)) {\n\t\t\tqueryClient = await this.client.$instance();\n\t\t}\n\t\tconst request = queryClient.request() as Request & {\n\t\t\tarrayRowMode: boolean;\n\t\t};\n\t\trequest.arrayRowMode = true;\n\n\t\tfor (const [index, param] of params.entries()) {\n\t\t\trequest.input(`par${index}`, param);\n\t\t}\n\n\t\treturn request.query(query);\n\t}\n\n\toverride async all<T = unknown>(query: SQL): Promise<T[]> {\n\t\tconst querySql = this.dialect.sqlToQuery(query);\n\t\tthis.logger.logQuery(querySql.sql, querySql.params);\n\t\treturn await this.query(querySql.sql, querySql.params).then(\n\t\t\t(result) => result.recordset,\n\t\t);\n\t}\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeMsSqlTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t\tconfig?: MsSqlTransactionConfig,\n\t): Promise<T> {\n\t\tconst mssqlTransaction = (this.client as ConnectionPool).transaction();\n\t\tconst session = new NodeMsSqlSession(\n\t\t\tmssqlTransaction,\n\t\t\tthis.dialect,\n\t\t\tthis.schema,\n\t\t\tthis.options,\n\t\t);\n\t\tconst tx = new NodeMsSqlTransaction(\n\t\t\tthis.dialect,\n\t\t\tsession as MsSqlSession<any, any, any, any>,\n\t\t\tthis.schema,\n\t\t\t0,\n\t\t);\n\n\t\tawait mssqlTransaction.begin(\n\t\t\tconfig?.isolationLevel\n\t\t\t\t? isolationLevelMap[config.isolationLevel]\n\t\t\t\t: undefined,\n\t\t);\n\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait mssqlTransaction.commit();\n\t\t\treturn result;\n\t\t} catch (err) {\n\t\t\tawait mssqlTransaction.rollback();\n\t\t\tthrow err;\n\t\t}\n\t}\n}\n\nexport class NodeMsSqlTransaction<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends TablesRelationalConfig,\n> extends MsSqlTransaction<\n\tNodeMsSqlQueryResultHKT,\n\tNodeMsSqlPreparedQueryHKT,\n\tTFullSchema,\n\tTSchema\n> {\n\tstatic override readonly [entityKind]: string = 'NodeMsSqlTransaction';\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeMsSqlTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t): Promise<T> {\n\t\tconst savepointName = `sp${this.nestedIndex + 1}`;\n\t\tconst tx = new NodeMsSqlTransaction(\n\t\t\tthis.dialect,\n\t\t\tthis.session,\n\t\t\tthis.schema,\n\t\t\tthis.nestedIndex + 1,\n\t\t);\n\n\t\tawait tx.execute(sql.raw(`save transaction ${savepointName}`));\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\treturn result;\n\t\t} catch (err) {\n\t\t\tawait tx.execute(sql.raw(`rollback transaction ${savepointName}`));\n\t\t\tthrow err;\n\t\t}\n\t}\n}\n\nconst isolationLevelMap: Record<\n\tMsSqlTransactionConfig['isolationLevel'],\n\t(typeof mssql.ISOLATION_LEVEL)[keyof (typeof mssql)['ISOLATION_LEVEL']]\n> = {\n\t'read uncommitted': mssql.ISOLATION_LEVEL.READ_UNCOMMITTED,\n\t'read committed': mssql.ISOLATION_LEVEL.READ_COMMITTED,\n\t'repeatable read': mssql.ISOLATION_LEVEL.REPEATABLE_READ,\n\tserializable: mssql.ISOLATION_LEVEL.SERIALIZABLE,\n\tsnapshot: mssql.ISOLATION_LEVEL.SNAPSHOT,\n};\n\nexport interface NodeMsSqlQueryResultHKT extends QueryResultHKT {\n\ttype: MsSqlQueryResult<this['row']>;\n}\n\nexport interface NodeMsSqlPreparedQueryHKT extends PreparedQueryHKT {\n\ttype: NodeMsSqlPreparedQuery<Assume<this['config'], PreparedQueryConfig>>;\n}\n"],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AACA,mBAAkB;AAClB,yBAAqB;AACrB,oBAA+B;AAE/B,oBAA2B;AAG3B,qBASO;AAEP,iBAA4D;AAC5D,mBAA0C;AAC1C,kBAAyB;AAMlB,MAAM,+BAEH,6BAAiB;AAAA,EAQ1B,YACS,QACR,aACQ,QACA,QACA,QACA,oBACP;AACD,UAAM;AAPE;AAEA;AACA;AACA;AACA;AAGR,SAAK,WAAW;AAAA,MACf,KAAK;AAAA,MACL,YAAY;AAAA,IACb;AAAA,EACD;AAAA,EApBA,QAA0B,wBAAU,IAAY;AAAA,EAExC;AAAA,EAoBR,MAAM,QACL,oBAA6C,CAAC,GACtB;AACxB,UAAM,aAAS,6BAAiB,KAAK,QAAQ,iBAAiB;AAE9D,SAAK,OAAO,SAAS,KAAK,SAAS,KAAK,MAAM;AAE9C,UAAM;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACD,IAAI;AACJ,QAAI,cAAc;AAClB,YAAI,kBAAG,QAAQ,oBAAQ,GAAG;AACzB,oBAAc,MAAM,OAAO,UAAU;AAAA,IACtC;AACA,UAAM,UAAU,YAAY,QAAQ;AACpC,eAAW,CAAC,OAAO,KAAK,KAAK,OAAO,QAAQ,GAAG;AAC9C,cAAQ,MAAM,MAAM,KAAK,IAAI,KAAK;AAAA,IACnC;AAEA,QAAI,CAAC,UAAU,CAAC,oBAAoB;AACnC,aAAO,QAAQ,MAAM,SAAS,GAAG;AAAA,IAClC;AAEA,YAAQ,eAAe;AACvB,UAAM,OAAO,MAAM,QAAQ,MAAa,SAAS,GAAG;AAEpD,QAAI,oBAAoB;AACvB,aAAO,mBAAmB,KAAK,SAAS;AAAA,IACzC;AAEA,WAAO,KAAK,UAAU,IAAI,CAAC,YAAQ,2BAA2B,QAAS,KAAK,mBAAmB,CAAC;AAAA,EACjG;AAAA,EAEA,OAAO,SACN,oBAA6C,CAAC,GAG7C;AACD,UAAM,aAAS,6BAAiB,KAAK,QAAQ,iBAAiB;AAE9D,UAAM;AAAA,MACL;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,MACA;AAAA,IACD,IAAI;AACJ,QAAI,cAAc;AAClB,YAAI,kBAAG,QAAQ,oBAAQ,GAAG;AACzB,oBAAc,MAAM,OAAO,UAAU;AAAA,IACtC;AACA,UAAM,UAAU,YAAY,QAAQ;AACpC,YAAQ,SAAS;AACjB,UAAM,gBAAgB,QAAQ,UAAU,kBAAkB;AAE1D,QAAI,eAAe;AAClB,cAAQ,eAAe;AAAA,IACxB;AAEA,eAAW,CAAC,OAAO,KAAK,KAAK,OAAO,QAAQ,GAAG;AAC9C,cAAQ,MAAM,MAAM,KAAK,IAAI,KAAK;AAAA,IACnC;AAEA,UAAM,SAAS,QAAQ,iBAAiB;AAExC,YAAQ,MAAM,SAAS,GAAG;AAE1B,aAAS,eAAe;AACvB,aAAO,MAAM;AAAA,IACd;AAEA,WAAO,GAAG,QAAQ,YAAY;AAE9B,QAAI;AACH,YAAM,YAAQ,yBAAK,QAAQ,KAAK;AAChC,YAAM,cAAU,yBAAK,QAAQ,OAAO;AAEpC,aAAO,MAAM;AACZ,eAAO,OAAO;AACd,cAAM,MAAM,MAAM,QAAQ,KAAK;AAAA,UAC9B;AAAA,UACA;AAAA,UACA,IAAI,QAAQ,CAAC,YAAY,OAAO,KAAK,QAAQ,OAAO,CAAC;AAAA,QACtD,CAAC;AACD,YAAI,QAAQ,UAAc,MAAM,QAAQ,GAAG,KAAK,IAAI,WAAW,GAAI;AAClE;AAAA,QAED,WAAW,eAAe,OAAO;AAChC,gBAAM;AAAA,QACP,OAAO;AACN,cAAI,eAAe;AAClB,gBAAI,oBAAoB;AACvB,oBAAM,YAAY,mBAAmB,CAAC,GAAgB,CAAC;AACvD,oBAAM,MAAM,QAAQ,SAAS,IAAI,UAAU,CAAC,IAAI;AAAA,YACjD,OAAO;AACN,wBAAM;AAAA,gBACL;AAAA,gBACA;AAAA,gBACA;AAAA,cACD;AAAA,YACD;AAAA,UACD,OAAO;AACN,kBAAM;AAAA,UACP;AAAA,QACD;AAAA,MACD;AAAA,IACD,UAAE;AACD,aAAO,IAAI,QAAQ,YAAY;AAC/B,cAAQ,OAAO;AAAA,IAChB;AAAA,EACD;AACD;AAMO,MAAM,yBAGH,4BAKR;AAAA,EAKD,YACS,QACR,SACQ,QACA,SACP;AACD,UAAM,OAAO;AALL;AAEA;AACA;AAGR,SAAK,SAAS,QAAQ,UAAU,IAAI,yBAAW;AAAA,EAChD;AAAA,EAZA,QAA0B,wBAAU,IAAY;AAAA,EAExC;AAAA,EAYR,aACC,OACA,QACA,oBACkD;AAClD,WAAO,IAAI;AAAA,MACV,KAAK;AAAA,MACL,MAAM;AAAA,MACN,MAAM;AAAA,MACN,KAAK;AAAA,MACL;AAAA,MACA;AAAA,IACD;AAAA,EACD;AAAA;AAAA;AAAA;AAAA;AAAA,EAMA,MAAM,MAAM,OAAe,QAA8C;AACxE,SAAK,OAAO,SAAS,OAAO,MAAM;AAElC,QAAI,cAAc,KAAK;AACvB,YAAI,kBAAG,KAAK,QAAQ,oBAAQ,GAAG;AAC9B,oBAAc,MAAM,KAAK,OAAO,UAAU;AAAA,IAC3C;AACA,UAAM,UAAU,YAAY,QAAQ;AAGpC,YAAQ,eAAe;AAEvB,eAAW,CAAC,OAAO,KAAK,KAAK,OAAO,QAAQ,GAAG;AAC9C,cAAQ,MAAM,MAAM,KAAK,IAAI,KAAK;AAAA,IACnC;AAEA,WAAO,QAAQ,MAAM,KAAK;AAAA,EAC3B;AAAA,EAEA,MAAe,IAAiB,OAA0B;AACzD,UAAM,WAAW,KAAK,QAAQ,WAAW,KAAK;AAC9C,SAAK,OAAO,SAAS,SAAS,KAAK,SAAS,MAAM;AAClD,WAAO,MAAM,KAAK,MAAM,SAAS,KAAK,SAAS,MAAM,EAAE;AAAA,MACtD,CAAC,WAAW,OAAO;AAAA,IACpB;AAAA,EACD;AAAA,EAEA,MAAe,YACd,aACA,QACa;AACb,UAAM,mBAAoB,KAAK,OAA0B,YAAY;AACrE,UAAM,UAAU,IAAI;AAAA,MACnB;AAAA,MACA,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,IACN;AACA,UAAM,KAAK,IAAI;AAAA,MACd,KAAK;AAAA,MACL;AAAA,MACA,KAAK;AAAA,MACL;AAAA,IACD;AAEA,UAAM,iBAAiB;AAAA,MACtB,QAAQ,iBACL,kBAAkB,OAAO,cAAc,IACvC;AAAA,IACJ;AAEA,QAAI;AACH,YAAM,SAAS,MAAM,YAAY,EAAE;AACnC,YAAM,iBAAiB,OAAO;AAC9B,aAAO;AAAA,IACR,SAAS,KAAK;AACb,YAAM,iBAAiB,SAAS;AAChC,YAAM;AAAA,IACP;AAAA,EACD;AACD;AAEO,MAAM,6BAGH,gCAKR;AAAA,EACD,QAA0B,wBAAU,IAAY;AAAA,EAEhD,MAAe,YACd,aACa;AACb,UAAM,gBAAgB,KAAK,KAAK,cAAc,CAAC;AAC/C,UAAM,KAAK,IAAI;AAAA,MACd,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK;AAAA,MACL,KAAK,cAAc;AAAA,IACpB;AAEA,UAAM,GAAG,QAAQ,eAAI,IAAI,oBAAoB,aAAa,EAAE,CAAC;AAC7D,QAAI;AACH,YAAM,SAAS,MAAM,YAAY,EAAE;AACnC,aAAO;AAAA,IACR,SAAS,KAAK;AACb,YAAM,GAAG,QAAQ,eAAI,IAAI,wBAAwB,aAAa,EAAE,CAAC;AACjE,YAAM;AAAA,IACP;AAAA,EACD;AACD;AAEA,MAAM,oBAGF;AAAA,EACH,oBAAoB,aAAAA,QAAM,gBAAgB;AAAA,EAC1C,kBAAkB,aAAAA,QAAM,gBAAgB;AAAA,EACxC,mBAAmB,aAAAA,QAAM,gBAAgB;AAAA,EACzC,cAAc,aAAAA,QAAM,gBAAgB;AAAA,EACpC,UAAU,aAAAA,QAAM,gBAAgB;AACjC;","names":["mssql"]}

package/node-mssql/session.d.cts

@@ -0,0 +1,48 @@
+import type { ConnectionPool, IResult } from 'mssql';
+import { entityKind } from "../entity.cjs";
+import type { Logger } from "../logger.cjs";
+import type { MsSqlDialect } from "../mssql-core/dialect.cjs";
+import type { SelectedFieldsOrdered } from "../mssql-core/query-builders/select.types.cjs";
+import { MsSqlSession, MsSqlTransaction, type MsSqlTransactionConfig, PreparedQuery, type PreparedQueryConfig, type PreparedQueryHKT, type PreparedQueryKind, type QueryResultHKT } from "../mssql-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";
+import { AutoPool } from "./pool.cjs";
+export type NodeMsSqlClient = Pick<ConnectionPool, 'request'> | AutoPool;
+export type MsSqlQueryResult<T extends unknown | unknown[] = any> = IResult<T>;
+export declare class NodeMsSqlPreparedQuery<T extends PreparedQueryConfig> extends PreparedQuery<T> {
+    private client;
+    private params;
+    private logger;
+    private fields;
+    private customResultMapper?;
+    static readonly [entityKind]: string;
+    private rawQuery;
+    constructor(client: NodeMsSqlClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined);
+    execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
+    iterator(placeholderValues?: Record<string, unknown>): AsyncGenerator<T['execute'] extends any[] ? T['execute'][number] : T['execute']>;
+}
+export interface NodeMsSqlSessionOptions {
+    logger?: Logger;
+}
+export declare class NodeMsSqlSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MsSqlSession<NodeMsSqlQueryResultHKT, NodeMsSqlPreparedQueryHKT, TFullSchema, TSchema> {
+    private client;
+    private schema;
+    private options;
+    static readonly [entityKind]: string;
+    private logger;
+    constructor(client: NodeMsSqlClient, dialect: MsSqlDialect, schema: RelationalSchemaConfig<TSchema> | undefined, options: NodeMsSqlSessionOptions);
+    prepareQuery<T extends PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute']): PreparedQueryKind<NodeMsSqlPreparedQueryHKT, T>;
+    all<T = unknown>(query: SQL): Promise<T[]>;
+    transaction<T>(transaction: (tx: NodeMsSqlTransaction<TFullSchema, TSchema>) => Promise<T>, config?: MsSqlTransactionConfig): Promise<T>;
+}
+export declare class NodeMsSqlTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MsSqlTransaction<NodeMsSqlQueryResultHKT, NodeMsSqlPreparedQueryHKT, TFullSchema, TSchema> {
+    static readonly [entityKind]: string;
+    transaction<T>(transaction: (tx: NodeMsSqlTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}
+export interface NodeMsSqlQueryResultHKT extends QueryResultHKT {
+    type: MsSqlQueryResult<this['row']>;
+}
+export interface NodeMsSqlPreparedQueryHKT extends PreparedQueryHKT {
+    type: NodeMsSqlPreparedQuery<Assume<this['config'], PreparedQueryConfig>>;
+}

package/node-mssql/session.d.ts

@@ -0,0 +1,48 @@
+import type { ConnectionPool, IResult } from 'mssql';
+import { entityKind } from "../entity.js";
+import type { Logger } from "../logger.js";
+import type { MsSqlDialect } from "../mssql-core/dialect.js";
+import type { SelectedFieldsOrdered } from "../mssql-core/query-builders/select.types.js";
+import { MsSqlSession, MsSqlTransaction, type MsSqlTransactionConfig, PreparedQuery, type PreparedQueryConfig, type PreparedQueryHKT, type PreparedQueryKind, type QueryResultHKT } from "../mssql-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";
+import { AutoPool } from "./pool.js";
+export type NodeMsSqlClient = Pick<ConnectionPool, 'request'> | AutoPool;
+export type MsSqlQueryResult<T extends unknown | unknown[] = any> = IResult<T>;
+export declare class NodeMsSqlPreparedQuery<T extends PreparedQueryConfig> extends PreparedQuery<T> {
+    private client;
+    private params;
+    private logger;
+    private fields;
+    private customResultMapper?;
+    static readonly [entityKind]: string;
+    private rawQuery;
+    constructor(client: NodeMsSqlClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined);
+    execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
+    iterator(placeholderValues?: Record<string, unknown>): AsyncGenerator<T['execute'] extends any[] ? T['execute'][number] : T['execute']>;
+}
+export interface NodeMsSqlSessionOptions {
+    logger?: Logger;
+}
+export declare class NodeMsSqlSession<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MsSqlSession<NodeMsSqlQueryResultHKT, NodeMsSqlPreparedQueryHKT, TFullSchema, TSchema> {
+    private client;
+    private schema;
+    private options;
+    static readonly [entityKind]: string;
+    private logger;
+    constructor(client: NodeMsSqlClient, dialect: MsSqlDialect, schema: RelationalSchemaConfig<TSchema> | undefined, options: NodeMsSqlSessionOptions);
+    prepareQuery<T extends PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, customResultMapper?: (rows: unknown[][]) => T['execute']): PreparedQueryKind<NodeMsSqlPreparedQueryHKT, T>;
+    all<T = unknown>(query: SQL): Promise<T[]>;
+    transaction<T>(transaction: (tx: NodeMsSqlTransaction<TFullSchema, TSchema>) => Promise<T>, config?: MsSqlTransactionConfig): Promise<T>;
+}
+export declare class NodeMsSqlTransaction<TFullSchema extends Record<string, unknown>, TSchema extends TablesRelationalConfig> extends MsSqlTransaction<NodeMsSqlQueryResultHKT, NodeMsSqlPreparedQueryHKT, TFullSchema, TSchema> {
+    static readonly [entityKind]: string;
+    transaction<T>(transaction: (tx: NodeMsSqlTransaction<TFullSchema, TSchema>) => Promise<T>): Promise<T>;
+}
+export interface NodeMsSqlQueryResultHKT extends QueryResultHKT {
+    type: MsSqlQueryResult<this['row']>;
+}
+export interface NodeMsSqlPreparedQueryHKT extends PreparedQueryHKT {
+    type: NodeMsSqlPreparedQuery<Assume<this['config'], PreparedQueryConfig>>;
+}

package/node-mssql/session.js

@@ -0,0 +1,224 @@
+import mssql from "mssql";
+import { once } from "node:events";
+import { entityKind, is } from "../entity.js";
+import { NoopLogger } from "../logger.js";
+import {
+  MsSqlSession,
+  MsSqlTransaction,
+  PreparedQuery
+} from "../mssql-core/session.js";
+import { fillPlaceholders, sql } from "../sql/sql.js";
+import { mapResultRow } from "../utils.js";
+import { AutoPool } from "./pool.js";
+class NodeMsSqlPreparedQuery extends PreparedQuery {
+  constructor(client, queryString, params, logger, fields, customResultMapper) {
+    super();
+    this.client = client;
+    this.params = params;
+    this.logger = logger;
+    this.fields = fields;
+    this.customResultMapper = customResultMapper;
+    this.rawQuery = {
+      sql: queryString,
+      parameters: params
+    };
+  }
+  static [entityKind] = "NodeMsSqlPreparedQuery";
+  rawQuery;
+  async execute(placeholderValues = {}) {
+    const params = fillPlaceholders(this.params, placeholderValues);
+    this.logger.logQuery(this.rawQuery.sql, params);
+    const {
+      fields,
+      client,
+      rawQuery,
+      joinsNotNullableMap,
+      customResultMapper
+    } = this;
+    let queryClient = client;
+    if (is(client, AutoPool)) {
+      queryClient = await client.$instance();
+    }
+    const request = queryClient.request();
+    for (const [index, param] of params.entries()) {
+      request.input(`par${index}`, param);
+    }
+    if (!fields && !customResultMapper) {
+      return request.query(rawQuery.sql);
+    }
+    request.arrayRowMode = true;
+    const rows = await request.query(rawQuery.sql);
+    if (customResultMapper) {
+      return customResultMapper(rows.recordset);
+    }
+    return rows.recordset.map((row) => mapResultRow(fields, row, joinsNotNullableMap));
+  }
+  async *iterator(placeholderValues = {}) {
+    const params = fillPlaceholders(this.params, placeholderValues);
+    const {
+      fields,
+      rawQuery,
+      joinsNotNullableMap,
+      client,
+      customResultMapper
+    } = this;
+    let queryClient = client;
+    if (is(client, AutoPool)) {
+      queryClient = await client.$instance();
+    }
+    const request = queryClient.request();
+    request.stream = true;
+    const hasRowsMapper = Boolean(fields || customResultMapper);
+    if (hasRowsMapper) {
+      request.arrayRowMode = true;
+    }
+    for (const [index, param] of params.entries()) {
+      request.input(`par${index}`, param);
+    }
+    const stream = request.toReadableStream();
+    request.query(rawQuery.sql);
+    function dataListener() {
+      stream.pause();
+    }
+    stream.on("data", dataListener);
+    try {
+      const onEnd = once(stream, "end");
+      const onError = once(stream, "error");
+      while (true) {
+        stream.resume();
+        const row = await Promise.race([
+          onEnd,
+          onError,
+          new Promise((resolve) => stream.once("data", resolve))
+        ]);
+        if (row === void 0 || Array.isArray(row) && row.length === 0) {
+          break;
+        } else if (row instanceof Error) {
+          throw row;
+        } else {
+          if (hasRowsMapper) {
+            if (customResultMapper) {
+              const mappedRow = customResultMapper([row]);
+              yield Array.isArray(mappedRow) ? mappedRow[0] : mappedRow;
+            } else {
+              yield mapResultRow(
+                fields,
+                row,
+                joinsNotNullableMap
+              );
+            }
+          } else {
+            yield row;
+          }
+        }
+      }
+    } finally {
+      stream.off("data", dataListener);
+      request.cancel();
+    }
+  }
+}
+class NodeMsSqlSession extends MsSqlSession {
+  constructor(client, dialect, schema, options) {
+    super(dialect);
+    this.client = client;
+    this.schema = schema;
+    this.options = options;
+    this.logger = options.logger ?? new NoopLogger();
+  }
+  static [entityKind] = "NodeMsSqlSession";
+  logger;
+  prepareQuery(query, fields, customResultMapper) {
+    return new NodeMsSqlPreparedQuery(
+      this.client,
+      query.sql,
+      query.params,
+      this.logger,
+      fields,
+      customResultMapper
+    );
+  }
+  /**
+   * @internal
+   * What is its purpose?
+   */
+  async query(query, params) {
+    this.logger.logQuery(query, params);
+    let queryClient = this.client;
+    if (is(this.client, AutoPool)) {
+      queryClient = await this.client.$instance();
+    }
+    const request = queryClient.request();
+    request.arrayRowMode = true;
+    for (const [index, param] of params.entries()) {
+      request.input(`par${index}`, param);
+    }
+    return request.query(query);
+  }
+  async all(query) {
+    const querySql = this.dialect.sqlToQuery(query);
+    this.logger.logQuery(querySql.sql, querySql.params);
+    return await this.query(querySql.sql, querySql.params).then(
+      (result) => result.recordset
+    );
+  }
+  async transaction(transaction, config) {
+    const mssqlTransaction = this.client.transaction();
+    const session = new NodeMsSqlSession(
+      mssqlTransaction,
+      this.dialect,
+      this.schema,
+      this.options
+    );
+    const tx = new NodeMsSqlTransaction(
+      this.dialect,
+      session,
+      this.schema,
+      0
+    );
+    await mssqlTransaction.begin(
+      config?.isolationLevel ? isolationLevelMap[config.isolationLevel] : void 0
+    );
+    try {
+      const result = await transaction(tx);
+      await mssqlTransaction.commit();
+      return result;
+    } catch (err) {
+      await mssqlTransaction.rollback();
+      throw err;
+    }
+  }
+}
+class NodeMsSqlTransaction extends MsSqlTransaction {
+  static [entityKind] = "NodeMsSqlTransaction";
+  async transaction(transaction) {
+    const savepointName = `sp${this.nestedIndex + 1}`;
+    const tx = new NodeMsSqlTransaction(
+      this.dialect,
+      this.session,
+      this.schema,
+      this.nestedIndex + 1
+    );
+    await tx.execute(sql.raw(`save transaction ${savepointName}`));
+    try {
+      const result = await transaction(tx);
+      return result;
+    } catch (err) {
+      await tx.execute(sql.raw(`rollback transaction ${savepointName}`));
+      throw err;
+    }
+  }
+}
+const isolationLevelMap = {
+  "read uncommitted": mssql.ISOLATION_LEVEL.READ_UNCOMMITTED,
+  "read committed": mssql.ISOLATION_LEVEL.READ_COMMITTED,
+  "repeatable read": mssql.ISOLATION_LEVEL.REPEATABLE_READ,
+  serializable: mssql.ISOLATION_LEVEL.SERIALIZABLE,
+  snapshot: mssql.ISOLATION_LEVEL.SNAPSHOT
+};
+export {
+  NodeMsSqlPreparedQuery,
+  NodeMsSqlSession,
+  NodeMsSqlTransaction
+};
+//# sourceMappingURL=session.js.map