Partition Removal

linkedin · Dec 4, 2024 · e7bf0c7 · e7bf0c7
1 parent 321fb40
commit e7bf0c7
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/clients/da-vinci-client/src/main/java/com/linkedin/davinci/DaVinciBackend.java b/clients/da-vinci-client/src/main/java/com/linkedin/davinci/DaVinciBackend.java
@@ -476,10 +476,7 @@ private synchronized void bootstrap() {
       aggVersionedStorageEngineStats.setStorageEngine(versionTopic, storageEngine);
       StoreBackend storeBackend = getStoreOrThrow(storeName);
       ComplementSet<Integer> persistedPartitionIds = ComplementSet.newSet(storageEngine.getPersistedPartitionIds());
-      ComplementSet<Integer> unassignedPartitionSet = ComplementSet.newSet(storeBackend.getSubscription());
-      // Set of partitions that are not included in the submitted set of partitions.
-      unassignedPartitionSet.removeAll(persistedPartitionIds);
-      storeBackend.unsubscribe(unassignedPartitionSet);
+      storeBackend.unsubscribe(persistedPartitionIds);
       storeBackend.subscribe(ComplementSet.newSet(partitions), Optional.of(version));
     });
   }