feat(example): extend batcher examples to use bindings (D1)

sor4chi · Dec 3, 2023 · fa4166d · fa4166d
1 parent 504a315
commit fa4166d
Show file tree

Hide file tree

Showing 4 changed files with 64 additions and 26 deletions.
diff --git a/examples/batcher/README.md b/examples/batcher/README.md
@@ -5,6 +5,25 @@ The code is based on [Cloudflare's Alarm example](https://developers.cloudflare.
 
 ```
 pnpm install
-pnpm dev
 ```
 
+Setup the database:
+
+```
+pnpm create-db
+```
+
+Copy the outputted database configuration.
+Add the database configuration to the `wrangler.toml` file.
+
+Add the following to the `wrangler.toml` file's `[[d1_databases]]` section:
+
+```
+preview_database_id = "DB"
+```
+
+Run the following to start the batcher:
+
+```
+pnpm dev
+```
diff --git a/examples/batcher/package.json b/examples/batcher/package.json
@@ -8,7 +8,8 @@
     "format": "prettier --write \"src/**/*.{ts,tsx}\"",
     "format:check": "prettier --check \"src/**/*.{ts,tsx}\"",
     "dev": "wrangler dev src/index.ts",
-    "deploy": "wrangler deploy --minify src/index.ts"
+    "deploy": "wrangler deploy --minify src/index.ts",
+    "create-db": "wrangler d1 create batch-db"
   },
   "dependencies": {
     "hono": "^3.6.0",

diff --git a/examples/batcher/src/batcher.ts b/examples/batcher/src/batcher.ts
@@ -2,6 +2,8 @@
 
 import { generateHonoObject } from "hono-do";
 
+import { Env } from ".";
+
 const SECONDS = 1000;
 
 declare module "hono-do" {
@@ -10,29 +12,42 @@ declare module "hono-do" {
   }
 }
 
-export const Batcher = generateHonoObject(
-  "/batcher",
-  async (app, state, vars) => {
-    const { storage } = state;
-    const vals = await storage.list({ reverse: true, limit: 1 });
-    vars.count = vals.size === 0 ? 0 : parseInt(vals.keys().next().value);
-
-    app.post("/", async (c) => {
-      vars.count++;
-
-      const currentAlarm = await storage.getAlarm();
-      if (currentAlarm == null) {
-        await storage.setAlarm(Date.now() + 10 * SECONDS);
-      }
-
-      await storage.put(vars.count.toString(), await c.req.text());
-      return c.json({ queued: vars.count });
-    });
-  },
-).alarm(async (state, vars) => {
+export const Batcher = generateHonoObject<
+  Env,
+  Record<string, never>,
+  "/batcher"
+>("/batcher", async (app, state, vars) => {
+  await vars.env.DB.prepare(
+    "CREATE TABLE IF NOT EXISTS batcher (id INTEGER PRIMARY KEY AUTOINCREMENT, data TEXT)",
+  ).run();
+
   const { storage } = state;
-  const vals = await storage.list();
-  console.log(Array.from(vals.values())); // eg: POST other upstream service
+  const vals = await storage.list<string>({ reverse: true, limit: 1 });
+  vars.count = vals.size === 0 ? 0 : parseInt(vals.keys().next().value);
+
+  app.post("/", async (c) => {
+    vars.count++;
+
+    const currentAlarm = await storage.getAlarm();
+    if (currentAlarm == null) {
+      await storage.setAlarm(Date.now() + 10 * SECONDS);
+    }
+
+    await storage.put(vars.count.toString(), await c.req.text());
+    return c.json({ queued: vars.count });
+  });
+}).alarm(async ({ storage }, vars) => {
+  const vals = await storage.list<string>();
+  let query = "INSERT INTO batcher (data) VALUES ";
+  const params: string[] = [];
+  for (const value of vals.values()) {
+    query += "(?),";
+    params.push(value);
+  }
+  query = query.slice(0, -1);
+  await vars.env.DB.prepare(query)
+    .bind(...params)
+    .run();
   await storage.deleteAll();
   vars.count = 0;
 });
diff --git a/examples/batcher/src/index.ts b/examples/batcher/src/index.ts
@@ -1,10 +1,13 @@
 import { Hono } from "hono";
 
-const app = new Hono<{
+export interface Env {
   Bindings: {
     BATCHER: DurableObjectNamespace;
+    DB: D1Database;
   };
-}>();
+}
+
+const app = new Hono<Env>();
 
 app.all("/batcher/*", (c) => {
   const id = c.env.BATCHER.idFromName("Batcher");