drizzle-orm 1.0.0-beta.21 → 1.0.0-beta.21-4fa69ab

package/cockroach/driver.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"driver.cjs","names":["entityKind","NodeCockroachSession","CockroachDatabase","CockroachDialect","DefaultLogger","V1"],"sources":["../../src/cockroach/driver.ts"],"sourcesContent":["import pg, { type Pool, type PoolConfig } from 'pg';\nimport * as V1 from '~/_relations.ts';\nimport { CockroachDatabase } from '~/cockroach-core/db.ts';\nimport { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { entityKind } from '~/entity.ts';\nimport type { Logger } from '~/logger.ts';\nimport { DefaultLogger } from '~/logger.ts';\nimport type { DrizzleConfig } from '~/utils.ts';\nimport type { NodeCockroachClient, NodeCockroachQueryResultHKT } from './session.ts';\nimport { NodeCockroachSession } from './session.ts';\n\nexport interface CockroachDriverOptions {\n\tlogger?: Logger;\n}\n\nexport class NodeCockroachDriver {\n\tstatic readonly [entityKind]: string = 'NodeCockroachDriver';\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tprivate dialect: CockroachDialect,\n\t\tprivate options: CockroachDriverOptions = {},\n\t) {\n\t}\n\n\tcreateSession(\n\t\tschema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined,\n\t): NodeCockroachSession<Record<string, unknown>, V1.TablesRelationalConfig> {\n\t\treturn new NodeCockroachSession(this.client, this.dialect, schema, { logger: this.options.logger });\n\t}\n}\n\nexport class NodeCockroachDatabase<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n> extends CockroachDatabase<NodeCockroachQueryResultHKT, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachDatabase';\n}\n\nfunction construct<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = NodeCockroachClient,\n>(\n\tclient: TClient,\n\tconfig: DrizzleConfig<TSchema> = {},\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tconst dialect = new CockroachDialect({ casing: config.casing });\n\tlet logger;\n\tif (config.logger === true) {\n\t\tlogger = new DefaultLogger();\n\t} else if (config.logger !== false) {\n\t\tlogger = config.logger;\n\t}\n\n\tlet schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined;\n\tif (config.schema) {\n\t\tconst tablesConfig = V1.extractTablesRelationalConfig(\n\t\t\tconfig.schema,\n\t\t\tV1.createTableRelationsHelpers,\n\t\t);\n\t\tschema = {\n\t\t\tfullSchema: config.schema,\n\t\t\tschema: tablesConfig.tables,\n\t\t\ttableNamesMap: tablesConfig.tableNamesMap,\n\t\t};\n\t}\n\n\tconst driver = new NodeCockroachDriver(client, dialect, { logger });\n\tconst session = driver.createSession(schema);\n\tconst db = new NodeCockroachDatabase(dialect, session, schema as any) as NodeCockroachDatabase<TSchema>;\n\t(<any> db).$client = client;\n\n\treturn db as any;\n}\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = Pool,\n>(\n\t...params:\n\t\t| [\n\t\t\tstring,\n\t\t]\n\t\t| [\n\t\t\tstring,\n\t\t\tDrizzleConfig<TSchema>,\n\t\t]\n\t\t| [\n\t\t\t(\n\t\t\t\t& DrizzleConfig<TSchema>\n\t\t\t\t& ({\n\t\t\t\t\tconnection: string | PoolConfig;\n\t\t\t\t} | {\n\t\t\t\t\tclient: TClient;\n\t\t\t\t})\n\t\t\t),\n\t\t]\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tif (typeof params[0] === 'string') {\n\t\tconst instance = new pg.Pool({\n\t\t\tconnectionString: params[0],\n\t\t});\n\n\t\treturn construct(instance, params[1] as DrizzleConfig<TSchema> | undefined) as any;\n\t}\n\n\tconst { connection, client, ...drizzleConfig } = params[0] as (\n\t\t& ({ connection?: PoolConfig | string; client?: TClient })\n\t\t& DrizzleConfig<TSchema>\n\t);\n\n\tif (client) return construct(client, drizzleConfig);\n\n\tconst instance = typeof connection === 'string'\n\t\t? new pg.Pool({\n\t\t\tconnectionString: connection,\n\t\t})\n\t\t: new pg.Pool(connection!);\n\n\treturn construct(instance, drizzleConfig) as any;\n}\n\nexport namespace drizzle {\n\texport function mock<TSchema extends Record<string, unknown> = Record<string, never>>(\n\t\tconfig?: DrizzleConfig<TSchema>,\n\t): NodeCockroachDatabase<TSchema> & {\n\t\t$client: '$client is not available on drizzle.mock()';\n\t} {\n\t\treturn construct({} as any, config) as any;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;AAeA,IAAa,sBAAb,MAAiC;CAChC,QAAiBA,0BAAsB;CAEvC,YACC,AAAQ,QACR,AAAQ,SACR,AAAQ,UAAkC,EAAE,EAC3C;EAHO;EACA;EACA;;CAIT,cACC,QAC2E;AAC3E,SAAO,IAAIC,+CAAqB,KAAK,QAAQ,KAAK,SAAS,QAAQ,EAAE,QAAQ,KAAK,QAAQ,QAAQ,CAAC;;;AAIrG,IAAa,wBAAb,cAEUC,yCAAwD;CACjE,QAA0BF,0BAAsB;;AAGjD,SAAS,UAIR,QACA,SAAiC,EAAE,EAGlC;CACD,MAAM,UAAU,IAAIG,6CAAiB,EAAE,QAAQ,OAAO,QAAQ,CAAC;CAC/D,IAAI;AACJ,KAAI,OAAO,WAAW,KACrB,UAAS,IAAIC,2BAAe;UAClB,OAAO,WAAW,MAC5B,UAAS,OAAO;CAGjB,IAAI;AACJ,KAAI,OAAO,QAAQ;EAClB,MAAM,eAAeC,gBAAG,8BACvB,OAAO,QACPA,gBAAG,4BACH;AACD,WAAS;GACR,YAAY,OAAO;GACnB,QAAQ,aAAa;GACrB,eAAe,aAAa;GAC5B;;CAKF,MAAM,KAAK,IAAI,sBAAsB,SAFtB,IAAI,oBAAoB,QAAQ,SAAS,EAAE,QAAQ,CAAC,CAC5C,cAAc,OAAO,EACW,OAAc;AACrE,CAAO,GAAI,UAAU;AAErB,QAAO;;AAGR,SAAgB,QAIf,GAAG,QAoBF;AACD,KAAI,OAAO,OAAO,OAAO,SAKxB,QAAO,UAJU,IAAI,WAAG,KAAK,EAC5B,kBAAkB,OAAO,IACzB,CAAC,EAEyB,OAAO,GAAyC;CAG5E,MAAM,EAAE,YAAY,QAAQ,GAAG,kBAAkB,OAAO;AAKxD,KAAI,OAAQ,QAAO,UAAU,QAAQ,cAAc;AAQnD,QAAO,UANU,OAAO,eAAe,WACpC,IAAI,WAAG,KAAK,EACb,kBAAkB,YAClB,CAAC,GACA,IAAI,WAAG,KAAK,WAAY,EAEA,cAAc;;;CAIlC,SAAS,KACf,QAGC;AACD,SAAO,UAAU,EAAE,EAAS,OAAO"}
+{"version":3,"file":"driver.cjs","names":["entityKind","NodeCockroachSession","CockroachDatabase","CockroachDialect","DefaultLogger","V1"],"sources":["../../src/cockroach/driver.ts"],"sourcesContent":["import pg, { type Pool, type PoolConfig } from 'pg';\nimport * as V1 from '~/_relations.ts';\nimport { CockroachDatabase } from '~/cockroach-core/db.ts';\nimport { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { entityKind } from '~/entity.ts';\nimport type { Logger } from '~/logger.ts';\nimport { DefaultLogger } from '~/logger.ts';\nimport type { DrizzleConfig } from '~/utils.ts';\nimport type { NodeCockroachClient, NodeCockroachQueryResultHKT } from './session.ts';\nimport { NodeCockroachSession } from './session.ts';\n\nexport interface CockroachDriverOptions {\n\tlogger?: Logger;\n\tuseJitMapper?: boolean;\n}\n\nexport class NodeCockroachDriver {\n\tstatic readonly [entityKind]: string = 'NodeCockroachDriver';\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tprivate dialect: CockroachDialect,\n\t\tprivate options: CockroachDriverOptions = {},\n\t) {\n\t}\n\n\tcreateSession(\n\t\tschema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined,\n\t): NodeCockroachSession<Record<string, unknown>, V1.TablesRelationalConfig> {\n\t\treturn new NodeCockroachSession(this.client, this.dialect, schema, {\n\t\t\tlogger: this.options.logger,\n\t\t\tuseJitMapper: this.options.useJitMapper,\n\t\t});\n\t}\n}\n\nexport class NodeCockroachDatabase<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n> extends CockroachDatabase<NodeCockroachQueryResultHKT, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachDatabase';\n}\n\nfunction construct<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = NodeCockroachClient,\n>(\n\tclient: TClient,\n\tconfig: DrizzleConfig<TSchema> = {},\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tconst dialect = new CockroachDialect({ casing: config.casing });\n\tlet logger;\n\tif (config.logger === true) {\n\t\tlogger = new DefaultLogger();\n\t} else if (config.logger !== false) {\n\t\tlogger = config.logger;\n\t}\n\n\tlet schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined;\n\tif (config.schema) {\n\t\tconst tablesConfig = V1.extractTablesRelationalConfig(\n\t\t\tconfig.schema,\n\t\t\tV1.createTableRelationsHelpers,\n\t\t);\n\t\tschema = {\n\t\t\tfullSchema: config.schema,\n\t\t\tschema: tablesConfig.tables,\n\t\t\ttableNamesMap: tablesConfig.tableNamesMap,\n\t\t};\n\t}\n\n\tconst driver = new NodeCockroachDriver(client, dialect, { logger, useJitMapper: config.useJitMappers });\n\tconst session = driver.createSession(schema);\n\tconst db = new NodeCockroachDatabase(dialect, session, schema as any) as NodeCockroachDatabase<TSchema>;\n\t(<any> db).$client = client;\n\n\treturn db as any;\n}\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = Pool,\n>(\n\t...params:\n\t\t| [\n\t\t\tstring,\n\t\t]\n\t\t| [\n\t\t\tstring,\n\t\t\tDrizzleConfig<TSchema>,\n\t\t]\n\t\t| [\n\t\t\t(\n\t\t\t\t& DrizzleConfig<TSchema>\n\t\t\t\t& ({\n\t\t\t\t\tconnection: string | PoolConfig;\n\t\t\t\t} | {\n\t\t\t\t\tclient: TClient;\n\t\t\t\t})\n\t\t\t),\n\t\t]\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tif (typeof params[0] === 'string') {\n\t\tconst instance = new pg.Pool({\n\t\t\tconnectionString: params[0],\n\t\t});\n\n\t\treturn construct(instance, params[1] as DrizzleConfig<TSchema> | undefined) as any;\n\t}\n\n\tconst { connection, client, ...drizzleConfig } = params[0] as (\n\t\t& ({ connection?: PoolConfig | string; client?: TClient })\n\t\t& DrizzleConfig<TSchema>\n\t);\n\n\tif (client) return construct(client, drizzleConfig);\n\n\tconst instance = typeof connection === 'string'\n\t\t? new pg.Pool({\n\t\t\tconnectionString: connection,\n\t\t})\n\t\t: new pg.Pool(connection!);\n\n\treturn construct(instance, drizzleConfig) as any;\n}\n\nexport namespace drizzle {\n\texport function mock<TSchema extends Record<string, unknown> = Record<string, never>>(\n\t\tconfig?: DrizzleConfig<TSchema>,\n\t): NodeCockroachDatabase<TSchema> & {\n\t\t$client: '$client is not available on drizzle.mock()';\n\t} {\n\t\treturn construct({} as any, config) as any;\n\t}\n}\n"],"mappings":";;;;;;;;;;;;;AAgBA,IAAa,sBAAb,MAAiC;CAChC,QAAiBA,0BAAsB;CAEvC,YACC,AAAQ,QACR,AAAQ,SACR,AAAQ,UAAkC,EAAE,EAC3C;EAHO;EACA;EACA;;CAIT,cACC,QAC2E;AAC3E,SAAO,IAAIC,+CAAqB,KAAK,QAAQ,KAAK,SAAS,QAAQ;GAClE,QAAQ,KAAK,QAAQ;GACrB,cAAc,KAAK,QAAQ;GAC3B,CAAC;;;AAIJ,IAAa,wBAAb,cAEUC,yCAAwD;CACjE,QAA0BF,0BAAsB;;AAGjD,SAAS,UAIR,QACA,SAAiC,EAAE,EAGlC;CACD,MAAM,UAAU,IAAIG,6CAAiB,EAAE,QAAQ,OAAO,QAAQ,CAAC;CAC/D,IAAI;AACJ,KAAI,OAAO,WAAW,KACrB,UAAS,IAAIC,2BAAe;UAClB,OAAO,WAAW,MAC5B,UAAS,OAAO;CAGjB,IAAI;AACJ,KAAI,OAAO,QAAQ;EAClB,MAAM,eAAeC,gBAAG,8BACvB,OAAO,QACPA,gBAAG,4BACH;AACD,WAAS;GACR,YAAY,OAAO;GACnB,QAAQ,aAAa;GACrB,eAAe,aAAa;GAC5B;;CAKF,MAAM,KAAK,IAAI,sBAAsB,SAFtB,IAAI,oBAAoB,QAAQ,SAAS;EAAE;EAAQ,cAAc,OAAO;EAAe,CAAC,CAChF,cAAc,OAAO,EACW,OAAc;AACrE,CAAO,GAAI,UAAU;AAErB,QAAO;;AAGR,SAAgB,QAIf,GAAG,QAoBF;AACD,KAAI,OAAO,OAAO,OAAO,SAKxB,QAAO,UAJU,IAAI,WAAG,KAAK,EAC5B,kBAAkB,OAAO,IACzB,CAAC,EAEyB,OAAO,GAAyC;CAG5E,MAAM,EAAE,YAAY,QAAQ,GAAG,kBAAkB,OAAO;AAKxD,KAAI,OAAQ,QAAO,UAAU,QAAQ,cAAc;AAQnD,QAAO,UANU,OAAO,eAAe,WACpC,IAAI,WAAG,KAAK,EACb,kBAAkB,YAClB,CAAC,GACA,IAAI,WAAG,KAAK,WAAY,EAEA,cAAc;;;CAIlC,SAAS,KACf,QAGC;AACD,SAAO,UAAU,EAAE,EAAS,OAAO"}

