Dont wait on a disconnected follower

There is no need to wait for a disconnected follower to be synced.
cloudamqp · Dec 11, 2024 · 5c12a40 · 5c12a40
1 parent 256e916
commit 5c12a40
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/src/lavinmq/clustering/follower.cr b/src/lavinmq/clustering/follower.cr
@@ -177,6 +177,7 @@ module LavinMQ
       def close(timeout : Time::Span = 30.seconds)
         @closed = true
         @actions.close
+        return if @socket.closed?
         if lag_in_bytes > 0
           Log.info { "Waiting for follower to be in sync" }
           select

diff --git a/src/lavinmq/clustering/server.cr b/src/lavinmq/clustering/server.cr
@@ -160,12 +160,16 @@ module LavinMQ
         end
       rescue ex : AuthenticationError
         Log.warn { "Follower negotiation error" }
+        socket.close
       rescue ex : InvalidStartHeaderError
         Log.warn { ex.message }
+        socket.close
       rescue ex : IO::EOFError
         Log.info { "Follower disconnected" }
+        socket.close
       rescue ex : IO::Error
         Log.warn { "Follower disonnected: #{ex.message}" }
+        socket.close
       ensure
         follower.try &.close
       end