tsdbTests.cpp

#include <gtest/gtest.h>
#include <stdlib.h>

#include "tsdb.h"
#include "dataformat.h"

TEST(TsdbTest, createRepo) {
  STsdbCfg config;

  // 1. Create a tsdb repository
  tsdbSetDefaultCfg(&config);
  tsdb_repo_t *pRepo = tsdbCreateRepo("/home/ubuntu/work/ttest/vnode0", &config, NULL);
  ASSERT_NE(pRepo, nullptr);

  // 2. Create a normal table
  STableCfg tCfg;
  ASSERT_EQ(tsdbInitTableCfg(&tCfg, TSDB_SUPER_TABLE, 987607499877672L, 0), -1);
  ASSERT_EQ(tsdbInitTableCfg(&tCfg, TSDB_NTABLE, 987607499877672L, 0), 0);

  int       nCols = 5;
  STSchema *schema = tdNewSchema(nCols);

  for (int i = 0; i < nCols; i++) {
    if (i == 0) {
      tdSchemaAppendCol(schema, TSDB_DATA_TYPE_TIMESTAMP, i, -1);
    } else {
      tdSchemaAppendCol(schema, TSDB_DATA_TYPE_INT, i, -1);
    }
  }

  tsdbTableSetSchema(&tCfg, schema, true);

  tsdbCreateTable(pRepo, &tCfg);

  // // 3. Loop to write some simple data
  int nRows = 10;
  SSubmitMsg *pMsg = (SSubmitMsg *)malloc(sizeof(SSubmitMsg) + sizeof(SSubmitBlk) + tdMaxRowBytesFromSchema(schema) * nRows);

  SSubmitBlk *pBlock = pMsg->blocks;
  pBlock->tableId = {.uid = 987607499877672L, .tid = 0};
  pBlock->sversion = 0;
  pBlock->len = 0;
  int64_t start_time = 1584081000000;
  for (int i = 0; i < nRows; i++) {
    int64_t ttime = start_time + 1000 * i;
    SDataRow row = (SDataRow)(pBlock->data + pBlock->len);
    tdInitDataRow(row, schema);

    for (int j = 0; j < schemaNCols(schema); j++) {
      if (j == 0) { // Just for timestamp
        tdAppendColVal(row, (void *)(&time), schemaColAt(schema, j));
      } else { // For int
        int val = 10;
        tdAppendColVal(row, (void *)(&val), schemaColAt(schema, j));
      }

    }
    pBlock->len += dataRowLen(row);

  }
  pMsg->length = pMsg->length + sizeof(SSubmitBlk) + pBlock->len;

  tsdbInsertData(pRepo, pMsg);
}