[Stats] Improve Stats::Init & Add it to GCS server (#9563)

python/ray/node.py

@@ -635,7 +635,8 @@ class Node:
             redis_password=self._ray_params.redis_password,
             config=self._config,
             fate_share=self.kernel_fate_share,
-            gcs_server_port=self._ray_params.gcs_server_port)
+            gcs_server_port=self._ray_params.gcs_server_port,
+            metrics_agent_port=self._ray_params.metrics_agent_port)
         assert (
             ray_constants.PROCESS_TYPE_GCS_SERVER not in self.all_processes)
         self.all_processes[ray_constants.PROCESS_TYPE_GCS_SERVER] = [

python/ray/services.py

@@ -1203,7 +1203,8 @@ def start_gcs_server(redis_address,
                      redis_password=None,
                      config=None,
                      fate_share=None,
-                     gcs_server_port=None):
+                     gcs_server_port=None,
+                     metrics_agent_port=None):
     """Start a gcs server.
     Args:
         redis_address (str): The address that the Redis server is listening on.
@@ -1215,6 +1216,7 @@ def start_gcs_server(redis_address,
         config (dict|None): Optional configuration that will
             override defaults in RayConfig.
         gcs_server_port (int): Port number of the gcs server.
+        metrics_agent_port(int): The port where metrics agent is bound to.
     Returns:
         ProcessInfo for the process that was started.
     """
@@ -1230,6 +1232,7 @@ def start_gcs_server(redis_address,
         "--redis_port={}".format(gcs_port),
         "--config_list={}".format(config_str),
         "--gcs_server_port={}".format(gcs_server_port),
+        "--metrics-agent-port={}".format(metrics_agent_port),
     ]
     if redis_password:
         command += ["--redis_password={}".format(redis_password)]