[BUG] error occurs when using the LIMIT clause with AthenaGoogleBigQu…

…eryConnector (#1808)
awslabs · Mar 8, 2024 · 646e686 · 646e686
1 parent ba27e15
commit 646e686
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.../src/main/java/com/amazonaws/athena/connectors/google/bigquery/BigQueryRecordHandler.java b/.../src/main/java/com/amazonaws/athena/connectors/google/bigquery/BigQueryRecordHandler.java
@@ -232,7 +232,7 @@ private void getTableData(BlockSpiller spiller, ReadRecordsRequest recordsReques
                 for (ReadRowsResponse response : stream) {
                     Preconditions.checkState(response.hasArrowRecordBatch());
                     VectorSchemaRoot root = reader.processRows(response.getArrowRecordBatch());
-                    long rowLimit = (recordsRequest.getConstraints().getLimit() > 0) ? recordsRequest.getConstraints().getLimit() : root.getRowCount();
+                    long rowLimit = (recordsRequest.getConstraints().getLimit() > 0 && recordsRequest.getConstraints().getLimit() < root.getRowCount()) ? recordsRequest.getConstraints().getLimit() : root.getRowCount();
                     for (int rowIndex = 0; rowIndex < rowLimit; rowIndex++) {
                         outputResults(spiller, recordsRequest, root, rowIndex);
                     }