package/cockroach/driver.d.cts

@@ -10,6 +10,7 @@ import { CockroachDatabase } from "../cockroach-core/db.cjs";
 //#region src/cockroach/driver.d.ts
 interface CockroachDriverOptions {
   logger?: Logger;
+  useJitMapper?: boolean;
 }
 declare class NodeCockroachDriver {
   private client;

package/cockroach/driver.d.ts

@@ -1,8 +1,8 @@
 import { NodeCockroachClient, NodeCockroachQueryResultHKT, NodeCockroachSession } from "./session.js";
 import { entityKind } from "../entity.js";
 import { DrizzleConfig } from "../utils.js";
-import * as V1 from "../_relations.js";
 import { Logger } from "../logger.js";
+import * as V1 from "../_relations.js";
 import { Pool, PoolConfig } from "pg";
 import { CockroachDatabase } from "../cockroach-core/db.js";
 import { CockroachDialect } from "../cockroach-core/dialect.js";
@@ -10,6 +10,7 @@ import { CockroachDialect } from "../cockroach-core/dialect.js";
 //#region src/cockroach/driver.d.ts
 interface CockroachDriverOptions {
   logger?: Logger;
+  useJitMapper?: boolean;
 }
 declare class NodeCockroachDriver {
   private client;

package/cockroach/driver.js

@@ -1,7 +1,7 @@
 import { NodeCockroachSession } from "./session.js";
 import { entityKind } from "../entity.js";
-import * as V1 from "../_relations.js";
 import { DefaultLogger } from "../logger.js";
+import * as V1 from "../_relations.js";
 import pg from "pg";
 import { CockroachDatabase } from "../cockroach-core/db.js";
 import { CockroachDialect } from "../cockroach-core/dialect.js";
@@ -15,7 +15,10 @@ var NodeCockroachDriver = class {
 		this.options = options;
 	}
 	createSession(schema) {
-		return new NodeCockroachSession(this.client, this.dialect, schema, { logger: this.options.logger });
+		return new NodeCockroachSession(this.client, this.dialect, schema, {
+			logger: this.options.logger,
+			useJitMapper: this.options.useJitMapper
+		});
 	}
 };
 var NodeCockroachDatabase = class extends CockroachDatabase {
@@ -35,7 +38,10 @@ function construct(client, config = {}) {
 			tableNamesMap: tablesConfig.tableNamesMap
 		};
 	}
-	const db = new NodeCockroachDatabase(dialect, new NodeCockroachDriver(client, dialect, { logger }).createSession(schema), schema);
+	const db = new NodeCockroachDatabase(dialect, new NodeCockroachDriver(client, dialect, {
+		logger,
+		useJitMapper: config.useJitMappers
+	}).createSession(schema), schema);
 	db.$client = client;
 	return db;
 }

package/cockroach/driver.js.map

@@ -1 +1 @@
-{"version":3,"file":"driver.js","names":[],"sources":["../../src/cockroach/driver.ts"],"sourcesContent":["import pg, { type Pool, type PoolConfig } from 'pg';\nimport * as V1 from '~/_relations.ts';\nimport { CockroachDatabase } from '~/cockroach-core/db.ts';\nimport { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { entityKind } from '~/entity.ts';\nimport type { Logger } from '~/logger.ts';\nimport { DefaultLogger } from '~/logger.ts';\nimport type { DrizzleConfig } from '~/utils.ts';\nimport type { NodeCockroachClient, NodeCockroachQueryResultHKT } from './session.ts';\nimport { NodeCockroachSession } from './session.ts';\n\nexport interface CockroachDriverOptions {\n\tlogger?: Logger;\n}\n\nexport class NodeCockroachDriver {\n\tstatic readonly [entityKind]: string = 'NodeCockroachDriver';\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tprivate dialect: CockroachDialect,\n\t\tprivate options: CockroachDriverOptions = {},\n\t) {\n\t}\n\n\tcreateSession(\n\t\tschema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined,\n\t): NodeCockroachSession<Record<string, unknown>, V1.TablesRelationalConfig> {\n\t\treturn new NodeCockroachSession(this.client, this.dialect, schema, { logger: this.options.logger });\n\t}\n}\n\nexport class NodeCockroachDatabase<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n> extends CockroachDatabase<NodeCockroachQueryResultHKT, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachDatabase';\n}\n\nfunction construct<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = NodeCockroachClient,\n>(\n\tclient: TClient,\n\tconfig: DrizzleConfig<TSchema> = {},\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tconst dialect = new CockroachDialect({ casing: config.casing });\n\tlet logger;\n\tif (config.logger === true) {\n\t\tlogger = new DefaultLogger();\n\t} else if (config.logger !== false) {\n\t\tlogger = config.logger;\n\t}\n\n\tlet schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined;\n\tif (config.schema) {\n\t\tconst tablesConfig = V1.extractTablesRelationalConfig(\n\t\t\tconfig.schema,\n\t\t\tV1.createTableRelationsHelpers,\n\t\t);\n\t\tschema = {\n\t\t\tfullSchema: config.schema,\n\t\t\tschema: tablesConfig.tables,\n\t\t\ttableNamesMap: tablesConfig.tableNamesMap,\n\t\t};\n\t}\n\n\tconst driver = new NodeCockroachDriver(client, dialect, { logger });\n\tconst session = driver.createSession(schema);\n\tconst db = new NodeCockroachDatabase(dialect, session, schema as any) as NodeCockroachDatabase<TSchema>;\n\t(<any> db).$client = client;\n\n\treturn db as any;\n}\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = Pool,\n>(\n\t...params:\n\t\t| [\n\t\t\tstring,\n\t\t]\n\t\t| [\n\t\t\tstring,\n\t\t\tDrizzleConfig<TSchema>,\n\t\t]\n\t\t| [\n\t\t\t(\n\t\t\t\t& DrizzleConfig<TSchema>\n\t\t\t\t& ({\n\t\t\t\t\tconnection: string | PoolConfig;\n\t\t\t\t} | {\n\t\t\t\t\tclient: TClient;\n\t\t\t\t})\n\t\t\t),\n\t\t]\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tif (typeof params[0] === 'string') {\n\t\tconst instance = new pg.Pool({\n\t\t\tconnectionString: params[0],\n\t\t});\n\n\t\treturn construct(instance, params[1] as DrizzleConfig<TSchema> | undefined) as any;\n\t}\n\n\tconst { connection, client, ...drizzleConfig } = params[0] as (\n\t\t& ({ connection?: PoolConfig | string; client?: TClient })\n\t\t& DrizzleConfig<TSchema>\n\t);\n\n\tif (client) return construct(client, drizzleConfig);\n\n\tconst instance = typeof connection === 'string'\n\t\t? new pg.Pool({\n\t\t\tconnectionString: connection,\n\t\t})\n\t\t: new pg.Pool(connection!);\n\n\treturn construct(instance, drizzleConfig) as any;\n}\n\nexport namespace drizzle {\n\texport function mock<TSchema extends Record<string, unknown> = Record<string, never>>(\n\t\tconfig?: DrizzleConfig<TSchema>,\n\t): NodeCockroachDatabase<TSchema> & {\n\t\t$client: '$client is not available on drizzle.mock()';\n\t} {\n\t\treturn construct({} as any, config) as any;\n\t}\n}\n"],"mappings":";;;;;;;;;AAeA,IAAa,sBAAb,MAAiC;CAChC,QAAiB,cAAsB;CAEvC,YACC,AAAQ,QACR,AAAQ,SACR,AAAQ,UAAkC,EAAE,EAC3C;EAHO;EACA;EACA;;CAIT,cACC,QAC2E;AAC3E,SAAO,IAAI,qBAAqB,KAAK,QAAQ,KAAK,SAAS,QAAQ,EAAE,QAAQ,KAAK,QAAQ,QAAQ,CAAC;;;AAIrG,IAAa,wBAAb,cAEU,kBAAwD;CACjE,QAA0B,cAAsB;;AAGjD,SAAS,UAIR,QACA,SAAiC,EAAE,EAGlC;CACD,MAAM,UAAU,IAAI,iBAAiB,EAAE,QAAQ,OAAO,QAAQ,CAAC;CAC/D,IAAI;AACJ,KAAI,OAAO,WAAW,KACrB,UAAS,IAAI,eAAe;UAClB,OAAO,WAAW,MAC5B,UAAS,OAAO;CAGjB,IAAI;AACJ,KAAI,OAAO,QAAQ;EAClB,MAAM,eAAe,GAAG,8BACvB,OAAO,QACP,GAAG,4BACH;AACD,WAAS;GACR,YAAY,OAAO;GACnB,QAAQ,aAAa;GACrB,eAAe,aAAa;GAC5B;;CAKF,MAAM,KAAK,IAAI,sBAAsB,SAFtB,IAAI,oBAAoB,QAAQ,SAAS,EAAE,QAAQ,CAAC,CAC5C,cAAc,OAAO,EACW,OAAc;AACrE,CAAO,GAAI,UAAU;AAErB,QAAO;;AAGR,SAAgB,QAIf,GAAG,QAoBF;AACD,KAAI,OAAO,OAAO,OAAO,SAKxB,QAAO,UAJU,IAAI,GAAG,KAAK,EAC5B,kBAAkB,OAAO,IACzB,CAAC,EAEyB,OAAO,GAAyC;CAG5E,MAAM,EAAE,YAAY,QAAQ,GAAG,kBAAkB,OAAO;AAKxD,KAAI,OAAQ,QAAO,UAAU,QAAQ,cAAc;AAQnD,QAAO,UANU,OAAO,eAAe,WACpC,IAAI,GAAG,KAAK,EACb,kBAAkB,YAClB,CAAC,GACA,IAAI,GAAG,KAAK,WAAY,EAEA,cAAc;;;CAIlC,SAAS,KACf,QAGC;AACD,SAAO,UAAU,EAAE,EAAS,OAAO"}
+{"version":3,"file":"driver.js","names":[],"sources":["../../src/cockroach/driver.ts"],"sourcesContent":["import pg, { type Pool, type PoolConfig } from 'pg';\nimport * as V1 from '~/_relations.ts';\nimport { CockroachDatabase } from '~/cockroach-core/db.ts';\nimport { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { entityKind } from '~/entity.ts';\nimport type { Logger } from '~/logger.ts';\nimport { DefaultLogger } from '~/logger.ts';\nimport type { DrizzleConfig } from '~/utils.ts';\nimport type { NodeCockroachClient, NodeCockroachQueryResultHKT } from './session.ts';\nimport { NodeCockroachSession } from './session.ts';\n\nexport interface CockroachDriverOptions {\n\tlogger?: Logger;\n\tuseJitMapper?: boolean;\n}\n\nexport class NodeCockroachDriver {\n\tstatic readonly [entityKind]: string = 'NodeCockroachDriver';\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tprivate dialect: CockroachDialect,\n\t\tprivate options: CockroachDriverOptions = {},\n\t) {\n\t}\n\n\tcreateSession(\n\t\tschema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined,\n\t): NodeCockroachSession<Record<string, unknown>, V1.TablesRelationalConfig> {\n\t\treturn new NodeCockroachSession(this.client, this.dialect, schema, {\n\t\t\tlogger: this.options.logger,\n\t\t\tuseJitMapper: this.options.useJitMapper,\n\t\t});\n\t}\n}\n\nexport class NodeCockroachDatabase<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n> extends CockroachDatabase<NodeCockroachQueryResultHKT, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachDatabase';\n}\n\nfunction construct<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = NodeCockroachClient,\n>(\n\tclient: TClient,\n\tconfig: DrizzleConfig<TSchema> = {},\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tconst dialect = new CockroachDialect({ casing: config.casing });\n\tlet logger;\n\tif (config.logger === true) {\n\t\tlogger = new DefaultLogger();\n\t} else if (config.logger !== false) {\n\t\tlogger = config.logger;\n\t}\n\n\tlet schema: V1.RelationalSchemaConfig<V1.TablesRelationalConfig> | undefined;\n\tif (config.schema) {\n\t\tconst tablesConfig = V1.extractTablesRelationalConfig(\n\t\t\tconfig.schema,\n\t\t\tV1.createTableRelationsHelpers,\n\t\t);\n\t\tschema = {\n\t\t\tfullSchema: config.schema,\n\t\t\tschema: tablesConfig.tables,\n\t\t\ttableNamesMap: tablesConfig.tableNamesMap,\n\t\t};\n\t}\n\n\tconst driver = new NodeCockroachDriver(client, dialect, { logger, useJitMapper: config.useJitMappers });\n\tconst session = driver.createSession(schema);\n\tconst db = new NodeCockroachDatabase(dialect, session, schema as any) as NodeCockroachDatabase<TSchema>;\n\t(<any> db).$client = client;\n\n\treturn db as any;\n}\n\nexport function drizzle<\n\tTSchema extends Record<string, unknown> = Record<string, never>,\n\tTClient extends NodeCockroachClient = Pool,\n>(\n\t...params:\n\t\t| [\n\t\t\tstring,\n\t\t]\n\t\t| [\n\t\t\tstring,\n\t\t\tDrizzleConfig<TSchema>,\n\t\t]\n\t\t| [\n\t\t\t(\n\t\t\t\t& DrizzleConfig<TSchema>\n\t\t\t\t& ({\n\t\t\t\t\tconnection: string | PoolConfig;\n\t\t\t\t} | {\n\t\t\t\t\tclient: TClient;\n\t\t\t\t})\n\t\t\t),\n\t\t]\n): NodeCockroachDatabase<TSchema> & {\n\t$client: TClient;\n} {\n\tif (typeof params[0] === 'string') {\n\t\tconst instance = new pg.Pool({\n\t\t\tconnectionString: params[0],\n\t\t});\n\n\t\treturn construct(instance, params[1] as DrizzleConfig<TSchema> | undefined) as any;\n\t}\n\n\tconst { connection, client, ...drizzleConfig } = params[0] as (\n\t\t& ({ connection?: PoolConfig | string; client?: TClient })\n\t\t& DrizzleConfig<TSchema>\n\t);\n\n\tif (client) return construct(client, drizzleConfig);\n\n\tconst instance = typeof connection === 'string'\n\t\t? new pg.Pool({\n\t\t\tconnectionString: connection,\n\t\t})\n\t\t: new pg.Pool(connection!);\n\n\treturn construct(instance, drizzleConfig) as any;\n}\n\nexport namespace drizzle {\n\texport function mock<TSchema extends Record<string, unknown> = Record<string, never>>(\n\t\tconfig?: DrizzleConfig<TSchema>,\n\t): NodeCockroachDatabase<TSchema> & {\n\t\t$client: '$client is not available on drizzle.mock()';\n\t} {\n\t\treturn construct({} as any, config) as any;\n\t}\n}\n"],"mappings":";;;;;;;;;AAgBA,IAAa,sBAAb,MAAiC;CAChC,QAAiB,cAAsB;CAEvC,YACC,AAAQ,QACR,AAAQ,SACR,AAAQ,UAAkC,EAAE,EAC3C;EAHO;EACA;EACA;;CAIT,cACC,QAC2E;AAC3E,SAAO,IAAI,qBAAqB,KAAK,QAAQ,KAAK,SAAS,QAAQ;GAClE,QAAQ,KAAK,QAAQ;GACrB,cAAc,KAAK,QAAQ;GAC3B,CAAC;;;AAIJ,IAAa,wBAAb,cAEU,kBAAwD;CACjE,QAA0B,cAAsB;;AAGjD,SAAS,UAIR,QACA,SAAiC,EAAE,EAGlC;CACD,MAAM,UAAU,IAAI,iBAAiB,EAAE,QAAQ,OAAO,QAAQ,CAAC;CAC/D,IAAI;AACJ,KAAI,OAAO,WAAW,KACrB,UAAS,IAAI,eAAe;UAClB,OAAO,WAAW,MAC5B,UAAS,OAAO;CAGjB,IAAI;AACJ,KAAI,OAAO,QAAQ;EAClB,MAAM,eAAe,GAAG,8BACvB,OAAO,QACP,GAAG,4BACH;AACD,WAAS;GACR,YAAY,OAAO;GACnB,QAAQ,aAAa;GACrB,eAAe,aAAa;GAC5B;;CAKF,MAAM,KAAK,IAAI,sBAAsB,SAFtB,IAAI,oBAAoB,QAAQ,SAAS;EAAE;EAAQ,cAAc,OAAO;EAAe,CAAC,CAChF,cAAc,OAAO,EACW,OAAc;AACrE,CAAO,GAAI,UAAU;AAErB,QAAO;;AAGR,SAAgB,QAIf,GAAG,QAoBF;AACD,KAAI,OAAO,OAAO,OAAO,SAKxB,QAAO,UAJU,IAAI,GAAG,KAAK,EAC5B,kBAAkB,OAAO,IACzB,CAAC,EAEyB,OAAO,GAAyC;CAG5E,MAAM,EAAE,YAAY,QAAQ,GAAG,kBAAkB,OAAO;AAKxD,KAAI,OAAQ,QAAO,UAAU,QAAQ,cAAc;AAQnD,QAAO,UANU,OAAO,eAAe,WACpC,IAAI,GAAG,KAAK,EACb,kBAAkB,YAClB,CAAC,GACA,IAAI,GAAG,KAAK,WAAY,EAEA,cAAc;;;CAIlC,SAAS,KACf,QAGC;AACD,SAAO,UAAU,EAAE,EAAS,OAAO"}

