test fix

linkedin · Jan 30, 2024 · 2a054fd · 2a054fd
1 parent 4018e69
commit 2a054fd
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/.../main/java/com/linkedin/davinci/stats/ingestion/heartbeat/HeartbeatMonitoringService.java b/.../main/java/com/linkedin/davinci/stats/ingestion/heartbeat/HeartbeatMonitoringService.java
@@ -132,7 +132,9 @@ private void recordHeartbeat(
     if (heartbeatTimestamps.get(store) != null) {
       if (heartbeatTimestamps.get(store).get(version) != null) {
         if (heartbeatTimestamps.get(store).get(version).get(partition) != null) {
-          heartbeatTimestamps.get(store).get(version).get(partition).put(region, timestamp);
+          if (region != null) {
+            heartbeatTimestamps.get(store).get(version).get(partition).put(region, timestamp);
+          }
         }
       }
     }