Add header selection for a version

biangeevibegx · Aug 25, 2023 · 4428905 · 4428905
1 parent 04a0c49
commit 4428905
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/wasm.go b/wasm.go
@@ -141,7 +141,11 @@ func (w *WebAssembly) Provision(ctx caddy.Context) error {
 }
 
 func (w *WebAssembly) ServeHTTP(res http.ResponseWriter, req *http.Request, next caddyhttp.Handler) error {
-	instance, err := w.loader.GetOrLoad(req.Context(), w.defaultVersion)
+	version := req.Header.Get("x-caddy-wasm-version")
+	if len(version) == 0 {
+		version = w.defaultVersion
+	}
+	instance, err := w.loader.GetOrLoad(req.Context(), version)
 	if instance == nil {
 		res.WriteHeader(http.StatusInternalServerError)
 		res.Write([]byte("Failed to load wasm: " + err.Error()))