drizzle-orm 1.0.0-beta.21 → 1.0.0-beta.21-af6b4cb

package/pg-core/async/session.js

@@ -3,32 +3,51 @@ 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 { getMigrationsToRun } from "../../migrator.utils.js";
-import { upgradeIfNeeded } from "../../up-migrations/pg.js";
+import { fillPlaceholders, sql } from "../../sql/sql.js";
 import { DrizzleQueryError, TransactionRollbackError } from "../../errors.js";
 import { NoopCache, strategyFor } from "../../cache/core/cache.js";
+import { getMigrationsToRun } from "../../migrator.utils.js";
+import { upgradeIfNeeded } from "../../up-migrations/pg.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"}

package/pg-core/codecs.cjs

@@ -0,0 +1,156 @@
+Object.defineProperty(exports, Symbol.toStringTag, { value: 'Module' });
+const require_runtime = require('../_virtual/_rolldown/runtime.cjs');
+const require_pg_core_array = require('./array.cjs');
+let __codecs_ts = require("../codecs.cjs");
+let __sql_sql_ts = require("../sql/sql.cjs");
+
+//#region src/pg-core/codecs.ts
+const PG_ALIAS_TO_TYPE_MAP = {
+	int2: "smallint",
+	integer: "int",
+	int4: "int",
+	int8: "bigint",
+	decimal: "numeric",
+	real: "float4",
+	double: "float8",
+	"double precision": "float8",
+	serial2: "smallserial",
+	serial4: "serial",
+	serial8: "bigserial",
+	character: "char",
+	"character varying": "varchar",
+	"time with time zone": "timetz",
+	"timestamp with time zone": "timestamptz",
+	boolean: "bool",
+	"bit varying": "varbit"
+};
+function resolvePgType(type) {
+	return PG_ALIAS_TO_TYPE_MAP[type] ?? type;
+}
+const castToText = (name) => __sql_sql_ts.sql`${name}::text`;
+const castToTextArr = (name, arrayDimensions) => __sql_sql_ts.sql`${name}::text${__sql_sql_ts.sql.raw("[]".repeat(arrayDimensions))}`;
+/** Used for cases when casting requires to unwrap and rebuild arrays
+*
+* @example
+* string_mtx::text[][] // can be casted to array directly
+*
+* encode(bytea_mtx, 'base64')[][] // invalid syntax, cast requires unwrapping and rebuilding array
+*/
+const arrayCompatCast = (cast) => (name, arrayDimensions) => {
+	if (!arrayDimensions) return cast(name);
+	const aliases = [];
+	for (let i = 0; i < arrayDimensions; i++) aliases.push(__sql_sql_ts.sql.identifier(`s${i}`));
+	let indexed = name;
+	for (const alias of aliases) indexed = __sql_sql_ts.sql`${indexed}[${alias}]`;
+	let expression = __sql_sql_ts.sql`array(\
+select ${cast(indexed)} \
+from generate_subscripts(${name}, ${__sql_sql_ts.sql.raw(arrayDimensions.toString())}) ${aliases[arrayDimensions - 1]} \
+order by ${aliases[arrayDimensions - 1]})`;
+	for (let dim = arrayDimensions - 1; dim > 0; dim--) expression = __sql_sql_ts.sql`array(\
+select ${expression} \
+from generate_subscripts(${name}, ${__sql_sql_ts.sql.raw(dim.toString())}) ${aliases[dim - 1]} \
+order by ${aliases[dim - 1]})`;
+	return __sql_sql_ts.sql`case when ${name} is null then null else ${expression} end`;
+};
+/** Used to recursively apply value normalizer to array of unknown dimensions */
+const arrayCompatNormalize = (normalize, transformToPgArray = false) => {
+	const loop = (value, arrayDimensions) => {
+		const innerDimensions = arrayDimensions - 1;
+		if (arrayDimensions > 1) for (let i = 0; i < value.length; ++i) loop(value[i], innerDimensions);
+		else for (let i = 0; i < value.length; ++i) value[i] = normalize(value[i]);
+		return value;
+	};
+	return transformToPgArray ? (v, d) => require_pg_core_array.makePgArray(loop(v, d)) : loop;
+};
+const genericPgCodecs = {
+	bytea: {
+		castInJson: (name) => __sql_sql_ts.sql`encode(${name}, 'base64')`,
+		castArrayInJson: arrayCompatCast((name) => __sql_sql_ts.sql`encode(${name}, 'base64')`),
+		normalizeInJson: (v) => Buffer.from(v, "base64"),
+		normalizeArrayInJson: arrayCompatNormalize((v) => Buffer.from(v, "base64"))
+	},
+	bigint: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		normalizeInJson: BigInt,
+		normalizeArrayInJson: arrayCompatNormalize(BigInt),
+		normalize: BigInt,
+		normalizeArray: arrayCompatNormalize(BigInt)
+	},
+	bigserial: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		normalizeInJson: BigInt,
+		normalizeArrayInJson: arrayCompatNormalize(BigInt),
+		normalize: BigInt,
+		normalizeArray: arrayCompatNormalize(BigInt)
+	},
+	date: { castArray: castToTextArr },
+	enum: {
+		castArray: castToTextArr,
+		normalizeParamArray: require_pg_core_array.makePgArray
+	},
+	geometry: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		normalizeArray: require_pg_core_array.parsePgArray
+	},
+	interval: {
+		castArrayInJson: castToTextArr,
+		castArray: castToTextArr
+	},
+	json: {
+		normalizeParam: (v) => typeof v === "object" && !Array.isArray(v) ? v : JSON.stringify(v),
+		normalizeParamArray: arrayCompatNormalize((v) => JSON.stringify(v), true)
+	},
+	jsonb: {
+		normalizeParam: (v) => typeof v === "object" && !Array.isArray(v) ? v : JSON.stringify(v),
+		normalizeParamArray: arrayCompatNormalize((v) => JSON.stringify(v), true)
+	},
+	line: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		cast: castToText,
+		castArray: castToTextArr
+	},
+	macaddr8: {
+		castArrayInJson: castToTextArr,
+		castArray: castToTextArr
+	},
+	numeric: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		castArray: castToTextArr
+	},
+	point: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		cast: castToText,
+		castArray: castToTextArr
+	},
+	timestamp: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		castArray: castToTextArr
+	},
+	timestamptz: {
+		castInJson: castToText,
+		castArrayInJson: castToTextArr,
+		castArray: castToTextArr
+	},
+	bit: { normalizeArray: require_pg_core_array.parsePgArray },
+	halfvec: { normalizeArray: require_pg_core_array.parsePgArray },
+	sparsevec: { normalizeArray: require_pg_core_array.parsePgArray },
+	vector: { normalizeArray: require_pg_core_array.parsePgArray }
+};
+const refineGenericPgCodecs = (extension) => (0, __codecs_ts.refineCodecs)(genericPgCodecs, extension);
+
+//#endregion
+exports.arrayCompatCast = arrayCompatCast;
+exports.arrayCompatNormalize = arrayCompatNormalize;
+exports.castToText = castToText;
+exports.castToTextArr = castToTextArr;
+exports.genericPgCodecs = genericPgCodecs;
+exports.refineGenericPgCodecs = refineGenericPgCodecs;
+exports.resolvePgType = resolvePgType;
+//# sourceMappingURL=codecs.cjs.map

