Rebasely Beta: data may be purged

examples/live-notes Public

TypeScript · 96 lines · 3.6 KiB · at 359d966

// NoteRoom: the Durable Object behind one note. It owns the note's storage (SQLite) and its// editors' WebSockets, and hands every edit to a Room (room.ts), which decides what happens.//// Sockets are accepted with the Hibernation API, so an idle note costs nothing: the runtime// keeps the connections open while the object is evicted, and wakes it with webSocketMessage// when a frame arrives. Everything the room needs lives in SQLite, so the constructor below is// also the wake-up path.import { DurableObject } from "cloudflare:workers";import type { Op } from "./ops.ts";import { LOG_SIZE, Room, type Note, type NoteStore } from "./room.ts";export class NoteRoom extends DurableObject {  #room: Room;  constructor(ctx: DurableObjectState, env: Cloudflare.Env) {    super(ctx, env);    this.#room = new Room(new SqlNoteStore(ctx.storage));  }  // GET /notes/:id, over RPC from the Worker.  read(): Note {    return this.#room.note;  }  // GET /notes/:id/ws: the Worker forwards the upgrade request here.  async fetch(request: Request): Promise<Response> {    if (request.headers.get("Upgrade") !== "websocket") return new Response("expected a WebSocket upgrade", { status: 426 });    const { 0: client, 1: server } = new WebSocketPair();    this.ctx.acceptWebSocket(server);    server.send(JSON.stringify(this.#room.snapshot()));    return new Response(null, { status: 101, webSocket: client });  }  async webSocketMessage(ws: WebSocket, data: string | ArrayBuffer): Promise<void> {    if (typeof data !== "string") {      ws.send(JSON.stringify({ type: "error", message: "frames are JSON text" }));      return;    }    const out = this.#room.receive(data);    ws.send(JSON.stringify(out.reply));    if (!out.broadcast) return;    const frame = JSON.stringify(out.broadcast);    for (const other of this.ctx.getWebSockets()) {      if (other === ws) continue;      try {        other.send(frame);      } catch {        // The socket is closing; webSocketClose will follow.      }    }  }  async webSocketClose(ws: WebSocket, code: number): Promise<void> {    // 1005 and 1006 report that no close code was sent; they can't be sent back.    ws.close(code === 1005 || code === 1006 ? 1000 : code);  }}// The note's row and its op log in the object's SQLite database.class SqlNoteStore implements NoteStore {  #storage: DurableObjectStorage;  constructor(storage: DurableObjectStorage) {    this.#storage = storage;    storage.sql.exec(`      CREATE TABLE IF NOT EXISTS note (id INTEGER PRIMARY KEY CHECK (id = 1), text TEXT NOT NULL, version INTEGER NOT NULL);      CREATE TABLE IF NOT EXISTS ops (version INTEGER PRIMARY KEY, op TEXT NOT NULL);    `);  }  load(): Note {    const row = this.#storage.sql.exec<{ text: string; version: number }>("SELECT text, version FROM note WHERE id = 1").toArray()[0];    return row ? { text: row.text, version: row.version } : { text: "", version: 0 };  }  opsSince(version: number): Op[] {    return this.#storage.sql      .exec<{ op: string }>("SELECT op FROM ops WHERE version > ? ORDER BY version", version)      .toArray()      .map((r) => JSON.parse(r.op) as Op);  }  commit(note: Note, op: Op): void {    this.#storage.transactionSync(() => {      const sql = this.#storage.sql;      sql.exec(        "INSERT INTO note (id, text, version) VALUES (1, ?, ?) ON CONFLICT (id) DO UPDATE SET text = excluded.text, version = excluded.version",        note.text,        note.version,      );      sql.exec("INSERT INTO ops (version, op) VALUES (?, ?)", note.version, JSON.stringify(op));      sql.exec("DELETE FROM ops WHERE version <= ?", note.version - LOG_SIZE);    });  }}