From f080911d9bf397e7fd3c492652303cc141fa32e1 Mon Sep 17 00:00:00 2001
From: Alex Wu <alex@anyscale.io>
Date: Fri, 21 May 2021 09:26:37 -0700
Subject: [PATCH] [dashboard] include worker id in actor snapshot (#15967)

Co-authored-by: Alex Wu <alex@anyscale.com>
---
 dashboard/modules/snapshot/snapshot_head.py     | 2 ++
 dashboard/modules/snapshot/snapshot_schema.json | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/dashboard/modules/snapshot/snapshot_head.py b/dashboard/modules/snapshot/snapshot_head.py
index d1fad44ac..95a4b8d00 100644
--- a/dashboard/modules/snapshot/snapshot_head.py
+++ b/dashboard/modules/snapshot/snapshot_head.py
@@ -73,6 +73,8 @@ class LogicalViewHead(dashboard_utils.DashboardHeadModule):
                 "resources": dict(
                     actor_table_entry.task_spec.required_resources),
                 "actor_class": actor_table_entry.class_name,
+                "current_worker_id": actor_table_entry.address.worker_id.hex(),
+                "current_raylet_id": actor_table_entry.address.raylet_id.hex(),
                 "ip_address": actor_table_entry.address.ip_address,
                 "port": actor_table_entry.address.port,
             }
diff --git a/dashboard/modules/snapshot/snapshot_schema.json b/dashboard/modules/snapshot/snapshot_schema.json
index b37f9f34d..5c8068a0c 100644
--- a/dashboard/modules/snapshot/snapshot_schema.json
+++ b/dashboard/modules/snapshot/snapshot_schema.json
@@ -102,6 +102,12 @@
                   "actorClass": {
                     "type": "string"
                   },
+                  "currentWorkerId": {
+                    "type": "string"
+                  },
+                  "currentRayletId": {
+                    "type": "string"
+                  },
                   "ipAddress": {
                     "type": "string"
                   },
@@ -118,6 +124,8 @@
                     "endTime",
                     "isDetached",
                     "resources",
+                    "currentWorkerId",
+                    "currentRayletId",
                     "actorClass",
                     "ipAddress",
                     "port"