From 907677f835fc8f5dc6d6381bc11680c6d24d1466 Mon Sep 17 00:00:00 2001
From: Viktor Liu <viktor@netbird.io>
Date: Wed, 4 Feb 2026 22:28:53 +0800
Subject: [PATCH] Set readiness false on disconnect right away

---
 proxy/server.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/proxy/server.go b/proxy/server.go
index a2123d4e8..dc1a6b0e5 100644
--- a/proxy/server.go
+++ b/proxy/server.go
@@ -314,6 +314,11 @@ func (s *Server) newManagementMappingWorker(ctx context.Context, client proto.Pr
 		s.Logger.Debug("Got mapping updates client from management server")
 
 		err = s.handleMappingStream(ctx, mappingClient, &initialSyncDone)
+
+		if s.healthChecker != nil {
+			s.healthChecker.SetManagementConnected(false)
+		}
+
 		backoffDuration := b.Duration()
 		switch {
 		case errors.Is(err, context.Canceled),