GenerateNumber Class — drizzle-orm Architecture

Source Code

drizzle-seed/src/services/Generators.ts lines 464–518

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

	private state: {
		rng: prand.RandomGenerator;
		minValue: number;
		maxValue: number;
		precision: number;
	} | undefined;
	override uniqueVersionOfGen = GenerateUniqueNumber;

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

		let { minValue, maxValue, precision } = this.params;
		if (precision === undefined) {
			precision = 100;
		}

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

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

		const rng = prand.xoroshiro128plus(seed);

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

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

		let value: number;

		[value, this.state.rng] = prand.uniformIntDistribution(this.state.minValue, this.state.maxValue, this.state.rng);
		return value / this.state.precision;
	}
}