Adding websocket locking in lameDuckMode() too.

Signed-off-by: Ivan Kozlovic <ivan@synadia.com>
nats-io · May 8, 2024 · 5df83a1 · 5df83a1
1 parent 968451a
commit 5df83a1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/server/server.go b/server/server.go
@@ -4138,8 +4138,10 @@ func (s *Server) lameDuckMode() {
 	if s.websocket.server != nil {
 		expected++
 		s.websocket.server.Close()
+		s.websocket.mu.Lock()
 		s.websocket.server = nil
 		s.websocket.listener = nil
+		s.websocket.mu.Unlock()
 	}
 	s.ldmCh = make(chan bool, expected)
 	opts := s.getOpts()