package/pg-core/codecs.cjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"codecs.cjs","names":["sql","makePgArray","parsePgArray"],"sources":["../../src/pg-core/codecs.ts"],"sourcesContent":["import {\n\ttype CastArrayCodec,\n\ttype CastCodec,\n\ttype Codecs,\n\ttype NormalizeArrayCodec,\n\ttype NormalizeCodec,\n\trefineCodecs,\n} from '~/codecs.ts';\nimport { type Name, sql, type SQLChunk } from '~/sql/sql.ts';\nimport type { PartialWithUndefined } from '~/utils.ts';\nimport { makePgArray, parsePgArray } from './array';\n\nexport type PostgresType =\n\t// Numeric\n\t| 'smallint'\n\t| 'int'\n\t| 'bigint'\n\t| 'numeric'\n\t| 'float4'\n\t| 'float8'\n\t| 'money'\n\t| 'smallserial'\n\t| 'serial'\n\t| 'bigserial'\n\t// Text\n\t| 'char'\n\t| 'varchar'\n\t| 'text'\n\t// Binary\n\t| 'bytea'\n\t// Datetime\n\t| 'date'\n\t| 'time'\n\t| 'timetz'\n\t| 'timestamp'\n\t| 'timestamptz'\n\t| 'interval'\n\t// Boolean\n\t| 'bool'\n\t// Enumerated\n\t| 'enum'\n\t// Geometric\n\t| 'point'\n\t| 'line'\n\t| 'lseg'\n\t| 'box'\n\t| 'path'\n\t| 'polygon'\n\t| 'circle'\n\t// Network Address\n\t| 'cidr'\n\t| 'inet'\n\t| 'macaddr'\n\t| 'macaddr8'\n\t// Bit String\n\t| 'bit'\n\t| 'varbit'\n\t// Full-Text Search\n\t| 'tsvector'\n\t| 'tsquery'\n\t// UUID\n\t| 'uuid'\n\t// XML\n\t| 'xml'\n\t// JSON\n\t| 'json'\n\t| 'jsonb'\n\t// Range\n\t| 'int4range'\n\t| 'int8range'\n\t| 'numrange'\n\t| 'tsrange'\n\t| 'tstzrange'\n\t| 'daterange'\n\t// Multirange\n\t| 'int4multirange'\n\t| 'int8multirange'\n\t| 'nummultirange'\n\t| 'tsmultirange'\n\t| 'tstzmultirange'\n\t| 'datemultirange'\n\t// Object Identifier / System Reference\n\t| 'oid'\n\t| 'regproc'\n\t| 'regprocedure'\n\t| 'regoper'\n\t| 'regoperator'\n\t| 'regclass'\n\t| 'regtype'\n\t| 'regrole'\n\t| 'regnamespace'\n\t| 'regconfig'\n\t| 'regdictionary'\n\t// PostGIS\n\t| 'geometry'\n\t| 'geography'\n\t| 'box2d'\n\t| 'box3d'\n\t| 'raster'\n\t// pgvector\n\t| 'halfvec'\n\t| 'sparsevec'\n\t| 'vector';\n\n// Some originals were swapped with aliases for simpler keys\nexport type PostgresAliasType =\n\t// Numeric\n\t| 'int2' // smallint\n\t| 'integer' // int\n\t| 'int4' // int\n\t| 'int8' // bigint\n\t| 'decimal' // numeric\n\t| 'real' // float4\n\t| 'double' // float8\n\t| 'double precision' // float8\n\t| 'serial2' // smallserial\n\t| 'serial4' // serial\n\t| 'serial8' // bigserial\n\t// Text\n\t| 'character' // char\n\t| 'character varying' // varchar\n\t// Datetime\n\t| 'time with time zone' // timetz\n\t| 'timestamp with time zone' // timestamptz\n\t// Boolean\n\t| 'boolean' // bool\n\t// Bit String\n\t| 'bit varying'; // varbit;\n\nexport type PostgresColumnType =\n\t| PostgresType\n\t| PostgresAliasType;\n\nconst PG_ALIAS_TO_TYPE_MAP: Record<PostgresAliasType, PostgresType> = {\n\tint2: 'smallint',\n\tinteger: 'int',\n\tint4: 'int',\n\tint8: 'bigint',\n\tdecimal: 'numeric',\n\treal: 'float4',\n\tdouble: 'float8',\n\t'double precision': 'float8',\n\tserial2: 'smallserial',\n\tserial4: 'serial',\n\tserial8: 'bigserial',\n\tcharacter: 'char',\n\t'character varying': 'varchar',\n\t'time with time zone': 'timetz',\n\t'timestamp with time zone': 'timestamptz',\n\tboolean: 'bool',\n\t'bit varying': 'varbit',\n};\n\nexport function resolvePgType(type: string) {\n\treturn PG_ALIAS_TO_TYPE_MAP[type as keyof typeof PG_ALIAS_TO_TYPE_MAP] ?? type;\n}\n\nexport type PgCodecs = Codecs<PostgresType>;\n\nexport const castToText: CastCodec = (name) => sql`${name}::text`;\nexport const castToTextArr: CastArrayCodec = (name, arrayDimensions) =>\n\tsql`${name}::text${sql.raw('[]'.repeat(arrayDimensions))}`;\n\n/** Used for cases when casting requires to unwrap and rebuild arrays\n *\n * @example\n * string_mtx::text[][] // can be casted to array directly\n *\n * encode(bytea_mtx, 'base64')[][] // invalid syntax, cast requires unwrapping and rebuilding array\n */\nexport const arrayCompatCast = (cast: CastCodec) =>\n(\n\tname: SQLChunk,\n\tarrayDimensions: number | undefined,\n): SQLChunk => {\n\tif (!arrayDimensions) return cast(name);\n\n\tconst aliases: Name[] = [];\n\tfor (let i = 0; i < arrayDimensions; i++) {\n\t\taliases.push(sql.identifier(`s${i}`));\n\t}\n\n\tlet indexed = name;\n\tfor (const alias of aliases) {\n\t\tindexed = sql`${indexed}[${alias}]`;\n\t}\n\n\tlet expression = sql`array(\\\nselect ${cast(indexed)} \\\nfrom generate_subscripts(${name}, ${sql.raw(arrayDimensions.toString())}) ${aliases[arrayDimensions - 1]} \\\norder by ${aliases[arrayDimensions - 1]})`;\n\n\tfor (let dim = arrayDimensions - 1; dim > 0; dim--) {\n\t\texpression = sql`array(\\\nselect ${expression} \\\nfrom generate_subscripts(${name}, ${sql.raw(dim.toString())}) ${aliases[dim - 1]} \\\norder by ${aliases[dim - 1]})`;\n\t}\n\n\t// Otherwise null returns as []\n\treturn sql`case when ${name} is null then null else ${expression} end`;\n};\n\n/** Used to recursively apply value normalizer to array of unknown dimensions */\nexport const arrayCompatNormalize = (normalize: NormalizeCodec, transformToPgArray = false) => {\n\tconst loop: NormalizeArrayCodec = (value, arrayDimensions) => {\n\t\tconst innerDimensions = arrayDimensions - 1;\n\t\tif (arrayDimensions > 1) {\n\t\t\tfor (let i = 0; i < (value as unknown[][]).length; ++i) {\n\t\t\t\tloop((value as unknown[][])[i]!, innerDimensions);\n\t\t\t}\n\t\t} else {\n\t\t\tfor (let i = 0; i < (value as unknown[][]).length; ++i) {\n\t\t\t\tvalue[i] = normalize((value as unknown[][])[i]!);\n\t\t\t}\n\t\t}\n\n\t\treturn value;\n\t};\n\n\treturn transformToPgArray ? (v: any, d: number) => makePgArray(loop(v, d)) : loop;\n};\n\nexport const genericPgCodecs: PgCodecs = {\n\tbytea: {\n\t\tcastInJson: (name) => sql`encode(${name}, 'base64')`,\n\t\tcastArrayInJson: arrayCompatCast((name) => sql`encode(${name}, 'base64')`),\n\t\tnormalizeInJson: (v: string) => Buffer.from(v, 'base64'),\n\t\tnormalizeArrayInJson: arrayCompatNormalize((v: string) => Buffer.from(v, 'base64')),\n\t},\n\tbigint: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tnormalizeInJson: BigInt,\n\t\tnormalizeArrayInJson: arrayCompatNormalize(BigInt),\n\t\tnormalize: BigInt,\n\t\tnormalizeArray: arrayCompatNormalize(BigInt),\n\t},\n\tbigserial: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tnormalizeInJson: BigInt,\n\t\tnormalizeArrayInJson: arrayCompatNormalize(BigInt),\n\t\tnormalize: BigInt,\n\t\tnormalizeArray: arrayCompatNormalize(BigInt),\n\t},\n\tdate: {\n\t\tcastArray: castToTextArr,\n\t},\n\tenum: {\n\t\tcastArray: castToTextArr,\n\t\tnormalizeParamArray: makePgArray,\n\t},\n\tgeometry: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tnormalizeArray: parsePgArray,\n\t},\n\tinterval: {\n\t\tcastArrayInJson: castToTextArr,\n\t\tcastArray: castToTextArr,\n\t},\n\tjson: {\n\t\tnormalizeParam: (v) => typeof v === 'object' && !Array.isArray(v) ? v : JSON.stringify(v),\n\t\tnormalizeParamArray: arrayCompatNormalize((v) => JSON.stringify(v), true),\n\t},\n\tjsonb: {\n\t\tnormalizeParam: (v) => typeof v === 'object' && !Array.isArray(v) ? v : JSON.stringify(v),\n\t\tnormalizeParamArray: arrayCompatNormalize((v) => JSON.stringify(v), true),\n\t},\n\tline: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tcast: castToText,\n\t\tcastArray: castToTextArr,\n\t},\n\tmacaddr8: {\n\t\tcastArrayInJson: castToTextArr,\n\t\tcastArray: castToTextArr,\n\t},\n\tnumeric: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tcastArray: castToTextArr,\n\t},\n\tpoint: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tcast: castToText,\n\t\tcastArray: castToTextArr,\n\t},\n\ttimestamp: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tcastArray: castToTextArr,\n\t},\n\ttimestamptz: {\n\t\tcastInJson: castToText,\n\t\tcastArrayInJson: castToTextArr,\n\t\tcastArray: castToTextArr,\n\t},\n\tbit: {\n\t\tnormalizeArray: parsePgArray,\n\t},\n\thalfvec: {\n\t\tnormalizeArray: parsePgArray,\n\t},\n\tsparsevec: {\n\t\tnormalizeArray: parsePgArray,\n\t},\n\tvector: {\n\t\tnormalizeArray: parsePgArray,\n\t},\n};\n\nexport const refineGenericPgCodecs = (extension?: PartialWithUndefined<PgCodecs>): PgCodecs =>\n\trefineCodecs<PostgresType>(genericPgCodecs, extension);\n"],"mappings":";;;;;;;AAqIA,MAAM,uBAAgE;CACrE,MAAM;CACN,SAAS;CACT,MAAM;CACN,MAAM;CACN,SAAS;CACT,MAAM;CACN,QAAQ;CACR,oBAAoB;CACpB,SAAS;CACT,SAAS;CACT,SAAS;CACT,WAAW;CACX,qBAAqB;CACrB,uBAAuB;CACvB,4BAA4B;CAC5B,SAAS;CACT,eAAe;CACf;AAED,SAAgB,cAAc,MAAc;AAC3C,QAAO,qBAAqB,SAA8C;;AAK3E,MAAa,cAAyB,SAAS,gBAAG,GAAG,KAAK;AAC1D,MAAa,iBAAiC,MAAM,oBACnD,gBAAG,GAAG,KAAK,QAAQA,iBAAI,IAAI,KAAK,OAAO,gBAAgB,CAAC;;;;;;;;AASzD,MAAa,mBAAmB,UAE/B,MACA,oBACc;AACd,KAAI,CAAC,gBAAiB,QAAO,KAAK,KAAK;CAEvC,MAAM,UAAkB,EAAE;AAC1B,MAAK,IAAI,IAAI,GAAG,IAAI,iBAAiB,IACpC,SAAQ,KAAKA,iBAAI,WAAW,IAAI,IAAI,CAAC;CAGtC,IAAI,UAAU;AACd,MAAK,MAAM,SAAS,QACnB,WAAU,gBAAG,GAAG,QAAQ,GAAG,MAAM;CAGlC,IAAI,aAAa,gBAAG;SACZ,KAAK,QAAQ,CAAC;2BACI,KAAK,IAAIA,iBAAI,IAAI,gBAAgB,UAAU,CAAC,CAAC,IAAI,QAAQ,kBAAkB,GAAG;WAC9F,QAAQ,kBAAkB,GAAG;AAEvC,MAAK,IAAI,MAAM,kBAAkB,GAAG,MAAM,GAAG,MAC5C,cAAa,gBAAG;SACT,WAAW;2BACO,KAAK,IAAIA,iBAAI,IAAI,IAAI,UAAU,CAAC,CAAC,IAAI,QAAQ,MAAM,GAAG;WACtE,QAAQ,MAAM,GAAG;AAI3B,QAAO,gBAAG,aAAa,KAAK,0BAA0B,WAAW;;;AAIlE,MAAa,wBAAwB,WAA2B,qBAAqB,UAAU;CAC9F,MAAM,QAA6B,OAAO,oBAAoB;EAC7D,MAAM,kBAAkB,kBAAkB;AAC1C,MAAI,kBAAkB,EACrB,MAAK,IAAI,IAAI,GAAG,IAAK,MAAsB,QAAQ,EAAE,EACpD,MAAM,MAAsB,IAAK,gBAAgB;MAGlD,MAAK,IAAI,IAAI,GAAG,IAAK,MAAsB,QAAQ,EAAE,EACpD,OAAM,KAAK,UAAW,MAAsB,GAAI;AAIlD,SAAO;;AAGR,QAAO,sBAAsB,GAAQ,MAAcC,kCAAY,KAAK,GAAG,EAAE,CAAC,GAAG;;AAG9E,MAAa,kBAA4B;CACxC,OAAO;EACN,aAAa,SAAS,gBAAG,UAAU,KAAK;EACxC,iBAAiB,iBAAiB,SAAS,gBAAG,UAAU,KAAK,aAAa;EAC1E,kBAAkB,MAAc,OAAO,KAAK,GAAG,SAAS;EACxD,sBAAsB,sBAAsB,MAAc,OAAO,KAAK,GAAG,SAAS,CAAC;EACnF;CACD,QAAQ;EACP,YAAY;EACZ,iBAAiB;EACjB,iBAAiB;EACjB,sBAAsB,qBAAqB,OAAO;EAClD,WAAW;EACX,gBAAgB,qBAAqB,OAAO;EAC5C;CACD,WAAW;EACV,YAAY;EACZ,iBAAiB;EACjB,iBAAiB;EACjB,sBAAsB,qBAAqB,OAAO;EAClD,WAAW;EACX,gBAAgB,qBAAqB,OAAO;EAC5C;CACD,MAAM,EACL,WAAW,eACX;CACD,MAAM;EACL,WAAW;EACX,qBAAqBA;EACrB;CACD,UAAU;EACT,YAAY;EACZ,iBAAiB;EACjB,gBAAgBC;EAChB;CACD,UAAU;EACT,iBAAiB;EACjB,WAAW;EACX;CACD,MAAM;EACL,iBAAiB,MAAM,OAAO,MAAM,YAAY,CAAC,MAAM,QAAQ,EAAE,GAAG,IAAI,KAAK,UAAU,EAAE;EACzF,qBAAqB,sBAAsB,MAAM,KAAK,UAAU,EAAE,EAAE,KAAK;EACzE;CACD,OAAO;EACN,iBAAiB,MAAM,OAAO,MAAM,YAAY,CAAC,MAAM,QAAQ,EAAE,GAAG,IAAI,KAAK,UAAU,EAAE;EACzF,qBAAqB,sBAAsB,MAAM,KAAK,UAAU,EAAE,EAAE,KAAK;EACzE;CACD,MAAM;EACL,YAAY;EACZ,iBAAiB;EACjB,MAAM;EACN,WAAW;EACX;CACD,UAAU;EACT,iBAAiB;EACjB,WAAW;EACX;CACD,SAAS;EACR,YAAY;EACZ,iBAAiB;EACjB,WAAW;EACX;CACD,OAAO;EACN,YAAY;EACZ,iBAAiB;EACjB,MAAM;EACN,WAAW;EACX;CACD,WAAW;EACV,YAAY;EACZ,iBAAiB;EACjB,WAAW;EACX;CACD,aAAa;EACZ,YAAY;EACZ,iBAAiB;EACjB,WAAW;EACX;CACD,KAAK,EACJ,gBAAgBA,oCAChB;CACD,SAAS,EACR,gBAAgBA,oCAChB;CACD,WAAW,EACV,gBAAgBA,oCAChB;CACD,QAAQ,EACP,gBAAgBA,oCAChB;CACD;AAED,MAAa,yBAAyB,4CACV,iBAAiB,UAAU"}

