(flink) branch master 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 master
in repository https://gitbox.apache.org/repos/asf/flink.git


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

commit 181d522598f5071590669681b131bce1f76a70cd
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     | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

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 73de3838117..1dc57a8c4ce 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
@@ -562,7 +562,8 @@ public class UnalignedCheckpointRescaleITCase extends 
UnalignedCheckpointTestBas
         Object[][] parameters =
                 new Object[][] {
                     // Disable CUSTOM_PARTITIONER since it does not work well, 
see FLINK-39162
-                    // new Object[] {"downscale", Topology.CUSTOM_PARTITIONER, 
3, 2, 0L},
+                    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},