Merge pull request #6 from muzzammilshahid/fix-message-on-cbor

Handle serializer on message receive
xconnio · May 3, 2024 · 1563845 · 1563845
2 parents 3915d82 + 3b8288e
commit 1563845
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/lib/src/session.dart b/lib/src/session.dart
@@ -5,6 +5,7 @@ import "package:wamp/src/helpers.dart";
 import "package:wamp/src/types.dart";
 import "package:wampproto/idgen.dart";
 import "package:wampproto/messages.dart" as msg;
+import "package:wampproto/serializers.dart";
 import "package:wampproto/session.dart";
 
 class Session {
@@ -13,7 +14,12 @@ class Session {
     Future.microtask(() async {
       while (true) {
         var message = await _baseSession.receive();
-        var decodedMessage = Uint8List.fromList((message as String).codeUnits);
+        Uint8List decodedMessage;
+        if (_baseSession.serializer == JSONSerializer()) {
+          decodedMessage = Uint8List.fromList((message as String).codeUnits);
+        } else {
+          decodedMessage = message as Uint8List;
+        }
         _processIncomingMessage(_wampSession.receive(decodedMessage));
       }
     });