diff --git a/internal/middleware/twirp.go b/internal/middleware/twirp.go index 8b60dc6..d6a3169 100644 --- a/internal/middleware/twirp.go +++ b/internal/middleware/twirp.go @@ -1,6 +1,7 @@ package middleware import ( + "context" "net/http" "github.com/Xe/ln" @@ -11,9 +12,14 @@ import ( func Twirp(next http.Handler) http.Handler { return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { ctx := r.Context() + + if ctx == nil { + panic("context is nil") + } + ctx, err := twirp.WithHTTPRequestHeaders(ctx, r.Header) if err != nil { - ln.Error(ctx, err, ln.Action("can't get request headers")) + ln.Error(context.Background(), err, ln.Action("can't get request headers")) http.Error(w, err.Error(), http.StatusInternalServerError) }