Merge pull request #26 from aki-0421/wait-until

Add WaitUntil

cloudflare/context.go

@@ -0,0 +1,18 @@
+package cloudflare
+
+import (
+	"context"
+	"syscall/js"
+
+	"github.com/syumai/workers/cloudflare/internal/cfruntimecontext"
+	"github.com/syumai/workers/internal/jsutil"
+)
+
+func WaitUntil(ctx context.Context, task func()) {
+	executionContext := cfruntimecontext.GetExecutionContext(ctx)
+
+	executionContext.Call("waitUntil", jsutil.NewPromise(js.FuncOf(func(this js.Value, args []js.Value) any {
+		task()
+		return nil
+	})))
+}

examples/wait-until/.gitignore

@@ -0,0 +1 @@
+dist

examples/wait-until/Makefile

@@ -0,0 +1,12 @@
+.PHONY: dev
+dev:
+	wrangler dev
+
+.PHONY: build
+build:
+	mkdir -p dist
+	tinygo build -o ./dist/app.wasm -target wasm ./...
+
+.PHONY: publish
+publish:
+	wrangler publish

examples/wait-until/README.md

@@ -0,0 +1,25 @@
+# wait-until
+
+* Executes a task in worker even after the server has returned the response.
+* This example executes 5-second task after responding and outputs the console log.
+
+## Document
+
+* https://developers.cloudflare.com/workers/runtime-apis/fetch-event/#waituntil
+
+## Development
+
+### Requirements
+
+This project requires these tools to be installed globally.
+
+* wrangler
+* tinygo
+
+### Commands
+
+```
+make dev     # run dev server
+make build   # build Go Wasm binary
+make publish # publish worker
+```

examples/wait-until/go.mod

@@ -0,0 +1,7 @@
+module github.com/syumai/hello
+
+go 1.18
+
+require github.com/syumai/workers v0.0.0
+
+replace github.com/syumai/workers => ../../

examples/wait-until/go.sum

@@ -0,0 +1,2 @@
+github.com/syumai/workers v0.1.0 h1:z5QfQR2X+PCKzom7RodpI5J4D5YF7NT7Qwzb9AM9dgY=
+github.com/syumai/workers v0.1.0/go.mod h1:alXIDhTyeTwSzh0ZgQ3cb9HQPyyYfIejupE4Z3efr14=

examples/wait-until/main.go

@@ -0,0 +1,25 @@
+package main
+
+import (
+	"fmt"
+	"net/http"
+	"time"
+
+	"github.com/syumai/workers"
+	"github.com/syumai/workers/cloudflare"
+)
+
+func main() {
+	handler := http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
+		cloudflare.WaitUntil(req.Context(), func() {
+			for i := 0; i < 5; i++ {
+				time.Sleep(time.Second)
+			}
+
+			fmt.Println("5-second task completed")
+		})
+
+		w.Write([]byte("http response done"))
+	})
+	workers.Serve(handler)
+}

examples/wait-until/worker.mjs

@@ -0,0 +1,22 @@
+import "../assets/polyfill_performance.js";
+import "../assets/wasm_exec.js";
+import mod from "./dist/app.wasm";
+
+const go = new Go();
+
+const readyPromise = new Promise((resolve) => {
+  globalThis.ready = resolve;
+});
+
+const load = WebAssembly.instantiate(mod, go.importObject).then((instance) => {
+  go.run(instance);
+  return instance;
+});
+
+export default {
+  async fetch(req, env, ctx) {
+    await load;
+    await readyPromise;
+    return handleRequest(req, { env, ctx });
+  }
+}

examples/wait-until/wrangler.toml

@@ -0,0 +1,6 @@
+name = "wait-until"
+main = "./worker.mjs"
+compatibility_date = "2023-02-24"
+
+[build]
+command = "make build"