support batch list latest jobs by flow id list

Author: gy2006

core/src/main/java/com/flowci/core/job/controller/JobController.java

@@ -35,6 +35,8 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.util.Base64;
+import java.util.Collection;
+import java.util.Collections;
 import java.util.List;
 
 /**
@@ -56,6 +58,15 @@ public class JobController extends BaseController {
     @Autowired
     private TaskExecutor appTaskExecutor;
 
+    @PostMapping("/batch/latest")
+    @Action(JobAction.LIST)
+    public List<Job> listLatest(@RequestBody Collection<String> flowIdList) {
+        if (flowIdList.isEmpty()) {
+            return Collections.emptyList();
+        }
+        return jobService.listLatest(flowIdList);
+    }
+
     @GetMapping("/{flow}")
     @Action(JobAction.LIST)
     public Page<JobItem> list(@PathVariable("flow") String name,

core/src/main/java/com/flowci/core/job/dao/CustomJobDao.java

@@ -1,6 +1,14 @@
 package com.flowci.core.job.dao;
 
+import com.flowci.core.job.domain.Job;
+import com.flowci.core.job.domain.JobNumber;
+
+import java.util.Collection;
+import java.util.List;
+
 public interface CustomJobDao {
 
     void increaseNumOfArtifact(String jobId);
+
+    List<Job> list(Collection<JobNumber> numbers);
 }

core/src/main/java/com/flowci/core/job/dao/CustomJobDaoImpl.java

@@ -1,10 +1,16 @@
 package com.flowci.core.job.dao;
 
 import com.flowci.core.job.domain.Job;
+import com.flowci.core.job.domain.JobNumber;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.data.mongodb.core.MongoOperations;
+import org.springframework.data.mongodb.core.query.Criteria;
 import org.springframework.data.mongodb.core.query.Update;
 
+import java.util.ArrayList;
+import java.util.Collection;
+import java.util.List;
+
 import static org.springframework.data.mongodb.core.query.Criteria.where;
 import static org.springframework.data.mongodb.core.query.Query.query;
 
@@ -20,4 +26,20 @@ public void increaseNumOfArtifact(String jobId) {
                 new Update().inc("numOfArtifact", 1),
                 Job.class);
     }
+
+    @Override
+    public List<Job> list(Collection<JobNumber> numbers) {
+        List<String> flowIdList = new ArrayList<>(numbers.size());
+        List<Long> buildNumList = new ArrayList<>(numbers.size());
+
+        for (JobNumber num : numbers) {
+            flowIdList.add(num.getFlowId());
+            buildNumList.add(num.getNumber());
+        }
+
+        Criteria criteria = where("flowId").in(flowIdList)
+                .andOperator(where("buildNumber").in(buildNumList));
+
+        return operations.find(query(criteria), Job.class);
+    }
 }

core/src/main/java/com/flowci/core/job/dao/JobNumberDao.java

@@ -20,6 +20,8 @@
 import org.springframework.data.mongodb.repository.MongoRepository;
 import org.springframework.stereotype.Repository;
 
+import java.util.Collection;
+import java.util.List;
 import java.util.Optional;
 
 /**
@@ -30,5 +32,7 @@ public interface JobNumberDao extends MongoRepository<JobNumber, String>, Custom
 
     Optional<JobNumber> findByFlowId(String flowId);
 
+    List<JobNumber> findAllByFlowIdIn(Collection<String> ids);
+
     void deleteAllByFlowId(String flowId);
 }

core/src/main/java/com/flowci/core/job/service/JobService.java

@@ -26,6 +26,7 @@
 import org.springframework.data.domain.Page;
 
 import javax.annotation.Nullable;
+import java.util.Collection;
 import java.util.List;
 
 /**
@@ -61,10 +62,15 @@ public interface JobService {
     JobYml getYml(Job job);
 
     /**
-     * Get latest job
+     * Get the latest job
      */
     Job getLatest(String flowId);
 
+    /**
+     * List latest jobs from flow id list
+     */
+    List<Job> listLatest(Collection<String> flowIdList);
+
     /**
      * List jobs with fields only shown on the list
      */

core/src/main/java/com/flowci/core/job/service/JobServiceImpl.java

@@ -199,6 +199,12 @@ public Job getLatest(String flowId) {
         throw new NotFoundException("No jobs for flow {0}", flowId);
     }
 
+    @Override
+    public List<Job> listLatest(Collection<String> flowIdList) {
+        List<JobNumber> numberList = jobNumberDao.findAllByFlowIdIn(flowIdList);
+        return jobDao.list(numberList);
+    }
+
     @Override
     public Page<JobItem> list(Flow flow, int page, int size) {
         PageRequest pageable = PageRequest.of(page, size, SortByBuildNumber);