All files / third-parties/drizzle-adapter/src/ydb-core unique-constraint.ts

83.33% Statements 15/18
50% Branches 2/4
80% Functions 8/10
83.33% Lines 15/18

Press n or j to go to the next uncovered block, b, p or k for the previous block.

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80            3x                     7x   7x 7x                             34x         2x       2x         34x           7x 7x       7x   7x                 34x   7x       2x    
import { entityKind } from 'drizzle-orm/entity'
import { getTableName } from 'drizzle-orm/table'
import type { YdbColumn } from './columns/common.js'
import type { YdbTable } from './table.js'
 
export function uniqueKeyName(table: YdbTable, columns: readonly string[]): string {
	return `${getTableName(table)}_${columns.join('_')}_unique`
}
 
function defaultUniqueName(table: YdbTable, columns: readonly YdbColumn[]): string {
	return uniqueKeyName(
		table,
		columns.map((column) => column.name)
	)
}
 
function assertColumnsBelongToTable(table: YdbTable, columns: readonly YdbColumn[]): void {
	let tableName = getTableName(table)
 
	for (let column of columns) {
		Iif (column.table !== table) {
			throw new Error(
				`Unique constraint column "${column.name}" does not belong to table "${tableName}"`
			)
		}
	}
}
 
export interface YdbUniqueConstraintConfig {
	readonly table: YdbTable
	readonly columns: readonly YdbColumn[]
	readonly name: string
}
 
export class YdbUniqueConstraintBuilderOn {
	static readonly [entityKind] = 'YdbUniqueConstraintBuilderOn'
 
	readonly #name: string | undefined
 
	constructor(name: string | undefined) {
		this.#name = name
	}
 
	on(...columns: [YdbColumn, ...YdbColumn[]]): YdbUniqueConstraintBuilder {
		return new YdbUniqueConstraintBuilder(this.#name, columns)
	}
}
 
export class YdbUniqueConstraintBuilder {
	static readonly [entityKind] = 'YdbUniqueConstraintBuilder'
 
	readonly #name: string | undefined
	readonly #columns: [YdbColumn, ...YdbColumn[]]
 
	constructor(name: string | undefined, columns: [YdbColumn, ...YdbColumn[]]) {
		this.#name = name
		this.#columns = columns
	}
 
	build(table: YdbTable): YdbUniqueConstraint {
		assertColumnsBelongToTable(table, this.#columns)
 
		return new YdbUniqueConstraint({
			table,
			columns: [...this.#columns],
			name: this.#name ?? defaultUniqueName(table, this.#columns),
		})
	}
}
 
export class YdbUniqueConstraint {
	static readonly [entityKind] = 'YdbUniqueConstraint'
 
	constructor(readonly config: YdbUniqueConstraintConfig) {}
}
 
export function unique(name?: string): YdbUniqueConstraintBuilderOn {
	return new YdbUniqueConstraintBuilderOn(name)
}