session.ts — drizzle-orm Source File

Architecture documentation for session.ts, a typescript file in the drizzle-orm codebase. 13 imports, 1 dependents.

File typescript DrizzleORM SQLDialects 13 imports 1 dependents 3 classes

Entity Profile

DrizzleORM→ SQLDialects→ session.ts — drizzle-orm Source File

Dependency Diagram

graph LR
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5["session.ts"]
  3566df6c_fc73_10e3_ceca_d9d9b2ec98b1["bun"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 3566df6c_fc73_10e3_ceca_d9d9b2ec98b1
  363fdd67_2503_7ef7_e1f3_a1ef31694431["index.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 363fdd67_2503_7ef7_e1f3_a1ef31694431
  47a9c453_10f1_a759_6246_695fcce2b20a["types.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 47a9c453_10f1_a759_6246_695fcce2b20a
  27705a9d_afe9_57dd_8c97_e52d8a67d426["entity.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 27705a9d_afe9_57dd_8c97_e52d8a67d426
  220c512d_350b_24bf_2142_53ec35a980ac["logger.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 220c512d_350b_24bf_2142_53ec35a980ac
  65dc222f_bc11_439f_d65b_20d7052b24ec["dialect.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 65dc222f_bc11_439f_d65b_20d7052b24ec
  fa14e9c0_b73d_4bcb_463b_adf18df8a285["index.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> fa14e9c0_b73d_4bcb_463b_adf18df8a285
  79ece9bc_4c30_3a6b_c173_2b7b1842a99b["select.types.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 79ece9bc_4c30_3a6b_c173_2b7b1842a99b
  0e227879_31b1_8596_726d_80119efbf846["session.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 0e227879_31b1_8596_726d_80119efbf846
  e4d6a0ab_9aa2_13a6_a2f1_58d94314c3f2["relations.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> e4d6a0ab_9aa2_13a6_a2f1_58d94314c3f2
  be483a7f_d5d7_7a9b_9a13_44a4a6aafbbd["sql.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> be483a7f_d5d7_7a9b_9a13_44a4a6aafbbd
  051cc2d2_fe6f_16df_aaec_3b341d2c2576["tracing.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> 051cc2d2_fe6f_16df_aaec_3b341d2c2576
  ecce3253_1e75_a87f_27b3_ca87e81a3024["utils.ts"]
  18315257_3b48_a60d_f8cf_ea5ba47f9fd5 --> ecce3253_1e75_a87f_27b3_ca87e81a3024
  053b47f2_f302_5369_9ab2_7667cbadf49f["driver.ts"]
  053b47f2_f302_5369_9ab2_7667cbadf49f --> 18315257_3b48_a60d_f8cf_ea5ba47f9fd5
  style 18315257_3b48_a60d_f8cf_ea5ba47f9fd5 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

/// <reference types="bun-types" />

import type { SavepointSQL, SQL, TransactionSQL } from 'bun';
import { type Cache, NoopCache } from '~/cache/core/index.ts';
import type { WithCacheConfig } from '~/cache/core/types.ts';
import { entityKind } from '~/entity.ts';
import type { Logger } from '~/logger.ts';
import { NoopLogger } from '~/logger.ts';
import type { PgDialect } from '~/pg-core/dialect.ts';
import { PgTransaction } from '~/pg-core/index.ts';
import type { SelectedFieldsOrdered } from '~/pg-core/query-builders/select.types.ts';
import type { PgQueryResultHKT, PgTransactionConfig, PreparedQueryConfig } from '~/pg-core/session.ts';
import { PgPreparedQuery, PgSession } from '~/pg-core/session.ts';
import type { RelationalSchemaConfig, TablesRelationalConfig } from '~/relations.ts';
import { fillPlaceholders, type Query } from '~/sql/sql.ts';
import { tracer } from '~/tracing.ts';
import { type Assume, mapResultRow } from '~/utils.ts';

export class BunSQLPreparedQuery<T extends PreparedQueryConfig> extends PgPreparedQuery<T> {
	static override readonly [entityKind]: string = 'BunSQLPreparedQuery';

	constructor(
		private client: SQL,
		private queryString: string,
		private params: unknown[],
		private logger: Logger,
		cache: Cache,
		queryMetadata: {
			type: 'select' | 'update' | 'delete' | 'insert';
			tables: string[];
		} | undefined,
		cacheConfig: WithCacheConfig | undefined,
		private fields: SelectedFieldsOrdered | undefined,
		private _isResponseInArrayMode: boolean,
		private customResultMapper?: (rows: unknown[][]) => T['execute'],
	) {
		super({ sql: queryString, params }, cache, queryMetadata, cacheConfig);
	}

	async execute(placeholderValues: Record<string, unknown> | undefined = {}): Promise<T['execute']> {
		return tracer.startActiveSpan('drizzle.execute', async (span) => {
			const params = fillPlaceholders(this.params, placeholderValues);

			span?.setAttributes({
				'drizzle.query.text': this.queryString,
				'drizzle.query.params': JSON.stringify(params),
			});

			this.logger.logQuery(this.queryString, params);

			const { fields, queryString: query, client, joinsNotNullableMap, customResultMapper } = this;
			if (!fields && !customResultMapper) {
				return tracer.startActiveSpan('drizzle.driver.execute', async () => {
					return await this.queryWithCache(query, params, async () => {
						return await client.unsafe(query, params as any[]);
					});
				});
			}

			const rows: any[] = await tracer.startActiveSpan('drizzle.driver.execute', async () => {
// ... (165 more lines)