db2 fix

vinlee19 · Apr 19, 2024 · 0f81da2 · 0f81da2
1 parent 3cebd37
commit 0f81da2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/fe/be-java-extensions/jdbc-scanner/src/main/java/org/apache/doris/jdbc/BaseJdbcExecutor.java b/fe/be-java-extensions/jdbc-scanner/src/main/java/org/apache/doris/jdbc/BaseJdbcExecutor.java
@@ -236,7 +236,7 @@ public long getBlockAddress(int batchSize, Map<String, String> outputParams) thr
     }
 
     protected void initializeBlock(int columnCount, String[] replaceStringList, int batchSizeNum,
-            VectorTable outputTable) {
+                                   VectorTable outputTable) {
         for (int i = 0; i < columnCount; ++i) {
             block.add(outputTable.getColumn(i).newObjectContainerArray(batchSizeNum));
         }
@@ -293,7 +293,7 @@ public int getCurBlockRows() {
 
     public boolean hasNext() throws UdfRuntimeException {
         try {
-            if (resultSet == null) {
+            if (resultSet == null || resultSet.isClosed()) {
                 return false;
             }
             return resultSet.next();

diff --git a/...java-extensions/jdbc-scanner/src/main/java/org/apache/doris/jdbc/DefaultJdbcExecutor.java b/...java-extensions/jdbc-scanner/src/main/java/org/apache/doris/jdbc/DefaultJdbcExecutor.java
@@ -332,7 +332,7 @@ public int getCurBlockRows() {
 
     public boolean hasNext() throws UdfRuntimeException {
         try {
-            if (resultSet == null) {
+            if (resultSet == null || resultSet.isClosed()) {
                 return false;
             }
             return resultSet.next();
@@ -866,7 +866,7 @@ private List<Object> convertArray(List<Object> list, ColumnType childType) {
                 convertedList.add(convertArray((List<Object>) element, childType));
             } else {
                 if (converter != null) {
-                    convertedList.add(converter.convert(new Object[] {element})[0]);
+                    convertedList.add(converter.convert(new Object[]{element})[0]);
                 } else {
                     convertedList.add(element);
                 }