You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
26 lines
485 B
26 lines
485 B
package request_id
|
|
|
|
import (
|
|
"context"
|
|
"net/http"
|
|
)
|
|
|
|
const AmzRequestIDHeader = "x-amz-request-id"
|
|
|
|
func Set(ctx context.Context, id string) context.Context {
|
|
return context.WithValue(ctx, AmzRequestIDHeader, id)
|
|
}
|
|
|
|
func Get(ctx context.Context) string {
|
|
if ctx == nil {
|
|
return ""
|
|
}
|
|
id, _ := ctx.Value(AmzRequestIDHeader).(string)
|
|
return id
|
|
}
|
|
|
|
func InjectToRequest(ctx context.Context, req *http.Request) {
|
|
if req != nil {
|
|
req.Header.Set(AmzRequestIDHeader, Get(ctx))
|
|
}
|
|
}
|