mateczagany commented on code in PR #860:
URL: 
https://github.com/apache/flink-kubernetes-operator/pull/860#discussion_r1712595662


##########
flink-kubernetes-operator/src/main/java/org/apache/flink/kubernetes/operator/observer/SnapshotObserver.java:
##########
@@ -220,63 +235,179 @@ private void 
observeTriggeredCheckpoint(FlinkResourceContext<CR> ctx, String job
 
     /** Clean up and dispose savepoints according to the configured max 
size/age. */
     @VisibleForTesting
-    void cleanupSavepointHistory(FlinkResourceContext<CR> ctx, SavepointInfo 
currentSavepointInfo) {
+    void cleanupSavepointHistory(FlinkResourceContext<CR> ctx) {
+        if (!FlinkStateSnapshotUtils.isSnapshotResourceEnabled(
+                ctx.getOperatorConfig(), ctx.getObserveConfig())) {
+            cleanupSavepointHistoryLegacy(ctx);
+            return;
+        }
+
+        var snapshots = 
ctx.getJosdkContext().getSecondaryResources(FlinkStateSnapshot.class);
+        if (CollectionUtil.isNullOrEmpty(snapshots)) {
+            return;
+        }
+        if (ctx.getObserveConfig().get(OPERATOR_SAVEPOINT_CLEANUP_ENABLED)) {
+            var savepointsToDelete =
+                    getFlinkStateSnapshotsToCleanUp(
+                            snapshots, ctx.getObserveConfig(), 
ctx.getOperatorConfig(), SAVEPOINT);
+            var checkpointsToDelete =
+                    getFlinkStateSnapshotsToCleanUp(
+                            snapshots, ctx.getObserveConfig(), 
ctx.getOperatorConfig(), CHECKPOINT);
+            Stream.concat(savepointsToDelete.stream(), 
checkpointsToDelete.stream())
+                    .forEach(
+                            snapshot ->
+                                    ctx.getKubernetesClient()
+                                            .resource(snapshot)
+                                            .withTimeoutInMillis(0L)
+                                            .delete());
+        }
+    }
+
+    @VisibleForTesting
+    Set<FlinkStateSnapshot> getFlinkStateSnapshotsToCleanUp(
+            Collection<FlinkStateSnapshot> snapshots,
+            Configuration observeConfig,
+            FlinkOperatorConfiguration operatorConfig,
+            SnapshotType snapshotType) {
+        var snapshotList =
+                snapshots.stream()
+                        .filter(s -> s.getStatus() != null)
+                        .filter(s -> 
COMPLETED.equals(s.getStatus().getState()))
+                        .filter(
+                                s ->
+                                        SnapshotTriggerType.PERIODIC.equals(
+                                                
FlinkStateSnapshotUtils.getSnapshotTriggerType(s)))

Review Comment:
   You are right. I will add a check to make sure that during cleanup the 
latest COMPLETED snapshot won't be deleted. Then it should be safe to add 
upgrade savepoints as well.



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: issues-unsubscr...@flink.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to