SOLR-18248 : Migration of Tasks V2 APIs to JAX-RS Constructs

solr/api/src/java/org/apache/solr/client/api/endpoint/ListActiveTasksApi.java

@@ -0,0 +1,21 @@
+package org.apache.solr.client.api.endpoint;
+
+import io.swagger.v3.oas.annotations.Operation;
+import jakarta.ws.rs.GET;
+import jakarta.ws.rs.Path;
+import jakarta.ws.rs.QueryParam;
+import org.apache.solr.client.api.model.ListActiveTaskResponse;
+import org.apache.solr.client.api.util.StoreApiParameters;
+
+import static org.apache.solr.client.api.util.Constants.INDEX_PATH_PREFIX;
+
+@Path(INDEX_PATH_PREFIX + "/tasks/listjalaz")
+public interface ListActiveTasksApi {
+  @GET
+  @StoreApiParameters
+  @Operation(
+      summary = "Lists all the currently running tasks or status of any taskUUID being passed as queryParam",
+      tags = {"tasks"})
+  ListActiveTaskResponse listActiveTasks(
+      @QueryParam("taskUUID") String taskUUID) throws Exception;
+}

solr/api/src/java/org/apache/solr/client/api/model/ListActiveTaskResponse.java

@@ -0,0 +1,9 @@
+package org.apache.solr.client.api.model;
+
+import com.fasterxml.jackson.annotation.JsonProperty;
+import java.util.Map;
+
+public class ListActiveTaskResponse extends SolrJerseyResponse {
+  @JsonProperty
+  public Map<String, String> taskList;
+}

solr/core/src/java/org/apache/solr/handler/admin/api/ListActiveTasks.java

@@ -0,0 +1,82 @@
+package org.apache.solr.handler.admin.api;
+
+import jakarta.inject.Inject;
+import org.apache.solr.api.JerseyResource;
+import org.apache.solr.client.api.endpoint.ListActiveTasksApi;
+import org.apache.solr.client.api.model.ListActiveTaskResponse;
+import org.apache.solr.core.CoreContainer;
+import org.apache.solr.jersey.PermissionName;
+import org.apache.solr.request.SolrQueryRequest;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+import java.lang.invoke.MethodHandles;
+import java.util.HashMap;
+import java.util.Iterator;
+import java.util.Map;
+
+import static org.apache.solr.security.PermissionNameProvider.Name.READ_PERM;
+
+public class ListActiveTasks extends JerseyResource implements ListActiveTasksApi {
+
+  private static final Logger log = LoggerFactory.getLogger(MethodHandles.lookup().lookupClass());
+
+  private final SolrQueryRequest solrQueryRequest;
+
+  @Inject
+  public ListActiveTasks(
+      SolrQueryRequest solrQueryRequest) {
+    this.solrQueryRequest = solrQueryRequest;
+  }
+
+
+
+  @Override
+  @PermissionName(READ_PERM)
+  public ListActiveTaskResponse listActiveTasks(String taskUUID) throws Exception {
+
+    final ListActiveTaskResponse response = instantiateJerseyResponse(ListActiveTaskResponse.class);
+    CoreContainer coreContainer = solrQueryRequest.getCoreContainer();
+
+    if (coreContainer.isZooKeeperAware()) {
+      if (log.isDebugEnabled()) {
+        log.debug("solr cloud");
+      }
+      handleSolrCloudMode(response, taskUUID);
+    } else {
+      if (log.isDebugEnabled()) {
+        log.debug("standalone solr");
+      }
+      handleStandAloneMode(response, taskUUID);
+    }
+
+    log.debug("something random");
+
+    return response;
+
+  }
+
+  private void handleStandAloneMode(ListActiveTaskResponse response, String taskUUID) {
+    Iterator<Map.Entry<String, String>> iterator = solrQueryRequest.getCore().getCancellableQueryTracker().getActiveQueriesGenerated();
+
+    Map<String, String> taskList = new HashMap<>();
+    while (iterator.hasNext()) {
+      Map.Entry<String, String> entry = iterator.next();
+      taskList.put(entry.getKey(), entry.getValue());
+    }
+
+    response.taskList = taskList;
+  }
+
+  private void handleSolrCloudMode(ListActiveTaskResponse response, String taskUUID) {
+    Iterator<Map.Entry<String, String>> iterator = solrQueryRequest.getCore().getCancellableQueryTracker().getActiveQueriesGenerated();
+
+    Map<String, String> taskList = new HashMap<>();
+    while (iterator.hasNext()) {
+      Map.Entry<String, String> entry = iterator.next();
+      taskList.put(entry.getKey(), entry.getValue());
+    }
+
+    response.taskList = taskList;
+  }
+}

solr/core/src/java/org/apache/solr/handler/component/TaskManagementHandler.java

@@ -21,14 +21,17 @@
 
 import java.io.IOException;
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.List;
 import java.util.Map;
+import org.apache.solr.api.JerseyResource;
 import org.apache.solr.common.SolrException;
 import org.apache.solr.common.params.CommonParams;
 import org.apache.solr.common.params.ModifiableSolrParams;
 import org.apache.solr.core.CoreContainer;
 import org.apache.solr.core.SolrCore;
 import org.apache.solr.handler.RequestHandlerBase;
+import org.apache.solr.handler.admin.api.ListActiveTasks;
 import org.apache.solr.request.SolrQueryRequest;
 import org.apache.solr.response.SolrQueryResponse;
 import org.apache.solr.security.PermissionNameProvider;
@@ -132,4 +135,9 @@ public static ResponseBuilder buildResponseBuilder(
 
     return rb;
   }
+
+  @Override
+  public Collection<Class<? extends JerseyResource>> getJerseyResources() {
+    return List.of(ListActiveTasks.class);
+  }
 }