fix: 🐛 【frontend-gray插件】针对fetch的请求，强制不缓存

heimanba · Mar 6, 2025 · 1549e4d · 1549e4d
1 parent d721c23
commit 1549e4d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 0 deletions.
diff --git a/plugins/wasm-go/extensions/frontend-gray/config/config.go b/plugins/wasm-go/extensions/frontend-gray/config/config.go
@@ -13,6 +13,7 @@ const (
 	IsPageRequest   = "is-page-request"
 	IsNotFound      = "is-not-found"
 	EnabledGray     = "enabled-gray"
+	SecFetchMode    = "sec-fetch-mode"
 )
 
 type LogInfo func(format string, args ...interface{})

diff --git a/plugins/wasm-go/extensions/frontend-gray/main.go b/plugins/wasm-go/extensions/frontend-gray/main.go
@@ -44,6 +44,8 @@ func onHttpRequestHeaders(ctx wrapper.HttpContext, grayConfig config.GrayConfig,
 	}
 	enabledGray := util.IsGrayEnabled(grayConfig, requestPath)
 	ctx.SetContext(config.EnabledGray, enabledGray)
+	secFetchMode, _ := proxywasm.GetHttpRequestHeader("sec-fetch-mode")
+	ctx.SetContext(config.SecFetchMode, secFetchMode)
 
 	if !enabledGray {
 		log.Infof("gray not enabled")
@@ -133,6 +135,10 @@ func onHttpResponseHeader(ctx wrapper.HttpContext, grayConfig config.GrayConfig,
 		ctx.DontReadResponseBody()
 		return types.ActionContinue
 	}
+	secFetchMode, isSecFetchModeOk := ctx.GetContext(config.SecFetchMode).(string)
+	if isSecFetchModeOk && secFetchMode == "cors" {
+		proxywasm.ReplaceHttpResponseHeader("cache-control", "no-cache, no-store, max-age=0, must-revalidate")
+	}
 	isPageRequest, ok := ctx.GetContext(config.IsPageRequest).(bool)
 	if !ok {
 		isPageRequest = false // 默认值