[server] tweak heartbeat reporting to use completion check (#1207)

linkedin · Sep 30, 2024 · 6057b1f · 6057b1f
1 parent b932806
commit 6057b1f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...t/src/main/java/com/linkedin/davinci/kafka/consumer/LeaderFollowerStoreIngestionTask.java b/...t/src/main/java/com/linkedin/davinci/kafka/consumer/LeaderFollowerStoreIngestionTask.java
@@ -2146,15 +2146,15 @@ protected void recordHeartbeatReceived(
           partitionConsumptionState.getPartition(),
           serverConfig.getKafkaClusterUrlToAliasMap().get(kafkaUrl),
           consumerRecord.getValue().producerMetadata.messageTimestamp,
-          partitionConsumptionState.isWaitingForReplicationLag());
+          partitionConsumptionState.isComplete());
     } else {
       heartbeatMonitoringService.recordFollowerHeartbeat(
           storeName,
           versionNumber,
           partitionConsumptionState.getPartition(),
           serverConfig.getKafkaClusterUrlToAliasMap().get(kafkaUrl),
           consumerRecord.getValue().producerMetadata.messageTimestamp,
-          partitionConsumptionState.isWaitingForReplicationLag());
+          partitionConsumptionState.isComplete());
     }
   }