ServerSimulator Class — drizzle-orm Architecture

Source Code

integration-tests/tests/sqlite/sqlite-proxy-batch.test.ts lines 136–212

class ServerSimulator {
	constructor(private db: BetterSqlite3.Database) {}

	async batch(queries: { sql: string; params: any[]; method: string }[]) {
		const results: { rows: any }[] = [];
		for (const query of queries) {
			const { method, sql, params } = query;

			if (method === 'run') {
				try {
					const result = this.db.prepare(sql).run(params);
					results.push(result as any);
				} catch (e: any) {
					return { error: e.message };
				}
			} else if (method === 'all' || method === 'values') {
				try {
					const rows = this.db.prepare(sql).raw().all(params);
					results.push({ rows: rows });
				} catch (e: any) {
					return { error: e.message };
				}
			} else if (method === 'get') {
				try {
					const row = this.db.prepare(sql).raw().get(params);
					results.push({ rows: row });
				} catch (e: any) {
					return { error: e.message };
				}
			} else {
				return { error: 'Unknown method value' };
			}
		}
		return results;
	}

	async query(sql: string, params: any[], method: string) {
		if (method === 'run') {
			try {
				const result = this.db.prepare(sql).run(params);
				return { data: result as any };
			} catch (e: any) {
				return { error: e.message };
			}
		} else if (method === 'all' || method === 'values') {
			try {
				const rows = this.db.prepare(sql).raw().all(params);
				return { data: rows };
			} catch (e: any) {
				return { error: e.message };
			}
		} else if (method === 'get') {
			try {
				const row = this.db.prepare(sql).raw().get(params);
				return { data: row };
			} catch (e: any) {
				return { error: e.message };
			}
		} else {
			return { error: 'Unknown method value' };
		}
	}

	migrations(queries: string[]) {
		this.db.exec('BEGIN');
		try {
			for (const query of queries) {
				this.db.exec(query);
			}
			this.db.exec('COMMIT');
		} catch {
			this.db.exec('ROLLBACK');
		}

		return {};
	}
}