Merge branch 'main' into merge/to3.0

taosdata · Jun 28, 2024 · 79327c8 · 79327c8
2 parents 00cb835 + 6d74823
commit 79327c8
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/src/benchData.c b/src/benchData.c
@@ -1991,6 +1991,20 @@ uint32_t bindParamBatch(threadInfo *pThreadInfo,
         }
     }
 
+    /*
+      1. The last batch size may be smaller than the previous batch size.
+      2. When inserting another table, the batch size reset again(bigger than lastBatchSize)
+    */        
+    int lastBatchSize = ((TAOS_MULTI_BIND *) pThreadInfo->bindParams)->num;
+    if (batch != lastBatchSize) {
+        for (int c = 0; c < columnCount + 1; c++) {
+            TAOS_MULTI_BIND *param =
+                    (TAOS_MULTI_BIND *) (pThreadInfo->bindParams +
+                    sizeof(TAOS_MULTI_BIND) * c);
+            param->num = batch;
+        }
+    }
+
     int64_t start = toolsGetTimestampUs();
     if (taos_stmt_bind_param_batch(
             stmt, (TAOS_MULTI_BIND *)pThreadInfo->bindParams)) {