def _connect_session_impl(self,
cleanup_instance=True,
dangling_timeout_seconds=60):
"""
Args:
cleanup_instance (bool, optional): If True, also delete graphscope
instance (such as pod) in closing process.
dangling_timeout_seconds (int, optional): After seconds of client
disconnect, coordinator will kill this graphscope instance.
Disable dangling check by setting -1.
"""
request = message_pb2.ConnectSessionRequest(
cleanup_instance=cleanup_instance,
dangling_timeout_seconds=dangling_timeout_seconds,
)
response = self._stub.ConnectSession(request)
response = check_grpc_response(response)
self._session_id = response.session_id
return (
response.session_id,
response.session_type,
json.loads(response.engine_config),
response.pod_name_list,
response.num_workers,
response.namespace,
)
def _fetch_logs_impl(self):
request = message_pb2.FetchLogsRequest(session_id=self._session_id)
responses = self._stub.FetchLogs(request)
for resp in responses:
resp = check_grpc_response(resp)
message = resp.message.rstrip()
if message:
logger.info(message, extra={"simple": True})
def _fetch_logs_impl(self):
request = message_pb2.FetchLogsRequest(session_id=self._session_id)
responses = self._stub.FetchLogs(request)
for resp in responses:
resp = check_grpc_response(resp)
message = resp.message.rstrip()
if message:
print(message, end="\n", file=sys.stdout, flush=True)
def create_learning_engine(self, object_id, handle, config):
request = message_pb2.CreateLearningInstanceRequest(
object_id=object_id,
handle=handle,
config=config,
)
response = self._stub.CreateLearningInstance(request)
response = check_grpc_response(response)
return response.endpoints
def create_interactive_engine(self, object_id, schema_path,
gremlin_server_cpu, gremlin_server_mem):
request = message_pb2.CreateInteractiveRequest(
object_id=object_id,
schema_path=schema_path,
gremlin_server_cpu=gremlin_server_cpu,
gremlin_server_mem=gremlin_server_mem,
)
response = self._stub.CreateInteractiveInstance(request)
return check_grpc_response(response)
def _connect_session_impl(self):
request = message_pb2.ConnectSessionRequest()
response = self._stub.ConnectSession(request)
response = check_grpc_response(response)
self._session_id = response.session_id
return (
response.session_id,
json.loads(response.engine_config),
response.pod_name_list,
)
def _run_step_impl(self, dag_def):
request = message_pb2.RunStepRequest(session_id=self._session_id,
dag_def=dag_def)
response = self._stub.RunStep(request)
return check_grpc_response(response)
def _close_session_impl(self):
request = message_pb2.CloseSessionRequest(session_id=self._session_id)
response = self._stub.CloseSession(request)
return check_grpc_response(response)
def close_learning_engine(self, object_id):
request = message_pb2.CloseLearningInstanceRequest(object_id=object_id)
response = self._stub.CloseLearningInstance(request)
return check_grpc_response(response)
def close_interactive_engine(self, object_id):
request = message_pb2.CloseInteractiveRequest(object_id=object_id)
response = self._stub.CloseInteractiveInstance(request)
return check_grpc_response(response)
def _close_session_impl(self, stop_instance=True):
request = message_pb2.CloseSessionRequest(session_id=self._session_id,
stop_instance=stop_instance)
response = self._stub.CloseSession(request)
return check_grpc_response(response)
def create_interactive_engine(self, object_id, schema_path):
request = message_pb2.CreateInteractiveRequest(object_id=object_id,
schema_path=schema_path)
response = self._stub.CreateInteractiveInstance(request)
return check_grpc_response(response)
