Never release store using CancellableThreads

elastic · Aug 9, 2019 · 30339ac · 30339ac
1 parent 67a81ab
commit 30339ac
Show file tree

Hide file tree

Showing 5 changed files with 148 additions and 136 deletions.
diff --git a/server/src/main/java/org/elasticsearch/common/util/CancellableThreads.java b/server/src/main/java/org/elasticsearch/common/util/CancellableThreads.java
@@ -206,4 +206,12 @@ public interface OnCancel {
          */
         void onCancel(String reason, @Nullable Exception beforeCancelException);
     }
+
+    /**
+     * Returns {@code true} if the current thread is being executed using {@link CancellableThreads} or was interrupted.
+     */
+    public synchronized boolean isCurrentThreadCancellable() {
+        final Thread currentThread = Thread.currentThread();
+        return currentThread.isInterrupted() || threads.contains(currentThread);
+    }
 }