session.ts — drizzle-orm Source File

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

File typescript DrizzleORM DatabaseDrivers 5 imports 1 dependents 2 classes

Entity Profile

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

Dependency Diagram

graph LR
  3260517d_a12c_a392_fb71_4aac0c61619f["session.ts"]
  041a3b87_0dbb_2d78_884f_fadf4eaefa72["extension"]
  3260517d_a12c_a392_fb71_4aac0c61619f --> 041a3b87_0dbb_2d78_884f_fadf4eaefa72
  27705a9d_afe9_57dd_8c97_e52d8a67d426["entity.ts"]
  3260517d_a12c_a392_fb71_4aac0c61619f --> 27705a9d_afe9_57dd_8c97_e52d8a67d426
  220c512d_350b_24bf_2142_53ec35a980ac["logger.ts"]
  3260517d_a12c_a392_fb71_4aac0c61619f --> 220c512d_350b_24bf_2142_53ec35a980ac
  fa14e9c0_b73d_4bcb_463b_adf18df8a285["index.ts"]
  3260517d_a12c_a392_fb71_4aac0c61619f --> fa14e9c0_b73d_4bcb_463b_adf18df8a285
  be483a7f_d5d7_7a9b_9a13_44a4a6aafbbd["sql.ts"]
  3260517d_a12c_a392_fb71_4aac0c61619f --> be483a7f_d5d7_7a9b_9a13_44a4a6aafbbd
  2aa3de16_49e0_facf_4e57_192a571e49d5["driver.ts"]
  2aa3de16_49e0_facf_4e57_192a571e49d5 --> 3260517d_a12c_a392_fb71_4aac0c61619f
  style 3260517d_a12c_a392_fb71_4aac0c61619f fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

import type { PrismaClient } from '@prisma/client/extension';

import { entityKind } from '~/entity.ts';
import { type Logger, NoopLogger } from '~/logger.ts';
import type {
	PgDialect,
	PgQueryResultHKT,
	PgTransaction,
	PgTransactionConfig,
	PreparedQueryConfig,
} from '~/pg-core/index.ts';
import { PgPreparedQuery, PgSession } from '~/pg-core/index.ts';
import type { Query, SQL } from '~/sql/sql.ts';
import { fillPlaceholders } from '~/sql/sql.ts';

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

	constructor(
		private readonly prisma: PrismaClient,
		query: Query,
		private readonly logger: Logger,
	) {
		super(query, undefined, undefined, undefined);
	}

	override execute(placeholderValues?: Record<string, unknown>): Promise<T> {
		const params = fillPlaceholders(this.query.params, placeholderValues ?? {});
		this.logger.logQuery(this.query.sql, params);
		return this.prisma.$queryRawUnsafe(this.query.sql, ...params);
	}

	override all(): Promise<unknown> {
		throw new Error('Method not implemented.');
	}

	override isResponseInArrayMode(): boolean {
		return false;
	}
}

export interface PrismaPgSessionOptions {
	logger?: Logger;
}

export class PrismaPgSession extends PgSession {
	static override readonly [entityKind]: string = 'PrismaPgSession';

	private readonly logger: Logger;

	constructor(
		dialect: PgDialect,
		private readonly prisma: PrismaClient,
		private readonly options: PrismaPgSessionOptions,
	) {
		super(dialect);
		this.logger = options.logger ?? new NoopLogger();
	}

	override execute<T>(query: SQL): Promise<T> {
		return this.prepareQuery<PreparedQueryConfig & { execute: T }>(this.dialect.sqlToQuery(query)).execute();
	}

	override prepareQuery<T extends PreparedQueryConfig = PreparedQueryConfig>(query: Query): PgPreparedQuery<T> {
		return new PrismaPgPreparedQuery(this.prisma, query, this.logger);
	}

	override transaction<T>(
		_transaction: (tx: PgTransaction<PgQueryResultHKT, Record<string, never>, Record<string, never>>) => Promise<T>,
		_config?: PgTransactionConfig,
	): Promise<T> {
		throw new Error('Method not implemented.');
	}
}

export interface PrismaPgQueryResultHKT extends PgQueryResultHKT {
	type: [];
}