Added middleware for processing request_id grpc and http requests (#6805)
This commit is contained in:
20
weed/util/request_id.go
Normal file
20
weed/util/request_id.go
Normal file
@@ -0,0 +1,20 @@
|
||||
package util
|
||||
|
||||
import "context"
|
||||
|
||||
const (
|
||||
RequestIdHttpHeader = "X-Request-ID"
|
||||
RequestIDKey = "x-request-id"
|
||||
)
|
||||
|
||||
func GetRequestID(ctx context.Context) string {
|
||||
if ctx == nil {
|
||||
return ""
|
||||
}
|
||||
id, _ := ctx.Value(RequestIDKey).(string)
|
||||
return id
|
||||
}
|
||||
|
||||
func WithRequestID(ctx context.Context, id string) context.Context {
|
||||
return context.WithValue(ctx, RequestIDKey, id)
|
||||
}
|
||||
Reference in New Issue
Block a user