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

package/pg-core/async/session.d.cts

@@ -1,58 +1,52 @@
-import { SelectedFieldsOrdered } from "../query-builders/select.types.cjs";
 import { PgDialect } from "../dialect.cjs";
 import { PgBasePreparedQuery, PgQueryResultHKT, PgSession, PgTransactionConfig, PreparedQueryConfig } from "../session.cjs";
 import { PgAsyncDatabase } from "./db.cjs";
 import { entityKind } from "../../entity.cjs";
 import { Query, SQL } from "../../sql/sql.cjs";
-import * as V1 from "../../_relations.cjs";
 import { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from "../../migrator.cjs";
 import { Cache } from "../../cache/core/cache.cjs";
 import { WithCacheConfig } from "../../cache/core/types.cjs";
 import { AnyRelations, EmptyRelations } from "../../relations.cjs";
+import { Logger } from "../../logger.cjs";
 
 //#region src/pg-core/async/session.d.ts
-declare abstract class PgAsyncPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {
-  private cache;
-  private queryMetadata;
-  private cacheConfig?;
+declare class PgAsyncPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {
+  protected executor: (params?: unknown[]) => Promise<any>;
+  readonly mode: 'arrays' | 'objects' | 'raw';
+  protected logger: Logger;
+  protected cache: Cache | undefined;
+  protected queryMetadata: {
+    type: 'select' | 'update' | 'delete' | 'insert';
+    tables: string[];
+  } | undefined;
+  protected cacheConfig: WithCacheConfig | undefined;
   static readonly [entityKind]: string;
-  constructor(query: Query, cache: Cache | undefined, queryMetadata: {
+  constructor(executor: (params?: unknown[]) => Promise<any>, query: Query, mapper: ((rows: any[]) => any) | undefined, mode: 'arrays' | 'objects' | 'raw', logger: Logger, cache: Cache | undefined, queryMetadata: {
     type: 'select' | 'update' | 'delete' | 'insert';
     tables: string[];
-  } | undefined, cacheConfig?: WithCacheConfig | undefined);
-  abstract execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
+  } | undefined, cacheConfig: WithCacheConfig | undefined);
+  execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
 }
-declare abstract class PgAsyncSession<TQueryResult extends PgQueryResultHKT = PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgSession {
+declare abstract class PgAsyncSession<TQueryResult extends PgQueryResultHKT = PgQueryResultHKT, TRelations extends AnyRelations = EmptyRelations> extends PgSession {
   static readonly [entityKind]: string;
-  constructor(dialect: PgDialect);
-  abstract prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, isResponseInArrayMode: boolean, customResultMapper?: (rows: unknown[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'], queryMetadata?: {
+  abstract prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, mode: 'arrays' | 'objects' | 'raw', name: string | boolean, mapper?: (rows: any[]) => any, queryMetadata?: {
     type: 'select' | 'update' | 'delete' | 'insert';
     tables: string[];
   }, cacheConfig?: WithCacheConfig): PgAsyncPreparedQuery<T>;
-  abstract prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, customResultMapper: (rows: Record<string, unknown>[], mapColumnValue?: (value: unknown) => unknown) => T['execute']): PgAsyncPreparedQuery<T>;
-  execute<T>(query: SQL): Promise<T>;
-  all<T = unknown>(query: SQL): Promise<T[]>;
-  abstract transaction<T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Promise<T>, config?: PgTransactionConfig): Promise<T>;
+  execute<T>(query: SQL): Promise<T[]>;
+  arrays<T>(query: SQL): Promise<T[]>;
+  objects<T>(query: SQL): Promise<T[]>;
+  abstract transaction<T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TRelations>) => Promise<T>, config?: PgTransactionConfig): Promise<T>;
 }
-declare abstract class PgAsyncTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgAsyncDatabase<TQueryResult, TFullSchema, TRelations, TSchema> {
-  protected relations: TRelations;
-  protected schema: {
-    fullSchema: Record<string, unknown>;
-    schema: TSchema;
-    tableNamesMap: Record<string, string>;
-  } | undefined;
+declare abstract class PgAsyncTransaction<TQueryResult extends PgQueryResultHKT, TRelations extends AnyRelations = EmptyRelations> extends PgAsyncDatabase<TQueryResult, TRelations> {
   protected readonly nestedIndex: number;
   static readonly [entityKind]: string;
-  constructor(dialect: PgDialect, session: PgAsyncSession<any, any, any, any>, relations: TRelations, schema: {
-    fullSchema: Record<string, unknown>;
-    schema: TSchema;
-    tableNamesMap: Record<string, string>;
-  } | undefined, nestedIndex?: number, parseRqbJson?: boolean);
+  constructor(dialect: PgDialect, session: PgAsyncSession<any, any>, relations: TRelations, nestedIndex: number | undefined, parseRqbJson: boolean | undefined);
   rollback(): never;
-  setTransaction(config: PgTransactionConfig): Promise<void>;
-  abstract transaction<T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Promise<T>): Promise<T>;
+  setTransaction(config: PgTransactionConfig): Promise<unknown>;
+  abstract transaction: <T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TRelations>) => Promise<T>) => Promise<T>;
 }
-declare function migrate(migrations: MigrationMeta[], db: PgAsyncDatabase<PgQueryResultHKT, any, any, any>, config: string | MigrationConfig): Promise<void | MigratorInitFailResponse>;
+declare function migrate(migrations: MigrationMeta[], db: PgAsyncDatabase<PgQueryResultHKT, any>, config: string | MigrationConfig): Promise<void | MigratorInitFailResponse>;
 //#endregion
 export { PgAsyncPreparedQuery, PgAsyncSession, PgAsyncTransaction, migrate };
 //# sourceMappingURL=session.d.cts.map

package/pg-core/async/session.d.ts

@@ -1,58 +1,52 @@
-import { SelectedFieldsOrdered } from "../query-builders/select.types.js";
 import { PgDialect } from "../dialect.js";
 import { PgBasePreparedQuery, PgQueryResultHKT, PgSession, PgTransactionConfig, PreparedQueryConfig } from "../session.js";
 import { PgAsyncDatabase } from "./db.js";
 import { entityKind } from "../../entity.js";
 import { Query, SQL } from "../../sql/sql.js";
-import * as V1 from "../../_relations.js";
+import { Logger } from "../../logger.js";
 import { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from "../../migrator.js";
-import { Cache } from "../../cache/core/cache.js";
 import { AnyRelations, EmptyRelations } from "../../relations.js";
+import { Cache } from "../../cache/core/cache.js";
 import { WithCacheConfig } from "../../cache/core/types.js";
 
 //#region src/pg-core/async/session.d.ts
-declare abstract class PgAsyncPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {
-  private cache;
-  private queryMetadata;
-  private cacheConfig?;
+declare class PgAsyncPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {
+  protected executor: (params?: unknown[]) => Promise<any>;
+  readonly mode: 'arrays' | 'objects' | 'raw';
+  protected logger: Logger;
+  protected cache: Cache | undefined;
+  protected queryMetadata: {
+    type: 'select' | 'update' | 'delete' | 'insert';
+    tables: string[];
+  } | undefined;
+  protected cacheConfig: WithCacheConfig | undefined;
   static readonly [entityKind]: string;
-  constructor(query: Query, cache: Cache | undefined, queryMetadata: {
+  constructor(executor: (params?: unknown[]) => Promise<any>, query: Query, mapper: ((rows: any[]) => any) | undefined, mode: 'arrays' | 'objects' | 'raw', logger: Logger, cache: Cache | undefined, queryMetadata: {
     type: 'select' | 'update' | 'delete' | 'insert';
     tables: string[];
-  } | undefined, cacheConfig?: WithCacheConfig | undefined);
-  abstract execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
+  } | undefined, cacheConfig: WithCacheConfig | undefined);
+  execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;
 }
-declare abstract class PgAsyncSession<TQueryResult extends PgQueryResultHKT = PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgSession {
+declare abstract class PgAsyncSession<TQueryResult extends PgQueryResultHKT = PgQueryResultHKT, TRelations extends AnyRelations = EmptyRelations> extends PgSession {
   static readonly [entityKind]: string;
-  constructor(dialect: PgDialect);
-  abstract prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, isResponseInArrayMode: boolean, customResultMapper?: (rows: unknown[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'], queryMetadata?: {
+  abstract prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, mode: 'arrays' | 'objects' | 'raw', name: string | boolean, mapper?: (rows: any[]) => any, queryMetadata?: {
     type: 'select' | 'update' | 'delete' | 'insert';
     tables: string[];
   }, cacheConfig?: WithCacheConfig): PgAsyncPreparedQuery<T>;
-  abstract prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query, fields: SelectedFieldsOrdered | undefined, name: string | undefined, customResultMapper: (rows: Record<string, unknown>[], mapColumnValue?: (value: unknown) => unknown) => T['execute']): PgAsyncPreparedQuery<T>;
-  execute<T>(query: SQL): Promise<T>;
-  all<T = unknown>(query: SQL): Promise<T[]>;
-  abstract transaction<T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Promise<T>, config?: PgTransactionConfig): Promise<T>;
+  execute<T>(query: SQL): Promise<T[]>;
+  arrays<T>(query: SQL): Promise<T[]>;
+  objects<T>(query: SQL): Promise<T[]>;
+  abstract transaction<T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TRelations>) => Promise<T>, config?: PgTransactionConfig): Promise<T>;
 }
-declare abstract class PgAsyncTransaction<TQueryResult extends PgQueryResultHKT, TFullSchema extends Record<string, unknown> = Record<string, never>, TRelations extends AnyRelations = EmptyRelations, TSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>> extends PgAsyncDatabase<TQueryResult, TFullSchema, TRelations, TSchema> {
-  protected relations: TRelations;
-  protected schema: {
-    fullSchema: Record<string, unknown>;
-    schema: TSchema;
-    tableNamesMap: Record<string, string>;
-  } | undefined;
+declare abstract class PgAsyncTransaction<TQueryResult extends PgQueryResultHKT, TRelations extends AnyRelations = EmptyRelations> extends PgAsyncDatabase<TQueryResult, TRelations> {
   protected readonly nestedIndex: number;
   static readonly [entityKind]: string;
-  constructor(dialect: PgDialect, session: PgAsyncSession<any, any, any, any>, relations: TRelations, schema: {
-    fullSchema: Record<string, unknown>;
-    schema: TSchema;
-    tableNamesMap: Record<string, string>;
-  } | undefined, nestedIndex?: number, parseRqbJson?: boolean);
+  constructor(dialect: PgDialect, session: PgAsyncSession<any, any>, relations: TRelations, nestedIndex: number | undefined, parseRqbJson: boolean | undefined);
   rollback(): never;
-  setTransaction(config: PgTransactionConfig): Promise<void>;
-  abstract transaction<T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Promise<T>): Promise<T>;
+  setTransaction(config: PgTransactionConfig): Promise<unknown>;
+  abstract transaction: <T>(transaction: (tx: PgAsyncTransaction<TQueryResult, TRelations>) => Promise<T>) => Promise<T>;
 }
-declare function migrate(migrations: MigrationMeta[], db: PgAsyncDatabase<PgQueryResultHKT, any, any, any>, config: string | MigrationConfig): Promise<void | MigratorInitFailResponse>;
+declare function migrate(migrations: MigrationMeta[], db: PgAsyncDatabase<PgQueryResultHKT, any>, config: string | MigrationConfig): Promise<void | MigratorInitFailResponse>;
 //#endregion
 export { PgAsyncPreparedQuery, PgAsyncSession, PgAsyncTransaction, migrate };
 //# sourceMappingURL=session.d.ts.map

