From 15f39b7620817292cfca39c71455ce3f01f3261b Mon Sep 17 00:00:00 2001
From: Hailey <me@haileyok.com>
Date: Sun, 4 Jan 2026 14:11:56 -0800
Subject: [PATCH] fix websocket disconnect

---
 server/handle_sync_subscribe_repos.go | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/server/handle_sync_subscribe_repos.go b/server/handle_sync_subscribe_repos.go
index d768a82..9bdcd34 100644
--- a/server/handle_sync_subscribe_repos.go
+++ b/server/handle_sync_subscribe_repos.go
@@ -43,9 +43,15 @@ func (s *Server) handleSyncSubscribeRepos(e echo.Context) error {
 	// drop the connection whenever a subscriber disconnects from the socket, we should get errors
 	go func() {
 		for {
-			if _, _, err := conn.ReadMessage(); err != nil {
-				logger.Warn("websocket error", "err", err)
-				cancel()
+			select {
+			case <-ctx.Done():
+				return
+			default:
+				if _, _, err := conn.ReadMessage(); err != nil {
+					logger.Warn("websocket error", "err", err)
+					cancel()
+					return
+				}
 			}
 		}
 	}()
-- 
2.51.2