fix: Fix the incorrect usage of DisableReroute (#991)

alibaba · May 23, 2024 · 7fcb608 · 7fcb608
1 parent 10f1adc
commit 7fcb608
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/plugins/wasm-go/extensions/ai-proxy/main.go b/plugins/wasm-go/extensions/ai-proxy/main.go
@@ -70,6 +70,9 @@ func onHttpRequestHeader(ctx wrapper.HttpContext, pluginConfig config.PluginConf
 	ctx.SetContext(ctxKeyApiName, apiName)
 
 	if handler, ok := activeProvider.(provider.RequestHeadersHandler); ok {
+		// Disable the route re-calculation since the plugin may modify some headers related to  the chosen route.
+		ctx.DisableReroute()
+
 		action, err := handler.OnRequestHeaders(ctx, apiName, log)
 		if err == nil {
 			return action
@@ -153,9 +156,6 @@ func onHttpResponseHeaders(ctx wrapper.HttpContext, pluginConfig config.PluginCo
 		ctx.BufferResponseBody()
 	}
 
-	// Disable the route re-calculation since the plugin may modify some headers related to  the chosen route.
-	ctx.DisableReroute()
-
 	return types.ActionContinue
 }