generateNumber() — drizzle-orm Function Reference

Architecture documentation for the generateNumber() function in Generators.ts from the drizzle-orm codebase.

Function typescript DrizzleSeed SeedOrchestration called by 1

Entity Profile

DrizzleSeed→ SeedOrchestration→ generateNumber() — drizzle-orm Function Reference

Dependency Diagram

graph TD
  2a2f15f9_4451_afca_fcc5_0b46ffae1a97["generateNumber()"]
  e2c413a5_f1cc_23b4_acd7_ab626330b0d7["GenerateUniqueInt"]
  2a2f15f9_4451_afca_fcc5_0b46ffae1a97 -->|defined in| e2c413a5_f1cc_23b4_acd7_ab626330b0d7
  cf7d0536_2315_3048_f863_848c58a61729["generate()"]
  cf7d0536_2315_3048_f863_848c58a61729 -->|calls| 2a2f15f9_4451_afca_fcc5_0b46ffae1a97
  style 2a2f15f9_4451_afca_fcc5_0b46ffae1a97 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

drizzle-seed/src/services/Generators.ts lines 750–782

	generateNumber(currMinNumb: number, currMaxNumb: number, intervalsToAdd: number[][], intervalIdx: number) {
		let numb: number;

		[numb, this.state!.rng] = prand.uniformIntDistribution(currMinNumb, currMaxNumb, this.state!.rng);

		if (this.genMaxRepeatedValuesCount !== undefined) {
			if (this.state!.integersCount.get(numb) === undefined) {
				this.state!.integersCount.set(numb, this.genMaxRepeatedValuesCount.generate() as number);
			}
			this.state!.integersCount.set(numb, this.state!.integersCount.get(numb)! - 1);
		}

		if (this.state!.integersCount.get(numb) === undefined || this.state!.integersCount.get(numb) === 0) {
			if (numb === currMinNumb) {
				intervalsToAdd = numb + 1 <= currMaxNumb ? [[numb + 1, currMaxNumb]] : [];
			} else if (numb === currMaxNumb) {
				intervalsToAdd = [[currMinNumb, numb - 1]];
			} else {
				intervalsToAdd = [
					[currMinNumb, numb - 1],
					[numb + 1, currMaxNumb],
				];
			}

			const t0 = new Date();
			this.state!.intervals[intervalIdx] = this.state!.intervals[this.state!.intervals.length - 1]!;
			this.state?.intervals.pop();
			this.timeSpent += (Date.now() - t0.getTime()) / 1000;
			this.state!.intervals.push(...intervalsToAdd);
		}

		return numb;
	}