package/pg-core/codecs.d.cts

@@ -0,0 +1,27 @@
+import { SQLChunk } from "../sql/sql.cjs";
+import { PartialWithUndefined } from "../utils.cjs";
+import { CastArrayCodec, CastCodec, Codecs, NormalizeArrayCodec, NormalizeCodec } from "../codecs.cjs";
+
+//#region src/pg-core/codecs.d.ts
+type PostgresType = 'smallint' | 'int' | 'bigint' | 'numeric' | 'float4' | 'float8' | 'money' | 'smallserial' | 'serial' | 'bigserial' | 'char' | 'varchar' | 'text' | 'bytea' | 'date' | 'time' | 'timetz' | 'timestamp' | 'timestamptz' | 'interval' | 'bool' | 'enum' | 'point' | 'line' | 'lseg' | 'box' | 'path' | 'polygon' | 'circle' | 'cidr' | 'inet' | 'macaddr' | 'macaddr8' | 'bit' | 'varbit' | 'tsvector' | 'tsquery' | 'uuid' | 'xml' | 'json' | 'jsonb' | 'int4range' | 'int8range' | 'numrange' | 'tsrange' | 'tstzrange' | 'daterange' | 'int4multirange' | 'int8multirange' | 'nummultirange' | 'tsmultirange' | 'tstzmultirange' | 'datemultirange' | 'oid' | 'regproc' | 'regprocedure' | 'regoper' | 'regoperator' | 'regclass' | 'regtype' | 'regrole' | 'regnamespace' | 'regconfig' | 'regdictionary' | 'geometry' | 'geography' | 'box2d' | 'box3d' | 'raster' | 'halfvec' | 'sparsevec' | 'vector';
+type PostgresAliasType = 'int2' | 'integer' | 'int4' | 'int8' | 'decimal' | 'real' | 'double' | 'double precision' | 'serial2' | 'serial4' | 'serial8' | 'character' | 'character varying' | 'time with time zone' | 'timestamp with time zone' | 'boolean' | 'bit varying';
+type PostgresColumnType = PostgresType | PostgresAliasType;
+declare function resolvePgType(type: string): PostgresType;
+type PgCodecs = Codecs<PostgresType>;
+declare const castToText: CastCodec;
+declare const castToTextArr: CastArrayCodec;
+/** Used for cases when casting requires to unwrap and rebuild arrays
+ *
+ * @example
+ * string_mtx::text[][] // can be casted to array directly
+ *
+ * encode(bytea_mtx, 'base64')[][] // invalid syntax, cast requires unwrapping and rebuilding array
+ */
+declare const arrayCompatCast: (cast: CastCodec) => (name: SQLChunk, arrayDimensions: number | undefined) => SQLChunk;
+/** Used to recursively apply value normalizer to array of unknown dimensions */
+declare const arrayCompatNormalize: (normalize: NormalizeCodec, transformToPgArray?: boolean) => NormalizeArrayCodec;
+declare const genericPgCodecs: PgCodecs;
+declare const refineGenericPgCodecs: (extension?: PartialWithUndefined<PgCodecs>) => PgCodecs;
+//#endregion
+export { PgCodecs, PostgresAliasType, PostgresColumnType, PostgresType, arrayCompatCast, arrayCompatNormalize, castToText, castToTextArr, genericPgCodecs, refineGenericPgCodecs, resolvePgType };
+//# sourceMappingURL=codecs.d.cts.map

