From 96d8b9e1afecccae803ec4b956ada745dbe71d9f Mon Sep 17 00:00:00 2001 From: Tom Clegg Date: Thu, 8 Jun 2023 10:30:35 -0400 Subject: [PATCH] 20602: Attach assigned priority to response log entry. Arvados-DCO-1.1-Signed-off-by: Tom Clegg --- sdk/go/httpserver/request_limiter.go | 2 ++ 1 file changed, 2 insertions(+) diff --git a/sdk/go/httpserver/request_limiter.go b/sdk/go/httpserver/request_limiter.go index 988643c746..2bb0a56730 100644 --- a/sdk/go/httpserver/request_limiter.go +++ b/sdk/go/httpserver/request_limiter.go @@ -11,6 +11,7 @@ import ( "time" "github.com/prometheus/client_golang/prometheus" + "github.com/sirupsen/logrus" ) // RequestLimiter wraps http.Handler, limiting the number of @@ -258,6 +259,7 @@ func (rl *RequestLimiter) remove(ent *qent) { func (rl *RequestLimiter) ServeHTTP(resp http.ResponseWriter, req *http.Request) { rl.setupOnce.Do(rl.setup) ent := rl.enqueue(req) + SetResponseLogFields(req.Context(), logrus.Fields{"priority": ent.priority}) var ok bool select { case <-req.Context().Done(): -- 2.30.2