Skip to content

Commit 87367d7

Browse files
committed
Prepare the python client for changes in RemoteOps protocol V3
(cherry picked from commit c10149e)
1 parent a46cf2e commit 87367d7

File tree

1 file changed

+6
-5
lines changed

1 file changed

+6
-5
lines changed

graphdatascience/query_runner/protocol/project_protocols.py

Lines changed: 6 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -132,12 +132,13 @@ def is_not_done(result: DataFrame) -> bool:
132132

133133
logger = getLogger()
134134

135-
member_address = query_runner.call_procedure(
135+
# We need to pin the driver to a specific cluster member
136+
response = query_runner.call_procedure(
136137
ProtocolVersion.V3.versioned_procedure_name(endpoint), params, yields, database, logging, False
137-
).squeeze()["host"]
138-
139-
# TODO: retrieve the port from the server
140-
projection_query_runner = query_runner.clone(member_address, 7687)
138+
).squeeze()
139+
member_host = response["host"]
140+
member_port = response["port"] if ("port" in response.index) else 7687
141+
projection_query_runner = query_runner.clone(member_host, member_port)
141142

142143
@retry(
143144
reraise=True,

0 commit comments

Comments
 (0)