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");