From fca384aff4f8d8924afc3597c354c56e8f9260d2 Mon Sep 17 00:00:00 2001
From: Guy Rozendorn <guy@rzn.co.il>
Date: Tue, 11 Aug 2020 23:43:36 +0300
Subject: [PATCH] Adpating to changes in gremlinpython-3.4.4

---
 aiogremlin/driver/protocol.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/aiogremlin/driver/protocol.py b/aiogremlin/driver/protocol.py
index 5a53ef5..8f3cd51 100644
--- a/aiogremlin/driver/protocol.py
+++ b/aiogremlin/driver/protocol.py
@@ -42,8 +42,7 @@ class GremlinServerWSProtocol(protocol.AbstractBaseProtocol):
             await func
 
     async def data_received(self, data, results_dict):
-        data = data.decode('utf-8')
-        message = self._message_serializer.deserialize_message(json.loads(data))
+        message = self._message_serializer.deserialize_message(data)
         request_id = message['requestId']
         status_code = message['status']['code']
         data = message['result']['data']
@@ -65,7 +64,6 @@ class GremlinServerWSProtocol(protocol.AbstractBaseProtocol):
             else:
                 if data:
                     for result in data:
-                        result = self._message_serializer.deserialize_message(result)
                         message = Message(status_code, result, msg)
                         result_set.queue_result(message)
                 else:
-- 
GitLab