From 40cb2f55bb9e53353e24ac1e12855e95d8eb6ced Mon Sep 17 00:00:00 2001
From: zhaoyanggh <yzhao@taosdata.com>
Date: Mon, 21 Jun 2021 10:13:27 +0800
Subject: [PATCH] change performance test tool to optimize write test

---
 tests/comparisonTest/tdengine/tdengineTest.c | 84 ++++++++++++++++++--
 1 file changed, 78 insertions(+), 6 deletions(-)

diff --git a/tests/comparisonTest/tdengine/tdengineTest.c b/tests/comparisonTest/tdengine/tdengineTest.c
index d1cf3a1f98..0de419e036 100644
--- a/tests/comparisonTest/tdengine/tdengineTest.c
+++ b/tests/comparisonTest/tdengine/tdengineTest.c
@@ -181,8 +181,8 @@ void writeDataImp(void *param) {
 
       if (lastMachineid != machineid) {
         lastMachineid = machineid;
-        sqlLen += sprintf(sql + sqlLen, " dev%d using devices tags(%d,'%s',%d) values",
-                          machineid, machineid, machinename, machinegroup);
+        sqlLen += sprintf(sql + sqlLen, " dev%d values",
+                          machineid);
       }
 
       sqlLen += sprintf(sql + sqlLen, "(%" PRId64 ",%d,%f)", timestamp, temperature, humidity);
@@ -192,7 +192,8 @@ void writeDataImp(void *param) {
         result = taos_query(taos, sql);
         code = taos_errno(result);
         if (code != 0) {
-          printf("thread:%d error:%d reason:%s\n", pThread->threadId, code, taos_errstr(taos));
+          printf("insert into dev%d values (%" PRId64 ",%d,%f)\n",machineid, timestamp, temperature, humidity);
+          printf("thread:%d error:%d reason:%s\n", pThread->threadId, code, taos_errstr(result));
         }
         taos_free_result(result);
 
@@ -210,6 +211,7 @@ void writeDataImp(void *param) {
     result = taos_query(taos, sql);
     code = taos_errno(result);
     if (code != 0) {
+      // printf("insert into dev%d using devices tags(%d,'%s',%d) values (%" PRId64 ",%d,%f)",machineid, machineid, machinename, machinegroup, timestamp, temperature, humidity);
       printf("thread:%d error:%d reason:%s\n", pThread->threadId, code, taos_errstr(taos));
     }
     taos_free_result(result);
@@ -246,7 +248,7 @@ void writeData() {
   taos_free_result(result);
 
   result = taos_query(taos,
-                      "create table if not exists db.devices(ts timestamp, temperature int, humidity float) "
+                      "create stable if not exists db.devices(ts timestamp, temperature int, humidity float) "
                       "tags(devid int, devname binary(16), devgroup int)");
   code = taos_errno(result);
   if (code != 0) {
@@ -254,6 +256,77 @@ void writeData() {
   }
   taos_free_result(result);
 
+  //create tables before insert the data
+  result = taos_query(taos, "use db");
+  code = taos_errno(result);
+  if (code != 0) {
+    taos_error(result, taos);
+  }
+  taos_free_result(result);
+  
+  char *sql = calloc(1, 8*1024*1024);
+  int sqlLen = 0;
+  int lastMachineid = 0;
+  int counter = 0;
+  int totalRecords = 0;
+  for (int i = 0; i < arguments.filesNum; i++) {
+    char fileName[300];
+    sprintf(fileName, "%s/testdata%d.csv", arguments.dataDir, i);
+
+    FILE *fp = fopen(fileName, "r");
+    if (fp == NULL) {
+      printf("failed to open file %s\n", fileName);
+      exit(1);
+    }
+    printf("open file %s success\n", fileName);
+
+    char *line = NULL;
+    size_t len = 0;
+    while (!feof(fp)) {
+      free(line);
+      line = NULL;
+      len = 0;
+
+      getline(&line, &len, fp);
+      if (line == NULL) break;
+
+      if (strlen(line) < 10) continue;
+      
+      int machineid;
+      char machinename[16];
+      int machinegroup;
+      int64_t timestamp;
+      int temperature;
+      float humidity;
+      sscanf(line, "%d%s%d%" PRId64 "%d%f", &machineid, machinename, &machinegroup, &timestamp, &temperature, &humidity);
+
+      if (counter == 0) {
+        sqlLen = sprintf(sql, "create table if not exists");
+      }
+
+      if (lastMachineid != machineid) {
+        lastMachineid = machineid;
+        sqlLen += sprintf(sql + sqlLen, " dev%d using devices tags(%d,'%s',%d)", machineid, machineid, machinename, machinegroup);
+      }
+      counter++;
+
+      if (counter >= arguments.rowsPerRequest) {
+        result = taos_query(taos, sql);
+        code = taos_errno(result);
+        if (code != 0) {
+          printf("create table error:%d reason:%s\n", code, taos_errstr(result));
+        }
+        taos_free_result(result);
+
+        totalRecords += counter;
+        counter = 0;
+        lastMachineid = -1;
+        sqlLen = 0;
+      }
+    }
+    fclose(fp);    
+  }
+
   int64_t st = getTimeStampMs();
 
   int a = arguments.filesNum / arguments.clients;
@@ -379,5 +452,4 @@ void readData() {
   }
 
   free(threads);
-}
-
+}
\ No newline at end of file
-- 
GitLab