Skip to content

Commit 787c45c

Browse files
committed
Remove PCS if it's dvc client
1 parent d69180f commit 787c45c

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

clients/da-vinci-client/src/main/java/com/linkedin/davinci/kafka/consumer/StoreIngestionTask.java

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2368,7 +2368,7 @@ protected void processCommonConsumerAction(ConsumerAction consumerAction) throws
23682368
* {@link #kafkaDataValidationService}, we would like to drain all the buffered messages before cleaning up those
23692369
* two variables to avoid the race condition.
23702370
*/
2371-
if (consumerAction.isHelixTriggeredAction()) {
2371+
if (isDaVinciClient() || consumerAction.isHelixTriggeredAction()) {
23722372
partitionConsumptionStateMap.remove(partition);
23732373
LOGGER.info(
23742374
"Removing tracking of replica: {} from storage utilization manager as this UNSUBSCRIBE is helix triggered action",

0 commit comments

Comments
 (0)