Add env BULK_PACKET_SIZE (default 524880 ( = 5120*1024))

rockeet · rockeet · commit dee44f4d3088 · 2024-01-25T14:57:34.000+08:00
diff --git a/src/db_driver.c b/src/db_driver.c
@@ -40,7 +40,7 @@
 #include "sb_ck_pr.h"
 
 /* Query length limit for bulk insert queries */
-#define BULK_PACKET_SIZE (5120*1024)
+static unsigned int BULK_PACKET_SIZE = 5120*1024;
 
 /* How many rows to insert before COMMITs (used in bulk insert) */
 static int ROWS_BEFORE_COMMIT = 1000;
@@ -943,6 +943,8 @@ int db_bulk_insert_init(db_conn_t *con, const char *query, size_t query_len)
     return 1;
   }
 
+  BULK_PACKET_SIZE = atoi(getenv("BULK_PACKET_SIZE")?:"5242880");
+
   /* Allocate query buffer */
   if (query_len + 1 > BULK_PACKET_SIZE)
   {

Original file line number	Diff line number	Diff line change
`@@ -40,7 +40,7 @@`
`40`	`40`	`#include "sb_ck_pr.h"`
`41`	`41`
`42`	`42`	`/* Query length limit for bulk insert queries */`
`43`		`-#define BULK_PACKET_SIZE (5120*1024)`
	`43`	`+static unsigned int BULK_PACKET_SIZE = 5120*1024;`
`44`	`44`
`45`	`45`	`/* How many rows to insert before COMMITs (used in bulk insert) */`
`46`	`46`	`static int ROWS_BEFORE_COMMIT = 1000;`
`@@ -943,6 +943,8 @@ int db_bulk_insert_init(db_conn_t con, const char query, size_t query_len)`
`943`	`943`	`return 1;`
`944`	`944`	`}`
`945`	`945`
	`946`	`+ BULK_PACKET_SIZE = atoi(getenv("BULK_PACKET_SIZE")?:"5242880");`
	`947`	`+`
`946`	`948`	`/* Allocate query buffer */`
`947`	`949`	`if (query_len + 1 > BULK_PACKET_SIZE)`
`948`	`950`	`{`