fix: proxy start endeless loop

almeidapaulopt · Jan 25, 2025 · 95de7f5 · 95de7f5
1 parent c87537a
commit 95de7f5
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/internal/proxymanager/proxy.go b/internal/proxymanager/proxy.go
@@ -172,10 +172,12 @@ func (proxy *Proxy) Start() {
 	go func() {
 		go proxy.start()
 		for {
-			event := <-proxy.providerProxy.WatchEvents()
-			proxy.state.Store(int32(event.State))
-
-			// TODO
+			select {
+			case event := <-proxy.providerProxy.WatchEvents():
+				proxy.state.Store(int32(event.State))
+			case <-proxy.ctx.Done():
+				return
+			}
 		}
 	}()
 }