package/cockroach/session.cjs

@@ -16,7 +16,8 @@ var NodeCockroachPreparedQuery = class extends __cockroach_core_session_ts.Cockr
 	static [__entity_ts.entityKind] = "NodeCockroachPreparedQuery";
 	rawQueryConfig;
 	queryConfig;
-	constructor(client, queryString, params, logger, fields, name, _isResponseInArrayMode, customResultMapper) {
+	jitMapper;
+	constructor(client, queryString, params, logger, fields, name, useJitMapper, customResultMapper) {
 		super({
 			sql: queryString,
 			params
@@ -25,7 +26,7 @@ var NodeCockroachPreparedQuery = class extends __cockroach_core_session_ts.Cockr
 		this.params = params;
 		this.logger = logger;
 		this.fields = fields;
-		this._isResponseInArrayMode = _isResponseInArrayMode;
+		this.useJitMapper = useJitMapper;
 		this.customResultMapper = customResultMapper;
 		this.rawQueryConfig = {
 			name,
@@ -83,7 +84,8 @@ var NodeCockroachPreparedQuery = class extends __cockroach_core_session_ts.Cockr
 				return client.query(query, params);
 			});
 			return __tracing_ts.tracer.startActiveSpan("drizzle.mapResponse", () => {
-				return customResultMapper ? customResultMapper(result.rows) : result.rows.map((row) => (0, __utils_ts.mapResultRow)(fields, row, joinsNotNullableMap));
+				if (customResultMapper) return customResultMapper(result.rows);
+				return this.useJitMapper ? (this.jitMapper ??= (0, __utils_ts.makeJitQueryMapper)(fields, joinsNotNullableMap))(result.rows) : result.rows.map((row) => (0, __utils_ts.mapResultRow)(fields, row, joinsNotNullableMap));
 			});
 		});
 	}
@@ -101,10 +103,6 @@ var NodeCockroachPreparedQuery = class extends __cockroach_core_session_ts.Cockr
 			});
 		});
 	}
-	/** @internal */
-	isResponseInArrayMode() {
-		return this._isResponseInArrayMode;
-	}
 };
 var NodeCockroachSession = class NodeCockroachSession extends __cockroach_core_session_ts.CockroachSession {
 	static [__entity_ts.entityKind] = "NodeCockroachSession";
@@ -116,8 +114,8 @@ var NodeCockroachSession = class NodeCockroachSession extends __cockroach_core_s
 		this.options = options;
 		this.logger = options.logger ?? new __logger_ts.NoopLogger();
 	}
