MyDurableObject Class — drizzle-orm Architecture

Source Code

integration-tests/tests/sqlite/durable-objects/index.ts lines 194–3484

export class MyDurableObject extends DurableObject {
	storage: DurableObjectStorage;
	db: DrizzleSqliteDODatabase;
	constructor(ctx: DurableObjectState, env: Env) {
		super(ctx, env);
		this.storage = ctx.storage;
		this.db = drizzle(this.storage, { logger: false });
	}

	async migrate1(): Promise<void> {
		try {
			this.db.run(sql`drop table if exists another_users`);
			this.db.run(sql`drop table if exists users12`);
			this.db.run(sql`drop table if exists __drizzle_migrations`);

			migrate(this.db, migrations);

			this.db.insert(usersMigratorTable).values({ name: 'John', email: 'email' }).run();
			const result = this.db.select().from(usersMigratorTable).all();

			this.db.insert(anotherUsersMigratorTable).values({ name: 'John', email: 'email' }).run();
			const result2 = this.db.select().from(anotherUsersMigratorTable).all();

			expect(result).deep.equal([{ id: 1, name: 'John', email: 'email' }]);
			expect(result2).deep.equal([{ id: 1, name: 'John', email: 'email' }]);

			this.db.run(sql`drop table another_users`);
			this.db.run(sql`drop table users12`);
			this.db.run(sql`drop table __drizzle_migrations`);
		} catch {
			throw new Error('migrate1 has broken');
		}
	}

	async beforeEach(): Promise<void> {
		this.db.run(sql`drop table if exists ${usersTable}`);
		this.db.run(sql`drop table if exists ${users2Table}`);
		this.db.run(sql`drop table if exists ${citiesTable}`);
		this.db.run(sql`drop table if exists ${coursesTable}`);
		this.db.run(sql`drop table if exists ${courseCategoriesTable}`);
		this.db.run(sql`drop table if exists ${orders}`);
		this.db.run(sql`drop table if exists ${bigIntExample}`);
		this.db.run(sql`drop table if exists ${pkExampleTable}`);
		this.db.run(sql`drop table if exists user_notifications_insert_into`);
		this.db.run(sql`drop table if exists users_insert_into`);
		this.db.run(sql`drop table if exists notifications_insert_into`);

		this.db.run(sql`
			create table ${usersTable} (
				id integer primary key,
				name text not null,
				verified integer not null default 0,
				json blob,
				created_at integer not null default (strftime('%s', 'now'))
			)
		`);

		this.db.run(sql`
			create table ${citiesTable} (
				id integer primary key,
				name text not null
			)
		`);
		this.db.run(sql`
			create table ${courseCategoriesTable} (
				id integer primary key,
				name text not null
			)
		`);

		this.db.run(sql`
			create table ${users2Table} (
				id integer primary key,
				name text not null,
				city_id integer references ${citiesTable}(${sql.identifier(citiesTable.id.name)})
			)
		`);
		this.db.run(sql`
			create table ${coursesTable} (
				id integer primary key,
				name text not null,