package/pg-core/codecs.d.ts

@@ -0,0 +1,27 @@
+import { PartialWithUndefined } from "../utils.js";
+import { CastArrayCodec, CastCodec, Codecs, NormalizeArrayCodec, NormalizeCodec } from "../codecs.js";
+import { SQLChunk } from "../sql/sql.js";
+
+//#region src/pg-core/codecs.d.ts
+type PostgresType = 'smallint' | 'int' | 'bigint' | 'numeric' | 'float4' | 'float8' | 'money' | 'smallserial' | 'serial' | 'bigserial' | 'char' | 'varchar' | 'text' | 'bytea' | 'date' | 'time' | 'timetz' | 'timestamp' | 'timestamptz' | 'interval' | 'bool' | 'enum' | 'point' | 'line' | 'lseg' | 'box' | 'path' | 'polygon' | 'circle' | 'cidr' | 'inet' | 'macaddr' | 'macaddr8' | 'bit' | 'varbit' | 'tsvector' | 'tsquery' | 'uuid' | 'xml' | 'json' | 'jsonb' | 'int4range' | 'int8range' | 'numrange' | 'tsrange' | 'tstzrange' | 'daterange' | 'int4multirange' | 'int8multirange' | 'nummultirange' | 'tsmultirange' | 'tstzmultirange' | 'datemultirange' | 'oid' | 'regproc' | 'regprocedure' | 'regoper' | 'regoperator' | 'regclass' | 'regtype' | 'regrole' | 'regnamespace' | 'regconfig' | 'regdictionary' | 'geometry' | 'geography' | 'box2d' | 'box3d' | 'raster' | 'halfvec' | 'sparsevec' | 'vector';
+type PostgresAliasType = 'int2' | 'integer' | 'int4' | 'int8' | 'decimal' | 'real' | 'double' | 'double precision' | 'serial2' | 'serial4' | 'serial8' | 'character' | 'character varying' | 'time with time zone' | 'timestamp with time zone' | 'boolean' | 'bit varying';
+type PostgresColumnType = PostgresType | PostgresAliasType;
+declare function resolvePgType(type: string): PostgresType;
+type PgCodecs = Codecs<PostgresType>;
+declare const castToText: CastCodec;
+declare const castToTextArr: CastArrayCodec;
+/** Used for cases when casting requires to unwrap and rebuild arrays
+ *
+ * @example
+ * string_mtx::text[][] // can be casted to array directly
+ *
+ * encode(bytea_mtx, 'base64')[][] // invalid syntax, cast requires unwrapping and rebuilding array
+ */
+declare const arrayCompatCast: (cast: CastCodec) => (name: SQLChunk, arrayDimensions: number | undefined) => SQLChunk;
+/** Used to recursively apply value normalizer to array of unknown dimensions */
+declare const arrayCompatNormalize: (normalize: NormalizeCodec, transformToPgArray?: boolean) => NormalizeArrayCodec;
+declare const genericPgCodecs: PgCodecs;
+declare const refineGenericPgCodecs: (extension?: PartialWithUndefined<PgCodecs>) => PgCodecs;
+//#endregion
+export { PgCodecs, PostgresAliasType, PostgresColumnType, PostgresType, arrayCompatCast, arrayCompatNormalize, castToText, castToTextArr, genericPgCodecs, refineGenericPgCodecs, resolvePgType };
+//# sourceMappingURL=codecs.d.ts.map