-	prepareQuery(query, fields, name, isResponseInArrayMode, customResultMapper) {
-		return new NodeCockroachPreparedQuery(this.client, query.sql, query.params, this.logger, fields, name, isResponseInArrayMode, customResultMapper);
+	prepareQuery(query, fields, name, customResultMapper) {
+		return new NodeCockroachPreparedQuery(this.client, query.sql, query.params, this.logger, fields, name, this.options.useJitMapper, customResultMapper);
 	}
 	async transaction(transaction, config) {
 		const session = this.client instanceof Pool ? new NodeCockroachSession(await this.client.connect(), this.dialect, this.schema, this.options) : this;

package/cockroach/session.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"session.cjs","names":["CockroachPreparedQuery","entityKind","tracer","CockroachSession","NoopLogger","CockroachTransaction","sql"],"sources":["../../src/cockroach/session.ts"],"sourcesContent":["import type { Client, PoolClient, QueryArrayConfig, QueryConfig, QueryResult, QueryResultRow } from 'pg';\nimport pg from 'pg';\nimport type * as V1 from '~/_relations.ts';\nimport type { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { CockroachTransaction } from '~/cockroach-core/index.ts';\nimport type { SelectedFieldsOrdered } from '~/cockroach-core/query-builders/select.types.ts';\nimport type {\n\tCockroachQueryResultHKT,\n\tCockroachTransactionConfig,\n\tPreparedQueryConfig,\n} from '~/cockroach-core/session.ts';\nimport { CockroachPreparedQuery, CockroachSession } from '~/cockroach-core/session.ts';\nimport { entityKind } from '~/entity.ts';\nimport { type Logger, NoopLogger } from '~/logger.ts';\nimport { fillPlaceholders, type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { tracer } from '~/tracing.ts';\nimport { type Assume, mapResultRow } from '~/utils.ts';\n\nconst { Pool, types } = pg;\n\nexport type NodeCockroachClient = pg.Pool | PoolClient | Client;\n\nexport class NodeCockroachPreparedQuery<T extends PreparedQueryConfig> extends CockroachPreparedQuery<T> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachPreparedQuery';\n\n\tprivate rawQueryConfig: QueryConfig;\n\tprivate queryConfig: QueryArrayConfig;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tqueryString: string,\n\t\tprivate params: unknown[],\n\t\tprivate logger: Logger,\n\t\tprivate fields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tprivate _isResponseInArrayMode: boolean,\n\t\tprivate customResultMapper?: (rows: unknown[][]) => T['execute'],\n\t) {\n\t\tsuper({ sql: queryString, params });\n\t\tthis.rawQueryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t\tthis.queryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\trowMode: 'array',\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t}\n\n\tasync execute(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['execute']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', async () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\n\t\t\tconst { fields, rawQueryConfig: rawQuery, client, queryConfig: query, joinsNotNullableMap, customResultMapper } =\n\t\t\t\tthis;\n\t\t\tif (!fields && !customResultMapper) {\n\t\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', async (span) => {\n\t\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t\t'drizzle.query.name': rawQuery.name,\n\t\t\t\t\t\t'drizzle.query.text': rawQuery.text,\n\t\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t\t});\n\t\t\t\t\treturn client.query(rawQuery, params);\n\t\t\t\t});\n\t\t\t}\n\n\t\t\tconst result = await tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': query.name,\n\t\t\t\t\t'drizzle.query.text': query.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn client.query(query, params);\n\t\t\t});\n\n\t\t\treturn tracer.startActiveSpan('drizzle.mapResponse', () => {\n\t\t\t\treturn customResultMapper\n\t\t\t\t\t? customResultMapper(result.rows)\n\t\t\t\t\t: result.rows.map((row) => mapResultRow<T['execute']>(fields!, row, joinsNotNullableMap));\n\t\t\t});\n\t\t});\n\t}\n\n\tall(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['all']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': this.rawQueryConfig.name,\n\t\t\t\t\t'drizzle.query.text': this.rawQueryConfig.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn this.client.query(this.rawQueryConfig, params).then((result) => result.rows);\n\t\t\t});\n\t\t});\n\t}\n\n\t/** @internal */\n\tisResponseInArrayMode(): boolean {\n\t\treturn this._isResponseInArrayMode;\n\t}\n}\n\nexport interface NodeCockroachSessionOptions {\n\tlogger?: Logger;\n}\n\nexport class NodeCockroachSession<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachSession<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachSession';\n\n\tprivate logger: Logger;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tdialect: CockroachDialect,\n\t\tprivate schema: V1.RelationalSchemaConfig<TSchema> | undefined,\n\t\tprivate options: NodeCockroachSessionOptions = {},\n\t) {\n\t\tsuper(dialect);\n\t\tthis.logger = options.logger ?? new NoopLogger();\n\t}\n\n\tprepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tisResponseInArrayMode: boolean,\n\t\tcustomResultMapper?: (rows: unknown[][]) => T['execute'],\n\t): CockroachPreparedQuery<T> {\n\t\treturn new NodeCockroachPreparedQuery(\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\tname,\n\t\t\tisResponseInArrayMode,\n\t\t\tcustomResultMapper,\n\t\t);\n\t}\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t\tconfig?: CockroachTransactionConfig | undefined,\n\t): Promise<T> {\n\t\tconst session = this.client instanceof Pool // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t? new NodeCockroachSession(await this.client.connect(), this.dialect, this.schema, this.options)\n\t\t\t: this;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(this.dialect, session, this.schema);\n\t\tawait tx.execute(sql`begin${config ? sql` ${tx.getTransactionConfigSQL(config)}` : undefined}`);\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql`commit`);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tawait tx.execute(sql`rollback`);\n\t\t\tthrow error;\n\t\t} finally {\n\t\t\tif (this.client instanceof Pool) { // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t\t(session.client as PoolClient).release();\n\t\t\t}\n\t\t}\n\t}\n\n\toverride async count(sql: SQL): Promise<number> {\n\t\tconst res = await this.execute<{ rows: [{ count: string }] }>(sql);\n\t\treturn Number(\n\t\t\tres['rows'][0]['count'],\n\t\t);\n\t}\n}\n\nexport class NodeCockroachTransaction<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachTransaction<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachTransaction';\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t): Promise<T> {\n\t\tconst savepointName = `sp${this.nestedIndex + 1}`;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(\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\t\tawait tx.execute(sql.raw(`savepoint ${savepointName}`));\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql.raw(`release savepoint ${savepointName}`));\n\t\t\treturn result;\n\t\t} catch (err) {\n\t\t\tawait tx.execute(sql.raw(`rollback to savepoint ${savepointName}`));\n\t\t\tthrow err;\n\t\t}\n\t}\n}\n\nexport interface NodeCockroachQueryResultHKT extends CockroachQueryResultHKT {\n\ttype: QueryResult<Assume<this['row'], QueryResultRow>>;\n}\n"],"mappings":";;;;;;;;;;;;;AAkBA,MAAM,EAAE,MAAM,UAAU;AAIxB,IAAa,6BAAb,cAA+EA,mDAA0B;CACxG,QAA0BC,0BAAsB;CAEhD,AAAQ;CACR,AAAQ;CAER,YACC,AAAQ,QACR,aACA,AAAQ,QACR,AAAQ,QACR,AAAQ,QACR,MACA,AAAQ,wBACR,AAAQ,oBACP;AACD,QAAM;GAAE,KAAK;GAAa;GAAQ,CAAC;EAT3B;EAEA;EACA;EACA;EAEA;EACA;AAGR,OAAK,iBAAiB;GACrB;GACA,MAAM;GACN,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;AACD,OAAK,cAAc;GAClB;GACA,MAAM;GACN,SAAS;GACT,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;;CAGF,MAAM,QAAQ,oBAAyD,EAAE,EAAyB;AACjG,SAAOC,oBAAO,gBAAgB,mBAAmB,YAAY;GAC5D,MAAM,4CAA0B,KAAK,QAAQ,kBAAkB;AAE/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;GAEtD,MAAM,EAAE,QAAQ,gBAAgB,UAAU,QAAQ,aAAa,OAAO,qBAAqB,uBAC1F;AACD,OAAI,CAAC,UAAU,CAAC,mBACf,QAAOA,oBAAO,gBAAgB,0BAA0B,OAAO,SAAS;AACvE,UAAM,cAAc;KACnB,sBAAsB,SAAS;KAC/B,sBAAsB,SAAS;KAC/B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,UAAU,OAAO;KACpC;GAGH,MAAM,SAAS,MAAMA,oBAAO,gBAAgB,2BAA2B,SAAS;AAC/E,UAAM,cAAc;KACnB,sBAAsB,MAAM;KAC5B,sBAAsB,MAAM;KAC5B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,OAAO,OAAO;KACjC;AAEF,UAAOA,oBAAO,gBAAgB,6BAA6B;AAC1D,WAAO,qBACJ,mBAAmB,OAAO,KAAK,GAC/B,OAAO,KAAK,KAAK,qCAAmC,QAAS,KAAK,oBAAoB,CAAC;KACzF;IACD;;CAGH,IAAI,oBAAyD,EAAE,EAAqB;AACnF,SAAOA,oBAAO,gBAAgB,yBAAyB;GACtD,MAAM,4CAA0B,KAAK,QAAQ,kBAAkB;AAC/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;AACtD,UAAOA,oBAAO,gBAAgB,2BAA2B,SAAS;AACjE,UAAM,cAAc;KACnB,sBAAsB,KAAK,eAAe;KAC1C,sBAAsB,KAAK,eAAe;KAC1C,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,KAAK,OAAO,MAAM,KAAK,gBAAgB,OAAO,CAAC,MAAM,WAAW,OAAO,KAAK;KAClF;IACD;;;CAIH,wBAAiC;AAChC,SAAO,KAAK;;;AAQd,IAAa,uBAAb,MAAa,6BAGHC,6CAAoE;CAC7E,QAA0BF,0BAAsB;CAEhD,AAAQ;CAER,YACC,AAAQ,QACR,SACA,AAAQ,QACR,AAAQ,UAAuC,EAAE,EAChD;AACD,QAAM,QAAQ;EALN;EAEA;EACA;AAGR,OAAK,SAAS,QAAQ,UAAU,IAAIG,wBAAY;;CAGjD,aACC,OACA,QACA,MACA,uBACA,oBAC4B;AAC5B,SAAO,IAAI,2BACV,KAAK,QACL,MAAM,KACN,MAAM,QACN,KAAK,QACL,QACA,MACA,uBACA,mBACA;;CAGF,MAAe,YACd,aACA,QACa;EACb,MAAM,UAAU,KAAK,kBAAkB,OACpC,IAAI,qBAAqB,MAAM,KAAK,OAAO,SAAS,EAAE,KAAK,SAAS,KAAK,QAAQ,KAAK,QAAQ,GAC9F;EACH,MAAM,KAAK,IAAI,yBAA+C,KAAK,SAAS,SAAS,KAAK,OAAO;AACjG,QAAM,GAAG,QAAQ,gBAAG,QAAQ,SAAS,gBAAG,IAAI,GAAG,wBAAwB,OAAO,KAAK,SAAY;AAC/F,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQ,gBAAG,SAAS;AAC7B,UAAO;WACC,OAAO;AACf,SAAM,GAAG,QAAQ,gBAAG,WAAW;AAC/B,SAAM;YACG;AACT,OAAI,KAAK,kBAAkB,KAC1B,CAAC,QAAQ,OAAsB,SAAS;;;CAK3C,MAAe,MAAM,KAA2B;EAC/C,MAAM,MAAM,MAAM,KAAK,QAAuC,IAAI;AAClE,SAAO,OACN,IAAI,QAAQ,GAAG,SACf;;;AAIH,IAAa,2BAAb,MAAa,iCAGHC,+CAAwE;CACjF,QAA0BJ,0BAAsB;CAEhD,MAAe,YACd,aACa;EACb,MAAM,gBAAgB,KAAK,KAAK,cAAc;EAC9C,MAAM,KAAK,IAAI,yBACd,KAAK,SACL,KAAK,SACL,KAAK,QACL,KAAK,cAAc,EACnB;AACD,QAAM,GAAG,QAAQK,iBAAI,IAAI,aAAa,gBAAgB,CAAC;AACvD,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQA,iBAAI,IAAI,qBAAqB,gBAAgB,CAAC;AAC/D,UAAO;WACC,KAAK;AACb,SAAM,GAAG,QAAQA,iBAAI,IAAI,yBAAyB,gBAAgB,CAAC;AACnE,SAAM"}
+{"version":3,"file":"session.cjs","names":["CockroachPreparedQuery","entityKind","tracer","CockroachSession","NoopLogger","CockroachTransaction","sql"],"sources":["../../src/cockroach/session.ts"],"sourcesContent":["import type { Client, PoolClient, QueryArrayConfig, QueryConfig, QueryResult, QueryResultRow } from 'pg';\nimport pg from 'pg';\nimport type * as V1 from '~/_relations.ts';\nimport type { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { CockroachTransaction } from '~/cockroach-core/index.ts';\nimport type { SelectedFieldsOrdered } from '~/cockroach-core/query-builders/select.types.ts';\nimport type {\n\tCockroachQueryResultHKT,\n\tCockroachTransactionConfig,\n\tPreparedQueryConfig,\n} from '~/cockroach-core/session.ts';\nimport { CockroachPreparedQuery, CockroachSession } from '~/cockroach-core/session.ts';\nimport { entityKind } from '~/entity.ts';\nimport { type Logger, NoopLogger } from '~/logger.ts';\nimport { fillPlaceholders, type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { tracer } from '~/tracing.ts';\nimport { type Assume, makeJitQueryMapper, mapResultRow, type RowsMapper } from '~/utils.ts';\n\nconst { Pool, types } = pg;\n\nexport type NodeCockroachClient = pg.Pool | PoolClient | Client;\n\nexport class NodeCockroachPreparedQuery<T extends PreparedQueryConfig> extends CockroachPreparedQuery<T> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachPreparedQuery';\n\n\tprivate rawQueryConfig: QueryConfig;\n\tprivate queryConfig: QueryArrayConfig;\n\tprivate jitMapper?: RowsMapper<T['execute']>;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tqueryString: string,\n\t\tprivate params: unknown[],\n\t\tprivate logger: Logger,\n\t\tprivate fields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tprivate useJitMapper: boolean | undefined,\n\t\tprivate customResultMapper?: (rows: unknown[][]) => T['execute'],\n\t) {\n\t\tsuper({ sql: queryString, params });\n\t\tthis.rawQueryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t\tthis.queryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\trowMode: 'array',\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t}\n\n\tasync execute(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['execute']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', async () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\n\t\t\tconst { fields, rawQueryConfig: rawQuery, client, queryConfig: query, joinsNotNullableMap, customResultMapper } =\n\t\t\t\tthis;\n\t\t\tif (!fields && !customResultMapper) {\n\t\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', async (span) => {\n\t\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t\t'drizzle.query.name': rawQuery.name,\n\t\t\t\t\t\t'drizzle.query.text': rawQuery.text,\n\t\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t\t});\n\t\t\t\t\treturn client.query(rawQuery, params);\n\t\t\t\t});\n\t\t\t}\n\n\t\t\tconst result = await tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': query.name,\n\t\t\t\t\t'drizzle.query.text': query.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn client.query(query, params);\n\t\t\t});\n\n\t\t\treturn tracer.startActiveSpan('drizzle.mapResponse', () => {\n\t\t\t\tif (customResultMapper) {\n\t\t\t\t\treturn (customResultMapper as (rows: unknown[][]) => unknown)(result.rows);\n\t\t\t\t}\n\n\t\t\t\treturn this.useJitMapper\n\t\t\t\t\t? (this.jitMapper ??= makeJitQueryMapper(fields!, joinsNotNullableMap))(result.rows)\n\t\t\t\t\t: result.rows.map((row) => mapResultRow(fields!, row, joinsNotNullableMap));\n\t\t\t});\n\t\t});\n\t}\n\n\tall(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['all']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': this.rawQueryConfig.name,\n\t\t\t\t\t'drizzle.query.text': this.rawQueryConfig.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn this.client.query(this.rawQueryConfig, params).then((result) => result.rows);\n\t\t\t});\n\t\t});\n\t}\n}\n\nexport interface NodeCockroachSessionOptions {\n\tlogger?: Logger;\n\tuseJitMapper?: boolean;\n}\n\nexport class NodeCockroachSession<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachSession<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachSession';\n\n\tprivate logger: Logger;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tdialect: CockroachDialect,\n\t\tprivate schema: V1.RelationalSchemaConfig<TSchema> | undefined,\n\t\tprivate options: NodeCockroachSessionOptions = {},\n\t) {\n\t\tsuper(dialect);\n\t\tthis.logger = options.logger ?? new NoopLogger();\n\t}\n\n\tprepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tcustomResultMapper?: (rows: unknown[][]) => T['execute'],\n\t): CockroachPreparedQuery<T> {\n\t\treturn new NodeCockroachPreparedQuery(\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\tname,\n\t\t\tthis.options.useJitMapper,\n\t\t\tcustomResultMapper,\n\t\t);\n\t}\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t\tconfig?: CockroachTransactionConfig | undefined,\n\t): Promise<T> {\n\t\tconst session = this.client instanceof Pool // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t? new NodeCockroachSession(await this.client.connect(), this.dialect, this.schema, this.options)\n\t\t\t: this;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(this.dialect, session, this.schema);\n\t\tawait tx.execute(sql`begin${config ? sql` ${tx.getTransactionConfigSQL(config)}` : undefined}`);\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql`commit`);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tawait tx.execute(sql`rollback`);\n\t\t\tthrow error;\n\t\t} finally {\n\t\t\tif (this.client instanceof Pool) { // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t\t(session.client as PoolClient).release();\n\t\t\t}\n\t\t}\n\t}\n\n\toverride async count(sql: SQL): Promise<number> {\n\t\tconst res = await this.execute<{ rows: [{ count: string }] }>(sql);\n\t\treturn Number(\n\t\t\tres['rows'][0]['count'],\n\t\t);\n\t}\n}\n\nexport class NodeCockroachTransaction<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachTransaction<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachTransaction';\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t): Promise<T> {\n\t\tconst savepointName = `sp${this.nestedIndex + 1}`;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(\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\t\tawait tx.execute(sql.raw(`savepoint ${savepointName}`));\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql.raw(`release savepoint ${savepointName}`));\n\t\t\treturn result;\n\t\t} catch (err) {\n\t\t\tawait tx.execute(sql.raw(`rollback to savepoint ${savepointName}`));\n\t\t\tthrow err;\n\t\t}\n\t}\n}\n\nexport interface NodeCockroachQueryResultHKT extends CockroachQueryResultHKT {\n\ttype: QueryResult<Assume<this['row'], QueryResultRow>>;\n}\n"],"mappings":";;;;;;;;;;;;;AAkBA,MAAM,EAAE,MAAM,UAAU;AAIxB,IAAa,6BAAb,cAA+EA,mDAA0B;CACxG,QAA0BC,0BAAsB;CAEhD,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YACC,AAAQ,QACR,aACA,AAAQ,QACR,AAAQ,QACR,AAAQ,QACR,MACA,AAAQ,cACR,AAAQ,oBACP;AACD,QAAM;GAAE,KAAK;GAAa;GAAQ,CAAC;EAT3B;EAEA;EACA;EACA;EAEA;EACA;AAGR,OAAK,iBAAiB;GACrB;GACA,MAAM;GACN,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;AACD,OAAK,cAAc;GAClB;GACA,MAAM;GACN,SAAS;GACT,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;;CAGF,MAAM,QAAQ,oBAAyD,EAAE,EAAyB;AACjG,SAAOC,oBAAO,gBAAgB,mBAAmB,YAAY;GAC5D,MAAM,4CAA0B,KAAK,QAAQ,kBAAkB;AAE/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;GAEtD,MAAM,EAAE,QAAQ,gBAAgB,UAAU,QAAQ,aAAa,OAAO,qBAAqB,uBAC1F;AACD,OAAI,CAAC,UAAU,CAAC,mBACf,QAAOA,oBAAO,gBAAgB,0BAA0B,OAAO,SAAS;AACvE,UAAM,cAAc;KACnB,sBAAsB,SAAS;KAC/B,sBAAsB,SAAS;KAC/B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,UAAU,OAAO;KACpC;GAGH,MAAM,SAAS,MAAMA,oBAAO,gBAAgB,2BAA2B,SAAS;AAC/E,UAAM,cAAc;KACnB,sBAAsB,MAAM;KAC5B,sBAAsB,MAAM;KAC5B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,OAAO,OAAO;KACjC;AAEF,UAAOA,oBAAO,gBAAgB,6BAA6B;AAC1D,QAAI,mBACH,QAAQ,mBAAsD,OAAO,KAAK;AAG3E,WAAO,KAAK,gBACR,KAAK,iDAAiC,QAAS,oBAAoB,EAAE,OAAO,KAAK,GAClF,OAAO,KAAK,KAAK,qCAAqB,QAAS,KAAK,oBAAoB,CAAC;KAC3E;IACD;;CAGH,IAAI,oBAAyD,EAAE,EAAqB;AACnF,SAAOA,oBAAO,gBAAgB,yBAAyB;GACtD,MAAM,4CAA0B,KAAK,QAAQ,kBAAkB;AAC/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;AACtD,UAAOA,oBAAO,gBAAgB,2BAA2B,SAAS;AACjE,UAAM,cAAc;KACnB,sBAAsB,KAAK,eAAe;KAC1C,sBAAsB,KAAK,eAAe;KAC1C,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,KAAK,OAAO,MAAM,KAAK,gBAAgB,OAAO,CAAC,MAAM,WAAW,OAAO,KAAK;KAClF;IACD;;;AASJ,IAAa,uBAAb,MAAa,6BAGHC,6CAAoE;CAC7E,QAA0BF,0BAAsB;CAEhD,AAAQ;CAER,YACC,AAAQ,QACR,SACA,AAAQ,QACR,AAAQ,UAAuC,EAAE,EAChD;AACD,QAAM,QAAQ;EALN;EAEA;EACA;AAGR,OAAK,SAAS,QAAQ,UAAU,IAAIG,wBAAY;;CAGjD,aACC,OACA,QACA,MACA,oBAC4B;AAC5B,SAAO,IAAI,2BACV,KAAK,QACL,MAAM,KACN,MAAM,QACN,KAAK,QACL,QACA,MACA,KAAK,QAAQ,cACb,mBACA;;CAGF,MAAe,YACd,aACA,QACa;EACb,MAAM,UAAU,KAAK,kBAAkB,OACpC,IAAI,qBAAqB,MAAM,KAAK,OAAO,SAAS,EAAE,KAAK,SAAS,KAAK,QAAQ,KAAK,QAAQ,GAC9F;EACH,MAAM,KAAK,IAAI,yBAA+C,KAAK,SAAS,SAAS,KAAK,OAAO;AACjG,QAAM,GAAG,QAAQ,gBAAG,QAAQ,SAAS,gBAAG,IAAI,GAAG,wBAAwB,OAAO,KAAK,SAAY;AAC/F,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQ,gBAAG,SAAS;AAC7B,UAAO;WACC,OAAO;AACf,SAAM,GAAG,QAAQ,gBAAG,WAAW;AAC/B,SAAM;YACG;AACT,OAAI,KAAK,kBAAkB,KAC1B,CAAC,QAAQ,OAAsB,SAAS;;;CAK3C,MAAe,MAAM,KAA2B;EAC/C,MAAM,MAAM,MAAM,KAAK,QAAuC,IAAI;AAClE,SAAO,OACN,IAAI,QAAQ,GAAG,SACf;;;AAIH,IAAa,2BAAb,MAAa,iCAGHC,+CAAwE;CACjF,QAA0BJ,0BAAsB;CAEhD,MAAe,YACd,aACa;EACb,MAAM,gBAAgB,KAAK,KAAK,cAAc;EAC9C,MAAM,KAAK,IAAI,yBACd,KAAK,SACL,KAAK,SACL,KAAK,QACL,KAAK,cAAc,EACnB;AACD,QAAM,GAAG,QAAQK,iBAAI,IAAI,aAAa,gBAAgB,CAAC;AACvD,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQA,iBAAI,IAAI,qBAAqB,gBAAgB,CAAC;AAC/D,UAAO;WACC,KAAK;AACb,SAAM,GAAG,QAAQA,iBAAI,IAAI,yBAAyB,gBAAgB,CAAC;AACnE,SAAM"}

package/cockroach/session.d.cts

@@ -16,17 +16,19 @@ declare class NodeCockroachPreparedQuery<T extends PreparedQueryConfig> extends
   private params;
   private logger;
   private fields;
-  private _isResponseInArrayMode;
+  private useJitMapper;
   private customResultMapper?;
   static readonly [entityKind]: string;
   private rawQueryConfig;
   private queryConfig;
-  constructor(client: NodeCockroachClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, name: string | undefined, _isResponseInArrayMode: boolean, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined);
+  private jitMapper?;
+  constructor(client: NodeCockroachClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, name: string | undefined, useJitMapper: boolean | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined);
   execute(placeholderValues?: Record<string, unknown> | undefined): Promise<T['execute']>;
   all(placeholderValues?: Record<string, unknown> | undefined): Promise<T['all']>;
 }
 interface NodeCockroachSessionOptions {
   logger?: Logger;
+  useJitMapper?: boolean;
 }
 declare class NodeCockroachSession<TFullSchema extends Record<string, unknown>, TSchema extends V1.TablesRelationalConfig> extends CockroachSession<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {
   private client;
@@ -35,7 +37,7 @@ declare class NodeCockroachSession<TFullSchema extends Record<string, unknown>,
   static readonly [entityKind]: string;
   private logger;
   constructor(client: NodeCockroachClient, dialect: CockroachDialect, schema: V1.RelationalSchemaConfig<TSchema> | undefined, options?: NodeCockroachSessionOptions);
-  prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, isResponseInArrayMode: boolean, customResultMapper?: (rows: unknown[][]) => T['execute']): CockroachPreparedQuery<T>;
+  prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, customResultMapper?: (rows: unknown[][]) => T['execute']): CockroachPreparedQuery<T>;
   transaction<T>(transaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>, config?: CockroachTransactionConfig | undefined): Promise<T>;
   count(sql: SQL): Promise<number>;
 }

