From 05b3b30c9f077ffe0664d69c0537c0eb97c47981 Mon Sep 17 00:00:00 2001 From: michel-laterman Date: Thu, 12 Sep 2024 08:44:02 -0700 Subject: [PATCH] change to atomic.Uint64 --- testing/fleetservertest/server.go | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/testing/fleetservertest/server.go b/testing/fleetservertest/server.go index 70d68dfeb86..38dd50337e6 100644 --- a/testing/fleetservertest/server.go +++ b/testing/fleetservertest/server.go @@ -133,7 +133,7 @@ func NewRouter(handlers *Handlers) *mux.Router { route.Handler. ServeHTTP(ww, r) handlers.logFn("[%s] DONE %d - %s %s %s %s %d\n", - requestID, ww.statusCode, r.Method, r.URL, r.Proto, r.RemoteAddr, atomic.LoadUint64(&ww.byteCount)) + requestID, ww.statusCode, r.Method, r.URL, r.Proto, r.RemoteAddr, ww.byteCount.Load()) })) } @@ -505,7 +505,7 @@ func updateLocalMetaAgentID(data []byte, agentID string) ([]byte, error) { type statusResponseWriter struct { w http.ResponseWriter statusCode int - byteCount uint64 + byteCount atomic.Uint64 } func (s *statusResponseWriter) Header() http.Header { @@ -514,7 +514,7 @@ func (s *statusResponseWriter) Header() http.Header { func (s *statusResponseWriter) Write(bs []byte) (int, error) { n, err := s.w.Write(bs) - atomic.AddUint64(&s.byteCount, uint64(n)) + s.byteCount.Add(uint64(n)) return n, err }