(flink) branch release-2.2 updated: [FLINK-39162][checkpoints] Disable UC for CustomPartitioner

[roman]

This is an automated email from the ASF dual-hosted git repository.

roman pushed a commit to branch release-2.2
in repository https://gitbox.apache.org/repos/asf/flink.git


The following commit(s) were added to refs/heads/release-2.2 by this push:
     new b5bdb239505 [FLINK-39162][checkpoints] Disable UC for CustomPartitioner
b5bdb239505 is described below

commit b5bdb239505398a811099a6c551bcaf2437554f8
Author: Roman Khachatryan <khachatryan.ro...@gmail.com>
AuthorDate: Tue Mar 3 23:14:21 2026 +0100

    [FLINK-39162][checkpoints] Disable UC for CustomPartitioner
---
 .../flink/streaming/runtime/partitioner/CustomPartitionerWrapper.java    | 1 +
 .../flink/test/checkpointing/UnalignedCheckpointRescaleITCase.java       | 1 +
 2 files changed, 2 insertions(+)

diff --git 
a/flink-runtime/src/main/java/org/apache/flink/streaming/runtime/partitioner/CustomPartitionerWrapper.java
 
b/flink-runtime/src/main/java/org/apache/flink/streaming/runtime/partitioner/CustomPartitionerWrapper.java
index e3c5cb442a2..49ec4c5b6ea 100644
--- 
a/flink-runtime/src/main/java/org/apache/flink/streaming/runtime/partitioner/CustomPartitionerWrapper.java
+++ 
b/flink-runtime/src/main/java/org/apache/flink/streaming/runtime/partitioner/CustomPartitionerWrapper.java
@@ -44,6 +44,7 @@ public class CustomPartitionerWrapper<K, T> extends 
StreamPartitioner<T> {
     public CustomPartitionerWrapper(Partitioner<K> partitioner, KeySelector<T, 
K> keySelector) {
         this.partitioner = partitioner;
         this.keySelector = keySelector;
+        disableUnalignedCheckpoints();
     }
 
     @Override
diff --git 
a/flink-tests/src/test/java/org/apache/flink/test/checkpointing/UnalignedCheckpointRescaleITCase.java
 
b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/UnalignedCheckpointRescaleITCase.java
index 4a97e17cf7b..aabef0a049c 100644
--- 
a/flink-tests/src/test/java/org/apache/flink/test/checkpointing/UnalignedCheckpointRescaleITCase.java
+++ 
b/flink-tests/src/test/java/org/apache/flink/test/checkpointing/UnalignedCheckpointRescaleITCase.java
@@ -544,6 +544,7 @@ public class UnalignedCheckpointRescaleITCase extends 
UnalignedCheckpointTestBas
         Object[][] parameters =
                 new Object[][] {
                     new Object[] {"downscale", Topology.CUSTOM_PARTITIONER, 3, 
2, 0L},
+                    new Object[] {"upscale", Topology.CUSTOM_PARTITIONER, 2, 
3, 0L},
                     new Object[] {"downscale", 
Topology.KEYED_DIFFERENT_PARALLELISM, 12, 7, 0L},
                     new Object[] {"upscale", 
Topology.KEYED_DIFFERENT_PARALLELISM, 7, 12, 0L},
                     new Object[] {"downscale", 
Topology.KEYED_DIFFERENT_PARALLELISM, 5, 3, 5L},