Always close socket in Follower#close

cloudamqp · Dec 12, 2024 · fffd6db · fffd6db
1 parent 5c12a40
commit fffd6db
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/lavinmq/clustering/follower.cr b/src/lavinmq/clustering/follower.cr
@@ -59,7 +59,7 @@ module LavinMQ
         end
       ensure
         begin
-          @lz4.close
+          lz4.close unless lz4.closed?
           @socket.close
         rescue IO::Error
           # ignore connection errors while closing
@@ -186,6 +186,9 @@ module LavinMQ
             Log.warn { "Timeout waiting for follower to be in sync" }
           end
         end
+      ensure
+        @lz4.close unless @lz4.closed?
+        @socket.close
       end
 
       def to_json(json : JSON::Builder)