Merge pull request #337 from dvonthenen/improve-text-buffer-msg

Improve WebSocketMessage Buffer Length Handling
deepgram · Sep 27, 2024 · 7125208 · 7125208
2 parents bfcffd0 + 547e9cc
commit 7125208
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/Deepgram/Clients/Listen/v1/WebSocket/WebSocketMessage.cs b/Deepgram/Clients/Listen/v1/WebSocket/WebSocketMessage.cs
@@ -13,7 +13,7 @@ public WebSocketMessage(byte[] message, WebSocketMessageType type)
 
     public WebSocketMessage(byte[] message, WebSocketMessageType type, int length)
     {
-        if (length != Constants.UseArrayLengthForSend || length < Constants.UseArrayLengthForSend)
+        if (length != Constants.UseArrayLengthForSend && length <= message.Length && length > 0)
         {
             Message = new ArraySegment<byte>(message, 0, length);
         }
@@ -22,10 +22,9 @@ public WebSocketMessage(byte[] message, WebSocketMessageType type, int length)
             Message = new ArraySegment<byte>(message, 0, message.Length);
         }
         MessageType = type;
-        Length = length;
     }
 
-    public int Length { get; }
+    public int Length => Message.Count;
 
     public ArraySegment<byte> Message { get; }
 

diff --git a/Deepgram/Clients/Speak/v1/WebSocket/WebSocketMessage.cs b/Deepgram/Clients/Speak/v1/WebSocket/WebSocketMessage.cs
@@ -13,7 +13,7 @@ public WebSocketMessage(byte[] message, WebSocketMessageType type)
 
     public WebSocketMessage(byte[] message, WebSocketMessageType type, int length)
     {
-        if (length != Constants.UseArrayLengthForSend || length < Constants.UseArrayLengthForSend)
+        if (length != Constants.UseArrayLengthForSend && length <= message.Length && length > 0)
         {
             Message = new ArraySegment<byte>(message, 0, length);
         }
@@ -22,10 +22,9 @@ public WebSocketMessage(byte[] message, WebSocketMessageType type, int length)
             Message = new ArraySegment<byte>(message, 0, message.Length);
         }
         MessageType = type;
-        Length = length;
     }
 
-    public int Length { get; }
+    public int Length => Message.Count;
 
     public ArraySegment<byte> Message { get; }