Update storage client as well

vesoft-inc · Jan 18, 2023 · 97acc41 · 97acc41
1 parent b2ded23
commit 97acc41
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 6 deletions.
diff --git a/nebula3/gclient/net/Connection.py b/nebula3/gclient/net/Connection.py
@@ -79,7 +79,6 @@ def open_SSL(self, ip, port, timeout, ssl_config=None):
  s.setTimeout(timeout)
  transport = THeaderTransport.THeaderTransport(s)
  protocol = THeaderProtocol.THeaderProtocol(transport)
- # protocol = TCompactProtocol.TCompactProtocol(transport) # using compact protocol still cause nebula3.Exception.IOErrorException: Header transport frame was too large
 
  transport.open()
  self._connection = GraphService.Client(protocol)

diff --git a/nebula3/sclient/net/__init__.py b/nebula3/sclient/net/__init__.py
@@ -10,9 +10,8 @@
 
 from nebula3.Exception import InValidHostname
 from nebula3.storage import GraphStorageService
-from nebula3.storage.ttypes import ScanVertexRequest, ScanEdgeRequest
-from nebula3.fbthrift.transport import TSocket, TTransport
-from nebula3.fbthrift.protocol import TBinaryProtocol
+from nebula3.fbthrift.transport import TSocket, THeaderTransport
+from nebula3.fbthrift.protocol import THeaderProtocol
 
 
 class GraphStorageConnection(object):
@@ -35,8 +34,8 @@ def open(self):
  s = TSocket.TSocket(self._address.host, self._address.port)
  if self._timeout > 0:
  s.setTimeout(self._timeout)
- transport = TTransport.TBufferedTransport(s)
- protocol = TBinaryProtocol.TBinaryProtocol(transport)
+ transport = THeaderTransport.THeaderTransport(s)
+ protocol = THeaderProtocol.THeaderProtocol(transport)
  transport.open()
  self._connection = GraphStorageService.Client(protocol)
  except Exception: