mirror of
https://github.com/seaweedfs/seaweedfs.git
synced 2025-08-20 04:33:36 +08:00
28 lines
516 B
Go
28 lines
516 B
Go
package util
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
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)
|
|
}
|
|
|
|
func ReqWithRequestId(req *http.Request, ctx context.Context) {
|
|
req.Header.Set(RequestIdHttpHeader, GetRequestID(ctx))
|
|
}
|