rewrite插件修复context引用

This commit is contained in:
Liujian
2021-11-12 10:14:41 +08:00
parent f04c8862fa
commit e90aee63ab
3 changed files with 14 additions and 9 deletions

View File

@@ -20,6 +20,11 @@ type ProxyRequest struct {
method string
}
func (r *ProxyRequest) SetPath(s string) {
r.initUrl()
r.uri.Path = s
}
func NewProxyRequest(requestReader *RequestReader) *ProxyRequest {
return &ProxyRequest{RequestReader: requestReader}
}
@@ -91,19 +96,20 @@ func (r *ProxyRequest) SetRaw(contentType string, body []byte) {
}
func (r *ProxyRequest) TargetServer() string {
if r.uri == nil {
uri := r.URL()
r.uri = &uri
}
r.initUrl()
return r.uri.Host
}
func (r *ProxyRequest) TargetURL() string {
r.initUrl()
return r.uri.Path
}
func (r *ProxyRequest) initUrl() {
if r.uri == nil {
uri := r.URL()
r.uri = &uri
}
return r.uri.Path
}
func (r *ProxyRequest) SetMethod(s string) {