20602: Attach assigned priority to response log entry.
authorTom Clegg <tom@curii.com>
Thu, 8 Jun 2023 14:30:35 +0000 (10:30 -0400)
committerTom Clegg <tom@curii.com>
Thu, 8 Jun 2023 18:31:39 +0000 (14:31 -0400)
Arvados-DCO-1.1-Signed-off-by: Tom Clegg <tom@curii.com>

sdk/go/httpserver/request_limiter.go

index 988643c746f7897f1448c6a8303c8a26d4ec6b55..2bb0a567305ca880b59e71dda6329fd4927592ca 100644 (file)
@@ -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():