github/wonderwall
1
0
Fork 0
mirror of https://github.com/nais/wonderwall.git synced 2026-05-24 09:12:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(middleware/correlationid): use x-request-id header if found in request

Browse Source
This commit is contained in:
Trong Huu Nguyen
2022-09-16 19:23:21 +02:00
parent b68877b963
commit 889e0c8edf
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
pkg/middleware/correlationid.go
View File

@@ -10,7 +10,10 @@ import (
func CorrelationIDHandler(next http.Handler) http.Handler {
fn := func(w http.ResponseWriter, r *http.Request) {
id := uuid.New().String()
id := r.Header.Get(chi_middleware.RequestIDHeader)
if len(id) == 0 {
id = uuid.New().String()
}
ctx := r.Context()
ctx = context.WithValue(ctx, chi_middleware.RequestIDKey, id)