Github user xubo245 commented on a diff in the pull request:
https://github.com/apache/carbondata/pull/2816#discussion_r228390686
--- Diff:
core/src/main/java/org/apache/carbondata/core/scan/result/RowBatch.java ---
@@ -100,4 +100,24 @@ public int getSize() {
counter++;
return row;
}
+
+ /**
+ * read next batch
+ *
+ * @param batch batch size
+ * @return rows
+ */
+ public List<Object[]> nextBatch(int batch) {
+ if (!hasNext()) {
+ throw new NoSuchElementException();
+ }
+ List<Object[]> row;
+ if (counter + batch > rows.size()) {
+ row = rows.subList(counter, rows.size());
+ } else {
+ row = rows.subList(counter, counter + batch);
+ }
+ counter = counter + batch;
--- End diff --
ok, optimized
---
