diff options
Diffstat (limited to 'weed/server')
| -rw-r--r-- | weed/server/common.go | 20 | ||||
| -rw-r--r-- | weed/server/filer_server_handlers_proxy.go | 5 |
2 files changed, 13 insertions, 12 deletions
diff --git a/weed/server/common.go b/weed/server/common.go index 4041eca51..a5ebd45c8 100644 --- a/weed/server/common.go +++ b/weed/server/common.go @@ -7,9 +7,6 @@ import ( "encoding/json" "errors" "fmt" - "github.com/google/uuid" - "github.com/seaweedfs/seaweedfs/weed/util/version" - "google.golang.org/grpc/metadata" "io" "io/fs" "mime/multipart" @@ -21,18 +18,21 @@ import ( "sync" "time" + "github.com/google/uuid" + "github.com/seaweedfs/seaweedfs/weed/util/request_id" + "github.com/seaweedfs/seaweedfs/weed/util/version" + "google.golang.org/grpc/metadata" + "github.com/seaweedfs/seaweedfs/weed/filer" "github.com/seaweedfs/seaweedfs/weed/s3api/s3_constants" "google.golang.org/grpc" + "github.com/gorilla/mux" "github.com/seaweedfs/seaweedfs/weed/glog" "github.com/seaweedfs/seaweedfs/weed/operation" "github.com/seaweedfs/seaweedfs/weed/stats" "github.com/seaweedfs/seaweedfs/weed/storage/needle" - "github.com/seaweedfs/seaweedfs/weed/util" - - "github.com/gorilla/mux" ) var serverStats *stats.ServerStats @@ -429,18 +429,18 @@ func ProcessRangeRequest(r *http.Request, w http.ResponseWriter, totalSize int64 func requestIDMiddleware(h http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { - reqID := r.Header.Get(util.RequestIdHttpHeader) + reqID := r.Header.Get(request_id.AmzRequestIDHeader) if reqID == "" { reqID = uuid.New().String() } - ctx := context.WithValue(r.Context(), util.RequestIDKey, reqID) + ctx := context.WithValue(r.Context(), request_id.AmzRequestIDHeader, reqID) ctx = metadata.NewOutgoingContext(ctx, metadata.New(map[string]string{ - util.RequestIDKey: reqID, + request_id.AmzRequestIDHeader: reqID, })) - w.Header().Set(util.RequestIdHttpHeader, reqID) + w.Header().Set(request_id.AmzRequestIDHeader, reqID) h(w, r.WithContext(ctx)) } } diff --git a/weed/server/filer_server_handlers_proxy.go b/weed/server/filer_server_handlers_proxy.go index fd22ccd7f..5a0d76a0b 100644 --- a/weed/server/filer_server_handlers_proxy.go +++ b/weed/server/filer_server_handlers_proxy.go @@ -3,9 +3,10 @@ package weed_server import ( "github.com/seaweedfs/seaweedfs/weed/glog" "github.com/seaweedfs/seaweedfs/weed/security" - "github.com/seaweedfs/seaweedfs/weed/util" util_http "github.com/seaweedfs/seaweedfs/weed/util/http" "github.com/seaweedfs/seaweedfs/weed/util/mem" + "github.com/seaweedfs/seaweedfs/weed/util/request_id" + "io" "math/rand/v2" "net/http" @@ -54,7 +55,7 @@ func (fs *FilerServer) proxyToVolumeServer(w http.ResponseWriter, r *http.Reques proxyReq.Header.Set("Host", r.Host) proxyReq.Header.Set("X-Forwarded-For", r.RemoteAddr) - util.ReqWithRequestId(proxyReq, ctx) + request_id.InjectToRequest(ctx, proxyReq) for header, values := range r.Header { for _, value := range values { |
