Github user sihuazhou commented on a diff in the pull request:
https://github.com/apache/flink/pull/5578#discussion_r171133066
--- Diff:
flink-runtime/src/main/java/org/apache/flink/runtime/state/TaskLocalStateStoreImpl.java
---
@@ -300,6 +291,32 @@ private void deleteDirectory(File directory) throws
IOException {
}
}
+ /**
+ * Pruning the useless checkpoints.
+ */
+ private void pruneCheckpoints(long checkpointID, boolean
breakTheIteration) {
+
+ Iterator<Map.Entry<Long, TaskStateSnapshot>> entryIterator =
+ storedTaskStateByCheckpointID.entrySet().iterator();
+
+ final List<Map.Entry<Long, TaskStateSnapshot>> toRemove = new
ArrayList<>();
+
+ while (entryIterator.hasNext()) {
+
+ Map.Entry<Long, TaskStateSnapshot> snapshotEntry =
entryIterator.next();
+ long entryCheckpointId = snapshotEntry.getKey();
+
+ if (entryCheckpointId != checkpointID) {
--- End diff --
I agree with you that the breaking case looks a bit dangerous ... I think
maybe we could pass a `Predicate` for the `if` and let the caller side pass the
`Predicate` into this function. This could make it cleaner from the caller side
and don't need to mass the logic into the `if` to make it complex.
---