package/cockroach/session.d.ts

@@ -1,8 +1,8 @@
 import { entityKind } from "../entity.js";
 import { Assume } from "../utils.js";
 import { Query, SQL } from "../sql/sql.js";
-import * as V1 from "../_relations.js";
 import { Logger } from "../logger.js";
+import * as V1 from "../_relations.js";
 import pg, { Client, PoolClient, QueryResult, QueryResultRow } from "pg";
 import { CockroachDialect } from "../cockroach-core/dialect.js";
 import { CockroachTransaction } from "../cockroach-core/index.js";
@@ -16,17 +16,19 @@ declare class NodeCockroachPreparedQuery<T extends PreparedQueryConfig> extends
   private params;
   private logger;
   private fields;
-  private _isResponseInArrayMode;
+  private useJitMapper;
   private customResultMapper?;
   static readonly [entityKind]: string;
   private rawQueryConfig;
   private queryConfig;
-  constructor(client: NodeCockroachClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, name: string | undefined, _isResponseInArrayMode: boolean, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined);
+  private jitMapper?;
+  constructor(client: NodeCockroachClient, queryString: string, params: unknown[], logger: Logger, fields: SelectedFieldsOrdered | undefined, name: string | undefined, useJitMapper: boolean | undefined, customResultMapper?: ((rows: unknown[][]) => T["execute"]) | undefined);
   execute(placeholderValues?: Record<string, unknown> | undefined): Promise<T['execute']>;
   all(placeholderValues?: Record<string, unknown> | undefined): Promise<T['all']>;
 }
 interface NodeCockroachSessionOptions {
   logger?: Logger;
+  useJitMapper?: boolean;
 }
 declare class NodeCockroachSession<TFullSchema extends Record<string, unknown>, TSchema extends V1.TablesRelationalConfig> extends CockroachSession<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {
   private client;
@@ -35,7 +37,7 @@ declare class NodeCockroachSession<TFullSchema extends Record<string, unknown>,
   static readonly [entityKind]: string;
   private logger;
   constructor(client: NodeCockroachClient, dialect: CockroachDialect, schema: V1.RelationalSchemaConfig<TSchema> | undefined, options?: NodeCockroachSessionOptions);
-  prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, isResponseInArrayMode: boolean, customResultMapper?: (rows: unknown[][]) => T['execute']): CockroachPreparedQuery<T>;
+  prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, customResultMapper?: (rows: unknown[][]) => T['execute']): CockroachPreparedQuery<T>;
   transaction<T>(transaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>, config?: CockroachTransactionConfig | undefined): Promise<T>;
   count(sql: SQL): Promise<number>;
 }

package/cockroach/session.js

@@ -1,6 +1,6 @@
 import { entityKind } from "../entity.js";
 import { tracer } from "../tracing.js";
-import { mapResultRow } from "../utils.js";
+import { makeJitQueryMapper, mapResultRow } from "../utils.js";
 import { fillPlaceholders, sql } from "../sql/sql.js";
 import { NoopLogger } from "../logger.js";
 import pg from "pg";
@@ -13,7 +13,8 @@ var NodeCockroachPreparedQuery = class extends CockroachPreparedQuery {
 	static [entityKind] = "NodeCockroachPreparedQuery";
 	rawQueryConfig;
 	queryConfig;
-	constructor(client, queryString, params, logger, fields, name, _isResponseInArrayMode, customResultMapper) {
+	jitMapper;
+	constructor(client, queryString, params, logger, fields, name, useJitMapper, customResultMapper) {
 		super({
 			sql: queryString,
 			params
@@ -22,7 +23,7 @@ var NodeCockroachPreparedQuery = class extends CockroachPreparedQuery {
 		this.params = params;
 		this.logger = logger;
 		this.fields = fields;
-		this._isResponseInArrayMode = _isResponseInArrayMode;
+		this.useJitMapper = useJitMapper;
 		this.customResultMapper = customResultMapper;
 		this.rawQueryConfig = {
 			name,
@@ -80,7 +81,8 @@ var NodeCockroachPreparedQuery = class extends CockroachPreparedQuery {
 				return client.query(query, params);
 			});
 			return tracer.startActiveSpan("drizzle.mapResponse", () => {
-				return customResultMapper ? customResultMapper(result.rows) : result.rows.map((row) => mapResultRow(fields, row, joinsNotNullableMap));
+				if (customResultMapper) return customResultMapper(result.rows);
+				return this.useJitMapper ? (this.jitMapper ??= makeJitQueryMapper(fields, joinsNotNullableMap))(result.rows) : result.rows.map((row) => mapResultRow(fields, row, joinsNotNullableMap));
 			});
 		});
 	}
@@ -98,10 +100,6 @@ var NodeCockroachPreparedQuery = class extends CockroachPreparedQuery {
 			});
 		});
 	}
