def _connect(self):
self._channel.connect()
self._kv_stub = gcs_service_pb2_grpc.InternalKVGcsServiceStub(
self._channel.channel())
self._runtime_env_stub = gcs_service_pb2_grpc.RuntimeEnvGcsServiceStub(
self._channel.channel())
self._node_info_stub = gcs_service_pb2_grpc.NodeInfoGcsServiceStub(
self._channel.channel())
self._job_info_stub = gcs_service_pb2_grpc.JobInfoGcsServiceStub(
self._channel.channel())
def __init__(self, address):
from ray._private.utils import init_grpc_channel
logger.debug(f"Connecting to gcs address: {address}")
options = [
("grpc.enable_http_proxy", 0),
("grpc.max_send_message_length", GcsClient.MAX_MESSAGE_LENGTH),
("grpc.max_receive_message_length", GcsClient.MAX_MESSAGE_LENGTH)
]
channel = init_grpc_channel(address, options=options)
self._kv_stub = gcs_service_pb2_grpc.InternalKVGcsServiceStub(channel)
def _connect(self):
self._channel.connect()
self._kv_stub = gcs_service_pb2_grpc.InternalKVGcsServiceStub(
self._channel.channel())
def _connect(self):
self._channel.connect()
self._kv_stub = gcs_service_pb2_grpc.InternalKVGcsServiceStub(
self._channel.channel())
self._heartbeat_info_stub = gcs_service_pb2_grpc.HeartbeatInfoGcsServiceStub(
self._channel.channel())