TD-100

src/tsdb/tests/tsdbTests.cpp

@@ -4,6 +4,7 @@
 
 #include "tdataformat.h"
 #include "tsdbMain.h"
+#include "tskiplist.h"
 
 static double getCurTime() {
   struct timeval tv;
@@ -125,8 +126,8 @@ TEST(TsdbTest, DISABLED_tableEncodeDecode) {
   ASSERT_EQ(memcmp(pTable->schema, tTable->schema, sizeof(STSchema) + sizeof(STColumn) * nCols), 0);
 }
 
-// TEST(TsdbTest, DISABLED_createRepo) {
-TEST(TsdbTest, createRepo) {
+TEST(TsdbTest, DISABLED_createRepo) {
+// TEST(TsdbTest, createRepo) {
   STsdbCfg config;
   STsdbRepo *repo;
 
@@ -167,7 +168,7 @@ TEST(TsdbTest, createRepo) {
     .sversion = tCfg.sversion,
     .startTime = 1584081000000,
     .interval = 1000,
-    .totalRows = 5000000,
+    .totalRows = 10000000,
     .rowsPerSubmit = 1,
     .pSchema = schema
   };
@@ -263,3 +264,45 @@ TEST(TsdbTest, DISABLED_createFileGroup) {
 
   int k = 0;
 }
+
+static char *getTKey(const void *data) {
+  return (char *)data;
+}
+
+static void insertSkipList(bool isAscend) {
+  TSKEY start_time = 1587393453000;
+  TSKEY interval = 1000;
+
+  SSkipList *pList = tSkipListCreate(5, TSDB_DATA_TYPE_TIMESTAMP, sizeof(TSKEY), 0, 0, 1, getTKey);
+  ASSERT_NE(pList, nullptr);
+
+  for (size_t i = 0; i < 20000000; i++)
+  {
+    TSKEY time = isAscend ? (start_time + i * interval) : (start_time - i * interval);
+    int32_t level = 0;
+    int32_t headSize = 0;
+
+    tSkipListNewNodeInfo(pList, &level, &headSize);
+    SSkipListNode *pNode = (SSkipListNode *)malloc(headSize + sizeof(TSKEY));
+    ASSERT_NE(pNode, nullptr);
+    pNode->level = level;
+    *(TSKEY *)((char *)pNode + headSize) = time;
+    tSkipListPut(pList, pNode);
+  }
+
+  tSkipListDestroy(pList);
+}
+
+TEST(TsdbTest, testSkipList) {
+  double stime = getCurTime();
+  insertSkipList(true);
+  double etime = getCurTime();
+
+  printf("Time used to insert 100000000 records takes %f seconds\n", etime-stime);
+
+  stime = getCurTime();
+  insertSkipList(false);
+  etime = getCurTime();
+
+  printf("Time used to insert 100000000 records takes %f seconds\n", etime-stime);
+}
\ No newline at end of file