starred/godoxy
1
0
Fork 0
mirror of https://github.com/yusing/godoxy.git synced 2026-04-22 16:58:54 +02:00
Code Issues 18 Packages Projects Releases 10 Wiki Activity

fixed middleware implementation, added middleware tracing for easier debug

Browse Source
This commit is contained in:
yusing
2024-10-02 13:55:41 +08:00
parent d172552fb0
commit ba13b81b0e
31 changed files with 561 additions and 196 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
internal/net/http/middleware/x_forwarded.go
View File

@@ -2,6 +2,7 @@ package middleware
import (
"net"
"net/http"
)
const (
@@ -14,7 +15,7 @@ const (
)
var SetXForwarded = &Middleware{
rewrite: func(req *Request) {
before: func(next http.HandlerFunc, w ResponseWriter, req *Request) {
req.Header.Del("Forwarded")
req.Header.Del(xForwardedFor)
req.Header.Del(xForwardedHost)
@@ -23,7 +24,7 @@ var SetXForwarded = &Middleware{
if err == nil {
req.Header.Set(xForwardedFor, clientIP)
} else {
req.Header.Del(xForwardedFor)
req.Header.Set(xForwardedFor, req.RemoteAddr)
}
req.Header.Set(xForwardedHost, req.Host)
if req.TLS == nil {
@@ -31,14 +32,16 @@ var SetXForwarded = &Middleware{
} else {
req.Header.Set(xForwardedProto, "https")
}
next(w, req)
},
}
var HideXForwarded = &Middleware{
rewrite: func(req *Request) {
before: func(next http.HandlerFunc, w ResponseWriter, req *Request) {
req.Header.Del("Forwarded")
req.Header.Del(xForwardedFor)
req.Header.Del(xForwardedHost)
req.Header.Del(xForwardedProto)
next(w, req)
},
}