Implemented MaxMessages "limiter" for Queue Browsing.

GyroGearl00se · Jun 16, 2024 · ff506ec · ff506ec
1 parent 6438d70
commit ff506ec
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 3 deletions.
diff --git a/Components/Pages/QueueBrowser.razor b/Components/Pages/QueueBrowser.razor
@@ -52,6 +52,10 @@
     <label>Queue Name</label>
     <input @bind="queueName" name="queueName" autocomplete="on" placeholder="Queue Name" />
 </div>
+<div class="form-group">
+    <label>Max Messages</label>
+    <input @bind="maxMessages" type="number" name="maxMessages" min="1" />
+</div>
 <button class="btn btn-primary" @onclick="BrowseQueueAsync">Browse Queue</button>
 
 <div>
@@ -140,6 +144,7 @@
     private string username;
     private string password;
     private string queueName;
+    private int maxMessages = 50;
     private bool sslVerify = true;
     private bool useTcps = true;
     private List<MessageDetails> messages = new List<MessageDetails>();
@@ -162,7 +167,7 @@
         {
             string formattedHost = FormatHost(host, useTcps);
             isBrowsing = true;
-            messages = await QueueBrowserService.BrowseQueueAsync(formattedHost, vpnName, username, password, queueName, sslVerify);
+            messages = await QueueBrowserService.BrowseQueueAsync(formattedHost, vpnName, username, password, queueName, sslVerify, maxMessages);
             Logger.LogInformation("Messages retrieved: {count}", messages.Count);
         }
         catch (SolaceSystems.Solclient.Messaging.OperationErrorException ex)

diff --git a/Services/QueueBrowserService.cs b/Services/QueueBrowserService.cs
@@ -56,7 +56,7 @@ private void Disconnect()
 
 
 
-        public async Task<List<MessageDetails>> BrowseQueueAsync(string host, string vpnName, string username, string password, string queueName, bool sslVerify)
+        public async Task<List<MessageDetails>> BrowseQueueAsync(string host, string vpnName, string username, string password, string queueName, bool sslVerify, int maxMessages)
         {
             _logger.LogInformation("Queue Browsing started.");
             List<MessageDetails> messages = new List<MessageDetails>();
@@ -91,7 +91,8 @@ public async Task<List<MessageDetails>> BrowseQueueAsync(string host, string vpn
                     using (IBrowser browser = _session.CreateBrowser(queue, browserProps))
                     {
                         IMessage message;
-                        while ((message = await Task.Run(() => browser.GetNext())) != null)
+                        int messageCount = 0;
+                        while ((message = await Task.Run(() => browser.GetNext())) != null && messageCount < maxMessages)
                         {
                             //_logger.LogInformation("Message received: {message}", Encoding.UTF8.GetString(message.BinaryAttachment));
                             DateTimeOffset dateTimeOffset = DateTimeOffset.FromUnixTimeMilliseconds(message.SenderTimestamp);
@@ -129,6 +130,7 @@ public async Task<List<MessageDetails>> BrowseQueueAsync(string host, string vpn
                                 UserProperties = keyValuePairs,
                                 DeliveryMode = message.DeliveryMode.ToString()
                             });
+                            messageCount++;
                         }
                     }
                 }