GenerateInt Class — drizzle-orm Architecture

Source Code

drizzle-seed/src/services/Generators.ts lines 574–641

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

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

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

		let { minValue, maxValue } = this.params;

		if (maxValue === undefined) {
			maxValue = 1000;
		}

		if (minValue === undefined) {
			minValue = -maxValue;
		}

		if (typeof minValue === 'number' && typeof maxValue === 'number') {
			minValue = minValue >= 0 ? Math.ceil(minValue) : Math.floor(minValue);
			maxValue = maxValue >= 0 ? Math.floor(maxValue) : Math.ceil(maxValue);
		}

		const rng = prand.xoroshiro128plus(seed);

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

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

		let value: number | bigint;
		if (typeof this.state.minValue === 'bigint' && typeof this.state.maxValue === 'bigint') {
			[value, this.state.rng] = prand.uniformBigIntDistribution(
				this.state.minValue,
				this.state.maxValue,
				this.state.rng,
			);
		} else {
			[value, this.state.rng] = prand.uniformIntDistribution(
				this.state.minValue as number,
				this.state.maxValue as number,
				this.state.rng,
			);
		}

		if (this.dataType === 'string') {
			return String(value);
		}

		if (this.dataType === 'bigint') {
			value = BigInt(value);
		}
		return value;
	}
}