Revert "Titus: return done jobs for getJob method for v3 engine"

spinnaker · Mar 26, 2018 · 52a6956 · 52a6956
1 parent d28f5fd
commit 52a6956
Showing 1 changed file with 11 additions and 14 deletions.
diff --git a/...in/groovy/com/netflix/spinnaker/clouddriver/titus/v3client/RegionScopedV3TitusClient.java b/...in/groovy/com/netflix/spinnaker/clouddriver/titus/v3client/RegionScopedV3TitusClient.java
@@ -128,7 +128,7 @@ public RegionScopedV3TitusClient(TitusRegion titusRegion,
 
   @Override
   public Job getJob(String jobId) {
-    return new Job(grpcBlockingStub.findJob(JobId.newBuilder().setId(jobId).build()), getTasks(Arrays.asList(jobId), true).get(jobId));
+    return new Job(grpcBlockingStub.findJob(JobId.newBuilder().setId(jobId).build()), getTasks(Arrays.asList(jobId)).get(jobId));
   }
 
   @Override
@@ -257,23 +257,20 @@ private List<Job> getJobs(JobQuery.Builder jobQuery) {
     List<String> jobIds = grpcJobs.stream().map(grpcJob -> grpcJob.getId()).collect(
       Collectors.toList()
     );
-    Map<String, List<com.netflix.titus.grpc.protogen.Task>> tasks = getTasks(jobIds, false);
+    Map<String, List<com.netflix.titus.grpc.protogen.Task>> tasks = getTasks(jobIds);
     return grpcJobs.stream().map(grpcJob -> new Job(grpcJob, tasks.get(grpcJob.getId()))).collect(Collectors.toList());
   }
 
-  private Map<String, List<com.netflix.titus.grpc.protogen.Task>> getTasks(List<String> jobIds, boolean includeDoneJobs) {
+  private Map<String, List<com.netflix.titus.grpc.protogen.Task>> getTasks(List<String> jobIds) {
     List<com.netflix.titus.grpc.protogen.Task> tasks = new ArrayList<>();
     TaskQueryResult taskResults;
     int currentTaskPage = 0;
     do {
-      TaskQuery.Builder taskQueryBuilder = TaskQuery.newBuilder();
-      taskQueryBuilder.setPage(Page.newBuilder().setPageNumber(currentTaskPage).setPageSize(100));
-      taskQueryBuilder.putFilteringCriteria("jobIds", jobIds.stream().collect(Collectors.joining(",")));
-      if(includeDoneJobs) {
-        taskQueryBuilder.putFilteringCriteria("taskStates", "Accepted,Launched,StartInitiated,Started,KillInitiated,Finished");
-      }
       taskResults = grpcBlockingStub.findTasks(
-        taskQueryBuilder.build()
+        TaskQuery.newBuilder()
+          .putFilteringCriteria("jobIds", jobIds.stream().collect(Collectors.joining(",")))
+          .setPage(Page.newBuilder().setPageNumber(currentTaskPage).setPageSize(100)
+          ).build()
       );
       tasks.addAll(taskResults.getItemsList());
       currentTaskPage++;
@@ -291,11 +288,11 @@ public Object getTaskJson(String taskId) {
     return toJson(grpcBlockingStub.findTask(TaskId.newBuilder().setId(taskId).build()));
   }
 
-  private Object toJson(Message message) {
-    Object job = null;
-    try {
+  private Object toJson(Message message){
+    Object job =null;
+    try{
       job = objectMapper.readValue(JsonFormat.printer().print(message), Object.class);
-    } catch (Exception e) {
+    } catch(Exception e){
 
     }
     return job;