Ejemplo n.º 1
0
 def __init__(self,
              redis_address,
              dashboard_agent_port,
              redis_password=None,
              temp_dir=None,
              log_dir=None,
              metrics_export_port=None,
              node_manager_port=None,
              object_store_name=None,
              raylet_name=None):
     """Initialize the DashboardAgent object."""
     # Public attributes are accessible for all agent modules.
     self.redis_address = dashboard_utils.address_tuple(redis_address)
     self.redis_password = redis_password
     self.temp_dir = temp_dir
     self.log_dir = log_dir
     self.dashboard_agent_port = dashboard_agent_port
     self.metrics_export_port = metrics_export_port
     self.node_manager_port = node_manager_port
     self.object_store_name = object_store_name
     self.raylet_name = raylet_name
     self.node_id = os.environ["RAY_NODE_ID"]
     assert self.node_id, "Empty node id (RAY_NODE_ID)."
     self.ip = ray._private.services.get_node_ip_address()
     self.server = aiogrpc.server(options=(("grpc.so_reuseport", 0), ))
     self.grpc_port = self.server.add_insecure_port(
         f"[::]:{self.dashboard_agent_port}")
     logger.info("Dashboard agent grpc address: %s:%s", self.ip,
                 self.grpc_port)
     self.aioredis_client = None
     self.aiogrpc_raylet_channel = aiogrpc.insecure_channel("{}:{}".format(
         self.ip, self.node_manager_port))
     self.http_session = None
Ejemplo n.º 2
0
 def __init__(self,
              node_ip_address,
              redis_address,
              dashboard_agent_port,
              redis_password=None,
              temp_dir=None,
              session_dir=None,
              runtime_env_dir=None,
              log_dir=None,
              metrics_export_port=None,
              node_manager_port=None,
              listen_port=0,
              object_store_name=None,
              raylet_name=None,
              logging_params=None):
     """Initialize the DashboardAgent object."""
     # Public attributes are accessible for all agent modules.
     self.ip = node_ip_address
     self.redis_address = dashboard_utils.address_tuple(redis_address)
     self.redis_password = redis_password
     self.temp_dir = temp_dir
     self.session_dir = session_dir
     self.runtime_env_dir = runtime_env_dir
     self.log_dir = log_dir
     self.dashboard_agent_port = dashboard_agent_port
     self.metrics_export_port = metrics_export_port
     self.node_manager_port = node_manager_port
     self.listen_port = listen_port
     self.object_store_name = object_store_name
     self.raylet_name = raylet_name
     self.logging_params = logging_params
     self.node_id = os.environ["RAY_NODE_ID"]
     # TODO(edoakes): RAY_RAYLET_PID isn't properly set on Windows. This is
     # only used for fate-sharing with the raylet and we need a different
     # fate-sharing mechanism for Windows anyways.
     if sys.platform not in ["win32", "cygwin"]:
         self.ppid = int(os.environ["RAY_RAYLET_PID"])
         assert self.ppid > 0
         logger.info("Parent pid is %s", self.ppid)
     self.server = aiogrpc.server(options=(("grpc.so_reuseport", 0), ))
     self.grpc_port = self.server.add_insecure_port(
         f"[::]:{self.dashboard_agent_port}")
     logger.info("Dashboard agent grpc address: %s:%s", self.ip,
                 self.grpc_port)
     self.aioredis_client = None
     options = (("grpc.enable_http_proxy", 0), )
     self.aiogrpc_raylet_channel = aiogrpc.insecure_channel(
         f"{self.ip}:{self.node_manager_port}", options=options)
     self.http_session = None
     ip, port = redis_address.split(":")
     self.gcs_client = connect_to_gcs(ip, int(port), redis_password)
Ejemplo n.º 3
0
 def __init__(self, http_host, http_port, http_port_retries, redis_address,
              redis_password, log_dir):
     # Public attributes are accessible for all head modules.
     # Walkaround for issue: https://github.com/ray-project/ray/issues/7084
     self.http_host = "127.0.0.1" if http_host == "localhost" else http_host
     self.http_port = http_port
     self.http_port_retries = http_port_retries
     self.redis_address = dashboard_utils.address_tuple(redis_address)
     self.redis_password = redis_password
     self.log_dir = log_dir
     self.aioredis_client = None
     self.aiogrpc_gcs_channel = None
     self.http_session = None
     self.ip = ray.util.get_node_ip_address()
     self.server = aiogrpc.server(options=(("grpc.so_reuseport", 0), ))
     self.grpc_port = self.server.add_insecure_port("[::]:0")
     logger.info("Dashboard head grpc address: %s:%s", self.ip,
                 self.grpc_port)
Ejemplo n.º 4
0
Archivo: head.py Proyecto: zhengpw/ray
 def __init__(self, http_host, http_port, redis_address, redis_password):
     # NodeInfoGcsService
     self._gcs_node_info_stub = None
     self._gcs_rpc_error_counter = 0
     # Public attributes are accessible for all head modules.
     self.http_host = http_host
     self.http_port = http_port
     self.redis_address = dashboard_utils.address_tuple(redis_address)
     self.redis_password = redis_password
     self.aioredis_client = None
     self.aiogrpc_gcs_channel = None
     self.http_session = None
     self.ip = ray.services.get_node_ip_address()
     self.server = aiogrpc.server(options=(("grpc.so_reuseport", 0), ))
     self.grpc_port = self.server.add_insecure_port("[::]:0")
     logger.info("Dashboard head grpc address: %s:%s", self.ip,
                 self.grpc_port)
     logger.info("Dashboard head http address: %s:%s", self.http_host,
                 self.http_port)
Ejemplo n.º 5
0
 def __init__(self, http_host, http_port, http_port_retries, redis_address,
              redis_password, log_dir):
     # HeartbeatInfoGcsService
     self._gcs_heartbeat_info_stub = None
     self._gcs_check_alive_seq = 0
     self._gcs_rpc_error_counter = 0
     # Public attributes are accessible for all head modules.
     # Walkaround for issue: https://github.com/ray-project/ray/issues/7084
     self.http_host = "127.0.0.1" if http_host == "localhost" else http_host
     self.http_port = http_port
     self.http_port_retries = http_port_retries
     self.redis_address = dashboard_utils.address_tuple(redis_address)
     self.redis_password = redis_password
     self.log_dir = log_dir
     self.aioredis_client = None
     self.aiogrpc_gcs_channel = None
     self.http_session = None
     self.ip = ray.util.get_node_ip_address()
     ip, port = redis_address.split(":")
     self.gcs_client = connect_to_gcs(ip, int(port), redis_password)
     self.server = aiogrpc.server(options=(("grpc.so_reuseport", 0), ))
     self.grpc_port = self.server.add_insecure_port("[::]:0")
     logger.info("Dashboard head grpc address: %s:%s", self.ip,
                 self.grpc_port)