fix rewrite header name when SetOnResponse = true

isayme · web-flow · commit b853fd03172e · 2024-12-13T11:58:10.000+08:00
diff --git a/pkg/handler/rewrite/rewrite.go b/pkg/handler/rewrite/rewrite.go
@@ -81,7 +81,7 @@ func (r *Rewrite) Handle(rw http.ResponseWriter, req *http.Request) {
 		for _, headerValue := range headerValues {
 			replacedValue := r.replaceHeaderValue(headerValue)
 			if r.rule.SetOnResponse {
-				rw.Header().Add(r.rule.Header, replacedValue)
+				rw.Header().Add(headerName, replacedValue)
 			} else {
 				header.Add(req, headerName, replacedValue)
 			}

Original file line number	Diff line number	Diff line change
`@@ -81,7 +81,7 @@ func (r Rewrite) Handle(rw http.ResponseWriter, req http.Request) {`
`81`	`81`	`for _, headerValue := range headerValues {`
`82`	`82`	`replacedValue := r.replaceHeaderValue(headerValue)`
`83`	`83`	`if r.rule.SetOnResponse {`
`84`		`- rw.Header().Add(r.rule.Header, replacedValue)`
	`84`	`+ rw.Header().Add(headerName, replacedValue)`
`85`	`85`	`} else {`
`86`	`86`	`header.Add(req, headerName, replacedValue)`
`87`	`87`	`}`