1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
|
package correlation
import (
"net/http"
)
// InjectCorrelationID is an HTTP middleware to generate an Correlation-ID for the incoming request,
// or extract the existing Correlation-ID from the incoming request. By default, any upstream Correlation-ID,
// passed in via the `X-Request-ID` header will be ignored. To enable this behaviour, the `WithPropagation`
// option should be passed into the options.
// Whether the Correlation-ID is generated or propagated, once inside this handler the request context
// will have a Correlation-ID associated with it.
func InjectCorrelationID(h http.Handler, opts ...InboundHandlerOption) http.Handler {
config := applyInboundHandlerOptions(opts)
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
parent := r.Context()
correlationID := ""
clientName := ""
if config.shouldPropagate(r) {
correlationID = r.Header.Get(propagationHeader)
clientName = r.Header.Get(clientNameHeader)
}
if correlationID == "" {
correlationID = SafeRandomID()
}
ctx := ContextWithCorrelation(parent, correlationID)
if clientName != "" {
ctx = ContextWithClientName(ctx, clientName)
}
if config.sendResponseHeader {
// Set the response header.
w.Header().Set(propagationHeader, correlationID)
}
h.ServeHTTP(w, r.WithContext(ctx))
})
}
|
