GenerateJson Class — drizzle-orm Architecture

Source Code

drizzle-seed/src/services/Generators.ts lines 1100–1202

export class GenerateJson extends AbstractGenerator<{ arraySize?: number }> {
	static override readonly entityKind: string = 'GenerateJson';

	private state: {
		emailGeneratorObj: GenerateEmail;
		nameGeneratorObj: GenerateFirstName;
		booleanGeneratorObj: GenerateBoolean;
		salaryGeneratorObj: GenerateInt;
		dateGeneratorObj: GenerateDate;
		visitedCountriesNumberGeneratorObj: GenerateInt;
		seed: number;
	} | undefined;

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

		const emailGeneratorObj = new GenerateEmail({});
		emailGeneratorObj.init({ count, seed });

		const nameGeneratorObj = new GenerateFirstName({});
		nameGeneratorObj.init({ count, seed });

		const booleanGeneratorObj = new GenerateBoolean({});
		booleanGeneratorObj.init({
			count,
			seed,
		});

		const salaryGeneratorObj = new GenerateInt({ minValue: 200, maxValue: 4000 });
		salaryGeneratorObj.init({
			count,
			seed,
			...salaryGeneratorObj.params,
		});

		const dateGeneratorObj = new GenerateDate({});
		dateGeneratorObj.dataType = 'string';
		dateGeneratorObj.init({ count, seed });

		const visitedCountriesNumberGeneratorObj = new GenerateInt({ minValue: 0, maxValue: 4 });
		visitedCountriesNumberGeneratorObj.init(
			{ count, seed, ...visitedCountriesNumberGeneratorObj.params },
		);

		this.state = {
			emailGeneratorObj,
			nameGeneratorObj,
			booleanGeneratorObj,
			salaryGeneratorObj,
			dateGeneratorObj,
			visitedCountriesNumberGeneratorObj,
			seed,
		};
	}

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

		const visitedCountries: string[] = [];
		const email = this.state.emailGeneratorObj.generate();
		const name = this.state.nameGeneratorObj.generate();
		const isGraduated = this.state.booleanGeneratorObj.generate();
		const hasJob = this.state.booleanGeneratorObj.generate();
		const salary = this.state.salaryGeneratorObj.generate() as number;
		const startedWorking = this.state.dateGeneratorObj.generate() as string;
		const visitedCountriesNumber = this.state.visitedCountriesNumberGeneratorObj.generate() as number;

		const uniqueCountriesGeneratorObj = new GenerateUniqueCountry({});
		uniqueCountriesGeneratorObj.init({
			count: visitedCountriesNumber,
			seed: this.state.seed + i,
		});
		for (let j = 0; j < visitedCountriesNumber; j++) {
			visitedCountries.push(uniqueCountriesGeneratorObj.generate());
		}

		const returnJson = hasJob
			? {
				email,