go-packages/iris
1
0
Fork 0
mirror of https://github.com/kataras/iris.git synced 2025-03-14 11:06:27 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

update host.ProxyHandler to compatiable with different host in target url

Browse Source
This commit is contained in:
tuhao 2021-01-11 15:25:32 +08:00
parent f4989bd5aa
commit b6ebc39b9e
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
core/host/proxy.go
View File

@ -12,20 +12,27 @@ import (
)
// ProxyHandler returns a new ReverseProxy that rewrites
// URLs to the scheme, host, and base path provided in target. If the
// target's path is "/base" and the incoming request was for "/dir",
// URLs to the scheme, host, and base path provided in target.
// Case 1: req.Host == target.Host
// If the target's path is "/base" and the incoming request was for "/dir",
// the target request will be for /base/dir.
//
// Case 2: req.Host != target.Host
// the target request will be forwarded to the target's url
// Relative to httputil.NewSingleHostReverseProxy with some additions.
func ProxyHandler(target *url.URL) *httputil.ReverseProxy {
targetQuery := target.RawQuery
director := func(req *http.Request) {
req.URL.Scheme = target.Scheme
if req.Host != target.Host {
req.URL.Path = target.Path
} else {
req.URL.Path = path.Join(target.Path, req.URL.Path)
}
req.URL.Host = target.Host
req.Host = target.Host
req.URL.Path = path.Join(target.Path, req.URL.Path)
if targetQuery == "" || req.URL.RawQuery == "" {
req.URL.RawQuery = targetQuery + req.URL.RawQuery
} else {