[BugFix] Causing index out of bounds (#373)

Signed-off-by: kino <[email protected]>
StarRocks · Jul 15, 2024 · 51cc562 · 51cc562
1 parent cc8689d
commit 51cc562
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/com/starrocks/connector/flink/row/source/StarRocksSourceFlinkRows.java b/src/main/java/com/starrocks/connector/flink/row/source/StarRocksSourceFlinkRows.java
@@ -70,7 +70,7 @@ public void init(List<ArrowFieldConverter> fieldConverters) throws IOException {
             for (int i = 0; i < rowCountOfBatch; i ++) {
                 sourceFlinkRows.add(new GenericRowData(this.selectedColumns.length));
             }
-            for (int i = 0; i < fieldVectors.size(); i++) {
+            for (int i = 0; i < selectedColumns.length; i++) {
                 FieldVector fieldVector = fieldVectors.get(i);
                 ArrowFieldConverter converter = fieldConverters.get(i);
                 for (int rowIndex = 0; rowIndex < rowCountOfBatch; rowIndex++) {