Drizzle <> Cloudflare Durable Objects SQLite

#:schema node_modules/wrangler/config-schema.json
name = "sqlite-durable-objects"
main = "src/index.ts"
compatibility_date = "2024-11-12"
compatibility_flags = [ "nodejs_compat" ]

# Bind a Durable Object. Durable objects are a scale-to-zero compute primitive based on the actor model.
# Durable Objects can live for as long as needed. Use these when you need a long-running "server", such as in realtime apps.
# Docs: https://developers.cloudflare.com/workers/wrangler/configuration/#durable-objects
[[durable_objects.bindings]]
name = "MY_DURABLE_OBJECT"
class_name = "MyDurableObject"

# Durable Object migrations.
# Docs: https://developers.cloudflare.com/workers/wrangler/configuration/#migrations
[[migrations]]
tag = "v1"
new_sqlite_classes = ["MyDurableObject"]

# We need rules so we can import migrations in the next steps
[[rules]]
type = "Text"
globs = ["**/*.sql"]
fallthrough = true

/// <reference types="@cloudflare/workers-types" />
import { drizzle, DrizzleSqliteDODatabase } from 'drizzle-orm/durable-sqlite';
import { DurableObject } from 'cloudflare:workers'
import { migrate } from 'drizzle-orm/durable-sqlite/migrator';
import migrations from '../drizzle/migrations';
import { usersTable } from './db/schema';

export class MyDurableObject extends DurableObject {
	storage: DurableObjectStorage;
	db: DrizzleSqliteDODatabase<any>;

	constructor(ctx: DurableObjectState, env: Env) {
		super(ctx, env);
		this.storage = ctx.storage;
		this.db = drizzle(this.storage, { logger: false });

		// 쿼리를 수락하기 전에 모든 마이그레이션이 완료되었는지 확인합니다.
		// 그렇지 않으면 Drizzle 데이터베이스 `this.db`에 액세스하는 모든 함수에서 `this.migrate()`를 실행해야 합니다.
		ctx.blockConcurrencyWhile(async () => {
			await this._migrate();
		});
	}

	async insertAndList(user: typeof usersTable.$inferInsert) {
		await this.insert(user);
		return this.select();
	}

	async insert(user: typeof usersTable.$inferInsert) {
		await this.db.insert(usersTable).values(user);
	}

	async select() {
		return this.db.select().from(usersTable);
	}

	async _migrate() {
		migrate(this.db, migrations);
	}
}

export default {
	/**
	 * 이것은 Cloudflare Worker의 표준 fetch 핸들러입니다
	 *
	 * @param request - 클라이언트에서 Worker로 제출된 요청
	 * @param env - wrangler.toml에 선언된 바인딩을 참조하는 인터페이스
	 * @param ctx - Worker의 실행 컨텍스트
	 * @returns 클라이언트로 다시 전송될 응답
	 */
	async fetch(request: Request, env: Env): Promise<Response> {
		const id: DurableObjectId = env.MY_DURABLE_OBJECT.idFromName('durable-object');
		const stub = env.MY_DURABLE_OBJECT.get(id);

		// 옵션 A - 최고의 성능.
		// DO 내에서 데이터베이스 액세스가 빠르므로, 모든 데이터베이스 상호 작용을 단일 Durable Object 호출로 번들링하는 것이 최대 성능을 위해 선호됩니다.
		const usersAll = await stub.insertAndList({
			name: 'John',
			age: 30,
			email: 'john@example.com',
		});
		console.log('New user created. Getting all users from the database: ', users);

		// 옵션 B - 느리지만 때때로 디버깅에 유용할 수 있습니다.
		// 노출된 경우 개별 Drizzle 쿼리를 직접 호출할 수도 있지만, 모든 쿼리는 Durable Object 인스턴스로의 왕복이라는 점을 명심하세요.
		await stub.insert({
			name: 'John',
			age: 30,
			email: 'john@example.com',
		});
		console.log('New user created!');

		const users = await stub.select();
		console.log('Getting all users from the database: ', users);

		return Response.json(users);
	}
}