From e47a02841d87101678cd28347fef2d0af1add603 Mon Sep 17 00:00:00 2001
From: Viktor Erlingsson <viktor@84codes.com>
Date: Fri, 22 Nov 2024 15:46:17 +0100
Subject: [PATCH] yield on delete so lavinmq doesnt freeze during delete (#859)

Yields during MessageStore#delete so that LavinMQ does not freeze while deleting large queues.
---
 src/lavinmq/amqp/queue/message_store.cr | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/lavinmq/amqp/queue/message_store.cr b/src/lavinmq/amqp/queue/message_store.cr
index 41fc1fa1bd..62c0d28ca9 100644
--- a/src/lavinmq/amqp/queue/message_store.cr
+++ b/src/lavinmq/amqp/queue/message_store.cr
@@ -195,11 +195,17 @@ module LavinMQ
 
       def delete
         close
-        @segments.each_value { |f| @replicator.try &.delete_file(f.path); f.delete }
-        @acks.each_value { |f| @replicator.try &.delete_file(f.path); f.delete }
+        @segments.each_value { |f| delete_file(f) }
+        @acks.each_value { |f| delete_file(f) }
         FileUtils.rm_rf @queue_data_dir
       end
 
+      def delete_file(file)
+        @replicator.try &.delete_file(file.path)
+        file.delete(raise_on_missing: false)
+        Fiber.yield
+      end
+
       def empty?
         @size.zero?
       end