[FLINK-36714][state] Ignore interrupted in sstmerge.CompactionSchedul…

…er#stop
Shopify · Nov 20, 2024 · b48ccce · b48ccce
1 parent 27569da
commit b48ccce
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/...nd-rocksdb/src/main/java/org/apache/flink/state/rocksdb/sstmerge/CompactionScheduler.java b/...nd-rocksdb/src/main/java/org/apache/flink/state/rocksdb/sstmerge/CompactionScheduler.java
@@ -28,6 +28,7 @@
 import java.util.concurrent.TimeUnit;
 
 import static java.util.Collections.emptyList;
+import static org.apache.flink.shaded.guava32.com.google.common.util.concurrent.Uninterruptibles.awaitTerminationUninterruptibly;
 
 /**
  * Schedules manual compactions of small disjoint SST files created by RocksDB. It does so
@@ -88,7 +89,7 @@ public void stop() throws InterruptedException {
                 scheduledExecutor.shutdownNow();
             }
         }
-        if (!scheduledExecutor.awaitTermination(5, TimeUnit.SECONDS)) {
+        if (!awaitTerminationUninterruptibly(scheduledExecutor, 5, TimeUnit.SECONDS)) {
             LOG.warn("Unable to terminate scheduled tasks in 5s");
         }
     }