[bug] byteBuffer calculate new capacity bug fix.

apache · Sep 14, 2023 · b2bf29f · b2bf29f
1 parent 0daf6c4
commit b2bf29f
Showing 1 changed file with 5 additions and 6 deletions.
diff --git a/spark-doris-connector/src/main/java/org/apache/doris/spark/load/RecordBatchInputStream.java b/spark-doris-connector/src/main/java/org/apache/doris/spark/load/RecordBatchInputStream.java
@@ -170,14 +170,13 @@ private void ensureCapacity(int need) {
      * @return new capacity
      */
     private int calculateNewCapacity(int capacity, int minCapacity) {
-        int newCapacity;
+        int newCapacity = 0;
         if (capacity == 0) {
             newCapacity = DEFAULT_BUF_SIZE;
-            while (newCapacity < minCapacity) {
-                newCapacity = newCapacity << 1;
-            }
-        } else {
-            newCapacity = capacity << 1;
+
+        }
+        while (newCapacity < minCapacity) {
+            newCapacity = newCapacity << 1;
         }
         return newCapacity;
     }