mirror of
https://github.com/yusing/godoxy.git
synced 2026-04-23 01:08:47 +02:00
fix(middleware): fix route bypass matching
- replace upstream headers approach with context value
This commit is contained in:
61
internal/net/gphttp/reverseproxy/context.go
Normal file
61
internal/net/gphttp/reverseproxy/context.go
Normal file
@@ -0,0 +1,61 @@
|
||||
package reverseproxy
|
||||
|
||||
import (
|
||||
"context"
|
||||
"net/http"
|
||||
)
|
||||
|
||||
var reverseProxyContextKey = struct{}{}
|
||||
|
||||
func (rp *ReverseProxy) WithContextValue(r *http.Request) *http.Request {
|
||||
return r.WithContext(context.WithValue(r.Context(), reverseProxyContextKey, rp))
|
||||
}
|
||||
|
||||
func TryGetReverseProxy(r *http.Request) *ReverseProxy {
|
||||
if rp, ok := r.Context().Value(reverseProxyContextKey).(*ReverseProxy); ok {
|
||||
return rp
|
||||
}
|
||||
return nil
|
||||
}
|
||||
|
||||
func TryGetUpstreamName(r *http.Request) string {
|
||||
if rp := TryGetReverseProxy(r); rp != nil {
|
||||
return rp.TargetName
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func TryGetUpstreamScheme(r *http.Request) string {
|
||||
if rp := TryGetReverseProxy(r); rp != nil {
|
||||
return rp.TargetURL.Scheme
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func TryGetUpstreamHost(r *http.Request) string {
|
||||
if rp := TryGetReverseProxy(r); rp != nil {
|
||||
return rp.TargetURL.Hostname()
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func TryGetUpstreamPort(r *http.Request) string {
|
||||
if rp := TryGetReverseProxy(r); rp != nil {
|
||||
return rp.TargetURL.Port()
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func TryGetUpstreamAddr(r *http.Request) string {
|
||||
if rp := TryGetReverseProxy(r); rp != nil {
|
||||
return rp.TargetURL.Host
|
||||
}
|
||||
return ""
|
||||
}
|
||||
|
||||
func TryGetUpstreamURL(r *http.Request) string {
|
||||
if rp := TryGetReverseProxy(r); rp != nil {
|
||||
return rp.TargetURL.String()
|
||||
}
|
||||
return ""
|
||||
}
|
||||
Reference in New Issue
Block a user