package/pg-core/async/session.js

@@ -3,7 +3,7 @@ import { PgAsyncDatabase } from "./db.js";
 import { entityKind, is } from "../../entity.js";
 import { tracer } from "../../tracing.js";
 import { assertUnreachable } from "../../utils.js";
-import { sql } from "../../sql/sql.js";
+import { fillPlaceholders, sql } from "../../sql/sql.js";
 import { getMigrationsToRun } from "../../migrator.utils.js";
 import { upgradeIfNeeded } from "../../up-migrations/pg.js";
 import { DrizzleQueryError, TransactionRollbackError } from "../../errors.js";
@@ -12,23 +12,42 @@ import { NoopCache, strategyFor } from "../../cache/core/cache.js";
 //#region src/pg-core/async/session.ts
 var PgAsyncPreparedQuery = class extends PgBasePreparedQuery {
 	static [entityKind] = "PgAsyncPreparedQuery";
-	constructor(query, cache, queryMetadata, cacheConfig) {
+	/** @internal */
+	mapper;
+	constructor(executor, query, mapper, mode, logger, cache, queryMetadata, cacheConfig) {
 		super(query);
+		this.executor = executor;
+		this.mode = mode;
+		this.logger = logger;
 		this.cache = cache;
 		this.queryMetadata = queryMetadata;
 		this.cacheConfig = cacheConfig;
+		this.mapper = mapper;
 		if (cache && cache.strategy() === "all" && cacheConfig === void 0) this.cacheConfig = {
 			enabled: true,
 			autoInvalidate: true
 		};
 		if (!this.cacheConfig?.enabled) this.cacheConfig = void 0;
 	}
-	/** @internal */
-	authToken;
-	/** @internal */
-	setToken(token) {
-		this.authToken = token;
-		return this;
+	execute(placeholderValues = {}) {
+		return tracer.startActiveSpan("drizzle.execute", async (span) => {
+			const params = fillPlaceholders(this.query.params, placeholderValues);
+			const { query: { sql }, mapper } = this;
+			span?.setAttributes({
+				"drizzle.query.text": sql,
+				"drizzle.query.params": JSON.stringify(params)
+			});
+			this.logger.logQuery(sql, params);
+			const query = tracer.startActiveSpan("drizzle.driver.execute", async (span) => {
+				span?.setAttributes({
+					"drizzle.query.text": sql,
+					"drizzle.query.params": JSON.stringify(params)
+				});
+				return await this.queryWithCache(sql, params, () => this.executor(params));
+			});
+			if (!mapper) return query;
+			return query.then((rows) => tracer.startActiveSpan("drizzle.mapResponse", () => mapper(rows)));
+		});
 	}
 	/** @internal */
 	async queryWithCache(queryString, params, query) {
@@ -57,27 +76,32 @@ var PgAsyncPreparedQuery = class extends PgBasePreparedQuery {
 };
 var PgAsyncSession = class extends PgSession {
 	static [entityKind] = "PgAsyncSession";
-	constructor(dialect) {
-		super(dialect);
+	execute(query) {
+		return tracer.startActiveSpan("drizzle.operation", () => {
+			return tracer.startActiveSpan("drizzle.prepareQuery", () => {
+				return this.prepareQuery(this.dialect.sqlToQuery(query), "raw", false);
+			}).execute();
+		});
 	}
-	/** @internal */
-	execute(query, token) {
+	arrays(query) {
 		return tracer.startActiveSpan("drizzle.operation", () => {
 			return tracer.startActiveSpan("drizzle.prepareQuery", () => {
-				return this.prepareQuery(this.dialect.sqlToQuery(query), void 0, void 0, false);
-			}).setToken(token).execute();
+				return this.prepareQuery(this.dialect.sqlToQuery(query), "arrays", false);
+			}).execute();
 		});
 	}
-	all(query) {
-		return this.prepareQuery(this.dialect.sqlToQuery(query), void 0, void 0, false).all();
+	objects(query) {
+		return tracer.startActiveSpan("drizzle.operation", () => {
+			return tracer.startActiveSpan("drizzle.prepareQuery", () => {
+				return this.prepareQuery(this.dialect.sqlToQuery(query), "objects", false);
+			}).execute();
+		});
 	}
 };
 var PgAsyncTransaction = class extends PgAsyncDatabase {
 	static [entityKind] = "PgAsyncTransaction";
-	constructor(dialect, session, relations, schema, nestedIndex = 0, parseRqbJson) {
-		super(dialect, session, relations, schema, parseRqbJson);
-		this.relations = relations;
-		this.schema = schema;
+	constructor(dialect, session, relations, nestedIndex = 0, parseRqbJson) {
+		super(dialect, session, relations, parseRqbJson);
 		this.nestedIndex = nestedIndex;
 	}
 	rollback() {
@@ -112,7 +136,7 @@ async function migrate(migrations, db, config) {
 		`;
 		await db.execute(migrationTableCreate);
 	}
-	const dbMigrations = await db.session.all(sql`select id, hash, created_at, name from ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)}`);
+	const dbMigrations = await db.session.objects(sql`select id, hash, created_at, name from ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)}`);
 	if (typeof config === "object" && config.init) {
 		if (dbMigrations.length) return { exitCode: "databaseMigrations" };
 		if (migrations.length > 1) return { exitCode: "localMigrations" };

package/pg-core/async/session.js.map

@@ -1 +1 @@
-{"version":3,"file":"session.js","names":[],"sources":["../../../src/pg-core/async/session.ts"],"sourcesContent":["import type * as V1 from '~/_relations.ts';\nimport { type Cache, NoopCache, strategyFor } from '~/cache/core/cache.ts';\nimport type { WithCacheConfig } from '~/cache/core/types.ts';\nimport { entityKind } from '~/entity.ts';\nimport { is } from '~/entity.ts';\nimport { TransactionRollbackError } from '~/errors.ts';\nimport { DrizzleQueryError } from '~/errors.ts';\nimport type { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from '~/migrator.ts';\nimport { getMigrationsToRun } from '~/migrator.utils.ts';\nimport type { AnyRelations, EmptyRelations } from '~/relations.ts';\nimport { type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { tracer } from '~/tracing.ts';\nimport { upgradeIfNeeded } from '~/up-migrations/pg.ts';\nimport type { NeonAuthToken } from '~/utils.ts';\nimport { assertUnreachable } from '~/utils.ts';\nimport type { PgDialect } from '../dialect.ts';\nimport type { SelectedFieldsOrdered } from '../query-builders/select.types.ts';\nimport type { PgQueryResultHKT, PgTransactionConfig, PreparedQueryConfig } from '../session.ts';\nimport { PgBasePreparedQuery, PgSession } from '../session.ts';\nimport { PgAsyncDatabase } from './db.ts';\n\nexport abstract class PgAsyncPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {\n\tstatic override readonly [entityKind]: string = 'PgAsyncPreparedQuery';\n\n\tconstructor(\n\t\tquery: Query,\n\t\t// cache instance\n\t\tprivate cache: Cache | undefined,\n\t\t// per query related metadata\n\t\tprivate queryMetadata: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t} | undefined,\n\t\t// config that was passed through $withCache\n\t\tprivate cacheConfig?: WithCacheConfig,\n\t) {\n\t\tsuper(query);\n\t\tif (cache && cache.strategy() === 'all' && cacheConfig === undefined) {\n\t\t\tthis.cacheConfig = { enabled: true, autoInvalidate: true };\n\t\t}\n\t\tif (!this.cacheConfig?.enabled) {\n\t\t\tthis.cacheConfig = undefined;\n\t\t}\n\t}\n\n\t/** @internal */\n\tprotected authToken?: NeonAuthToken;\n\t/** @internal */\n\tsetToken(token?: NeonAuthToken) {\n\t\tthis.authToken = token;\n\t\treturn this;\n\t}\n\n\tabstract override execute(placeholderValues?: Record<string, unknown>): Promise<T['execute']>;\n\n\t/** @internal */\n\tabstract override all(placeholderValues?: Record<string, unknown>): Promise<T['all']>;\n\n\t/** @internal */\n\tprotected async queryWithCache<T>(\n\t\tqueryString: string,\n\t\tparams: any[],\n\t\tquery: () => Promise<T>,\n\t): Promise<T> {\n\t\tconst cacheStrat = this.cache !== undefined && !is(this.cache, NoopCache)\n\t\t\t? await strategyFor(queryString, params, this.queryMetadata, this.cacheConfig)\n\t\t\t: { type: 'skip' as const };\n\n\t\tif (cacheStrat.type === 'skip') {\n\t\t\treturn query().catch((e) => {\n\t\t\t\tthrow new DrizzleQueryError(queryString, params, e as Error);\n\t\t\t});\n\t\t}\n\n\t\tconst cache = this.cache!;\n\n\t\t// For mutate queries, we should query the database, wait for a response, and then perform invalidation\n\t\tif (cacheStrat.type === 'invalidate') {\n\t\t\treturn Promise.all([\n\t\t\t\tquery(),\n\t\t\t\tcache.onMutate({ tables: cacheStrat.tables }),\n\t\t\t]).then((res) => res[0]).catch((e) => {\n\t\t\t\tthrow new DrizzleQueryError(queryString, params, e as Error);\n\t\t\t});\n\t\t}\n\n\t\tif (cacheStrat.type === 'try') {\n\t\t\tconst { tables, key, isTag, autoInvalidate, config } = cacheStrat;\n\t\t\tconst fromCache = await cache.get(\n\t\t\t\tkey,\n\t\t\t\ttables,\n\t\t\t\tisTag,\n\t\t\t\tautoInvalidate,\n\t\t\t);\n\n\t\t\tif (fromCache === undefined) {\n\t\t\t\tconst result = await query().catch((e) => {\n\t\t\t\t\tthrow new DrizzleQueryError(queryString, params, e as Error);\n\t\t\t\t});\n\t\t\t\t// put actual key\n\t\t\t\tawait cache.put(\n\t\t\t\t\tkey,\n\t\t\t\t\tresult,\n\t\t\t\t\t// make sure we send tables that were used in a query only if user wants to invalidate it on each write\n\t\t\t\t\tautoInvalidate ? tables : [],\n\t\t\t\t\tisTag,\n\t\t\t\t\tconfig,\n\t\t\t\t);\n\t\t\t\t// put flag if we should invalidate or not\n\t\t\t\treturn result;\n\t\t\t}\n\n\t\t\treturn fromCache as unknown as T;\n\t\t}\n\n\t\tassertUnreachable(cacheStrat);\n\t}\n}\n\nexport abstract class PgAsyncSession<\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTFullSchema extends Record<string, unknown> = Record<string, never>,\n\tTRelations extends AnyRelations = EmptyRelations,\n\tTSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>,\n> extends PgSession {\n\tstatic override readonly [entityKind]: string = 'PgAsyncSession';\n\n\tconstructor(dialect: PgDialect) {\n\t\tsuper(dialect);\n\t}\n\n\tabstract override prepareQuery<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[][], mapColumnValue?: (value: unknown) => unknown) => T['execute'],\n\t\tqueryMetadata?: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t},\n\t\tcacheConfig?: WithCacheConfig,\n\t): PgAsyncPreparedQuery<T>;\n\n\tabstract override prepareRelationalQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tfields: SelectedFieldsOrdered | undefined,\n\t\tname: string | undefined,\n\t\tcustomResultMapper: (\n\t\t\trows: Record<string, unknown>[],\n\t\t\tmapColumnValue?: (value: unknown) => unknown,\n\t\t) => T['execute'],\n\t): PgAsyncPreparedQuery<T>;\n\n\toverride execute<T>(query: SQL): Promise<T>;\n\t/** @internal */\n\toverride execute<T>(query: SQL, token?: NeonAuthToken): Promise<T>;\n\t/** @internal */\n\toverride execute<T>(query: SQL, token?: NeonAuthToken): Promise<T> {\n\t\treturn tracer.startActiveSpan('drizzle.operation', () => {\n\t\t\tconst prepared = tracer.startActiveSpan('drizzle.prepareQuery', () => {\n\t\t\t\treturn this.prepareQuery<PreparedQueryConfig & { execute: T }>(\n\t\t\t\t\tthis.dialect.sqlToQuery(query),\n\t\t\t\t\tundefined,\n\t\t\t\t\tundefined,\n\t\t\t\t\tfalse,\n\t\t\t\t);\n\t\t\t});\n\n\t\t\treturn prepared.setToken(token).execute();\n\t\t});\n\t}\n\n\toverride all<T = unknown>(query: SQL): Promise<T[]> {\n\t\treturn this.prepareQuery<PreparedQueryConfig & { all: T[] }>(\n\t\t\tthis.dialect.sqlToQuery(query),\n\t\t\tundefined,\n\t\t\tundefined,\n\t\t\tfalse,\n\t\t).all();\n\t}\n\n\tabstract transaction<T>(\n\t\ttransaction: (tx: PgAsyncTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Promise<T>,\n\t\tconfig?: PgTransactionConfig,\n\t): Promise<T>;\n}\n\nexport abstract class PgAsyncTransaction<\n\tTQueryResult extends PgQueryResultHKT,\n\tTFullSchema extends Record<string, unknown> = Record<string, never>,\n\tTRelations extends AnyRelations = EmptyRelations,\n\tTSchema extends V1.TablesRelationalConfig = V1.ExtractTablesWithRelations<TFullSchema>,\n> extends PgAsyncDatabase<TQueryResult, TFullSchema, TRelations, TSchema> {\n\tstatic override readonly [entityKind]: string = 'PgAsyncTransaction';\n\n\tconstructor(\n\t\tdialect: PgDialect,\n\t\tsession: PgAsyncSession<any, any, any, any>,\n\t\tprotected relations: TRelations,\n\t\tprotected schema: {\n\t\t\tfullSchema: Record<string, unknown>;\n\t\t\tschema: TSchema;\n\t\t\ttableNamesMap: Record<string, string>;\n\t\t} | undefined,\n\t\tprotected readonly nestedIndex = 0,\n\t\tparseRqbJson?: boolean,\n\t) {\n\t\tsuper(dialect, session, relations, schema, parseRqbJson);\n\t}\n\n\trollback(): never {\n\t\tthrow new TransactionRollbackError();\n\t}\n\n\t/** @internal */\n\tgetTransactionConfigSQL(config: PgTransactionConfig): SQL {\n\t\tconst chunks: string[] = [];\n\t\tif (config.isolationLevel) {\n\t\t\tchunks.push(`isolation level ${config.isolationLevel}`);\n\t\t}\n\t\tif (config.accessMode) {\n\t\t\tchunks.push(config.accessMode);\n\t\t}\n\t\tif (typeof config.deferrable === 'boolean') {\n\t\t\tchunks.push(config.deferrable ? 'deferrable' : 'not deferrable');\n\t\t}\n\t\treturn sql.raw(chunks.join(' '));\n\t}\n\n\tsetTransaction(config: PgTransactionConfig): Promise<void> {\n\t\treturn this.session.execute(sql`set transaction ${this.getTransactionConfigSQL(config)}`);\n\t}\n\n\tabstract override transaction<T>(\n\t\ttransaction: (tx: PgAsyncTransaction<TQueryResult, TFullSchema, TRelations, TSchema>) => Promise<T>,\n\t): Promise<T>;\n}\n\nexport async function migrate(\n\tmigrations: MigrationMeta[],\n\tdb: PgAsyncDatabase<PgQueryResultHKT, any, any, any>,\n\tconfig: string | MigrationConfig,\n): Promise<void | MigratorInitFailResponse> {\n\tconst migrationsTable = typeof config === 'string'\n\t\t? '__drizzle_migrations'\n\t\t: config.migrationsTable ?? '__drizzle_migrations';\n\tconst migrationsSchema = typeof config === 'string' ? 'drizzle' : config.migrationsSchema ?? 'drizzle';\n\n\tawait db.execute(sql`CREATE SCHEMA IF NOT EXISTS ${sql.identifier(migrationsSchema)}`);\n\n\t// Detect DB version and upgrade table schema if needed\n\tconst { newDb } = await upgradeIfNeeded(migrationsSchema, migrationsTable, db, migrations);\n\n\t// Create table with latest schema (version 1) if this is a new database\n\tif (newDb) {\n\t\tconst migrationTableCreate = sql`\n\t\t\tCREATE TABLE IF NOT EXISTS ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} (\n\t\t\t\tid SERIAL PRIMARY KEY,\n\t\t\t\thash text NOT NULL,\n\t\t\t\tcreated_at bigint,\n\t\t\t\tname text,\n\t\t\t\tapplied_at timestamp with time zone DEFAULT now()\n\t\t\t)\n\t\t`;\n\t\tawait db.execute(migrationTableCreate);\n\t}\n\n\tconst dbMigrations = await db.session.all<{ id: number; hash: string; created_at: string; name: string }>(\n\t\tsql`select id, hash, created_at, name from ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)}`,\n\t);\n\n\tif (typeof config === 'object' && config.init) {\n\t\tif (dbMigrations.length) {\n\t\t\treturn { exitCode: 'databaseMigrations' as const };\n\t\t}\n\n\t\tif (migrations.length > 1) {\n\t\t\treturn { exitCode: 'localMigrations' as const };\n\t\t}\n\n\t\tconst [migration] = migrations;\n\n\t\tif (!migration) return;\n\n\t\tawait db.execute(\n\t\t\tsql`insert into ${sql.identifier(migrationsSchema)}.${\n\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t} (\"hash\", \"created_at\", \"name\") values(${migration.hash}, ${migration.folderMillis}, ${migration.name ?? null})`,\n\t\t);\n\n\t\treturn;\n\t}\n\n\tconst migrationsToRun = getMigrationsToRun({ localMigrations: migrations, dbMigrations });\n\tawait db.transaction(async (tx) => {\n\t\tfor (const migration of migrationsToRun) {\n\t\t\tfor (const stmt of migration.sql) {\n\t\t\t\tawait tx.execute(sql.raw(stmt));\n\t\t\t}\n\t\t\tawait tx.execute(\n\t\t\t\tsql`insert into ${sql.identifier(migrationsSchema)}.${\n\t\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t\t} (\"hash\", \"created_at\", \"name\") values(${migration.hash}, ${migration.folderMillis}, ${\n\t\t\t\t\tmigration.name ?? null\n\t\t\t\t})`,\n\t\t\t);\n\t\t}\n\t});\n}\n"],"mappings":";;;;;;;;;;;;AAqBA,IAAsB,uBAAtB,cAAkF,oBAAoB;CACrG,QAA0B,cAAsB;CAEhD,YACC,OAEA,AAAQ,OAER,AAAQ,eAKR,AAAQ,aACP;AACD,QAAM,MAAM;EATJ;EAEA;EAKA;AAGR,MAAI,SAAS,MAAM,UAAU,KAAK,SAAS,gBAAgB,OAC1D,MAAK,cAAc;GAAE,SAAS;GAAM,gBAAgB;GAAM;AAE3D,MAAI,CAAC,KAAK,aAAa,QACtB,MAAK,cAAc;;;CAKrB,AAAU;;CAEV,SAAS,OAAuB;AAC/B,OAAK,YAAY;AACjB,SAAO;;;CASR,MAAgB,eACf,aACA,QACA,OACa;EACb,MAAM,aAAa,KAAK,UAAU,UAAa,CAAC,GAAG,KAAK,OAAO,UAAU,GACtE,MAAM,YAAY,aAAa,QAAQ,KAAK,eAAe,KAAK,YAAY,GAC5E,EAAE,MAAM,QAAiB;AAE5B,MAAI,WAAW,SAAS,OACvB,QAAO,OAAO,CAAC,OAAO,MAAM;AAC3B,SAAM,IAAI,kBAAkB,aAAa,QAAQ,EAAW;IAC3D;EAGH,MAAM,QAAQ,KAAK;AAGnB,MAAI,WAAW,SAAS,aACvB,QAAO,QAAQ,IAAI,CAClB,OAAO,EACP,MAAM,SAAS,EAAE,QAAQ,WAAW,QAAQ,CAAC,CAC7C,CAAC,CAAC,MAAM,QAAQ,IAAI,GAAG,CAAC,OAAO,MAAM;AACrC,SAAM,IAAI,kBAAkB,aAAa,QAAQ,EAAW;IAC3D;AAGH,MAAI,WAAW,SAAS,OAAO;GAC9B,MAAM,EAAE,QAAQ,KAAK,OAAO,gBAAgB,WAAW;GACvD,MAAM,YAAY,MAAM,MAAM,IAC7B,KACA,QACA,OACA,eACA;AAED,OAAI,cAAc,QAAW;IAC5B,MAAM,SAAS,MAAM,OAAO,CAAC,OAAO,MAAM;AACzC,WAAM,IAAI,kBAAkB,aAAa,QAAQ,EAAW;MAC3D;AAEF,UAAM,MAAM,IACX,KACA,QAEA,iBAAiB,SAAS,EAAE,EAC5B,OACA,OACA;AAED,WAAO;;AAGR,UAAO;;AAGR,oBAAkB,WAAW;;;AAI/B,IAAsB,iBAAtB,cAKU,UAAU;CACnB,QAA0B,cAAsB;CAEhD,YAAY,SAAoB;AAC/B,QAAM,QAAQ;;;CA8Bf,AAAS,QAAW,OAAY,OAAmC;AAClE,SAAO,OAAO,gBAAgB,2BAA2B;AAUxD,UATiB,OAAO,gBAAgB,8BAA8B;AACrE,WAAO,KAAK,aACX,KAAK,QAAQ,WAAW,MAAM,EAC9B,QACA,QACA,MACA;KACA,CAEc,SAAS,MAAM,CAAC,SAAS;IACxC;;CAGH,AAAS,IAAiB,OAA0B;AACnD,SAAO,KAAK,aACX,KAAK,QAAQ,WAAW,MAAM,EAC9B,QACA,QACA,MACA,CAAC,KAAK;;;AAST,IAAsB,qBAAtB,cAKU,gBAAgE;CACzE,QAA0B,cAAsB;CAEhD,YACC,SACA,SACA,AAAU,WACV,AAAU,QAKV,AAAmB,cAAc,GACjC,cACC;AACD,QAAM,SAAS,SAAS,WAAW,QAAQ,aAAa;EAT9C;EACA;EAKS;;CAMpB,WAAkB;AACjB,QAAM,IAAI,0BAA0B;;;CAIrC,wBAAwB,QAAkC;EACzD,MAAM,SAAmB,EAAE;AAC3B,MAAI,OAAO,eACV,QAAO,KAAK,mBAAmB,OAAO,iBAAiB;AAExD,MAAI,OAAO,WACV,QAAO,KAAK,OAAO,WAAW;AAE/B,MAAI,OAAO,OAAO,eAAe,UAChC,QAAO,KAAK,OAAO,aAAa,eAAe,iBAAiB;AAEjE,SAAO,IAAI,IAAI,OAAO,KAAK,IAAI,CAAC;;CAGjC,eAAe,QAA4C;AAC1D,SAAO,KAAK,QAAQ,QAAQ,GAAG,mBAAmB,KAAK,wBAAwB,OAAO,GAAG;;;AAQ3F,eAAsB,QACrB,YACA,IACA,QAC2C;CAC3C,MAAM,kBAAkB,OAAO,WAAW,WACvC,yBACA,OAAO,mBAAmB;CAC7B,MAAM,mBAAmB,OAAO,WAAW,WAAW,YAAY,OAAO,oBAAoB;AAE7F,OAAM,GAAG,QAAQ,GAAG,+BAA+B,IAAI,WAAW,iBAAiB,GAAG;CAGtF,MAAM,EAAE,UAAU,MAAM,gBAAgB,kBAAkB,iBAAiB,IAAI,WAAW;AAG1F,KAAI,OAAO;EACV,MAAM,uBAAuB,GAAG;gCACF,IAAI,WAAW,iBAAiB,CAAC,GAAG,IAAI,WAAW,gBAAgB,CAAC;;;;;;;;AAQlG,QAAM,GAAG,QAAQ,qBAAqB;;CAGvC,MAAM,eAAe,MAAM,GAAG,QAAQ,IACrC,GAAG,0CAA0C,IAAI,WAAW,iBAAiB,CAAC,GAAG,IAAI,WAAW,gBAAgB,GAChH;AAED,KAAI,OAAO,WAAW,YAAY,OAAO,MAAM;AAC9C,MAAI,aAAa,OAChB,QAAO,EAAE,UAAU,sBAA+B;AAGnD,MAAI,WAAW,SAAS,EACvB,QAAO,EAAE,UAAU,mBAA4B;EAGhD,MAAM,CAAC,aAAa;AAEpB,MAAI,CAAC,UAAW;AAEhB,QAAM,GAAG,QACR,GAAG,eAAe,IAAI,WAAW,iBAAiB,CAAC,GAClD,IAAI,WAAW,gBAAgB,CAC/B,yCAAyC,UAAU,KAAK,IAAI,UAAU,aAAa,IAAI,UAAU,QAAQ,KAAK,GAC/G;AAED;;CAGD,MAAM,kBAAkB,mBAAmB;EAAE,iBAAiB;EAAY;EAAc,CAAC;AACzF,OAAM,GAAG,YAAY,OAAO,OAAO;AAClC,OAAK,MAAM,aAAa,iBAAiB;AACxC,QAAK,MAAM,QAAQ,UAAU,IAC5B,OAAM,GAAG,QAAQ,IAAI,IAAI,KAAK,CAAC;AAEhC,SAAM,GAAG,QACR,GAAG,eAAe,IAAI,WAAW,iBAAiB,CAAC,GAClD,IAAI,WAAW,gBAAgB,CAC/B,yCAAyC,UAAU,KAAK,IAAI,UAAU,aAAa,IACnF,UAAU,QAAQ,KAClB,GACD;;GAED"}
+{"version":3,"file":"session.js","names":[],"sources":["../../../src/pg-core/async/session.ts"],"sourcesContent":["import { type Cache, NoopCache, strategyFor } from '~/cache/core/cache.ts';\nimport type { WithCacheConfig } from '~/cache/core/types.ts';\nimport { entityKind } from '~/entity.ts';\nimport { is } from '~/entity.ts';\nimport { TransactionRollbackError } from '~/errors.ts';\nimport { DrizzleQueryError } from '~/errors.ts';\nimport type { Logger } from '~/logger.ts';\nimport type { MigrationConfig, MigrationMeta, MigratorInitFailResponse } from '~/migrator.ts';\nimport { getMigrationsToRun } from '~/migrator.utils.ts';\nimport type { AnyRelations, EmptyRelations } from '~/relations.ts';\nimport { fillPlaceholders, type Query, type SQL, sql } from '~/sql/sql.ts';\nimport { tracer } from '~/tracing.ts';\nimport { upgradeIfNeeded } from '~/up-migrations/pg.ts';\nimport { assertUnreachable } from '~/utils.ts';\nimport type { PgDialect } from '../dialect.ts';\nimport type { PgQueryResultHKT, PgTransactionConfig, PreparedQueryConfig } from '../session.ts';\nimport { PgBasePreparedQuery, PgSession } from '../session.ts';\nimport { PgAsyncDatabase } from './db.ts';\n\nexport class PgAsyncPreparedQuery<T extends PreparedQueryConfig> extends PgBasePreparedQuery {\n\tstatic override readonly [entityKind]: string = 'PgAsyncPreparedQuery';\n\n\t/** @internal */\n\treadonly mapper: ((rows: any[]) => any) | undefined;\n\n\tconstructor(\n\t\tprotected executor: (params?: unknown[]) => Promise<any>,\n\t\tquery: Query,\n\t\tmapper: ((rows: any[]) => any) | undefined,\n\t\treadonly mode: 'arrays' | 'objects' | 'raw',\n\t\tprotected logger: Logger,\n\t\t// cache instance\n\t\tprotected cache: Cache | undefined,\n\t\t// per query related metadata\n\t\tprotected queryMetadata: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t} | undefined,\n\t\t// config that was passed through $withCache\n\t\tprotected cacheConfig: WithCacheConfig | undefined,\n\t) {\n\t\tsuper(query);\n\t\tthis.mapper = mapper;\n\t\tif (cache && cache.strategy() === 'all' && cacheConfig === undefined) {\n\t\t\tthis.cacheConfig = { enabled: true, autoInvalidate: true };\n\t\t}\n\t\tif (!this.cacheConfig?.enabled) {\n\t\t\tthis.cacheConfig = undefined;\n\t\t}\n\t}\n\n\toverride execute(placeholderValues: Record<string, unknown> = {}): Promise<T['execute']> {\n\t\treturn tracer.startActiveSpan('drizzle.execute', async (span) => {\n\t\t\tconst params = fillPlaceholders(this.query.params, placeholderValues);\n\t\t\tconst { query: { sql }, mapper } = this;\n\n\t\t\tspan?.setAttributes({\n\t\t\t\t'drizzle.query.text': sql,\n\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t});\n\n\t\t\tthis.logger.logQuery(sql, params);\n\n\t\t\tconst query = tracer.startActiveSpan('drizzle.driver.execute', async (span) => {\n\t\t\t\tspan?.setAttributes({\n\t\t\t\t\t'drizzle.query.text': sql,\n\t\t\t\t\t'drizzle.query.params': JSON.stringify(params),\n\t\t\t\t});\n\n\t\t\t\t// return await so tracer captures time accurately\n\t\t\t\treturn await this.queryWithCache(sql, params, () => this.executor(params));\n\t\t\t});\n\n\t\t\tif (!mapper) return query;\n\n\t\t\treturn query.then((rows) => tracer.startActiveSpan('drizzle.mapResponse', () => mapper(rows as unknown[])));\n\t\t});\n\t}\n\n\t/** @internal */\n\tprotected async queryWithCache<T>(\n\t\tqueryString: string,\n\t\tparams: any[],\n\t\tquery: () => Promise<T>,\n\t): Promise<T> {\n\t\tconst cacheStrat = this.cache !== undefined && !is(this.cache, NoopCache)\n\t\t\t? await strategyFor(queryString, params, this.queryMetadata, this.cacheConfig)\n\t\t\t: { type: 'skip' as const };\n\n\t\tif (cacheStrat.type === 'skip') {\n\t\t\treturn query().catch((e) => {\n\t\t\t\tthrow new DrizzleQueryError(queryString, params, e as Error);\n\t\t\t});\n\t\t}\n\n\t\tconst cache = this.cache!;\n\n\t\t// For mutate queries, we should query the database, wait for a response, and then perform invalidation\n\t\tif (cacheStrat.type === 'invalidate') {\n\t\t\treturn Promise.all([\n\t\t\t\tquery(),\n\t\t\t\tcache.onMutate({ tables: cacheStrat.tables }),\n\t\t\t]).then((res) => res[0]).catch((e) => {\n\t\t\t\tthrow new DrizzleQueryError(queryString, params, e as Error);\n\t\t\t});\n\t\t}\n\n\t\tif (cacheStrat.type === 'try') {\n\t\t\tconst { tables, key, isTag, autoInvalidate, config } = cacheStrat;\n\t\t\tconst fromCache = await cache.get(\n\t\t\t\tkey,\n\t\t\t\ttables,\n\t\t\t\tisTag,\n\t\t\t\tautoInvalidate,\n\t\t\t);\n\n\t\t\tif (fromCache === undefined) {\n\t\t\t\tconst result = await query().catch((e) => {\n\t\t\t\t\tthrow new DrizzleQueryError(queryString, params, e as Error);\n\t\t\t\t});\n\t\t\t\t// put actual key\n\t\t\t\tawait cache.put(\n\t\t\t\t\tkey,\n\t\t\t\t\tresult,\n\t\t\t\t\t// make sure we send tables that were used in a query only if user wants to invalidate it on each write\n\t\t\t\t\tautoInvalidate ? tables : [],\n\t\t\t\t\tisTag,\n\t\t\t\t\tconfig,\n\t\t\t\t);\n\t\t\t\t// put flag if we should invalidate or not\n\t\t\t\treturn result;\n\t\t\t}\n\n\t\t\treturn fromCache as unknown as T;\n\t\t}\n\n\t\tassertUnreachable(cacheStrat);\n\t}\n}\n\nexport abstract class PgAsyncSession<\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTRelations extends AnyRelations = EmptyRelations,\n> extends PgSession {\n\tstatic override readonly [entityKind]: string = 'PgAsyncSession';\n\n\tabstract override prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(\n\t\tquery: Query,\n\t\tmode: 'arrays' | 'objects' | 'raw',\n\t\tname: string | boolean,\n\t\tmapper?: (rows: any[]) => any,\n\t\tqueryMetadata?: {\n\t\t\ttype: 'select' | 'update' | 'delete' | 'insert';\n\t\t\ttables: string[];\n\t\t},\n\t\tcacheConfig?: WithCacheConfig,\n\t): PgAsyncPreparedQuery<T>;\n\n\toverride execute<T>(query: SQL): Promise<T[]> {\n\t\treturn tracer.startActiveSpan('drizzle.operation', () => {\n\t\t\tconst prepared = tracer.startActiveSpan('drizzle.prepareQuery', () => {\n\t\t\t\treturn this.prepareQuery<PreparedQueryConfig & { execute: T[] }>(\n\t\t\t\t\tthis.dialect.sqlToQuery(query),\n\t\t\t\t\t'raw',\n\t\t\t\t\tfalse,\n\t\t\t\t);\n\t\t\t});\n\n\t\t\treturn prepared.execute();\n\t\t});\n\t}\n\n\toverride arrays<T>(query: SQL): Promise<T[]> {\n\t\treturn tracer.startActiveSpan('drizzle.operation', () => {\n\t\t\tconst prepared = tracer.startActiveSpan('drizzle.prepareQuery', () => {\n\t\t\t\treturn this.prepareQuery<PreparedQueryConfig & { execute: T[] }>(\n\t\t\t\t\tthis.dialect.sqlToQuery(query),\n\t\t\t\t\t'arrays',\n\t\t\t\t\tfalse,\n\t\t\t\t);\n\t\t\t});\n\n\t\t\treturn prepared.execute();\n\t\t});\n\t}\n\n\toverride objects<T>(query: SQL): Promise<T[]> {\n\t\treturn tracer.startActiveSpan('drizzle.operation', () => {\n\t\t\tconst prepared = tracer.startActiveSpan('drizzle.prepareQuery', () => {\n\t\t\t\treturn this.prepareQuery<PreparedQueryConfig & { execute: T[] }>(\n\t\t\t\t\tthis.dialect.sqlToQuery(query),\n\t\t\t\t\t'objects',\n\t\t\t\t\tfalse,\n\t\t\t\t);\n\t\t\t});\n\n\t\t\treturn prepared.execute();\n\t\t});\n\t}\n\n\tabstract transaction<T>(\n\t\ttransaction: (tx: PgAsyncTransaction<TQueryResult, TRelations>) => Promise<T>,\n\t\tconfig?: PgTransactionConfig,\n\t): Promise<T>;\n}\n\nexport abstract class PgAsyncTransaction<\n\tTQueryResult extends PgQueryResultHKT,\n\tTRelations extends AnyRelations = EmptyRelations,\n> extends PgAsyncDatabase<TQueryResult, TRelations> {\n\tstatic override readonly [entityKind]: string = 'PgAsyncTransaction';\n\n\tconstructor(\n\t\tdialect: PgDialect,\n\t\tsession: PgAsyncSession<any, any>,\n\t\trelations: TRelations,\n\t\tprotected readonly nestedIndex = 0,\n\t\tparseRqbJson: boolean | undefined,\n\t) {\n\t\tsuper(dialect, session, relations, parseRqbJson);\n\t}\n\n\trollback(): never {\n\t\tthrow new TransactionRollbackError();\n\t}\n\n\t/** @internal */\n\tgetTransactionConfigSQL(config: PgTransactionConfig): SQL {\n\t\tconst chunks: string[] = [];\n\t\tif (config.isolationLevel) {\n\t\t\tchunks.push(`isolation level ${config.isolationLevel}`);\n\t\t}\n\t\tif (config.accessMode) {\n\t\t\tchunks.push(config.accessMode);\n\t\t}\n\t\tif (typeof config.deferrable === 'boolean') {\n\t\t\tchunks.push(config.deferrable ? 'deferrable' : 'not deferrable');\n\t\t}\n\t\treturn sql.raw(chunks.join(' '));\n\t}\n\n\tsetTransaction(config: PgTransactionConfig): Promise<unknown> {\n\t\treturn this.session.execute<void>(sql`set transaction ${this.getTransactionConfigSQL(config)}`);\n\t}\n\n\tabstract override transaction: <T>(\n\t\ttransaction: (tx: PgAsyncTransaction<TQueryResult, TRelations>) => Promise<T>,\n\t) => Promise<T>;\n}\n\nexport async function migrate(\n\tmigrations: MigrationMeta[],\n\tdb: PgAsyncDatabase<PgQueryResultHKT, any>,\n\tconfig: string | MigrationConfig,\n): Promise<void | MigratorInitFailResponse> {\n\tconst migrationsTable = typeof config === 'string'\n\t\t? '__drizzle_migrations'\n\t\t: config.migrationsTable ?? '__drizzle_migrations';\n\tconst migrationsSchema = typeof config === 'string' ? 'drizzle' : config.migrationsSchema ?? 'drizzle';\n\n\tawait db.execute(sql`CREATE SCHEMA IF NOT EXISTS ${sql.identifier(migrationsSchema)}`);\n\n\t// Detect DB version and upgrade table schema if needed\n\tconst { newDb } = await upgradeIfNeeded(migrationsSchema, migrationsTable, db, migrations);\n\n\t// Create table with latest schema (version 1) if this is a new database\n\tif (newDb) {\n\t\tconst migrationTableCreate = sql`\n\t\t\tCREATE TABLE IF NOT EXISTS ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)} (\n\t\t\t\tid SERIAL PRIMARY KEY,\n\t\t\t\thash text NOT NULL,\n\t\t\t\tcreated_at bigint,\n\t\t\t\tname text,\n\t\t\t\tapplied_at timestamp with time zone DEFAULT now()\n\t\t\t)\n\t\t`;\n\t\tawait db.execute(migrationTableCreate);\n\t}\n\n\tconst dbMigrations = await db.session.objects<{ id: number; hash: string; created_at: string; name: string }>(\n\t\tsql`select id, hash, created_at, name from ${sql.identifier(migrationsSchema)}.${sql.identifier(migrationsTable)}`,\n\t);\n\n\tif (typeof config === 'object' && config.init) {\n\t\tif (dbMigrations.length) {\n\t\t\treturn { exitCode: 'databaseMigrations' as const };\n\t\t}\n\n\t\tif (migrations.length > 1) {\n\t\t\treturn { exitCode: 'localMigrations' as const };\n\t\t}\n\n\t\tconst [migration] = migrations;\n\n\t\tif (!migration) return;\n\n\t\tawait db.execute(\n\t\t\tsql`insert into ${sql.identifier(migrationsSchema)}.${\n\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t} (\"hash\", \"created_at\", \"name\") values(${migration.hash}, ${migration.folderMillis}, ${migration.name ?? null})`,\n\t\t);\n\n\t\treturn;\n\t}\n\n\tconst migrationsToRun = getMigrationsToRun({ localMigrations: migrations, dbMigrations });\n\tawait db.transaction(async (tx) => {\n\t\tfor (const migration of migrationsToRun) {\n\t\t\tfor (const stmt of migration.sql) {\n\t\t\t\tawait tx.execute(sql.raw(stmt));\n\t\t\t}\n\t\t\tawait tx.execute(\n\t\t\t\tsql`insert into ${sql.identifier(migrationsSchema)}.${\n\t\t\t\t\tsql.identifier(migrationsTable)\n\t\t\t\t} (\"hash\", \"created_at\", \"name\") values(${migration.hash}, ${migration.folderMillis}, ${\n\t\t\t\t\tmigration.name ?? null\n\t\t\t\t})`,\n\t\t\t);\n\t\t}\n\t});\n}\n"],"mappings":";;;;;;;;;;;;AAmBA,IAAa,uBAAb,cAAyE,oBAAoB;CAC5F,QAA0B,cAAsB;;CAGhD,AAAS;CAET,YACC,AAAU,UACV,OACA,QACA,AAAS,MACT,AAAU,QAEV,AAAU,OAEV,AAAU,eAKV,AAAU,aACT;AACD,QAAM,MAAM;EAfF;EAGD;EACC;EAEA;EAEA;EAKA;AAGV,OAAK,SAAS;AACd,MAAI,SAAS,MAAM,UAAU,KAAK,SAAS,gBAAgB,OAC1D,MAAK,cAAc;GAAE,SAAS;GAAM,gBAAgB;GAAM;AAE3D,MAAI,CAAC,KAAK,aAAa,QACtB,MAAK,cAAc;;CAIrB,AAAS,QAAQ,oBAA6C,EAAE,EAAyB;AACxF,SAAO,OAAO,gBAAgB,mBAAmB,OAAO,SAAS;GAChE,MAAM,SAAS,iBAAiB,KAAK,MAAM,QAAQ,kBAAkB;GACrE,MAAM,EAAE,OAAO,EAAE,OAAO,WAAW;AAEnC,SAAM,cAAc;IACnB,sBAAsB;IACtB,wBAAwB,KAAK,UAAU,OAAO;IAC9C,CAAC;AAEF,QAAK,OAAO,SAAS,KAAK,OAAO;GAEjC,MAAM,QAAQ,OAAO,gBAAgB,0BAA0B,OAAO,SAAS;AAC9E,UAAM,cAAc;KACnB,sBAAsB;KACtB,wBAAwB,KAAK,UAAU,OAAO;KAC9C,CAAC;AAGF,WAAO,MAAM,KAAK,eAAe,KAAK,cAAc,KAAK,SAAS,OAAO,CAAC;KACzE;AAEF,OAAI,CAAC,OAAQ,QAAO;AAEpB,UAAO,MAAM,MAAM,SAAS,OAAO,gBAAgB,6BAA6B,OAAO,KAAkB,CAAC,CAAC;IAC1G;;;CAIH,MAAgB,eACf,aACA,QACA,OACa;EACb,MAAM,aAAa,KAAK,UAAU,UAAa,CAAC,GAAG,KAAK,OAAO,UAAU,GACtE,MAAM,YAAY,aAAa,QAAQ,KAAK,eAAe,KAAK,YAAY,GAC5E,EAAE,MAAM,QAAiB;AAE5B,MAAI,WAAW,SAAS,OACvB,QAAO,OAAO,CAAC,OAAO,MAAM;AAC3B,SAAM,IAAI,kBAAkB,aAAa,QAAQ,EAAW;IAC3D;EAGH,MAAM,QAAQ,KAAK;AAGnB,MAAI,WAAW,SAAS,aACvB,QAAO,QAAQ,IAAI,CAClB,OAAO,EACP,MAAM,SAAS,EAAE,QAAQ,WAAW,QAAQ,CAAC,CAC7C,CAAC,CAAC,MAAM,QAAQ,IAAI,GAAG,CAAC,OAAO,MAAM;AACrC,SAAM,IAAI,kBAAkB,aAAa,QAAQ,EAAW;IAC3D;AAGH,MAAI,WAAW,SAAS,OAAO;GAC9B,MAAM,EAAE,QAAQ,KAAK,OAAO,gBAAgB,WAAW;GACvD,MAAM,YAAY,MAAM,MAAM,IAC7B,KACA,QACA,OACA,eACA;AAED,OAAI,cAAc,QAAW;IAC5B,MAAM,SAAS,MAAM,OAAO,CAAC,OAAO,MAAM;AACzC,WAAM,IAAI,kBAAkB,aAAa,QAAQ,EAAW;MAC3D;AAEF,UAAM,MAAM,IACX,KACA,QAEA,iBAAiB,SAAS,EAAE,EAC5B,OACA,OACA;AAED,WAAO;;AAGR,UAAO;;AAGR,oBAAkB,WAAW;;;AAI/B,IAAsB,iBAAtB,cAGU,UAAU;CACnB,QAA0B,cAAsB;CAchD,AAAS,QAAW,OAA0B;AAC7C,SAAO,OAAO,gBAAgB,2BAA2B;AASxD,UARiB,OAAO,gBAAgB,8BAA8B;AACrE,WAAO,KAAK,aACX,KAAK,QAAQ,WAAW,MAAM,EAC9B,OACA,MACA;KACA,CAEc,SAAS;IACxB;;CAGH,AAAS,OAAU,OAA0B;AAC5C,SAAO,OAAO,gBAAgB,2BAA2B;AASxD,UARiB,OAAO,gBAAgB,8BAA8B;AACrE,WAAO,KAAK,aACX,KAAK,QAAQ,WAAW,MAAM,EAC9B,UACA,MACA;KACA,CAEc,SAAS;IACxB;;CAGH,AAAS,QAAW,OAA0B;AAC7C,SAAO,OAAO,gBAAgB,2BAA2B;AASxD,UARiB,OAAO,gBAAgB,8BAA8B;AACrE,WAAO,KAAK,aACX,KAAK,QAAQ,WAAW,MAAM,EAC9B,WACA,MACA;KACA,CAEc,SAAS;IACxB;;;AASJ,IAAsB,qBAAtB,cAGU,gBAA0C;CACnD,QAA0B,cAAsB;CAEhD,YACC,SACA,SACA,WACA,AAAmB,cAAc,GACjC,cACC;AACD,QAAM,SAAS,SAAS,WAAW,aAAa;EAH7B;;CAMpB,WAAkB;AACjB,QAAM,IAAI,0BAA0B;;;CAIrC,wBAAwB,QAAkC;EACzD,MAAM,SAAmB,EAAE;AAC3B,MAAI,OAAO,eACV,QAAO,KAAK,mBAAmB,OAAO,iBAAiB;AAExD,MAAI,OAAO,WACV,QAAO,KAAK,OAAO,WAAW;AAE/B,MAAI,OAAO,OAAO,eAAe,UAChC,QAAO,KAAK,OAAO,aAAa,eAAe,iBAAiB;AAEjE,SAAO,IAAI,IAAI,OAAO,KAAK,IAAI,CAAC;;CAGjC,eAAe,QAA+C;AAC7D,SAAO,KAAK,QAAQ,QAAc,GAAG,mBAAmB,KAAK,wBAAwB,OAAO,GAAG;;;AAQjG,eAAsB,QACrB,YACA,IACA,QAC2C;CAC3C,MAAM,kBAAkB,OAAO,WAAW,WACvC,yBACA,OAAO,mBAAmB;CAC7B,MAAM,mBAAmB,OAAO,WAAW,WAAW,YAAY,OAAO,oBAAoB;AAE7F,OAAM,GAAG,QAAQ,GAAG,+BAA+B,IAAI,WAAW,iBAAiB,GAAG;CAGtF,MAAM,EAAE,UAAU,MAAM,gBAAgB,kBAAkB,iBAAiB,IAAI,WAAW;AAG1F,KAAI,OAAO;EACV,MAAM,uBAAuB,GAAG;gCACF,IAAI,WAAW,iBAAiB,CAAC,GAAG,IAAI,WAAW,gBAAgB,CAAC;;;;;;;;AAQlG,QAAM,GAAG,QAAQ,qBAAqB;;CAGvC,MAAM,eAAe,MAAM,GAAG,QAAQ,QACrC,GAAG,0CAA0C,IAAI,WAAW,iBAAiB,CAAC,GAAG,IAAI,WAAW,gBAAgB,GAChH;AAED,KAAI,OAAO,WAAW,YAAY,OAAO,MAAM;AAC9C,MAAI,aAAa,OAChB,QAAO,EAAE,UAAU,sBAA+B;AAGnD,MAAI,WAAW,SAAS,EACvB,QAAO,EAAE,UAAU,mBAA4B;EAGhD,MAAM,CAAC,aAAa;AAEpB,MAAI,CAAC,UAAW;AAEhB,QAAM,GAAG,QACR,GAAG,eAAe,IAAI,WAAW,iBAAiB,CAAC,GAClD,IAAI,WAAW,gBAAgB,CAC/B,yCAAyC,UAAU,KAAK,IAAI,UAAU,aAAa,IAAI,UAAU,QAAQ,KAAK,GAC/G;AAED;;CAGD,MAAM,kBAAkB,mBAAmB;EAAE,iBAAiB;EAAY;EAAc,CAAC;AACzF,OAAM,GAAG,YAAY,OAAO,OAAO;AAClC,OAAK,MAAM,aAAa,iBAAiB;AACxC,QAAK,MAAM,QAAQ,UAAU,IAC5B,OAAM,GAAG,QAAQ,IAAI,IAAI,KAAK,CAAC;AAEhC,SAAM,GAAG,QACR,GAAG,eAAe,IAAI,WAAW,iBAAiB,CAAC,GAClD,IAAI,WAAW,gBAAgB,CAC/B,yCAAyC,UAAU,KAAK,IAAI,UAAU,aAAa,IACnF,UAAU,QAAQ,KAClB,GACD;;GAED"}

package/pg-core/async/update.cjs

@@ -3,33 +3,29 @@ const require_runtime = require('../../_virtual/_rolldown/runtime.cjs');
 const require_pg_core_utils = require('../utils.cjs');
 const require_pg_core_query_builders_update = require('../query-builders/update.cjs');
 let __entity_ts = require("../../entity.cjs");
+let __tracing_ts = require("../../tracing.cjs");
 let __utils_ts = require("../../utils.cjs");
 let __query_promise_ts = require("../../query-promise.cjs");
-let __query_name_generator_ts = require("../../query-name-generator.cjs");
 
 //#region src/pg-core/async/update.ts
 var PgAsyncUpdateBase = class extends require_pg_core_query_builders_update.PgUpdateBase {
 	static [__entity_ts.entityKind] = "PgAsyncUpdate";
 	/** @internal */
 	_prepare(name, generateName = false) {
-		const query = this.dialect.sqlToQuery(this.getSQL());
-		const preparedQuery = this.session.prepareQuery(query, this.config.returning, name ?? (generateName ? (0, __query_name_generator_ts.preparedStatementName)(query.sql, query.params) : name), true, void 0, {
-			type: "insert",
-			tables: require_pg_core_utils.extractUsedTable(this.config.table)
-		}, this.cacheConfig);
-		preparedQuery.joinsNotNullableMap = this.joinsNotNullableMap;
-		return preparedQuery.setToken(this.authToken);
+		const { session, config, dialect, joinsNotNullableMap, cacheConfig } = this;
+		const { returning: fields } = config;
+		return __tracing_ts.tracer.startActiveSpan("drizzle.prepareQuery", () => {
+			const query = dialect.sqlToQuery(this.getSQL());
+			const mapper = fields ? this.dialect.mapperGenerators.rows(fields, joinsNotNullableMap) : void 0;
+			return session.prepareQuery(query, fields ? "arrays" : "raw", name ?? generateName, mapper, {
+				type: "update",
+				tables: [...require_pg_core_utils.extractUsedTable(this.config.table)]
+			}, cacheConfig);
+		});
 	}
 	prepare(name) {
 		return this._prepare(name, true);
 	}
-	/** @internal */
-	authToken;
-	/** @internal */
-	setToken(token) {
-		this.authToken = token;
-		return this;
-	}
 	execute = (placeholderValues = {}) => {
 		return this._prepare().execute(placeholderValues);
 	};

package/pg-core/async/update.cjs.map

@@ -1 +1 @@
-{"version":3,"file":"update.cjs","names":["PgUpdateBase","entityKind","extractUsedTable","QueryPromise"],"sources":["../../../src/pg-core/async/update.ts"],"sourcesContent":["import { entityKind } from '~/entity.ts';\nimport type { PgQueryResultHKT, PgQueryResultKind, PreparedQueryConfig } from '~/pg-core/session.ts';\nimport type { PgTable } from '~/pg-core/table.ts';\nimport type { JoinNullability } from '~/query-builders/select.types.ts';\nimport { preparedStatementName } from '~/query-name-generator.ts';\nimport { QueryPromise } from '~/query-promise.ts';\nimport type { RunnableQuery } from '~/runnable-query.ts';\nimport type { ColumnsSelection, SQL } from '~/sql/sql.ts';\nimport type { Subquery } from '~/subquery.ts';\nimport { applyMixins, type Assume, type NeonAuthToken } from '~/utils.ts';\nimport { type Join, PgUpdateBase, type PgUpdateHKTBase } from '../query-builders/update.ts';\nimport { extractUsedTable } from '../utils.ts';\nimport type { PgViewBase } from '../view-base.ts';\nimport type { PgAsyncPreparedQuery, PgAsyncSession } from './session.ts';\n\nexport type PgAsyncUpdatePrepare<T extends AnyPgAsyncUpdate> = PgAsyncPreparedQuery<\n\tPreparedQueryConfig & {\n\t\texecute: T['_']['returning'] extends undefined ? PgQueryResultKind<T['_']['queryResult'], never>\n\t\t\t: T['_']['returning'][];\n\t}\n>;\n\nexport type PgAsyncUpdate<\n\tTTable extends PgTable = PgTable,\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = Record<string, unknown> | undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n> = PgAsyncUpdateBase<\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\ttrue,\n\tnever\n>;\n\nexport interface PgAsyncUpdateHKT extends PgUpdateHKTBase {\n\t_type: PgAsyncUpdateBase<\n\t\tAssume<this['table'], PgTable>,\n\t\tAssume<this['queryResult'], PgQueryResultHKT>,\n\t\tAssume<this['from'], PgTable | Subquery | PgViewBase | SQL | undefined>,\n\t\tAssume<this['selectedFields'], ColumnsSelection | undefined>,\n\t\tAssume<this['returning'], Record<string, unknown> | undefined>,\n\t\tAssume<this['nullabilityMap'], Record<string, JoinNullability>>,\n\t\tAssume<this['joins'], Join[]>,\n\t\tthis['dynamic'],\n\t\tthis['excludedMethods']\n\t>;\n}\n\nexport type AnyPgAsyncUpdate = PgAsyncUpdateBase<any, any, any, any, any, any, any, any, any>;\n\nexport interface PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTJoins extends Join[] = [],\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTDynamic extends boolean = false,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTExcludedMethods extends string = never,\n> extends QueryPromise<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[]> {}\n\nexport class PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n\tTDynamic extends boolean = false,\n\tTExcludedMethods extends string = never,\n> extends PgUpdateBase<\n\tPgAsyncUpdateHKT,\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\tTDynamic,\n\tTExcludedMethods\n> implements RunnableQuery<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[], 'pg'> {\n\tstatic override readonly [entityKind]: string = 'PgAsyncUpdate';\n\n\tdeclare protected session: PgAsyncSession;\n\n\t/** @internal */\n\t_prepare(name?: string, generateName = false): PgAsyncUpdatePrepare<this> {\n\t\tconst query = this.dialect.sqlToQuery(this.getSQL());\n\t\tconst preparedQuery = this.session.prepareQuery<\n\t\t\tPreparedQueryConfig & { execute: TReturning[] }\n\t\t>(\n\t\t\tquery,\n\t\t\tthis.config.returning,\n\t\t\tname ?? (generateName ? preparedStatementName(query.sql, query.params) : name),\n\t\t\ttrue,\n\t\t\tundefined,\n\t\t\t{\n\t\t\t\ttype: 'insert',\n\t\t\t\ttables: extractUsedTable(this.config.table),\n\t\t\t},\n\t\t\tthis.cacheConfig,\n\t\t);\n\t\tpreparedQuery.joinsNotNullableMap = this.joinsNotNullableMap;\n\t\treturn preparedQuery.setToken(this.authToken);\n\t}\n\n\tprepare(name?: string): PgAsyncUpdatePrepare<this> {\n\t\treturn this._prepare(name, true);\n\t}\n\n\t/** @internal */\n\tprivate authToken?: NeonAuthToken;\n\t/** @internal */\n\tsetToken(token?: NeonAuthToken) {\n\t\tthis.authToken = token;\n\t\treturn this;\n\t}\n\n\texecute: ReturnType<this['prepare']>['execute'] = (placeholderValues: Record<string, unknown> = {}) => {\n\t\treturn this._prepare().execute(placeholderValues);\n\t};\n}\n\napplyMixins(PgAsyncUpdateBase, [QueryPromise]);\n"],"mappings":";;;;;;;;;;AA4EA,IAAa,oBAAb,cAUUA,mDAW6G;CACtH,QAA0BC,0BAAsB;;CAKhD,SAAS,MAAe,eAAe,OAAmC;EACzE,MAAM,QAAQ,KAAK,QAAQ,WAAW,KAAK,QAAQ,CAAC;EACpD,MAAM,gBAAgB,KAAK,QAAQ,aAGlC,OACA,KAAK,OAAO,WACZ,SAAS,oEAAqC,MAAM,KAAK,MAAM,OAAO,GAAG,OACzE,MACA,QACA;GACC,MAAM;GACN,QAAQC,uCAAiB,KAAK,OAAO,MAAM;GAC3C,EACD,KAAK,YACL;AACD,gBAAc,sBAAsB,KAAK;AACzC,SAAO,cAAc,SAAS,KAAK,UAAU;;CAG9C,QAAQ,MAA2C;AAClD,SAAO,KAAK,SAAS,MAAM,KAAK;;;CAIjC,AAAQ;;CAER,SAAS,OAAuB;AAC/B,OAAK,YAAY;AACjB,SAAO;;CAGR,WAAmD,oBAA6C,EAAE,KAAK;AACtG,SAAO,KAAK,UAAU,CAAC,QAAQ,kBAAkB;;;4BAIvC,mBAAmB,CAACC,gCAAa,CAAC"}
+{"version":3,"file":"update.cjs","names":["PgUpdateBase","entityKind","tracer","extractUsedTable","QueryPromise"],"sources":["../../../src/pg-core/async/update.ts"],"sourcesContent":["import { entityKind } from '~/entity.ts';\nimport type { PgQueryResultHKT, PgQueryResultKind, PreparedQueryConfig } from '~/pg-core/session.ts';\nimport type { PgTable } from '~/pg-core/table.ts';\nimport type { JoinNullability } from '~/query-builders/select.types.ts';\nimport { QueryPromise } from '~/query-promise.ts';\nimport type { RunnableQuery } from '~/runnable-query.ts';\nimport type { ColumnsSelection, SQL } from '~/sql/sql.ts';\nimport type { Subquery } from '~/subquery.ts';\nimport { tracer } from '~/tracing.ts';\nimport { applyMixins, type Assume } from '~/utils.ts';\nimport { type Join, PgUpdateBase, type PgUpdateHKTBase } from '../query-builders/update.ts';\nimport { extractUsedTable } from '../utils.ts';\nimport type { PgViewBase } from '../view-base.ts';\nimport type { PgAsyncPreparedQuery, PgAsyncSession } from './session.ts';\n\nexport type PgAsyncUpdatePrepare<T extends AnyPgAsyncUpdate> = PgAsyncPreparedQuery<\n\tPreparedQueryConfig & {\n\t\texecute: T['_']['returning'] extends undefined ? PgQueryResultKind<T['_']['queryResult'], never>\n\t\t\t: T['_']['returning'][];\n\t}\n>;\n\nexport type PgAsyncUpdate<\n\tTTable extends PgTable = PgTable,\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = Record<string, unknown> | undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n> = PgAsyncUpdateBase<\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\ttrue,\n\tnever\n>;\n\nexport interface PgAsyncUpdateHKT extends PgUpdateHKTBase {\n\t_type: PgAsyncUpdateBase<\n\t\tAssume<this['table'], PgTable>,\n\t\tAssume<this['queryResult'], PgQueryResultHKT>,\n\t\tAssume<this['from'], PgTable | Subquery | PgViewBase | SQL | undefined>,\n\t\tAssume<this['selectedFields'], ColumnsSelection | undefined>,\n\t\tAssume<this['returning'], Record<string, unknown> | undefined>,\n\t\tAssume<this['nullabilityMap'], Record<string, JoinNullability>>,\n\t\tAssume<this['joins'], Join[]>,\n\t\tthis['dynamic'],\n\t\tthis['excludedMethods']\n\t>;\n}\n\nexport type AnyPgAsyncUpdate = PgAsyncUpdateBase<any, any, any, any, any, any, any, any, any>;\n\nexport interface PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTJoins extends Join[] = [],\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTDynamic extends boolean = false,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTExcludedMethods extends string = never,\n> extends QueryPromise<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[]> {}\n\nexport class PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n\tTDynamic extends boolean = false,\n\tTExcludedMethods extends string = never,\n> extends PgUpdateBase<\n\tPgAsyncUpdateHKT,\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\tTDynamic,\n\tTExcludedMethods\n> implements RunnableQuery<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[], 'pg'> {\n\tstatic override readonly [entityKind]: string = 'PgAsyncUpdate';\n\n\tdeclare protected session: PgAsyncSession;\n\n\t/** @internal */\n\t_prepare(name?: string, generateName = false): PgAsyncUpdatePrepare<this> {\n\t\tconst { session, config, dialect, joinsNotNullableMap, cacheConfig } = this;\n\t\tconst { returning: fields } = config;\n\n\t\treturn tracer.startActiveSpan('drizzle.prepareQuery', () => {\n\t\t\tconst query = dialect.sqlToQuery(this.getSQL());\n\t\t\tconst mapper = fields\n\t\t\t\t? this.dialect.mapperGenerators.rows(fields, joinsNotNullableMap)\n\t\t\t\t: undefined;\n\n\t\t\tconst preparedQuery = session.prepareQuery<PreparedQueryConfig & { execute: any }>(\n\t\t\t\tquery,\n\t\t\t\tfields ? 'arrays' : 'raw',\n\t\t\t\tname ?? generateName,\n\t\t\t\tmapper,\n\t\t\t\t{ type: 'update', tables: [...extractUsedTable(this.config.table)] },\n\t\t\t\tcacheConfig,\n\t\t\t);\n\n\t\t\treturn preparedQuery;\n\t\t});\n\t}\n\n\tprepare(name?: string): PgAsyncUpdatePrepare<this> {\n\t\treturn this._prepare(name, true);\n\t}\n\n\texecute: ReturnType<this['prepare']>['execute'] = (placeholderValues: Record<string, unknown> = {}) => {\n\t\treturn this._prepare().execute(placeholderValues);\n\t};\n}\n\napplyMixins(PgAsyncUpdateBase, [QueryPromise]);\n"],"mappings":";;;;;;;;;;AA4EA,IAAa,oBAAb,cAUUA,mDAW6G;CACtH,QAA0BC,0BAAsB;;CAKhD,SAAS,MAAe,eAAe,OAAmC;EACzE,MAAM,EAAE,SAAS,QAAQ,SAAS,qBAAqB,gBAAgB;EACvE,MAAM,EAAE,WAAW,WAAW;AAE9B,SAAOC,oBAAO,gBAAgB,8BAA8B;GAC3D,MAAM,QAAQ,QAAQ,WAAW,KAAK,QAAQ,CAAC;GAC/C,MAAM,SAAS,SACZ,KAAK,QAAQ,iBAAiB,KAAK,QAAQ,oBAAoB,GAC/D;AAWH,UATsB,QAAQ,aAC7B,OACA,SAAS,WAAW,OACpB,QAAQ,cACR,QACA;IAAE,MAAM;IAAU,QAAQ,CAAC,GAAGC,uCAAiB,KAAK,OAAO,MAAM,CAAC;IAAE,EACpE,YACA;IAGA;;CAGH,QAAQ,MAA2C;AAClD,SAAO,KAAK,SAAS,MAAM,KAAK;;CAGjC,WAAmD,oBAA6C,EAAE,KAAK;AACtG,SAAO,KAAK,UAAU,CAAC,QAAQ,kBAAkB;;;4BAIvC,mBAAmB,CAACC,gCAAa,CAAC"}

package/pg-core/async/update.js

@@ -1,33 +1,29 @@
 import { extractUsedTable } from "../utils.js";
 import { PgUpdateBase } from "../query-builders/update.js";
 import { entityKind } from "../../entity.js";
+import { tracer } from "../../tracing.js";
 import { applyMixins } from "../../utils.js";
 import { QueryPromise } from "../../query-promise.js";
-import { preparedStatementName } from "../../query-name-generator.js";
 
 //#region src/pg-core/async/update.ts
 var PgAsyncUpdateBase = class extends PgUpdateBase {
 	static [entityKind] = "PgAsyncUpdate";
 	/** @internal */
 	_prepare(name, generateName = false) {
-		const query = this.dialect.sqlToQuery(this.getSQL());
-		const preparedQuery = this.session.prepareQuery(query, this.config.returning, name ?? (generateName ? preparedStatementName(query.sql, query.params) : name), true, void 0, {
-			type: "insert",
-			tables: extractUsedTable(this.config.table)
-		}, this.cacheConfig);
-		preparedQuery.joinsNotNullableMap = this.joinsNotNullableMap;
-		return preparedQuery.setToken(this.authToken);
+		const { session, config, dialect, joinsNotNullableMap, cacheConfig } = this;
+		const { returning: fields } = config;
+		return tracer.startActiveSpan("drizzle.prepareQuery", () => {
+			const query = dialect.sqlToQuery(this.getSQL());
+			const mapper = fields ? this.dialect.mapperGenerators.rows(fields, joinsNotNullableMap) : void 0;
+			return session.prepareQuery(query, fields ? "arrays" : "raw", name ?? generateName, mapper, {
+				type: "update",
+				tables: [...extractUsedTable(this.config.table)]
+			}, cacheConfig);
+		});
 	}
 	prepare(name) {
 		return this._prepare(name, true);
 	}
-	/** @internal */
-	authToken;
-	/** @internal */
-	setToken(token) {
-		this.authToken = token;
-		return this;
-	}
 	execute = (placeholderValues = {}) => {
 		return this._prepare().execute(placeholderValues);
 	};

package/pg-core/async/update.js.map

@@ -1 +1 @@
-{"version":3,"file":"update.js","names":[],"sources":["../../../src/pg-core/async/update.ts"],"sourcesContent":["import { entityKind } from '~/entity.ts';\nimport type { PgQueryResultHKT, PgQueryResultKind, PreparedQueryConfig } from '~/pg-core/session.ts';\nimport type { PgTable } from '~/pg-core/table.ts';\nimport type { JoinNullability } from '~/query-builders/select.types.ts';\nimport { preparedStatementName } from '~/query-name-generator.ts';\nimport { QueryPromise } from '~/query-promise.ts';\nimport type { RunnableQuery } from '~/runnable-query.ts';\nimport type { ColumnsSelection, SQL } from '~/sql/sql.ts';\nimport type { Subquery } from '~/subquery.ts';\nimport { applyMixins, type Assume, type NeonAuthToken } from '~/utils.ts';\nimport { type Join, PgUpdateBase, type PgUpdateHKTBase } from '../query-builders/update.ts';\nimport { extractUsedTable } from '../utils.ts';\nimport type { PgViewBase } from '../view-base.ts';\nimport type { PgAsyncPreparedQuery, PgAsyncSession } from './session.ts';\n\nexport type PgAsyncUpdatePrepare<T extends AnyPgAsyncUpdate> = PgAsyncPreparedQuery<\n\tPreparedQueryConfig & {\n\t\texecute: T['_']['returning'] extends undefined ? PgQueryResultKind<T['_']['queryResult'], never>\n\t\t\t: T['_']['returning'][];\n\t}\n>;\n\nexport type PgAsyncUpdate<\n\tTTable extends PgTable = PgTable,\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = Record<string, unknown> | undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n> = PgAsyncUpdateBase<\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\ttrue,\n\tnever\n>;\n\nexport interface PgAsyncUpdateHKT extends PgUpdateHKTBase {\n\t_type: PgAsyncUpdateBase<\n\t\tAssume<this['table'], PgTable>,\n\t\tAssume<this['queryResult'], PgQueryResultHKT>,\n\t\tAssume<this['from'], PgTable | Subquery | PgViewBase | SQL | undefined>,\n\t\tAssume<this['selectedFields'], ColumnsSelection | undefined>,\n\t\tAssume<this['returning'], Record<string, unknown> | undefined>,\n\t\tAssume<this['nullabilityMap'], Record<string, JoinNullability>>,\n\t\tAssume<this['joins'], Join[]>,\n\t\tthis['dynamic'],\n\t\tthis['excludedMethods']\n\t>;\n}\n\nexport type AnyPgAsyncUpdate = PgAsyncUpdateBase<any, any, any, any, any, any, any, any, any>;\n\nexport interface PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTJoins extends Join[] = [],\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTDynamic extends boolean = false,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTExcludedMethods extends string = never,\n> extends QueryPromise<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[]> {}\n\nexport class PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n\tTDynamic extends boolean = false,\n\tTExcludedMethods extends string = never,\n> extends PgUpdateBase<\n\tPgAsyncUpdateHKT,\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\tTDynamic,\n\tTExcludedMethods\n> implements RunnableQuery<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[], 'pg'> {\n\tstatic override readonly [entityKind]: string = 'PgAsyncUpdate';\n\n\tdeclare protected session: PgAsyncSession;\n\n\t/** @internal */\n\t_prepare(name?: string, generateName = false): PgAsyncUpdatePrepare<this> {\n\t\tconst query = this.dialect.sqlToQuery(this.getSQL());\n\t\tconst preparedQuery = this.session.prepareQuery<\n\t\t\tPreparedQueryConfig & { execute: TReturning[] }\n\t\t>(\n\t\t\tquery,\n\t\t\tthis.config.returning,\n\t\t\tname ?? (generateName ? preparedStatementName(query.sql, query.params) : name),\n\t\t\ttrue,\n\t\t\tundefined,\n\t\t\t{\n\t\t\t\ttype: 'insert',\n\t\t\t\ttables: extractUsedTable(this.config.table),\n\t\t\t},\n\t\t\tthis.cacheConfig,\n\t\t);\n\t\tpreparedQuery.joinsNotNullableMap = this.joinsNotNullableMap;\n\t\treturn preparedQuery.setToken(this.authToken);\n\t}\n\n\tprepare(name?: string): PgAsyncUpdatePrepare<this> {\n\t\treturn this._prepare(name, true);\n\t}\n\n\t/** @internal */\n\tprivate authToken?: NeonAuthToken;\n\t/** @internal */\n\tsetToken(token?: NeonAuthToken) {\n\t\tthis.authToken = token;\n\t\treturn this;\n\t}\n\n\texecute: ReturnType<this['prepare']>['execute'] = (placeholderValues: Record<string, unknown> = {}) => {\n\t\treturn this._prepare().execute(placeholderValues);\n\t};\n}\n\napplyMixins(PgAsyncUpdateBase, [QueryPromise]);\n"],"mappings":";;;;;;;;AA4EA,IAAa,oBAAb,cAUU,aAW6G;CACtH,QAA0B,cAAsB;;CAKhD,SAAS,MAAe,eAAe,OAAmC;EACzE,MAAM,QAAQ,KAAK,QAAQ,WAAW,KAAK,QAAQ,CAAC;EACpD,MAAM,gBAAgB,KAAK,QAAQ,aAGlC,OACA,KAAK,OAAO,WACZ,SAAS,eAAe,sBAAsB,MAAM,KAAK,MAAM,OAAO,GAAG,OACzE,MACA,QACA;GACC,MAAM;GACN,QAAQ,iBAAiB,KAAK,OAAO,MAAM;GAC3C,EACD,KAAK,YACL;AACD,gBAAc,sBAAsB,KAAK;AACzC,SAAO,cAAc,SAAS,KAAK,UAAU;;CAG9C,QAAQ,MAA2C;AAClD,SAAO,KAAK,SAAS,MAAM,KAAK;;;CAIjC,AAAQ;;CAER,SAAS,OAAuB;AAC/B,OAAK,YAAY;AACjB,SAAO;;CAGR,WAAmD,oBAA6C,EAAE,KAAK;AACtG,SAAO,KAAK,UAAU,CAAC,QAAQ,kBAAkB;;;AAInD,YAAY,mBAAmB,CAAC,aAAa,CAAC"}
+{"version":3,"file":"update.js","names":[],"sources":["../../../src/pg-core/async/update.ts"],"sourcesContent":["import { entityKind } from '~/entity.ts';\nimport type { PgQueryResultHKT, PgQueryResultKind, PreparedQueryConfig } from '~/pg-core/session.ts';\nimport type { PgTable } from '~/pg-core/table.ts';\nimport type { JoinNullability } from '~/query-builders/select.types.ts';\nimport { QueryPromise } from '~/query-promise.ts';\nimport type { RunnableQuery } from '~/runnable-query.ts';\nimport type { ColumnsSelection, SQL } from '~/sql/sql.ts';\nimport type { Subquery } from '~/subquery.ts';\nimport { tracer } from '~/tracing.ts';\nimport { applyMixins, type Assume } from '~/utils.ts';\nimport { type Join, PgUpdateBase, type PgUpdateHKTBase } from '../query-builders/update.ts';\nimport { extractUsedTable } from '../utils.ts';\nimport type { PgViewBase } from '../view-base.ts';\nimport type { PgAsyncPreparedQuery, PgAsyncSession } from './session.ts';\n\nexport type PgAsyncUpdatePrepare<T extends AnyPgAsyncUpdate> = PgAsyncPreparedQuery<\n\tPreparedQueryConfig & {\n\t\texecute: T['_']['returning'] extends undefined ? PgQueryResultKind<T['_']['queryResult'], never>\n\t\t\t: T['_']['returning'][];\n\t}\n>;\n\nexport type PgAsyncUpdate<\n\tTTable extends PgTable = PgTable,\n\tTQueryResult extends PgQueryResultHKT = PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = Record<string, unknown> | undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n> = PgAsyncUpdateBase<\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\ttrue,\n\tnever\n>;\n\nexport interface PgAsyncUpdateHKT extends PgUpdateHKTBase {\n\t_type: PgAsyncUpdateBase<\n\t\tAssume<this['table'], PgTable>,\n\t\tAssume<this['queryResult'], PgQueryResultHKT>,\n\t\tAssume<this['from'], PgTable | Subquery | PgViewBase | SQL | undefined>,\n\t\tAssume<this['selectedFields'], ColumnsSelection | undefined>,\n\t\tAssume<this['returning'], Record<string, unknown> | undefined>,\n\t\tAssume<this['nullabilityMap'], Record<string, JoinNullability>>,\n\t\tAssume<this['joins'], Join[]>,\n\t\tthis['dynamic'],\n\t\tthis['excludedMethods']\n\t>;\n}\n\nexport type AnyPgAsyncUpdate = PgAsyncUpdateBase<any, any, any, any, any, any, any, any, any>;\n\nexport interface PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTJoins extends Join[] = [],\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTDynamic extends boolean = false,\n\t// eslint-disable-next-line @typescript-eslint/no-unused-vars\n\tTExcludedMethods extends string = never,\n> extends QueryPromise<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[]> {}\n\nexport class PgAsyncUpdateBase<\n\tTTable extends PgTable,\n\tTQueryResult extends PgQueryResultHKT,\n\tTFrom extends PgTable | Subquery | PgViewBase | SQL | undefined = undefined,\n\tTSelectedFields extends ColumnsSelection | undefined = undefined,\n\tTReturning extends Record<string, unknown> | undefined = undefined,\n\tTNullabilityMap extends Record<string, JoinNullability> = Record<TTable['_']['name'], 'not-null'>,\n\tTJoins extends Join[] = [],\n\tTDynamic extends boolean = false,\n\tTExcludedMethods extends string = never,\n> extends PgUpdateBase<\n\tPgAsyncUpdateHKT,\n\tTTable,\n\tTQueryResult,\n\tTFrom,\n\tTSelectedFields,\n\tTReturning,\n\tTNullabilityMap,\n\tTJoins,\n\tTDynamic,\n\tTExcludedMethods\n> implements RunnableQuery<TReturning extends undefined ? PgQueryResultKind<TQueryResult, never> : TReturning[], 'pg'> {\n\tstatic override readonly [entityKind]: string = 'PgAsyncUpdate';\n\n\tdeclare protected session: PgAsyncSession;\n\n\t/** @internal */\n\t_prepare(name?: string, generateName = false): PgAsyncUpdatePrepare<this> {\n\t\tconst { session, config, dialect, joinsNotNullableMap, cacheConfig } = this;\n\t\tconst { returning: fields } = config;\n\n\t\treturn tracer.startActiveSpan('drizzle.prepareQuery', () => {\n\t\t\tconst query = dialect.sqlToQuery(this.getSQL());\n\t\t\tconst mapper = fields\n\t\t\t\t? this.dialect.mapperGenerators.rows(fields, joinsNotNullableMap)\n\t\t\t\t: undefined;\n\n\t\t\tconst preparedQuery = session.prepareQuery<PreparedQueryConfig & { execute: any }>(\n\t\t\t\tquery,\n\t\t\t\tfields ? 'arrays' : 'raw',\n\t\t\t\tname ?? generateName,\n\t\t\t\tmapper,\n\t\t\t\t{ type: 'update', tables: [...extractUsedTable(this.config.table)] },\n\t\t\t\tcacheConfig,\n\t\t\t);\n\n\t\t\treturn preparedQuery;\n\t\t});\n\t}\n\n\tprepare(name?: string): PgAsyncUpdatePrepare<this> {\n\t\treturn this._prepare(name, true);\n\t}\n\n\texecute: ReturnType<this['prepare']>['execute'] = (placeholderValues: Record<string, unknown> = {}) => {\n\t\treturn this._prepare().execute(placeholderValues);\n\t};\n}\n\napplyMixins(PgAsyncUpdateBase, [QueryPromise]);\n"],"mappings":";;;;;;;;AA4EA,IAAa,oBAAb,cAUU,aAW6G;CACtH,QAA0B,cAAsB;;CAKhD,SAAS,MAAe,eAAe,OAAmC;EACzE,MAAM,EAAE,SAAS,QAAQ,SAAS,qBAAqB,gBAAgB;EACvE,MAAM,EAAE,WAAW,WAAW;AAE9B,SAAO,OAAO,gBAAgB,8BAA8B;GAC3D,MAAM,QAAQ,QAAQ,WAAW,KAAK,QAAQ,CAAC;GAC/C,MAAM,SAAS,SACZ,KAAK,QAAQ,iBAAiB,KAAK,QAAQ,oBAAoB,GAC/D;AAWH,UATsB,QAAQ,aAC7B,OACA,SAAS,WAAW,OACpB,QAAQ,cACR,QACA;IAAE,MAAM;IAAU,QAAQ,CAAC,GAAG,iBAAiB,KAAK,OAAO,MAAM,CAAC;IAAE,EACpE,YACA;IAGA;;CAGH,QAAQ,MAA2C;AAClD,SAAO,KAAK,SAAS,MAAM,KAAK;;CAGjC,WAAmD,oBAA6C,EAAE,KAAK;AACtG,SAAO,KAAK,UAAU,CAAC,QAAQ,kBAAkB;;;AAInD,YAAY,mBAAmB,CAAC,aAAa,CAAC"}