From f10beae1d730bf6a23d08a9342a9843e6884133c Mon Sep 17 00:00:00 2001
From: Mark Vulfson <mvulfson@netflix.com>
Date: Mon, 8 Jul 2019 14:08:54 -0700
Subject: [PATCH] fix(titus): adjust page size for titus queries (#3854)

based on feedback from titus team, adjust page size for titus job/tasks
requests to 1k
---
 .../titus/client/RegionScopedTitusClient.java            | 9 ++-------
 1 file changed, 2 insertions(+), 7 deletions(-)

diff --git a/clouddriver-titus/src/main/groovy/com/netflix/spinnaker/clouddriver/titus/client/RegionScopedTitusClient.java b/clouddriver-titus/src/main/groovy/com/netflix/spinnaker/clouddriver/titus/client/RegionScopedTitusClient.java
index 77ab63c41ce..51fb149e702 100644
--- a/clouddriver-titus/src/main/groovy/com/netflix/spinnaker/clouddriver/titus/client/RegionScopedTitusClient.java
+++ b/clouddriver-titus/src/main/groovy/com/netflix/spinnaker/clouddriver/titus/client/RegionScopedTitusClient.java
@@ -416,8 +416,7 @@ public Map<String, List<String>> getTaskIdsForJobIds() {
         .addFields("id")
         .addFields("jobId");
 
-    List<com.netflix.titus.grpc.protogen.Task> grpcTasks =
-        getTasksWithFilter(taskQueryBuilder, 10000);
+    List<com.netflix.titus.grpc.protogen.Task> grpcTasks = getTasksWithFilter(taskQueryBuilder);
     return grpcTasks.stream()
         .collect(
             Collectors.groupingBy(
@@ -491,12 +490,8 @@ private List<com.netflix.titus.grpc.protogen.Job> getJobsWithFilter(
 
   private List<com.netflix.titus.grpc.protogen.Task> getTasksWithFilter(
       TaskQuery.Builder taskQueryBuilder) {
-    return getTasksWithFilter(
-        taskQueryBuilder, titusRegion.getFeatureFlags().contains("largePages") ? 2000 : 1000);
-  }
 
-  private List<com.netflix.titus.grpc.protogen.Task> getTasksWithFilter(
-      TaskQuery.Builder taskQueryBuilder, Integer pageSize) {
+    final int pageSize = 1000;
     List<com.netflix.titus.grpc.protogen.Task> grpcTasks = new ArrayList<>();
 
     TaskQueryResult taskResults;