slog/middleware.go at main · unrolled/slog · GitHub

This repository was archived by the owner on Dec 27, 2023. It is now read-only.

27 lines (21 loc) · 729 Bytes

package slog

import "net/http"

var (

// RequestHeaderKey is the key used when adding the header token.

RequestHeaderKey = "___slog_request_token___"

// RequestFieldKey is the key used in the Field output.

RequestFieldKey = "reqID"

// ResponseHeaderKey will send the token on the response as well (if set).

ResponseHeaderKey = ""

)

// Requestify adds a unique key to the request (header) and uses it for logging.

func Requestify(next http.Handler) http.Handler {

return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {

token := RequestToken.Generate()

r.Header.Add(RequestHeaderKey, token)

if ResponseHeaderKey != "" {

w.Header().Add(ResponseHeaderKey, token)

}

next.ServeHTTP(w, r)

})

}

Expand file tree