improved middleware variable subsititution

This commit is contained in:
yusing
2024-12-04 01:58:17 +08:00
parent fcfb7a0105
commit eabdd3de00
3 changed files with 109 additions and 86 deletions

View File

@@ -1,8 +1,6 @@
package middleware
import (
"net/http"
E "github.com/yusing/go-proxy/internal/error"
)
@@ -14,7 +12,7 @@ func NewModifyResponse(optsRaw OptionsRaw) (*Middleware, E.Error) {
mr := new(modifyResponse)
mr.m = &Middleware{
impl: mr,
modifyResponse: func(resp *http.Response) error {
modifyResponse: func(resp *Response) error {
mr.m.AddTraceResponse("before modify response", resp)
mr.modifyHeaders(resp.Request, resp, resp.Header)
mr.m.AddTraceResponse("after modify response", resp)