def get_all_node_info(
self,
timeout: Optional[float] = None
) -> gcs_service_pb2.GetAllNodeInfoReply:
req = gcs_service_pb2.GetAllNodeInfoRequest()
reply = self._node_info_stub.GetAllNodeInfo(req, timeout=timeout)
return reply
async def _get_nodes(self):
"""Read the client table.
Returns:
A dict of information about the nodes in the cluster.
"""
request = gcs_service_pb2.GetAllNodeInfoRequest()
reply = await self._gcs_node_info_stub.GetAllNodeInfo(request, timeout=2)
if reply.status.code == 0:
result = {}
for node_info in reply.node_info_list:
node_info_dict = gcs_node_info_to_dict(node_info)
result[node_info_dict["nodeId"]] = node_info_dict
return result
else:
logger.error("Failed to GetAllNodeInfo: %s", reply.status.message)
async def _get_nodes(self):
"""Read the client table.
Returns:
A list of information about the nodes in the cluster.
"""
request = gcs_service_pb2.GetAllNodeInfoRequest()
reply = await self._gcs_node_info_stub.GetAllNodeInfo(request,
timeout=2)
if reply.status.code == 0:
results = []
node_id_set = set()
for node_info in reply.node_info_list:
if node_info.node_id in node_id_set:
continue
node_id_set.add(node_info.node_id)
node_info_dict = gcs_node_info_to_dict(node_info)
results.append(node_info_dict)
return results
else:
logger.error("Failed to GetAllNodeInfo: %s", reply.status.message)
