Update BackgroundMqService.cs

mythz · mythz · commit 87293bee2171 · 2023-04-27T10:05:27.000+08:00
Add ability to traverse Background MQ messages
diff --git a/ServiceStack/src/ServiceStack/Messaging/BackgroundMqService.cs b/ServiceStack/src/ServiceStack/Messaging/BackgroundMqService.cs
@@ -492,6 +492,7 @@ public class BackgroundMqCollection<T> : IMqCollection
 
         private long totalOutQMessagesAdded = 0;
         private long totalDlQMessagesAdded = 0;
+        public string[] QueueNames { get; }
         
         public BackgroundMqCollection(BackgroundMqClient mqClient, IMessageHandlerFactory handlerFactory, int threadCount, int outQMaxSize)
         {
@@ -506,6 +507,13 @@ public BackgroundMqCollection(BackgroundMqClient mqClient, IMessageHandlerFactor
                 { QueueNames<T>.Dlq, new BlockingCollection<IMessage>() },
                 { QueueNames<T>.Out, new BlockingCollection<IMessage>() },
             };
+            QueueNames = new[]
+            {
+                QueueNames<T>.In,
+                QueueNames<T>.Priority,
+                QueueNames<T>.Dlq,
+                QueueNames<T>.Out,
+            };
         }
 
         public void Add(string queueName, IMessage message)
@@ -640,7 +648,20 @@ public Dictionary<string, long> GetDescriptionMap()
 
             return to;
         }
-        
+
+        public List<IMessage> GetQueueMessages(string queueName)
+        {
+            var to = new List<IMessage>();
+            if (queueMap.TryGetValue(queueName, out var mq))
+            {
+                foreach (var msg in mq)
+                {
+                    to.Add(msg);
+                }
+            }
+            return to;
+        }
+
         //Called when AppHost is disposing
         public void Dispose()
         {
