Record RM callback nodes report when nodes updated (#615)

tony-framework · Oct 29, 2021 · 1994467 · 1994467
1 parent 9a4f7a4
commit 1994467
Showing 1 changed file with 10 additions and 1 deletion.
diff --git a/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java b/tony-core/src/main/java/com/linkedin/tony/ApplicationMaster.java
@@ -1111,7 +1111,16 @@ public void onShutdownRequest() {
 
     @Override
     public void onNodesUpdated(List<NodeReport> list) {
-      LOG.info("onNodesUpdated called in RMCallbackHandler");
+      StringBuilder consoleMsgBuilder =
+              new StringBuilder("Received onNodesUpdated called in RMCallbackHandler, node reports as follows.");
+      if (list == null || list.size() == 0) {
+        LOG.info(consoleMsgBuilder.toString());
+        return;
+      }
+      for (NodeReport nodeReport : list) {
+        consoleMsgBuilder.append("\nnodeId = " + nodeReport.getNodeId() + ", healthReport = " + nodeReport.getHealthReport());
+      }
+      LOG.info(consoleMsgBuilder.toString());
     }
 
     @Override