10764: Add unit tests

[arvados.git] / services / ws / router.go

diff --git a/services/ws/router.go b/services/ws/router.go
index 04207455e640badbe3be6dc4f4ed25c4302466bd..15b825f2abfa293f8ba2734f57508ba9306a03af 100644 (file)
--- a/services/ws/router.go
+++ b/services/ws/router.go
@@ -9,6 +9,7 @@ import (
        "sync/atomic"
        "time"
 
+       "git.curoverse.com/arvados.git/sdk/go/ctxlog"
        "github.com/Sirupsen/logrus"
        "golang.org/x/net/websocket"
 )
@@ -115,7 +116,7 @@ func (rtr *router) ServeHTTP(resp http.ResponseWriter, req *http.Request) {
 
        logger := logger(req.Context()).
                WithField("RequestID", rtr.newReqID())
-       ctx := contextWithLogger(req.Context(), logger)
+       ctx := ctxlog.Context(req.Context(), logger)
        req = req.WithContext(ctx)
        logger.WithFields(logrus.Fields{
                "remoteAddr":      req.RemoteAddr,