-	/** @internal */
-	isResponseInArrayMode() {
-		return this._isResponseInArrayMode;
-	}
 };
 var NodeCockroachSession = class NodeCockroachSession extends CockroachSession {
 	static [entityKind] = "NodeCockroachSession";
@@ -113,8 +111,8 @@ var NodeCockroachSession = class NodeCockroachSession extends CockroachSession {
 		this.options = options;
 		this.logger = options.logger ?? new NoopLogger();
 	}
-	prepareQuery(query, fields, name, isResponseInArrayMode, customResultMapper) {
-		return new NodeCockroachPreparedQuery(this.client, query.sql, query.params, this.logger, fields, name, isResponseInArrayMode, customResultMapper);
+	prepareQuery(query, fields, name, customResultMapper) {
+		return new NodeCockroachPreparedQuery(this.client, query.sql, query.params, this.logger, fields, name, this.options.useJitMapper, customResultMapper);
 	}
 	async transaction(transaction, config) {
 		const session = this.client instanceof Pool$1 ? new NodeCockroachSession(await this.client.connect(), this.dialect, this.schema, this.options) : this;

package/cockroach/session.js.map

@@ -1 +1 @@
-{"version":3,"file":"session.js","names":["Pool"],"sources":["../../src/cockroach/session.ts"],"sourcesContent":["import type { Client, PoolClient, QueryArrayConfig, QueryConfig, QueryResult, QueryResultRow } from 'pg';\nimport pg from 'pg';\nimport type * as V1 from '~/_relations.ts';\nimport type { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { CockroachTransaction } from '~/cockroach-core/index.ts';\nimport type { SelectedFieldsOrdered } from '~/cockroach-core/query-builders/select.types.ts';\nimport type {\n\tCockroachQueryResultHKT,\n\tCockroachTransactionConfig,\n\tPreparedQueryConfig,\n} from '~/cockroach-core/session.ts';\nimport { CockroachPreparedQuery, CockroachSession } from '~/cockroach-core/session.ts';\nimport { entityKind } from '~/entity.ts';\nimport { type Logger, NoopLogger } from '~/logger.ts';\nimport { fillPlaceholders, type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { tracer } from '~/tracing.ts';\nimport { type Assume, mapResultRow } from '~/utils.ts';\n\nconst { Pool, types } = pg;\n\nexport type NodeCockroachClient = pg.Pool | PoolClient | Client;\n\nexport class NodeCockroachPreparedQuery<T extends PreparedQueryConfig> extends CockroachPreparedQuery<T> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachPreparedQuery';\n\n\tprivate rawQueryConfig: QueryConfig;\n\tprivate queryConfig: QueryArrayConfig;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tqueryString: string,\n\t\tprivate params: unknown[],\n\t\tprivate logger: Logger,\n\t\tprivate fields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tprivate _isResponseInArrayMode: boolean,\n\t\tprivate customResultMapper?: (rows: unknown[][]) => T['execute'],\n\t) {\n\t\tsuper({ sql: queryString, params });\n\t\tthis.rawQueryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t\tthis.queryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\trowMode: 'array',\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t}\n\n\tasync execute(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['execute']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', async () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\n\t\t\tconst { fields, rawQueryConfig: rawQuery, client, queryConfig: query, joinsNotNullableMap, customResultMapper } =\n\t\t\t\tthis;\n\t\t\tif (!fields && !customResultMapper) {\n\t\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', async (span) => {\n\t\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t\t'drizzle.query.name': rawQuery.name,\n\t\t\t\t\t\t'drizzle.query.text': rawQuery.text,\n\t\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t\t});\n\t\t\t\t\treturn client.query(rawQuery, params);\n\t\t\t\t});\n\t\t\t}\n\n\t\t\tconst result = await tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': query.name,\n\t\t\t\t\t'drizzle.query.text': query.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn client.query(query, params);\n\t\t\t});\n\n\t\t\treturn tracer.startActiveSpan('drizzle.mapResponse', () => {\n\t\t\t\treturn customResultMapper\n\t\t\t\t\t? customResultMapper(result.rows)\n\t\t\t\t\t: result.rows.map((row) => mapResultRow<T['execute']>(fields!, row, joinsNotNullableMap));\n\t\t\t});\n\t\t});\n\t}\n\n\tall(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['all']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': this.rawQueryConfig.name,\n\t\t\t\t\t'drizzle.query.text': this.rawQueryConfig.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn this.client.query(this.rawQueryConfig, params).then((result) => result.rows);\n\t\t\t});\n\t\t});\n\t}\n\n\t/** @internal */\n\tisResponseInArrayMode(): boolean {\n\t\treturn this._isResponseInArrayMode;\n\t}\n}\n\nexport interface NodeCockroachSessionOptions {\n\tlogger?: Logger;\n}\n\nexport class NodeCockroachSession<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachSession<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachSession';\n\n\tprivate logger: Logger;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tdialect: CockroachDialect,\n\t\tprivate schema: V1.RelationalSchemaConfig<TSchema> | undefined,\n\t\tprivate options: NodeCockroachSessionOptions = {},\n\t) {\n\t\tsuper(dialect);\n\t\tthis.logger = options.logger ?? new NoopLogger();\n\t}\n\n\tprepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tisResponseInArrayMode: boolean,\n\t\tcustomResultMapper?: (rows: unknown[][]) => T['execute'],\n\t): CockroachPreparedQuery<T> {\n\t\treturn new NodeCockroachPreparedQuery(\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\tname,\n\t\t\tisResponseInArrayMode,\n\t\t\tcustomResultMapper,\n\t\t);\n\t}\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t\tconfig?: CockroachTransactionConfig | undefined,\n\t): Promise<T> {\n\t\tconst session = this.client instanceof Pool // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t? new NodeCockroachSession(await this.client.connect(), this.dialect, this.schema, this.options)\n\t\t\t: this;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(this.dialect, session, this.schema);\n\t\tawait tx.execute(sql`begin${config ? sql` ${tx.getTransactionConfigSQL(config)}` : undefined}`);\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql`commit`);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tawait tx.execute(sql`rollback`);\n\t\t\tthrow error;\n\t\t} finally {\n\t\t\tif (this.client instanceof Pool) { // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t\t(session.client as PoolClient).release();\n\t\t\t}\n\t\t}\n\t}\n\n\toverride async count(sql: SQL): Promise<number> {\n\t\tconst res = await this.execute<{ rows: [{ count: string }] }>(sql);\n\t\treturn Number(\n\t\t\tres['rows'][0]['count'],\n\t\t);\n\t}\n}\n\nexport class NodeCockroachTransaction<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachTransaction<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachTransaction';\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t): Promise<T> {\n\t\tconst savepointName = `sp${this.nestedIndex + 1}`;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(\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\t\tawait tx.execute(sql.raw(`savepoint ${savepointName}`));\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql.raw(`release savepoint ${savepointName}`));\n\t\t\treturn result;\n\t\t} catch (err) {\n\t\t\tawait tx.execute(sql.raw(`rollback to savepoint ${savepointName}`));\n\t\t\tthrow err;\n\t\t}\n\t}\n}\n\nexport interface NodeCockroachQueryResultHKT extends CockroachQueryResultHKT {\n\ttype: QueryResult<Assume<this['row'], QueryResultRow>>;\n}\n"],"mappings":";;;;;;;;;;AAkBA,MAAM,EAAE,cAAM,UAAU;AAIxB,IAAa,6BAAb,cAA+E,uBAA0B;CACxG,QAA0B,cAAsB;CAEhD,AAAQ;CACR,AAAQ;CAER,YACC,AAAQ,QACR,aACA,AAAQ,QACR,AAAQ,QACR,AAAQ,QACR,MACA,AAAQ,wBACR,AAAQ,oBACP;AACD,QAAM;GAAE,KAAK;GAAa;GAAQ,CAAC;EAT3B;EAEA;EACA;EACA;EAEA;EACA;AAGR,OAAK,iBAAiB;GACrB;GACA,MAAM;GACN,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;AACD,OAAK,cAAc;GAClB;GACA,MAAM;GACN,SAAS;GACT,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;;CAGF,MAAM,QAAQ,oBAAyD,EAAE,EAAyB;AACjG,SAAO,OAAO,gBAAgB,mBAAmB,YAAY;GAC5D,MAAM,SAAS,iBAAiB,KAAK,QAAQ,kBAAkB;AAE/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;GAEtD,MAAM,EAAE,QAAQ,gBAAgB,UAAU,QAAQ,aAAa,OAAO,qBAAqB,uBAC1F;AACD,OAAI,CAAC,UAAU,CAAC,mBACf,QAAO,OAAO,gBAAgB,0BAA0B,OAAO,SAAS;AACvE,UAAM,cAAc;KACnB,sBAAsB,SAAS;KAC/B,sBAAsB,SAAS;KAC/B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,UAAU,OAAO;KACpC;GAGH,MAAM,SAAS,MAAM,OAAO,gBAAgB,2BAA2B,SAAS;AAC/E,UAAM,cAAc;KACnB,sBAAsB,MAAM;KAC5B,sBAAsB,MAAM;KAC5B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,OAAO,OAAO;KACjC;AAEF,UAAO,OAAO,gBAAgB,6BAA6B;AAC1D,WAAO,qBACJ,mBAAmB,OAAO,KAAK,GAC/B,OAAO,KAAK,KAAK,QAAQ,aAA2B,QAAS,KAAK,oBAAoB,CAAC;KACzF;IACD;;CAGH,IAAI,oBAAyD,EAAE,EAAqB;AACnF,SAAO,OAAO,gBAAgB,yBAAyB;GACtD,MAAM,SAAS,iBAAiB,KAAK,QAAQ,kBAAkB;AAC/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;AACtD,UAAO,OAAO,gBAAgB,2BAA2B,SAAS;AACjE,UAAM,cAAc;KACnB,sBAAsB,KAAK,eAAe;KAC1C,sBAAsB,KAAK,eAAe;KAC1C,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,KAAK,OAAO,MAAM,KAAK,gBAAgB,OAAO,CAAC,MAAM,WAAW,OAAO,KAAK;KAClF;IACD;;;CAIH,wBAAiC;AAChC,SAAO,KAAK;;;AAQd,IAAa,uBAAb,MAAa,6BAGH,iBAAoE;CAC7E,QAA0B,cAAsB;CAEhD,AAAQ;CAER,YACC,AAAQ,QACR,SACA,AAAQ,QACR,AAAQ,UAAuC,EAAE,EAChD;AACD,QAAM,QAAQ;EALN;EAEA;EACA;AAGR,OAAK,SAAS,QAAQ,UAAU,IAAI,YAAY;;CAGjD,aACC,OACA,QACA,MACA,uBACA,oBAC4B;AAC5B,SAAO,IAAI,2BACV,KAAK,QACL,MAAM,KACN,MAAM,QACN,KAAK,QACL,QACA,MACA,uBACA,mBACA;;CAGF,MAAe,YACd,aACA,QACa;EACb,MAAM,UAAU,KAAK,kBAAkBA,SACpC,IAAI,qBAAqB,MAAM,KAAK,OAAO,SAAS,EAAE,KAAK,SAAS,KAAK,QAAQ,KAAK,QAAQ,GAC9F;EACH,MAAM,KAAK,IAAI,yBAA+C,KAAK,SAAS,SAAS,KAAK,OAAO;AACjG,QAAM,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,GAAG,wBAAwB,OAAO,KAAK,SAAY;AAC/F,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQ,GAAG,SAAS;AAC7B,UAAO;WACC,OAAO;AACf,SAAM,GAAG,QAAQ,GAAG,WAAW;AAC/B,SAAM;YACG;AACT,OAAI,KAAK,kBAAkBA,OAC1B,CAAC,QAAQ,OAAsB,SAAS;;;CAK3C,MAAe,MAAM,KAA2B;EAC/C,MAAM,MAAM,MAAM,KAAK,QAAuC,IAAI;AAClE,SAAO,OACN,IAAI,QAAQ,GAAG,SACf;;;AAIH,IAAa,2BAAb,MAAa,iCAGH,qBAAwE;CACjF,QAA0B,cAAsB;CAEhD,MAAe,YACd,aACa;EACb,MAAM,gBAAgB,KAAK,KAAK,cAAc;EAC9C,MAAM,KAAK,IAAI,yBACd,KAAK,SACL,KAAK,SACL,KAAK,QACL,KAAK,cAAc,EACnB;AACD,QAAM,GAAG,QAAQ,IAAI,IAAI,aAAa,gBAAgB,CAAC;AACvD,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQ,IAAI,IAAI,qBAAqB,gBAAgB,CAAC;AAC/D,UAAO;WACC,KAAK;AACb,SAAM,GAAG,QAAQ,IAAI,IAAI,yBAAyB,gBAAgB,CAAC;AACnE,SAAM"}
+{"version":3,"file":"session.js","names":["Pool"],"sources":["../../src/cockroach/session.ts"],"sourcesContent":["import type { Client, PoolClient, QueryArrayConfig, QueryConfig, QueryResult, QueryResultRow } from 'pg';\nimport pg from 'pg';\nimport type * as V1 from '~/_relations.ts';\nimport type { CockroachDialect } from '~/cockroach-core/dialect.ts';\nimport { CockroachTransaction } from '~/cockroach-core/index.ts';\nimport type { SelectedFieldsOrdered } from '~/cockroach-core/query-builders/select.types.ts';\nimport type {\n\tCockroachQueryResultHKT,\n\tCockroachTransactionConfig,\n\tPreparedQueryConfig,\n} from '~/cockroach-core/session.ts';\nimport { CockroachPreparedQuery, CockroachSession } from '~/cockroach-core/session.ts';\nimport { entityKind } from '~/entity.ts';\nimport { type Logger, NoopLogger } from '~/logger.ts';\nimport { fillPlaceholders, type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { tracer } from '~/tracing.ts';\nimport { type Assume, makeJitQueryMapper, mapResultRow, type RowsMapper } from '~/utils.ts';\n\nconst { Pool, types } = pg;\n\nexport type NodeCockroachClient = pg.Pool | PoolClient | Client;\n\nexport class NodeCockroachPreparedQuery<T extends PreparedQueryConfig> extends CockroachPreparedQuery<T> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachPreparedQuery';\n\n\tprivate rawQueryConfig: QueryConfig;\n\tprivate queryConfig: QueryArrayConfig;\n\tprivate jitMapper?: RowsMapper<T['execute']>;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tqueryString: string,\n\t\tprivate params: unknown[],\n\t\tprivate logger: Logger,\n\t\tprivate fields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tprivate useJitMapper: boolean | undefined,\n\t\tprivate customResultMapper?: (rows: unknown[][]) => T['execute'],\n\t) {\n\t\tsuper({ sql: queryString, params });\n\t\tthis.rawQueryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t\tthis.queryConfig = {\n\t\t\tname,\n\t\t\ttext: queryString,\n\t\t\trowMode: 'array',\n\t\t\ttypes: {\n\t\t\t\t// @ts-ignore\n\t\t\t\tgetTypeParser: (typeId, format) => {\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMPTZ) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.TIMESTAMP) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.DATE) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\tif (typeId === types.builtins.INTERVAL) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// numeric[]\n\t\t\t\t\tif (typeId as number === 1231) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp[]\n\t\t\t\t\tif (typeId as number === 1115) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// timestamp with timezone[]\n\t\t\t\t\tif (typeId as number === 1185) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// interval[]\n\t\t\t\t\tif (typeId as number === 1187) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// date[]\n\t\t\t\t\tif (typeId as number === 1182) {\n\t\t\t\t\t\treturn (val: any) => val;\n\t\t\t\t\t}\n\t\t\t\t\t// @ts-ignore\n\t\t\t\t\treturn types.getTypeParser(typeId, format);\n\t\t\t\t},\n\t\t\t},\n\t\t};\n\t}\n\n\tasync execute(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['execute']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', async () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\n\t\t\tconst { fields, rawQueryConfig: rawQuery, client, queryConfig: query, joinsNotNullableMap, customResultMapper } =\n\t\t\t\tthis;\n\t\t\tif (!fields && !customResultMapper) {\n\t\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', async (span) => {\n\t\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t\t'drizzle.query.name': rawQuery.name,\n\t\t\t\t\t\t'drizzle.query.text': rawQuery.text,\n\t\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t\t});\n\t\t\t\t\treturn client.query(rawQuery, params);\n\t\t\t\t});\n\t\t\t}\n\n\t\t\tconst result = await tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': query.name,\n\t\t\t\t\t'drizzle.query.text': query.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn client.query(query, params);\n\t\t\t});\n\n\t\t\treturn tracer.startActiveSpan('drizzle.mapResponse', () => {\n\t\t\t\tif (customResultMapper) {\n\t\t\t\t\treturn (customResultMapper as (rows: unknown[][]) => unknown)(result.rows);\n\t\t\t\t}\n\n\t\t\t\treturn this.useJitMapper\n\t\t\t\t\t? (this.jitMapper ??= makeJitQueryMapper(fields!, joinsNotNullableMap))(result.rows)\n\t\t\t\t\t: result.rows.map((row) => mapResultRow(fields!, row, joinsNotNullableMap));\n\t\t\t});\n\t\t});\n\t}\n\n\tall(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['all']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', () => {\n\t\t\tconst params = fillPlaceholders(this.params, placeholderValues);\n\t\t\tthis.logger.logQuery(this.rawQueryConfig.text, params);\n\t\t\treturn tracer.startActiveSpan('drizzle.driver.execute', (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.name': this.rawQueryConfig.name,\n\t\t\t\t\t'drizzle.query.text': this.rawQueryConfig.text,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\t\t\t\treturn this.client.query(this.rawQueryConfig, params).then((result) => result.rows);\n\t\t\t});\n\t\t});\n\t}\n}\n\nexport interface NodeCockroachSessionOptions {\n\tlogger?: Logger;\n\tuseJitMapper?: boolean;\n}\n\nexport class NodeCockroachSession<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachSession<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachSession';\n\n\tprivate logger: Logger;\n\n\tconstructor(\n\t\tprivate client: NodeCockroachClient,\n\t\tdialect: CockroachDialect,\n\t\tprivate schema: V1.RelationalSchemaConfig<TSchema> | undefined,\n\t\tprivate options: NodeCockroachSessionOptions = {},\n\t) {\n\t\tsuper(dialect);\n\t\tthis.logger = options.logger ?? new NoopLogger();\n\t}\n\n\tprepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tcustomResultMapper?: (rows: unknown[][]) => T['execute'],\n\t): CockroachPreparedQuery<T> {\n\t\treturn new NodeCockroachPreparedQuery(\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\tname,\n\t\t\tthis.options.useJitMapper,\n\t\t\tcustomResultMapper,\n\t\t);\n\t}\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t\tconfig?: CockroachTransactionConfig | undefined,\n\t): Promise<T> {\n\t\tconst session = this.client instanceof Pool // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t? new NodeCockroachSession(await this.client.connect(), this.dialect, this.schema, this.options)\n\t\t\t: this;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(this.dialect, session, this.schema);\n\t\tawait tx.execute(sql`begin${config ? sql` ${tx.getTransactionConfigSQL(config)}` : undefined}`);\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql`commit`);\n\t\t\treturn result;\n\t\t} catch (error) {\n\t\t\tawait tx.execute(sql`rollback`);\n\t\t\tthrow error;\n\t\t} finally {\n\t\t\tif (this.client instanceof Pool) { // oxlint-disable-line drizzle-internal/no-instanceof\n\t\t\t\t(session.client as PoolClient).release();\n\t\t\t}\n\t\t}\n\t}\n\n\toverride async count(sql: SQL): Promise<number> {\n\t\tconst res = await this.execute<{ rows: [{ count: string }] }>(sql);\n\t\treturn Number(\n\t\t\tres['rows'][0]['count'],\n\t\t);\n\t}\n}\n\nexport class NodeCockroachTransaction<\n\tTFullSchema extends Record<string, unknown>,\n\tTSchema extends V1.TablesRelationalConfig,\n> extends CockroachTransaction<NodeCockroachQueryResultHKT, TFullSchema, TSchema> {\n\tstatic override readonly [entityKind]: string = 'NodeCockroachTransaction';\n\n\toverride async transaction<T>(\n\t\ttransaction: (tx: NodeCockroachTransaction<TFullSchema, TSchema>) => Promise<T>,\n\t): Promise<T> {\n\t\tconst savepointName = `sp${this.nestedIndex + 1}`;\n\t\tconst tx = new NodeCockroachTransaction<TFullSchema, TSchema>(\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\t\tawait tx.execute(sql.raw(`savepoint ${savepointName}`));\n\t\ttry {\n\t\t\tconst result = await transaction(tx);\n\t\t\tawait tx.execute(sql.raw(`release savepoint ${savepointName}`));\n\t\t\treturn result;\n\t\t} catch (err) {\n\t\t\tawait tx.execute(sql.raw(`rollback to savepoint ${savepointName}`));\n\t\t\tthrow err;\n\t\t}\n\t}\n}\n\nexport interface NodeCockroachQueryResultHKT extends CockroachQueryResultHKT {\n\ttype: QueryResult<Assume<this['row'], QueryResultRow>>;\n}\n"],"mappings":";;;;;;;;;;AAkBA,MAAM,EAAE,cAAM,UAAU;AAIxB,IAAa,6BAAb,cAA+E,uBAA0B;CACxG,QAA0B,cAAsB;CAEhD,AAAQ;CACR,AAAQ;CACR,AAAQ;CAER,YACC,AAAQ,QACR,aACA,AAAQ,QACR,AAAQ,QACR,AAAQ,QACR,MACA,AAAQ,cACR,AAAQ,oBACP;AACD,QAAM;GAAE,KAAK;GAAa;GAAQ,CAAC;EAT3B;EAEA;EACA;EACA;EAEA;EACA;AAGR,OAAK,iBAAiB;GACrB;GACA,MAAM;GACN,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;AACD,OAAK,cAAc;GAClB;GACA,MAAM;GACN,SAAS;GACT,OAAO,EAEN,gBAAgB,QAAQ,WAAW;AAClC,QAAI,WAAW,MAAM,SAAS,YAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,UAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,KAC7B,SAAQ,QAAa;AAEtB,QAAI,WAAW,MAAM,SAAS,SAC7B,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,QAAI,WAAqB,KACxB,SAAQ,QAAa;AAGtB,WAAO,MAAM,cAAc,QAAQ,OAAO;MAE3C;GACD;;CAGF,MAAM,QAAQ,oBAAyD,EAAE,EAAyB;AACjG,SAAO,OAAO,gBAAgB,mBAAmB,YAAY;GAC5D,MAAM,SAAS,iBAAiB,KAAK,QAAQ,kBAAkB;AAE/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;GAEtD,MAAM,EAAE,QAAQ,gBAAgB,UAAU,QAAQ,aAAa,OAAO,qBAAqB,uBAC1F;AACD,OAAI,CAAC,UAAU,CAAC,mBACf,QAAO,OAAO,gBAAgB,0BAA0B,OAAO,SAAS;AACvE,UAAM,cAAc;KACnB,sBAAsB,SAAS;KAC/B,sBAAsB,SAAS;KAC/B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,UAAU,OAAO;KACpC;GAGH,MAAM,SAAS,MAAM,OAAO,gBAAgB,2BAA2B,SAAS;AAC/E,UAAM,cAAc;KACnB,sBAAsB,MAAM;KAC5B,sBAAsB,MAAM;KAC5B,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,OAAO,MAAM,OAAO,OAAO;KACjC;AAEF,UAAO,OAAO,gBAAgB,6BAA6B;AAC1D,QAAI,mBACH,QAAQ,mBAAsD,OAAO,KAAK;AAG3E,WAAO,KAAK,gBACR,KAAK,cAAc,mBAAmB,QAAS,oBAAoB,EAAE,OAAO,KAAK,GAClF,OAAO,KAAK,KAAK,QAAQ,aAAa,QAAS,KAAK,oBAAoB,CAAC;KAC3E;IACD;;CAGH,IAAI,oBAAyD,EAAE,EAAqB;AACnF,SAAO,OAAO,gBAAgB,yBAAyB;GACtD,MAAM,SAAS,iBAAiB,KAAK,QAAQ,kBAAkB;AAC/D,QAAK,OAAO,SAAS,KAAK,eAAe,MAAM,OAAO;AACtD,UAAO,OAAO,gBAAgB,2BAA2B,SAAS;AACjE,UAAM,cAAc;KACnB,sBAAsB,KAAK,eAAe;KAC1C,sBAAsB,KAAK,eAAe;KAC1C,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AACF,WAAO,KAAK,OAAO,MAAM,KAAK,gBAAgB,OAAO,CAAC,MAAM,WAAW,OAAO,KAAK;KAClF;IACD;;;AASJ,IAAa,uBAAb,MAAa,6BAGH,iBAAoE;CAC7E,QAA0B,cAAsB;CAEhD,AAAQ;CAER,YACC,AAAQ,QACR,SACA,AAAQ,QACR,AAAQ,UAAuC,EAAE,EAChD;AACD,QAAM,QAAQ;EALN;EAEA;EACA;AAGR,OAAK,SAAS,QAAQ,UAAU,IAAI,YAAY;;CAGjD,aACC,OACA,QACA,MACA,oBAC4B;AAC5B,SAAO,IAAI,2BACV,KAAK,QACL,MAAM,KACN,MAAM,QACN,KAAK,QACL,QACA,MACA,KAAK,QAAQ,cACb,mBACA;;CAGF,MAAe,YACd,aACA,QACa;EACb,MAAM,UAAU,KAAK,kBAAkBA,SACpC,IAAI,qBAAqB,MAAM,KAAK,OAAO,SAAS,EAAE,KAAK,SAAS,KAAK,QAAQ,KAAK,QAAQ,GAC9F;EACH,MAAM,KAAK,IAAI,yBAA+C,KAAK,SAAS,SAAS,KAAK,OAAO;AACjG,QAAM,GAAG,QAAQ,GAAG,QAAQ,SAAS,GAAG,IAAI,GAAG,wBAAwB,OAAO,KAAK,SAAY;AAC/F,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQ,GAAG,SAAS;AAC7B,UAAO;WACC,OAAO;AACf,SAAM,GAAG,QAAQ,GAAG,WAAW;AAC/B,SAAM;YACG;AACT,OAAI,KAAK,kBAAkBA,OAC1B,CAAC,QAAQ,OAAsB,SAAS;;;CAK3C,MAAe,MAAM,KAA2B;EAC/C,MAAM,MAAM,MAAM,KAAK,QAAuC,IAAI;AAClE,SAAO,OACN,IAAI,QAAQ,GAAG,SACf;;;AAIH,IAAa,2BAAb,MAAa,iCAGH,qBAAwE;CACjF,QAA0B,cAAsB;CAEhD,MAAe,YACd,aACa;EACb,MAAM,gBAAgB,KAAK,KAAK,cAAc;EAC9C,MAAM,KAAK,IAAI,yBACd,KAAK,SACL,KAAK,SACL,KAAK,QACL,KAAK,cAAc,EACnB;AACD,QAAM,GAAG,QAAQ,IAAI,IAAI,aAAa,gBAAgB,CAAC;AACvD,MAAI;GACH,MAAM,SAAS,MAAM,YAAY,GAAG;AACpC,SAAM,GAAG,QAAQ,IAAI,IAAI,qBAAqB,gBAAgB,CAAC;AAC/D,UAAO;WACC,KAAK;AACb,SAAM,GAAG,QAAQ,IAAI,IAAI,yBAAyB,gBAAgB,CAAC;AACnE,SAAM"}

package/cockroach-core/columns/bigint.cjs

@@ -21,10 +21,10 @@ var CockroachBigInt53 = class extends require_cockroach_core_columns_common.Cock
 	getSQLType() {
 		return "int8";
 	}
-	mapFromDriverValue(value) {
+	mapFromDriverValue = (value) => {
 		if (typeof value === "number") return value;
 		return Number(value);
-	}
+	};
 };
 var CockroachBigInt64Builder = class extends require_cockroach_core_columns_int_common.CockroachIntColumnBaseBuilder {
 	static [__entity_ts.entityKind] = "CockroachBigInt64Builder";
@@ -41,9 +41,9 @@ var CockroachBigInt64 = class extends require_cockroach_core_columns_common.Cock
 	getSQLType() {
 		return "int8";
 	}
-	mapFromDriverValue(value) {
+	mapFromDriverValue = (value) => {
 		return BigInt(value);
-	}
+	};
 };
 function bigint(a, b) {
 	const { name, config } = (0, __utils_ts.getColumnNameAndConfig)(a, b);

package/cockroach-core/columns/bigint.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"bigint.cjs","names":["CockroachIntColumnBaseBuilder","entityKind","CockroachColumn"],"sources":["../../../src/cockroach-core/columns/bigint.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { getColumnNameAndConfig } from '~/utils.ts';\nimport { CockroachColumn } from './common.ts';\nimport { CockroachIntColumnBaseBuilder } from './int.common.ts';\n\nexport class CockroachBigInt53Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'number int53';\n\tdata: number;\n\tdriverParam: number | string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'number int53', 'CockroachBigInt53');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt53(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt53<T extends ColumnBaseConfig<'number int53'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\toverride mapFromDriverValue(value: number | string): number {\n\t\tif (typeof value === 'number') {\n\t\t\treturn value;\n\t\t}\n\t\treturn Number(value);\n\t}\n}\n\nexport class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'bigint int64';\n\tdata: bigint;\n\tdriverParam: string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'bigint int64', 'CockroachBigInt64');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt64(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt64<T extends ColumnBaseConfig<'bigint int64'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\t// eslint-disable-next-line unicorn/prefer-native-coercion-functions\n\toverride mapFromDriverValue(value: string): bigint {\n\t\treturn BigInt(value);\n\t}\n}\n\nexport interface CockroachBigIntConfig<T extends 'number' | 'bigint' = 'number' | 'bigint'> {\n\tmode: T;\n}\n\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\n"],"mappings":";;;;;;;;AAOA,IAAa,2BAAb,cAA8CA,wEAI3C;CACF,QAA0BC,0BAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmFC,sDAAmB;CACrG,QAA0BD,0BAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAGR,AAAS,mBAAmB,OAAgC;AAC3D,MAAI,OAAO,UAAU,SACpB,QAAO;AAER,SAAO,OAAO,MAAM;;;AAItB,IAAa,2BAAb,cAA8CD,wEAI3C;CACF,QAA0BC,0BAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmFC,sDAAmB;CACrG,QAA0BD,0BAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAIR,AAAS,mBAAmB,OAAuB;AAClD,SAAO,OAAO,MAAM;;;AAetB,SAAgB,OAAO,GAAmC,GAA2B;CACpF,MAAM,EAAE,MAAM,kDAAyD,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK;;AAS1C,SAAgB,KAAK,GAAmC,GAA2B;CAClF,MAAM,EAAE,MAAM,kDAAyD,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK"}
+{"version":3,"file":"bigint.cjs","names":["CockroachIntColumnBaseBuilder","entityKind","CockroachColumn"],"sources":["../../../src/cockroach-core/columns/bigint.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { getColumnNameAndConfig } from '~/utils.ts';\nimport { CockroachColumn } from './common.ts';\nimport { CockroachIntColumnBaseBuilder } from './int.common.ts';\n\nexport class CockroachBigInt53Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'number int53';\n\tdata: number;\n\tdriverParam: number | string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'number int53', 'CockroachBigInt53');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt53(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt53<T extends ColumnBaseConfig<'number int53'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\toverride mapFromDriverValue = (value: number | string): number => {\n\t\tif (typeof value === 'number') {\n\t\t\treturn value;\n\t\t}\n\t\treturn Number(value);\n\t};\n}\n\nexport class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'bigint int64';\n\tdata: bigint;\n\tdriverParam: string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'bigint int64', 'CockroachBigInt64');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt64(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt64<T extends ColumnBaseConfig<'bigint int64'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\t// eslint-disable-next-line unicorn/prefer-native-coercion-functions\n\toverride mapFromDriverValue = (value: string): bigint => {\n\t\treturn BigInt(value);\n\t};\n}\n\nexport interface CockroachBigIntConfig<T extends 'number' | 'bigint' = 'number' | 'bigint'> {\n\tmode: T;\n}\n\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\n"],"mappings":";;;;;;;;AAOA,IAAa,2BAAb,cAA8CA,wEAI3C;CACF,QAA0BC,0BAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmFC,sDAAmB;CACrG,QAA0BD,0BAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAGR,AAAS,sBAAsB,UAAmC;AACjE,MAAI,OAAO,UAAU,SACpB,QAAO;AAER,SAAO,OAAO,MAAM;;;AAItB,IAAa,2BAAb,cAA8CD,wEAI3C;CACF,QAA0BC,0BAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmFC,sDAAmB;CACrG,QAA0BD,0BAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAIR,AAAS,sBAAsB,UAA0B;AACxD,SAAO,OAAO,MAAM;;;AAetB,SAAgB,OAAO,GAAmC,GAA2B;CACpF,MAAM,EAAE,MAAM,kDAAyD,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK;;AAS1C,SAAgB,KAAK,GAAmC,GAA2B;CAClF,MAAM,EAAE,MAAM,kDAAyD,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK"}

package/cockroach-core/columns/bigint.d.cts

@@ -15,7 +15,7 @@ declare class CockroachBigInt53Builder extends CockroachIntColumnBaseBuilder<{
 declare class CockroachBigInt53<T extends ColumnBaseConfig<'number int53'>> extends CockroachColumn<T> {
   static readonly [entityKind]: string;
   getSQLType(): string;
-  mapFromDriverValue(value: number | string): number;
+  mapFromDriverValue: (value: number | string) => number;
 }
 declare class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{
   dataType: 'bigint int64';
@@ -28,7 +28,7 @@ declare class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{
 declare class CockroachBigInt64<T extends ColumnBaseConfig<'bigint int64'>> extends CockroachColumn<T> {
   static readonly [entityKind]: string;
   getSQLType(): string;
-  mapFromDriverValue(value: string): bigint;
+  mapFromDriverValue: (value: string) => bigint;
 }
 interface CockroachBigIntConfig<T extends 'number' | 'bigint' = 'number' | 'bigint'> {
   mode: T;

package/cockroach-core/columns/bigint.d.ts

@@ -15,7 +15,7 @@ declare class CockroachBigInt53Builder extends CockroachIntColumnBaseBuilder<{
 declare class CockroachBigInt53<T extends ColumnBaseConfig<'number int53'>> extends CockroachColumn<T> {
   static readonly [entityKind]: string;
   getSQLType(): string;
-  mapFromDriverValue(value: number | string): number;
+  mapFromDriverValue: (value: number | string) => number;
 }
 declare class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{
   dataType: 'bigint int64';
@@ -28,7 +28,7 @@ declare class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{
 declare class CockroachBigInt64<T extends ColumnBaseConfig<'bigint int64'>> extends CockroachColumn<T> {
   static readonly [entityKind]: string;
   getSQLType(): string;
-  mapFromDriverValue(value: string): bigint;
+  mapFromDriverValue: (value: string) => bigint;
 }
 interface CockroachBigIntConfig<T extends 'number' | 'bigint' = 'number' | 'bigint'> {
   mode: T;

package/cockroach-core/columns/bigint.js

@@ -19,10 +19,10 @@ var CockroachBigInt53 = class extends CockroachColumn {
 	getSQLType() {
 		return "int8";
 	}
-	mapFromDriverValue(value) {
+	mapFromDriverValue = (value) => {
 		if (typeof value === "number") return value;
 		return Number(value);
-	}
+	};
 };
 var CockroachBigInt64Builder = class extends CockroachIntColumnBaseBuilder {
 	static [entityKind] = "CockroachBigInt64Builder";
@@ -39,9 +39,9 @@ var CockroachBigInt64 = class extends CockroachColumn {
 	getSQLType() {
 		return "int8";
 	}
-	mapFromDriverValue(value) {
+	mapFromDriverValue = (value) => {
 		return BigInt(value);
-	}
+	};
 };
 function bigint(a, b) {
 	const { name, config } = getColumnNameAndConfig(a, b);

package/cockroach-core/columns/bigint.js.map

@@ -1 +1 @@
-{"version":3,"file":"bigint.js","names":[],"sources":["../../../src/cockroach-core/columns/bigint.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { getColumnNameAndConfig } from '~/utils.ts';\nimport { CockroachColumn } from './common.ts';\nimport { CockroachIntColumnBaseBuilder } from './int.common.ts';\n\nexport class CockroachBigInt53Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'number int53';\n\tdata: number;\n\tdriverParam: number | string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'number int53', 'CockroachBigInt53');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt53(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt53<T extends ColumnBaseConfig<'number int53'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\toverride mapFromDriverValue(value: number | string): number {\n\t\tif (typeof value === 'number') {\n\t\t\treturn value;\n\t\t}\n\t\treturn Number(value);\n\t}\n}\n\nexport class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'bigint int64';\n\tdata: bigint;\n\tdriverParam: string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'bigint int64', 'CockroachBigInt64');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt64(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt64<T extends ColumnBaseConfig<'bigint int64'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\t// eslint-disable-next-line unicorn/prefer-native-coercion-functions\n\toverride mapFromDriverValue(value: string): bigint {\n\t\treturn BigInt(value);\n\t}\n}\n\nexport interface CockroachBigIntConfig<T extends 'number' | 'bigint' = 'number' | 'bigint'> {\n\tmode: T;\n}\n\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\n"],"mappings":";;;;;;AAOA,IAAa,2BAAb,cAA8C,8BAI3C;CACF,QAA0B,cAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmF,gBAAmB;CACrG,QAA0B,cAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAGR,AAAS,mBAAmB,OAAgC;AAC3D,MAAI,OAAO,UAAU,SACpB,QAAO;AAER,SAAO,OAAO,MAAM;;;AAItB,IAAa,2BAAb,cAA8C,8BAI3C;CACF,QAA0B,cAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmF,gBAAmB;CACrG,QAA0B,cAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAIR,AAAS,mBAAmB,OAAuB;AAClD,SAAO,OAAO,MAAM;;;AAetB,SAAgB,OAAO,GAAmC,GAA2B;CACpF,MAAM,EAAE,MAAM,WAAW,uBAA8C,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK;;AAS1C,SAAgB,KAAK,GAAmC,GAA2B;CAClF,MAAM,EAAE,MAAM,WAAW,uBAA8C,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK"}
+{"version":3,"file":"bigint.js","names":[],"sources":["../../../src/cockroach-core/columns/bigint.ts"],"sourcesContent":["import type { AnyCockroachTable } from '~/cockroach-core/table.ts';\nimport type { ColumnBaseConfig } from '~/column.ts';\nimport { entityKind } from '~/entity.ts';\nimport { getColumnNameAndConfig } from '~/utils.ts';\nimport { CockroachColumn } from './common.ts';\nimport { CockroachIntColumnBaseBuilder } from './int.common.ts';\n\nexport class CockroachBigInt53Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'number int53';\n\tdata: number;\n\tdriverParam: number | string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'number int53', 'CockroachBigInt53');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt53(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt53<T extends ColumnBaseConfig<'number int53'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt53';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\toverride mapFromDriverValue = (value: number | string): number => {\n\t\tif (typeof value === 'number') {\n\t\t\treturn value;\n\t\t}\n\t\treturn Number(value);\n\t};\n}\n\nexport class CockroachBigInt64Builder extends CockroachIntColumnBaseBuilder<{\n\tdataType: 'bigint int64';\n\tdata: bigint;\n\tdriverParam: string;\n}> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64Builder';\n\n\tconstructor(name: string) {\n\t\tsuper(name, 'bigint int64', 'CockroachBigInt64');\n\t}\n\n\t/** @internal */\n\toverride build<TTableName extends string>(\n\t\ttable: AnyCockroachTable<{ name: TTableName }>,\n\t) {\n\t\treturn new CockroachBigInt64(\n\t\t\ttable,\n\t\t\tthis.config,\n\t\t);\n\t}\n}\n\nexport class CockroachBigInt64<T extends ColumnBaseConfig<'bigint int64'>> extends CockroachColumn<T> {\n\tstatic override readonly [entityKind]: string = 'CockroachBigInt64';\n\n\tgetSQLType(): string {\n\t\treturn 'int8';\n\t}\n\n\t// eslint-disable-next-line unicorn/prefer-native-coercion-functions\n\toverride mapFromDriverValue = (value: string): bigint => {\n\t\treturn BigInt(value);\n\t};\n}\n\nexport interface CockroachBigIntConfig<T extends 'number' | 'bigint' = 'number' | 'bigint'> {\n\tmode: T;\n}\n\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function bigint(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8<TMode extends CockroachBigIntConfig['mode']>(\n\tname: string,\n\tconfig: CockroachBigIntConfig<TMode>,\n): TMode extends 'number' ? CockroachBigInt53Builder : CockroachBigInt64Builder;\nexport function int8(a: string | CockroachBigIntConfig, b?: CockroachBigIntConfig) {\n\tconst { name, config } = getColumnNameAndConfig<CockroachBigIntConfig>(a, b);\n\tif (config.mode === 'number') {\n\t\treturn new CockroachBigInt53Builder(name);\n\t}\n\treturn new CockroachBigInt64Builder(name);\n}\n"],"mappings":";;;;;;AAOA,IAAa,2BAAb,cAA8C,8BAI3C;CACF,QAA0B,cAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmF,gBAAmB;CACrG,QAA0B,cAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAGR,AAAS,sBAAsB,UAAmC;AACjE,MAAI,OAAO,UAAU,SACpB,QAAO;AAER,SAAO,OAAO,MAAM;;;AAItB,IAAa,2BAAb,cAA8C,8BAI3C;CACF,QAA0B,cAAsB;CAEhD,YAAY,MAAc;AACzB,QAAM,MAAM,gBAAgB,oBAAoB;;;CAIjD,AAAS,MACR,OACC;AACD,SAAO,IAAI,kBACV,OACA,KAAK,OACL;;;AAIH,IAAa,oBAAb,cAAmF,gBAAmB;CACrG,QAA0B,cAAsB;CAEhD,aAAqB;AACpB,SAAO;;CAIR,AAAS,sBAAsB,UAA0B;AACxD,SAAO,OAAO,MAAM;;;AAetB,SAAgB,OAAO,GAAmC,GAA2B;CACpF,MAAM,EAAE,MAAM,WAAW,uBAA8C,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK;;AAS1C,SAAgB,KAAK,GAAmC,GAA2B;CAClF,MAAM,EAAE,MAAM,WAAW,uBAA8C,GAAG,EAAE;AAC5E,KAAI,OAAO,SAAS,SACnB,QAAO,IAAI,yBAAyB,KAAK;AAE1C,QAAO,IAAI,yBAAyB,KAAK"}