GenerateInterval Class — drizzle-orm Architecture

Source Code

drizzle-seed/src/services/Generators.ts lines 1227–1313

export class GenerateInterval 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;
	arraySize?: number;
}> {
	static override readonly entityKind: string = 'GenerateInterval';

	private state: {
		rng: prand.RandomGenerator;
		fieldsToGenerate: string[];
	} | undefined;
	override uniqueVersionOfGen: new(params: any) => AbstractGenerator<any> = GenerateUniqueInterval;
	private config: { [key: string]: { from: number; to: number } } = {
		year: {
			from: 0,
			to: 5,
		},
		month: {
			from: 0,
			to: 12,
		},
		day: {
			from: 1,
			to: 29,
		},
		hour: {
			from: 0,
			to: 24,
		},
		minute: {
			from: 0,
			to: 60,
		},
		second: {
			from: 0,
			to: 60,
		},
	};

	override init({ count, seed }: { count: number; seed: number }) {
		super.init({ count, seed });

		const allFields = ['year', 'month', 'day', 'hour', 'minute', 'second'];
		let fieldsToGenerate: string[] = allFields;

		if (this.params.fields !== undefined && this.params.fields?.includes(' to ')) {
			const tokens = this.params.fields.split(' to ');
			const endIdx = allFields.indexOf(tokens[1]!);
			fieldsToGenerate = allFields.slice(0, endIdx + 1);
		} else if (this.params.fields !== undefined) {
			const endIdx = allFields.indexOf(this.params.fields);
			fieldsToGenerate = allFields.slice(0, endIdx + 1);
		}

		const rng = prand.xoroshiro128plus(seed);
		this.state = { rng, fieldsToGenerate };
	}

	generate() {
		if (this.state === undefined) {
			throw new Error('state is not defined.');
		}

		let interval = '', numb: number;

		for (const field of this.state.fieldsToGenerate) {
			const from = this.config[field]!.from, to = this.config[field]!.to;
			[numb, this.state.rng] = prand.uniformIntDistribution(from, to, this.state.rng);