fix: include original code and message in stream errors for requestRe…

…sponse + requestStream
viglucci · Apr 17, 2022 · afd3435 · afd3435
1 parent 925b198
commit afd3435
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/UnityRSocket/Assets/Viglucci/UnityRSocket/Runtime/RSocketRequester.cs b/UnityRSocket/Assets/Viglucci/UnityRSocket/Runtime/RSocketRequester.cs
@@ -140,8 +140,10 @@ private void HandleErrorFrame(RSocketFrame.ErrorFrame abstractFrame)
         {
             _done = true;
 
-            // TODO: get actual error code and message from frame
-            _receiver.OnError(new RSocketError(RSocketErrorCodes.REJECTED, "An unexpected error occurred"));
+            RSocketError rSocketError = new RSocketError(
+                (RSocketErrorCodes) abstractFrame.Code,
+                abstractFrame.Message ?? "An unexpected error occurred");
+            _receiver.OnError(rSocketError);
         }
 
         private void HandlePayloadFrame(RSocketFrame.AbstractRequestFrame abstractFrame)
@@ -308,8 +310,10 @@ private void HandleErrorFrame(RSocketFrame.ErrorFrame abstractFrame)
         {
             _done = true;
 
-            // TODO: get actual error code and message from frame
-            _receiver.OnError(new RSocketError(abstractFrame.Code, abstractFrame.Message));
+            RSocketError rSocketError = new RSocketError(
+                (RSocketErrorCodes) abstractFrame.Code,
+                abstractFrame.Message ?? "An unexpected error occurred");
+            _receiver.OnError(rSocketError);
         }
 
         private void HandlePayloadFrame(RSocketFrame.AbstractRequestFrame abstractFrame)