root/go-reverseproxy
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

4

Browse source
This commit is contained in:
Zeev Diukman 2025-03-08 17:51:15 +00:00
parent e554570004
commit 0b5421edc2
1 changed files with 14 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
reverseproxy.go
View file

@ -28,7 +28,7 @@ func New(ctx context.Context, host string) *ReverseProxy {
targetQuery := target.RawQuery targetQuery := target.RawQuery
r.URL.Scheme = target.Scheme r.URL.Scheme = target.Scheme
r.URL.Host = target.Host r.URL.Host = target.Host
r.URL.Path, r.URL.RawPath = joinURLPath(target, r.URL) r.URL.Path, r.URL.RawPath = JoinURLPath(target, r.URL)
if targetQuery == "" || r.URL.RawQuery == "" { if targetQuery == "" || r.URL.RawQuery == "" {
r.URL.RawQuery = targetQuery + r.URL.RawQuery r.URL.RawQuery = targetQuery + r.URL.RawQuery
} else { } else {
@ -45,11 +45,11 @@ func (revereProxy *ReverseProxy) SetContext(ctx context.Context) {
revereProxy.Context = ctx revereProxy.Context = ctx
} }
func (revereProxy *ReverseProxy) DirectorFunc(df func(jup JoinURLPathFunc) DirectorFunc) { func (revereProxy *ReverseProxy) DirectorFunc(df func(jup JoinURLPathFunc) DirectorFunc) {
d := df(joinURLPath) d := df(JoinURLPath)
revereProxy.Director = d revereProxy.Director = d
} }
func singleJoiningSlash(a, b string) string { func SingleJoiningSlash(a, b string) string {
aslash := strings.HasSuffix(a, "/") aslash := strings.HasSuffix(a, "/")
bslash := strings.HasPrefix(b, "/") bslash := strings.HasPrefix(b, "/")
switch { switch {
@ -61,9 +61,9 @@ func singleJoiningSlash(a, b string) string {
return a + b return a + b
} }
func joinURLPath(a, b *url.URL) (path, rawpath string) { func JoinURLPath(a, b *url.URL) (path, rawpath string) {
if a.RawPath == "" && b.RawPath == "" { if a.RawPath == "" && b.RawPath == "" {
return singleJoiningSlash(a.Path, b.Path), "" return SingleJoiningSlash(a.Path, b.Path), ""
} }
// Same as singleJoiningSlash, but uses EscapedPath to determine // Same as singleJoiningSlash, but uses EscapedPath to determine
// whether a slash should be added // whether a slash should be added
@ -86,3 +86,12 @@ type JoinURLPathFunc func(*url.URL, *url.URL) (string, string)
// type SingleJoiningSlashFunc func(string, string) string // type SingleJoiningSlashFunc func(string, string) string
type DirectorFunc func(*http.Request) type DirectorFunc func(*http.Request)
func StripPrefix(r *http.Request, prefix string) *http.Request {
newPath := strings.TrimPrefix(r.URL.Path, prefix)
if newPath == "" {
newPath = "/"
}
r.URL.Path = newPath
return r
}