github
/
nps
mirror of https://github.com/ehang-io/nps
1
0
Fork 0
Code Issues Releases Wiki Activity

removed unnecessary host fetch in revereproxy.Director

pull/508/head
snowie2000 2020-04-15 11:08:05 +08:00
parent a732febf3b
commit 89e2a4c2eb
1 changed files with 2 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
server/proxy/http.go
View File

@ -155,13 +155,8 @@ func NewHttpReverseProxy(s *httpServer) *HttpReverseProxy {
local, _ := net.ResolveTCPAddr("tcp", "127.0.0.1") local, _ := net.ResolveTCPAddr("tcp", "127.0.0.1")
proxy := NewReverseProxy(&httputil.ReverseProxy{ proxy := NewReverseProxy(&httputil.ReverseProxy{
Director: func(r *http.Request) { Director: func(r *http.Request) {
r.URL.Host = r.Host host := r.Context().Value("host").(*file.Host)
if host, err := file.GetDb().GetInfoByHost(r.Host, r); err != nil { common.ChangeHostAndHeader(r, host.HostChange, host.HeaderChange, "", false)
logs.Notice("the url %s %s %s can't be parsed!", r.URL.Scheme, r.Host, r.RequestURI)
return
} else {
common.ChangeHostAndHeader(r, host.HostChange, host.HeaderChange, "", false)
}
}, },
Transport: &http.Transport{ Transport: &http.Transport{
ResponseHeaderTimeout: rp.responseHeaderTimeout, ResponseHeaderTimeout: rp.responseHeaderTimeout,