Add default value of Glue GetPartitions MaxResults

bloomberg · Mar 11, 2020 · 4f22b0e · 4f22b0e
1 parent bbeac7f
commit 4f22b0e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/presto-hive/src/main/java/io/prestosql/plugin/hive/metastore/glue/GlueHiveMetastore.java b/presto-hive/src/main/java/io/prestosql/plugin/hive/metastore/glue/GlueHiveMetastore.java
@@ -146,6 +146,7 @@ public class GlueHiveMetastore
     private static final String WILDCARD_EXPRESSION = "";
     private static final int BATCH_GET_PARTITION_MAX_PAGE_SIZE = 1000;
     private static final int BATCH_CREATE_PARTITION_MAX_PAGE_SIZE = 100;
+    private static final int AWS_GLUE_GET_PARTITIONS_MAX_RESULTS = 128;
     private static final Comparator<Partition> PARTITION_COMPARATOR =
             comparing(Partition::getValues, lexicographical(String.CASE_INSENSITIVE_ORDER));
 
@@ -749,7 +750,8 @@ private List<Partition> getPartitions(Table table, String expression, @Nullable
                             .withTableName(table.getTableName())
                             .withExpression(expression)
                             .withSegment(segment)
-                            .withNextToken(nextToken));
+                            .withNextToken(nextToken)
+                            .withMaxResults(AWS_GLUE_GET_PARTITIONS_MAX_RESULTS));
                     result.getPartitions()
                             .forEach(partition -> partitions.add(GlueToPrestoConverter.convertPartition(partition, table.getParameters())));
                     nextToken = result.getNextToken();