refactor: headers handling

This commit is contained in:
yusing
2025-05-26 19:57:39 +08:00
parent 216c03c5ff
commit 4a6bed7728
3 changed files with 41 additions and 33 deletions

View File

@@ -14,7 +14,11 @@ var ModifyResponse = NewMiddleware[modifyResponse]()
// modifyResponse implements ResponseModifier.
func (mr *modifyResponse) modifyResponse(resp *http.Response) error {
mr.AddTraceResponse("before modify response", resp)
mr.modifyHeaders(resp.Request, resp, resp.Header)
if !mr.needVarSubstitution {
mr.modifyHeaders(resp.Request, resp.Header)
} else {
mr.modifyHeadersWithVarSubstitution(resp.Request, resp, resp.Header)
}
mr.AddTraceResponse("after modify response", resp)
return nil
}