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)