yield on delete so lavinmq doesnt freeze during delete

cloudamqp · Nov 22, 2024 · faf8584 · faf8584
1 parent 19b2652
commit faf8584
Showing 1 changed file with 8 additions and 2 deletions.
diff --git 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