From f9391a78e713d30e13d26735d9614b8e50143cc0 Mon Sep 17 00:00:00 2001
From: fariel <hf@dorisdb.com>
Date: Fri, 23 Jul 2021 14:38:43 +0800
Subject: [PATCH] optimize buffer allocated logic

---
 .../writer/doriswriter/manager/DorisStreamLoadVisitor.java      | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/doriswriter/src/main/java/com/dorisdb/connector/datax/plugin/writer/doriswriter/manager/DorisStreamLoadVisitor.java b/doriswriter/src/main/java/com/dorisdb/connector/datax/plugin/writer/doriswriter/manager/DorisStreamLoadVisitor.java
index f792bfaf..0bea3bdc 100644
--- a/doriswriter/src/main/java/com/dorisdb/connector/datax/plugin/writer/doriswriter/manager/DorisStreamLoadVisitor.java
+++ b/doriswriter/src/main/java/com/dorisdb/connector/datax/plugin/writer/doriswriter/manager/DorisStreamLoadVisitor.java
@@ -106,7 +106,7 @@ public class DorisStreamLoadVisitor {
         }
        
         if (DorisWriterOptions.StreamLoadFormat.JSON.equals(writerOptions.getStreamLoadFormat())) {
-            ByteBuffer bos = ByteBuffer.allocate(totalBytes + rows.size() + 1);
+            ByteBuffer bos = ByteBuffer.allocate(totalBytes + (rows.isEmpty() ? 2 : rows.size() + 1));
             bos.put("[".getBytes(StandardCharsets.UTF_8));
             byte[] jsonDelimiter = ",".getBytes(StandardCharsets.UTF_8);
             boolean isFirstElement = true;