16361: using shared lock types by default for kill tasks unless markA…

…sUnused is on
apache · Apr 30, 2024 · 82b91a9 · 82b91a9
1 parent b18b192
commit 82b91a9
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/...g-service/src/main/java/org/apache/druid/indexing/common/task/KillUnusedSegmentsTask.java b/...g-service/src/main/java/org/apache/druid/indexing/common/task/KillUnusedSegmentsTask.java
@@ -32,6 +32,7 @@
 import org.apache.druid.indexer.report.KillTaskReport;
 import org.apache.druid.indexer.report.TaskReport;
 import org.apache.druid.indexing.common.TaskLock;
+import org.apache.druid.indexing.common.TaskLockType;
 import org.apache.druid.indexing.common.TaskToolbox;
 import org.apache.druid.indexing.common.actions.MarkSegmentsAsUnusedAction;
 import org.apache.druid.indexing.common.actions.RetrieveUnusedSegmentsAction;
@@ -343,6 +344,9 @@ private NavigableMap<DateTime, List<TaskLock>> getNonRevokedTaskLockMap(TaskActi
   @Override
   public boolean isReady(TaskActionClient taskActionClient) throws Exception
   {
-    return super.isReady(taskActionClient, getContextValue(Tasks.TASK_LOCK_TYPE, Tasks.DEFAULT_TASK_LOCK_TYPE));
+    if (markAsUnused) {
+      return super.isReady(taskActionClient);
+    }
+    return super.isReady(taskActionClient, getContextValue(Tasks.TASK_LOCK_TYPE, TaskLockType.SHARED));
   }
 }