Generators.ts — drizzle-orm Source File

Source Code

/* eslint-disable drizzle-internal/require-entity-kind */
import prand from 'pure-rand';
import adjectives, { maxStringLength as maxAdjectiveLength } from '../datasets/adjectives.ts';
import cityNames, { maxStringLength as maxCityNameLength } from '../datasets/cityNames.ts';
import companyNameSuffixes, { maxStringLength as maxCompanyNameSuffixLength } from '../datasets/companyNameSuffixes.ts';
import countries, { maxStringLength as maxCountryLength } from '../datasets/countries.ts';
import emailDomains, { maxStringLength as maxEmailDomainLength } from '../datasets/emailDomains.ts';
import firstNames, { maxStringLength as maxFirstNameLength } from '../datasets/firstNames.ts';
import jobsTitles, { maxStringLength as maxJobTitleLength } from '../datasets/jobsTitles.ts';
import lastNames, { maxStringLength as maxLastNameLength } from '../datasets/lastNames.ts';
import loremIpsumSentences, { maxStringLength as maxLoremIpsumLength } from '../datasets/loremIpsumSentences.ts';
import phonesInfo from '../datasets/phonesInfo.ts';
import states, { maxStringLength as maxStateLength } from '../datasets/states.ts';
import streetSuffix, { maxStringLength as maxStreetSuffixLength } from '../datasets/streetSuffix.ts';
import { fastCartesianProduct, fillTemplate, getWeightedIndices, isObject } from './utils.ts';

export abstract class AbstractGenerator<T = {}> {
	static readonly entityKind: string = 'AbstractGenerator';
	static readonly version: number = 1;

	public isUnique = false;
	public notNull = false;

	// param for generators which have a unique version of themselves
	public uniqueVersionOfGen?: new(params: T) => AbstractGenerator<T>;

	public dataType?: string;
	public timeSpent?: number;

	//
	public arraySize?: number;
	public baseColumnDataType?: string;

	// param for text-like generators
	public stringLength?: number;

	// params for GenerateValuesFromArray
	public weightedCountSeed?: number | undefined;
	public maxRepeatedValuesCount?: number | { weight: number; count: number | number[] }[] | undefined;

	public params: T;

	constructor(params?: T) {
		this.params = params === undefined ? {} as T : params as T;
	}

	init(params: { count: number | { weight: number; count: number | number[] }[]; seed: number }): void;
	init() {
		this.updateParams();
	}

	updateParams() {
		if ((this.params as any).arraySize !== undefined) {
			this.arraySize = (this.params as any).arraySize;
		}

		if ((this.params as any).isUnique !== undefined) {
			if ((this.params as any).isUnique === false && this.isUnique === true) {
				throw new Error('specifying non unique generator to unique column.');
			}
// ... (3060 more lines)