[dashboard] Remove unused fields in dashboard actor table for better memory footprint (#21919)

dashboard/client/src/type/actor.ts

@@ -67,28 +67,12 @@ export type TaskSpec = {
 
 export type Actor = {
   actorId: string;
-  children: { [key: string]: Actor };
-  taskSpec: TaskSpec;
-  ipAddress: string;
-  isDirectCall: boolean;
   jobId: string;
-  numExecutedTasks: number;
-  numLocalObjects: number;
-  numObjectIdsInScope: number;
   state: ActorEnum | string; // PENDING, ALIVE, RECONSTRUCTING, DEAD
-  taskQueueLength: number;
-  usedObjectStoreMemory: number;
-  usedResources: { [key: string]: string | number };
-  timestamp: number;
-  actorTitle: string;
-  averageTaskExecutionSpeed: number;
   nodeId: string;
   pid: number;
-  ownerAddress: Address;
   address: Address;
-  maxReconstructions: string;
-  remainingReconstructions: string;
-  isDetached: false;
   name: string;
   numRestarts: string;
+  taskSpec: TaskSpec;
 };

dashboard/modules/actor/actor_head.py

@@ -29,13 +29,41 @@ routes = dashboard_optional_utils.ClassMethodRouteTable
 
 
 def actor_table_data_to_dict(message):
-    return dashboard_utils.message_to_dict(
+    orig_message = dashboard_utils.message_to_dict(
         message, {
             "actorId", "parentId", "jobId", "workerId", "rayletId",
             "actorCreationDummyObjectId", "callerId", "taskId", "parentTaskId",
             "sourceActorId", "placementGroupId"
         },
         including_default_value_fields=True)
+    # The complete schema for actor table is here:
+    #     src/ray/protobuf/gcs.proto
+    # It is super big and for dashboard, we don't need that much information.
+    # Only preserve the necessary ones here for memory usage.
+    fields = {
+        "actorId",
+        "jobId",
+        "pid",
+        "address",
+        "state",
+        "name",
+        "numRestarts",
+        "taskSpec",
+        "timestamp",
+        "numExecutedTasks",
+    }
+    light_message = {k: v for (k, v) in orig_message.items() if k in fields}
+    if "taskSpec" in light_message:
+        actor_class = actor_classname_from_task_spec(light_message["taskSpec"])
+        light_message["actorClass"] = actor_class
+        if "functionDescriptor" in light_message["taskSpec"]:
+            light_message["taskSpec"] = {
+                "functionDescriptor": light_message["taskSpec"][
+                    "functionDescriptor"]
+            }
+        else:
+            light_message.pop("taskSpec")
+    return light_message
 
 
 class ActorHead(dashboard_utils.DashboardHeadModule):
@@ -62,13 +90,6 @@ class ActorHead(dashboard_utils.DashboardHeadModule):
             self._stubs[node_id] = stub
 
     async def _update_actors(self):
-        # TODO(fyrestone): Refactor code for updating actor / node / job.
-
-        def _process_actor_table_data(data):
-            actor_class = actor_classname_from_task_spec(
-                data.get("taskSpec", {}))
-            data["actorClass"] = actor_class
-
         # Get all actor info.
         while True:
             try:
@@ -80,7 +101,6 @@ class ActorHead(dashboard_utils.DashboardHeadModule):
                     actors = {}
                     for message in reply.actor_table_data:
                         actor_table_data = actor_table_data_to_dict(message)
-                        _process_actor_table_data(actor_table_data)
                         actors[actor_table_data["actorId"]] = actor_table_data
                     # Update actors.
                     DataSource.actors.reset(actors)
@@ -113,7 +133,6 @@ class ActorHead(dashboard_utils.DashboardHeadModule):
 
         def process_actor_data_from_pubsub(actor_id, actor_table_data):
             actor_table_data = actor_table_data_to_dict(actor_table_data)
-            _process_actor_table_data(actor_table_data)
             # If actor is not new registered but updated, we only update
             # states related fields.
             if actor_table_data["state"] != "DEPENDENCIES_UNREADY":