v2.ts — drizzle-orm Source File

Source Code

/* eslint-disable drizzle-internal/require-entity-kind */
import prand from 'pure-rand';
import { AbstractGenerator } from '../Generators.ts';

export class GenerateUniqueIntervalV2 extends AbstractGenerator<{
	fields?:
		| 'year'
		| 'month'
		| 'day'
		| 'hour'
		| 'minute'
		| 'second'
		| 'year to month'
		| 'day to hour'
		| 'day to minute'
		| 'day to second'
		| 'hour to minute'
		| 'hour to second'
		| 'minute to second';
	isUnique?: boolean;
}> {
	static override readonly 'entityKind': string = 'GenerateUniqueInterval';
	static override readonly version: number = 2;

	private state: {
		rng: prand.RandomGenerator;
		fieldsToGenerate: string[];
		intervalSet: Set<string>;
	} | undefined;
	public override isUnique = true;
	private config: { [key: string]: { from: number; to: number } } = {
		year: {
			from: 0,
			to: 5,
		},
		month: {
			from: 0,
			to: 11,
		},
		day: {
			from: 0,
			to: 29,
		},
		hour: {
			from: 0,
			to: 23,
		},
		minute: {
			from: 0,
			to: 59,
		},
		second: {
			from: 0,
			to: 59,
		},
	};

	override init({ count, seed }: { count: number; seed: number }) {
		const allFields = ['year', 'month', 'day', 'hour', 'minute', 'second'];
		let fieldsToGenerate: string[] = allFields;
// ... (173 more lines)