GenerateUniqueNumber Class — drizzle-orm Architecture

Source Code

drizzle-seed/src/services/Generators.ts lines 520–572

export class GenerateUniqueNumber extends AbstractGenerator<
	{
		minValue?: number;
		maxValue?: number;
		precision?: number;
		isUnique?: boolean;
	}
> {
	static override readonly entityKind: string = 'GenerateUniqueNumber';

	private state: {
		genUniqueIntObj: GenerateUniqueInt;
		minValue: number;
		maxValue: number;
		precision: number;
	} | undefined;
	public override isUnique = true;

	override init({ count, seed }: { count: number; seed: number }) {
		let { minValue, maxValue, precision } = this.params;

		if (precision === undefined) {
			precision = 100;
		}

		if (maxValue === undefined) {
			maxValue = count * precision;
		} else {
			maxValue *= precision;
		}

		if (minValue === undefined) {
			minValue = -maxValue;
		} else {
			minValue *= precision;
		}

		const genUniqueIntObj = new GenerateUniqueInt({ minValue, maxValue });
		genUniqueIntObj.init({ count, seed });

		this.state = { genUniqueIntObj, minValue, maxValue, precision };
	}

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

		const value = this.state.genUniqueIntObj.generate() as number / this.state.precision;

		return value;
	}
}