From a7173b5dee71468c6eef784d90509b385f8e730b Mon Sep 17 00:00:00 2001 From: Rajat Jaiswal Date: Fri, 16 Jul 2021 10:35:28 +0530 Subject: [PATCH] fix: timeout issues in gremlinpython==3.4.9 (#17) * Fix timeout issues in tornado Signed-off-by: Rajat Jaiswal * Bump the version Signed-off-by: Rajat Jaiswal Signed-off-by: Amom --- amundsen_gremlin/neptune_bulk_loader/api.py | 3 ++- setup.py | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/amundsen_gremlin/neptune_bulk_loader/api.py b/amundsen_gremlin/neptune_bulk_loader/api.py index 9c21bdd..21e04e4 100644 --- a/amundsen_gremlin/neptune_bulk_loader/api.py +++ b/amundsen_gremlin/neptune_bulk_loader/api.py @@ -22,6 +22,7 @@ from gremlin_python.driver.driver_remote_connection import ( DriverRemoteConnection ) +from gremlin_python.driver.tornado.transport import TornadoTransport from gremlin_python.process.anonymous_traversal import traversal from gremlin_python.process.graph_traversal import GraphTraversalSource from neptune_python_utils.endpoints import Endpoints, RequestParameters @@ -87,7 +88,7 @@ def create_graph_traversal_source(**kwargs: Any) -> GraphTraversalSource: prepared_request = override_prepared_request_parameters( endpoints.gremlin_endpoint().prepare_request(), override_uri=override_uri) kwargs['traversal_source'] = 'g' - remote_connection = DriverRemoteConnection(url=prepared_request, **kwargs) + remote_connection = DriverRemoteConnection(url=prepared_request, transport_factory=lambda: TornadoTransport(read_timeout=None, write_timeout=None), **kwargs) return traversal().withRemote(remote_connection) return create_graph_traversal_source diff --git a/setup.py b/setup.py index 37b02c5..8a32a98 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ setup( name='amundsen-gremlin', - version='0.0.9', + version='0.0.10', description='Gremlin code library for Amundsen', url='https://github.com/amundsen-io/amundsengremlin', maintainer='Amundsen TSC',