nats-io#636 - JetStream Batch Get Client support

* Added eod parse in headers
Ivandemidov00 · Nov 23, 2024 · 95a495f · 95a495f
1 parent 075f749
commit 95a495f
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/src/NATS.Client.Core/NatsHeaderParser.cs b/src/NATS.Client.Core/NatsHeaderParser.cs
@@ -119,6 +119,11 @@ private bool TryParseHeaderLine(ReadOnlySpan<byte> headerLine, NatsHeaders heade
                 headers.Message = NatsHeaders.Messages.MessageSizeExceedsMaxBytes;
                 headers.MessageText = NatsHeaders.MessageMessageSizeExceedsMaxBytesStr;
             }
+            else if (headerLine.SequenceEqual(NatsHeaders.MessageEobCode))
+            {
+                headers.Message = NatsHeaders.Messages.EobCode;
+                headers.MessageText = NatsHeaders.MessageEobCodeStr;
+            }
             else
             {
                 headers.Message = NatsHeaders.Messages.Text;

diff --git a/src/NATS.Client.Core/NatsHeaders.cs b/src/NATS.Client.Core/NatsHeaders.cs
@@ -25,6 +25,7 @@ public enum Messages
         NoMessages,
         RequestTimeout,
         MessageSizeExceedsMaxBytes,
+        EobCode,
     }
 
     // Uses C# compiler's optimization for static byte[] data
@@ -56,6 +57,10 @@ public enum Messages
     internal static ReadOnlySpan<byte> MessageMessageSizeExceedsMaxBytes => new byte[] { 77, 101, 115, 115, 97, 103, 101, 32, 83, 105, 122, 101, 32, 69, 120, 99, 101, 101, 100, 115, 32, 77, 97, 120, 66, 121, 116, 101, 115 };
     internal static readonly string MessageMessageSizeExceedsMaxBytesStr = "Message Size Exceeds MaxBytes";
 
+    // EOB
+    internal static ReadOnlySpan<byte> MessageEobCode => new byte[] { 69, 79, 66 };
+    internal static readonly string MessageEobCodeStr = "EOB";
+
     private static readonly string[] EmptyKeys = Array.Empty<string>();
     private static readonly StringValues[] EmptyValues = Array.Empty<StringValues>();
 

diff --git a/tests/NATS.Client.CoreUnit.Tests/NatsHeaderTest.cs b/tests/NATS.Client.CoreUnit.Tests/NatsHeaderTest.cs
@@ -96,6 +96,7 @@ public void ParserTests()
     [InlineData("Request Timeout", NatsHeaders.Messages.RequestTimeout)]
     [InlineData("Message Size Exceeds MaxBytes", NatsHeaders.Messages.MessageSizeExceedsMaxBytes)]
     [InlineData("test message", NatsHeaders.Messages.Text)]
+    [InlineData("EOB", NatsHeaders.Messages.EobCode)]
     public void ParserMessageEnumTests(string message, NatsHeaders.Messages result)
     {
         var parser = new NatsHeaderParser(Encoding.UTF8);