1643 lines
50 KiB
C++
1643 lines
50 KiB
C++
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#include <gtest/gtest.h>
|
|
#include <iostream>
|
|
#include "clientInt.h"
|
|
#include "osSemaphore.h"
|
|
#include "taoserror.h"
|
|
#include "tglobal.h"
|
|
#include "thash.h"
|
|
|
|
#pragma GCC diagnostic push
|
|
#pragma GCC diagnostic ignored "-Wwrite-strings"
|
|
#pragma GCC diagnostic ignored "-Wunused-function"
|
|
#pragma GCC diagnostic ignored "-Wunused-variable"
|
|
#pragma GCC diagnostic ignored "-Wsign-compare"
|
|
|
|
#include "executor.h"
|
|
#include "taos.h"
|
|
|
|
namespace {
|
|
|
|
void printSubResults(void* pRes, int32_t* totalRows) {
|
|
char buf[1024] = {0};
|
|
|
|
int32_t vgId = tmq_get_vgroup_id(pRes);
|
|
int64_t offset = tmq_get_vgroup_offset(pRes);
|
|
while (1) {
|
|
TAOS_ROW row = taos_fetch_row(pRes);
|
|
if (row == NULL) {
|
|
break;
|
|
}
|
|
|
|
TAOS_FIELD* fields = taos_fetch_fields(pRes);
|
|
if(fields == NULL) {
|
|
std::cout << "fields is null" << std::endl;
|
|
break;
|
|
}
|
|
int32_t numOfFields = taos_field_count(pRes);
|
|
int32_t precision = taos_result_precision(pRes);
|
|
(void)taos_print_row(buf, row, fields, numOfFields);
|
|
*totalRows += 1;
|
|
std::cout << "vgId:" << vgId << ", offset:" << offset << ", precision:" << precision << ", row content:" << buf
|
|
<< std::endl;
|
|
}
|
|
|
|
// taos_free_result(pRes);
|
|
}
|
|
|
|
void showDB(TAOS* pConn) {
|
|
TAOS_RES* pRes = taos_query(pConn, "show databases");
|
|
TAOS_ROW pRow = NULL;
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
}
|
|
}
|
|
|
|
void printResult(TAOS_RES* pRes) {
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
int32_t n = 0;
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
// int32_t* length = taos_fetch_lengths(pRes);
|
|
// for(int32_t i = 0; i < numOfFields; ++i) {
|
|
// (void)printf("(%d):%d " , i, length[i]);
|
|
// }
|
|
// (void)printf("\n");
|
|
//
|
|
// int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
// (void)printf("%s\n", str);
|
|
// memset(str, 0, sizeof(str));
|
|
}
|
|
}
|
|
|
|
void fetchCallback(void* param, void* res, int32_t numOfRow) {
|
|
#if 0
|
|
(void)printf("numOfRow = %d \n", numOfRow);
|
|
int numFields = taos_num_fields(res);
|
|
TAOS_FIELD *fields = taos_fetch_fields(res);
|
|
TAOS *_taos = (TAOS *)param;
|
|
if (numOfRow > 0) {
|
|
for (int i = 0; i < numOfRow; ++i) {
|
|
TAOS_ROW row = taos_fetch_row(res);
|
|
|
|
char temp[256] = {0};
|
|
taos_print_row(temp, row, fields, numFields);
|
|
(void)printf("%s\n", temp);
|
|
}
|
|
taos_fetch_rows_a(res, fetchCallback, _taos);
|
|
} else {
|
|
(void)printf("no more data, close the connection.\n");
|
|
// taos_free_result(res);
|
|
// taos_close(_taos);
|
|
// taos_cleanup();
|
|
}
|
|
#endif
|
|
if (numOfRow == 0) {
|
|
(void)printf("completed\n");
|
|
return;
|
|
}
|
|
|
|
taos_fetch_raw_block_a(res, fetchCallback, param);
|
|
}
|
|
|
|
void queryCallback(void* param, void* res, int32_t code) {
|
|
if (code != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to execute, reason:%s\n", taos_errstr(res));
|
|
taos_free_result(res);
|
|
tsem_t *sem = (tsem_t *)param;
|
|
tsem_post(sem);
|
|
return;
|
|
}
|
|
(void)printf("start to fetch data\n");
|
|
taos_fetch_raw_block_a(res, fetchCallback, param);
|
|
taos_free_result(res);
|
|
tsem_t *sem = (tsem_t *)param;
|
|
tsem_post(sem);
|
|
}
|
|
|
|
void createNewTable(TAOS* pConn, int32_t index, int32_t numOfRows, int64_t startTs, const char* pVarchar) {
|
|
char str[1024] = {0};
|
|
(void)sprintf(str, "create table if not exists tu%d using st2 tags(%d)", index, index);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, str);
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create table tu, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
if (startTs == 0) {
|
|
for (int32_t i = 0; i < numOfRows; i += 20) {
|
|
char sql[1024] = {0};
|
|
(void)sprintf(sql,
|
|
"insert into tu%d values(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)"
|
|
"(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)"
|
|
"(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)"
|
|
"(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)(now+%da, %d)",
|
|
index, i, i, i + 1, i + 1, i + 2, i + 2, i + 3, i + 3, i + 4, i + 4, i + 5, i + 5, i + 6, i + 6, i + 7,
|
|
i + 7, i + 8, i + 8, i + 9, i + 9, i + 10, i + 10, i + 11, i + 11, i + 12, i + 12, i + 13, i + 13, i + 14,
|
|
i + 14, i + 15, i + 15, i + 16, i + 16, i + 17, i + 17, i + 18, i + 18, i + 19, i + 19);
|
|
TAOS_RES* p = taos_query(pConn, sql);
|
|
if (taos_errno(p) != 0) {
|
|
(void)printf("failed to insert data, reason:%s\n", taos_errstr(p));
|
|
}
|
|
|
|
taos_free_result(p);
|
|
}
|
|
} else {
|
|
for (int32_t i = 0; i < numOfRows; i += 20) {
|
|
char sql[1024*50] = {0};
|
|
(void)sprintf(sql,
|
|
"insert into tu%d values(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, "
|
|
"%d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, "
|
|
"'%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')(%ld, %d, '%s')",
|
|
index, startTs, i, pVarchar, startTs + 1, i + 1, pVarchar, startTs + 2, i + 2, pVarchar, startTs + 3, i + 3, pVarchar, startTs + 4, i + 4,
|
|
pVarchar, startTs + 5, i + 5, pVarchar, startTs + 6, i + 6, pVarchar, startTs + 7, i + 7, pVarchar, startTs + 8, i + 8, pVarchar, startTs + 9, i + 9,
|
|
pVarchar, startTs + 10, i + 10, pVarchar, startTs + 11, i + 11, pVarchar, startTs + 12, i + 12, pVarchar, startTs + 13, i + 13, pVarchar, startTs + 14,
|
|
i + 14, pVarchar, startTs + 15, i + 15, pVarchar, startTs + 16, i + 16, pVarchar, startTs + 17, i + 17, pVarchar, startTs + 18, i + 18,
|
|
pVarchar, startTs + 19, i + 19, pVarchar);
|
|
TAOS_RES* p = taos_query(pConn, sql);
|
|
if (taos_errno(p) != 0) {
|
|
(void)printf("failed to insert data, reason:%s\n", taos_errstr(p));
|
|
}
|
|
|
|
// startTs += 20;
|
|
taos_free_result(p);
|
|
}
|
|
}
|
|
}
|
|
|
|
void* queryThread(void* arg) {
|
|
TAOS* pConn = taos_connect("192.168.0.209", "root", "taosdata", NULL, 0);
|
|
if (pConn == NULL) {
|
|
(void)printf("failed to connect to db, reason:%s", taos_errstr(pConn));
|
|
return NULL;
|
|
}
|
|
|
|
int64_t el = 0;
|
|
|
|
for (int32_t i = 0; i < 5; ++i) {
|
|
int64_t st = taosGetTimestampUs();
|
|
TAOS_RES* pRes = taos_query(pConn,
|
|
"SELECT _wstart as ts,max(usage_user) FROM benchmarkcpu.host_49 WHERE ts >= "
|
|
"1451618560000 AND ts < 1451622160000 INTERVAL(1m) ;");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed, reason:%s\n", taos_errstr(pRes));
|
|
} else {
|
|
printResult(pRes);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
el += (taosGetTimestampUs() - st);
|
|
if (i % 1000 == 0 && i != 0) {
|
|
(void)printf("total:%d, avg time:%.2fms\n", i, el / (double)(i * 1000));
|
|
}
|
|
}
|
|
|
|
taos_close(pConn);
|
|
return NULL;
|
|
}
|
|
|
|
int32_t numOfThreads = 1;
|
|
|
|
void tmq_commit_cb_print(tmq_t* pTmq, int32_t code, void* param) {
|
|
// (void)printf("auto commit success, code:%d\n", code);
|
|
}
|
|
|
|
void* doConsumeData(void* param) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
tmq_conf_t* conf = tmq_conf_new();
|
|
(void)tmq_conf_set(conf, "enable.auto.commit", "true");
|
|
(void)tmq_conf_set(conf, "auto.commit.interval.ms", "1000");
|
|
(void)tmq_conf_set(conf, "group.id", "cgrpName41");
|
|
(void)tmq_conf_set(conf, "td.connect.user", "root");
|
|
(void)tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
|
(void)tmq_conf_set(conf, "auto.offset.reset", "earliest");
|
|
(void)tmq_conf_set(conf, "experimental.snapshot.enable", "true");
|
|
(void)tmq_conf_set(conf, "msg.with.table.name", "true");
|
|
(void)tmq_conf_set_auto_commit_cb(conf, tmq_commit_cb_print, NULL);
|
|
|
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
|
tmq_conf_destroy(conf);
|
|
|
|
// 创建订阅 topics 列表
|
|
tmq_list_t* topicList = tmq_list_new();
|
|
(void)tmq_list_append(topicList, "topic_t2");
|
|
|
|
// 启动订阅
|
|
(void)tmq_subscribe(tmq, topicList);
|
|
|
|
tmq_list_destroy(topicList);
|
|
|
|
TAOS_FIELD* fields = NULL;
|
|
int32_t numOfFields = 0;
|
|
int32_t precision = 0;
|
|
int32_t totalRows = 0;
|
|
int32_t msgCnt = 0;
|
|
int32_t timeout = 25000;
|
|
|
|
int32_t count = 0;
|
|
|
|
while (1) {
|
|
TAOS_RES* pRes = tmq_consumer_poll(tmq, timeout);
|
|
if (pRes) {
|
|
char buf[1024] = {0};
|
|
|
|
const char* topicName = tmq_get_topic_name(pRes);
|
|
const char* dbName = tmq_get_db_name(pRes);
|
|
int32_t vgroupId = tmq_get_vgroup_id(pRes);
|
|
|
|
(void)printf("topic: %s\n", topicName);
|
|
(void)printf("db: %s\n", dbName);
|
|
(void)printf("vgroup id: %d\n", vgroupId);
|
|
|
|
while (1) {
|
|
TAOS_ROW row = taos_fetch_row(pRes);
|
|
if (row == NULL) {
|
|
break;
|
|
}
|
|
|
|
fields = taos_fetch_fields(pRes);
|
|
numOfFields = taos_field_count(pRes);
|
|
precision = taos_result_precision(pRes);
|
|
(void)taos_print_row(buf, row, fields, numOfFields);
|
|
totalRows += 1;
|
|
// (void)printf("precision: %d, row content: %s\n", precision, buf);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return NULL;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
int main(int argc, char** argv) {
|
|
testing::InitGoogleTest(&argc, argv);
|
|
if (argc > 1) {
|
|
//numOfThreads = atoi(argv[1]);
|
|
int32_t code = taosStr2int32(argv[1], &numOfThreads);
|
|
if (code != 0) {
|
|
return code;
|
|
}
|
|
|
|
|
|
}
|
|
|
|
numOfThreads = TMAX(numOfThreads, 1);
|
|
(void)printf("the runing threads is:%d", numOfThreads);
|
|
|
|
return RUN_ALL_TESTS();
|
|
}
|
|
|
|
TEST(clientCase, driverInit_Test) {
|
|
// taosInitGlobalCfg();
|
|
// taos_init();
|
|
}
|
|
|
|
TEST(clientCase, connect_Test) {
|
|
taos_options(TSDB_OPTION_CONFIGDIR, "~/first/cfg");
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
if (pConn == NULL) {
|
|
(void)printf("failed to connect to server, reason:%s\n", taos_errstr(NULL));
|
|
}
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "drop database abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in drop db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_user_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "create user abc pass 'abc'");
|
|
if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to create user, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_account_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "create account aabc pass 'abc'");
|
|
if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to create user, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, drop_account_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "drop account aabc");
|
|
if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to create user, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, show_user_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "show users");
|
|
TAOS_ROW pRow = NULL;
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
ASSERT_NE(pFields, nullptr);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, drop_user_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "drop user abc");
|
|
if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to create user, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, show_db_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "show databases");
|
|
TAOS_ROW pRow = NULL;
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
ASSERT_NE(pFields, nullptr);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
}
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_db_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "create database abc1 vgroups 2");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in create db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
ASSERT_TRUE(pFields == NULL);
|
|
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
ASSERT_EQ(numOfFields, 0);
|
|
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create database abc1 vgroups 4");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in create db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_dnode_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "create dnode abc1 port 7000");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in create dnode, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create dnode 1.1.1.1 port 9000");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create dnode, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, drop_dnode_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "drop dnode 3");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in drop dnode, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
ASSERT_TRUE(pFields == NULL);
|
|
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
ASSERT_EQ(numOfFields, 0);
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "drop dnode 4");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in drop dnode, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, use_db_test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in use db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
ASSERT_TRUE(pFields == NULL);
|
|
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
ASSERT_EQ(numOfFields, 0);
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_stable_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "create database if not exists abc1 vgroups 2");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in create db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "use abc1");
|
|
while (taos_errno(pRes) == TSDB_CODE_MND_DB_IN_CREATING || taos_errno(pRes) == TSDB_CODE_MND_DB_IN_DROPPING) {
|
|
taosMsleep(2000);
|
|
pRes = taos_query(pConn, "use abc1");
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table if not exists abc1.st1(ts timestamp, k int) tags(a int)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("error in create stable, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
ASSERT_TRUE(pFields == NULL);
|
|
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
ASSERT_EQ(numOfFields, 0);
|
|
taos_free_result(pRes);
|
|
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_table_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table if not exists tm0(ts timestamp, k int)");
|
|
ASSERT_EQ(taos_errno(pRes), 0);
|
|
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table if not exists tm0(ts timestamp, k blob)");
|
|
ASSERT_NE(taos_errno(pRes), 0);
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_ctable_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to use db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create stable if not exists st1 (ts timestamp, k int ) tags(a int)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create stable, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table tu using st1 tags('2021-10-10 1:1:1');");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create child table tm0, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, show_stable_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "show abc1.stables");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to show stables, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, show_vgroup_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to use db, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "show vgroups");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to show vgroups, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, create_multiple_tables) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "create database if not exists abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create db, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
return;
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "use abc1");
|
|
while (taos_errno(pRes) == TSDB_CODE_MND_DB_IN_CREATING || taos_errno(pRes) == TSDB_CODE_MND_DB_IN_DROPPING) {
|
|
taosMsleep(2000);
|
|
pRes = taos_query(pConn, "use abc1");
|
|
}
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to use db, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
return;
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create stable if not exists st1 (ts timestamp, k int) tags(a int)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create stable tables, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table if not exists t_2 using st1 tags(1)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create multiple tables, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
pRes = taos_query(pConn, "create table if not exists t_3 using st1 tags(2)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create multiple tables, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
|
|
for (int32_t i = 0; i < 500; i += 2) {
|
|
char sql[512] = {0};
|
|
(void)snprintf(sql, tListLen(sql), "create table t_x_%d using st1 tags(2) t_x_%d using st1 tags(5)", i, i + 1);
|
|
TAOS_RES* pres = taos_query(pConn, sql);
|
|
if (taos_errno(pres) != 0) {
|
|
(void)printf("failed to create table %d\n, reason:%s", i, taos_errstr(pres));
|
|
}
|
|
taos_free_result(pres);
|
|
}
|
|
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, show_table_Test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "show tables");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to show tables, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "use abc1");
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "show tables");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to show tables, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
}
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
int32_t count = 0;
|
|
char str[512] = {0};
|
|
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
if(code > 0) {
|
|
(void)printf("%d: %s\n", ++count, str);
|
|
}
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, generated_request_id_test) {
|
|
SHashObj* phash = taosHashInit(10000, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_ENTRY_LOCK);
|
|
|
|
for (int32_t i = 0; i < 50000; ++i) {
|
|
uint64_t v = generateRequestId();
|
|
void* result = taosHashGet(phash, &v, sizeof(v));
|
|
if (result != nullptr) {
|
|
// (void)printf("0x%llx, index:%d\n", v, i);
|
|
}
|
|
ASSERT_EQ(result, nullptr);
|
|
(void)taosHashPut(phash, &v, sizeof(v), NULL, 0);
|
|
}
|
|
|
|
taosHashCleanup(phash);
|
|
}
|
|
|
|
TEST(clientCase, insert_test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "insert into t_2 values(now, 1)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create into table t_2, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, projection_query_tables) {
|
|
taos_options(TSDB_OPTION_CONFIGDIR, "/home/lisa/first/cfg");
|
|
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = NULL;
|
|
|
|
pRes= taos_query(pConn, "use abc1");
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "select forecast(k,'algo=arima,wncheck=0') from t1 where ts<='2024-11-15 1:7:44'");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create table tu, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table tu using st2 tags(2)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create table tu, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table st2 (ts timestamp, v1 int) tags(t1 int)");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create table tu, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
int64_t start = 1685959190000;
|
|
const char* pstr =
|
|
"abcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefgh"
|
|
"ijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnop"
|
|
"qrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwx"
|
|
"yzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdef"
|
|
"ghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyzabcdefghijklmnopqrstuvwxyz!@#$%^&&*&^^%$#@!qQWERTYUIOPASDFGHJKL:"
|
|
"QWERTYUIOP{}";
|
|
|
|
for(int32_t i = 0; i < 1; ++i) {
|
|
char str[1024] = {0};
|
|
(void)sprintf(str, "create table if not exists tu%d using st2 tags(%d)", i, i);
|
|
|
|
TAOS_RES* px = taos_query(pConn, str);
|
|
if (taos_errno(px) != 0) {
|
|
(void)printf("failed to create table tu, reason:%s\n", taos_errstr(px));
|
|
}
|
|
taos_free_result(px);
|
|
}
|
|
|
|
for(int32_t j = 0; j < 1; ++j) {
|
|
start += 20;
|
|
for (int32_t i = 0; i < 1; ++i) {
|
|
createNewTable(pConn, i, 10000, 0, pstr);
|
|
}
|
|
}
|
|
|
|
pRes = taos_query(pConn, "select * from abc1.st2");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to select from table, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
// int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
// (void)printf("%s\n", str);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, tsbs_perf_test) {
|
|
TdThread qid[20] = {0};
|
|
|
|
for (int32_t i = 0; i < numOfThreads; ++i) {
|
|
(void)taosThreadCreate(&qid[i], NULL, queryThread, NULL);
|
|
}
|
|
for (int32_t i = 0; i < numOfThreads; ++i) {
|
|
(void)taosThreadJoin(qid[i], NULL);
|
|
}
|
|
}
|
|
|
|
TEST(clientCase, projection_query_stables) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "explain select * from dbvg.st where tbname='ct1'");
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
int32_t numOfRows = 0;
|
|
int32_t i = 0;
|
|
int32_t prev = 0;
|
|
|
|
char str[512] = {0};
|
|
while (1) {
|
|
pRow = taos_fetch_row(pRes);
|
|
if (pRow == NULL) {
|
|
break;
|
|
}
|
|
i += numOfRows;
|
|
|
|
if ( (i / 1000000) > prev) {
|
|
(void)printf("%d\n", i);
|
|
prev = i/1000000;
|
|
}
|
|
//(void)printf("%d\n", i);
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, agg_query_tables) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to use db, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
ASSERT_TRUE(false);
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
int64_t st = 1685959293299;
|
|
for (int32_t i = 0; i < 5; ++i) {
|
|
char s[256] = {0};
|
|
|
|
while (1) {
|
|
(void)sprintf(s, "insert into t1 values(%ld, %d)", st + i, i);
|
|
pRes = taos_query(pConn, s);
|
|
|
|
int32_t ret = taos_errno(pRes);
|
|
|
|
if (ret != 0) {
|
|
(void)printf("failed to insert into table, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
taos_free_result(pRes);
|
|
break;
|
|
}
|
|
}
|
|
taos_close(pConn);
|
|
}
|
|
|
|
// --- copy the following script in the shell to setup the environment ---
|
|
//
|
|
// create database test;
|
|
// use test;
|
|
// create table m1(ts timestamp, k int) tags(a int);
|
|
// create table tm0 using m1 tags(1);
|
|
// create table tm1 using m1 tags(2);
|
|
// insert into tm0 values('2021-1-1 1:1:1.120', 1) ('2021-1-1 1:1:2.9', 2) tm1 values('2021-1-1 1:1:1.120', 11) ('2021-1-1
|
|
// 1:1:2.99', 22);
|
|
|
|
TEST(clientCase, async_api_test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "use abc1");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to use db, reason:%s\n", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
return;
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "insert into tu(ts) values('2022-02-27 12:12:61')");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed, reason:%s\n", taos_errstr(pRes));
|
|
}
|
|
|
|
int32_t n = 0;
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
int32_t* length = taos_fetch_lengths(pRes);
|
|
for (int32_t i = 0; i < numOfFields; ++i) {
|
|
(void)printf("(%d):%d ", i, length[i]);
|
|
}
|
|
(void)printf("\n");
|
|
|
|
int32_t code = taos_print_row(str, pRow, pFields, numOfFields);
|
|
(void)printf("%s\n", str);
|
|
(void)memset(str, 0, sizeof(str));
|
|
}
|
|
taos_free_result(pRes);
|
|
tsem_t sem;
|
|
(void)tsem_init(&sem, 0, 0);
|
|
taos_query_a(pConn, "select count(*) from tu", queryCallback, &sem);
|
|
tsem_wait(&sem);
|
|
tsem_destroy(&sem);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, update_test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "select cast(0 as timestamp)-1y");
|
|
if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to create database, code:%s", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
return;
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "use abc1");
|
|
if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
(void)printf("failed to use db, code:%s", taos_errstr(pRes));
|
|
taos_free_result(pRes);
|
|
taos_close(pConn);
|
|
return;
|
|
}
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table tup (ts timestamp, k int);");
|
|
if (taos_errno(pRes) != 0) {
|
|
(void)printf("failed to create table, reason:%s", taos_errstr(pRes));
|
|
}
|
|
|
|
taos_free_result(pRes);
|
|
|
|
char s[256] = {0};
|
|
for (int32_t i = 0; i < 10; ++i) {
|
|
(void)sprintf(s, "insert into tup values(now+%da, %d)", i, i);
|
|
pRes = taos_query(pConn, s);
|
|
taos_free_result(pRes);
|
|
}
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, sub_db_test) {
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
// TAOS_RES* pRes = taos_query(pConn, "create topic topic_t1 as select * from t1");
|
|
// if (taos_errno(pRes) != TSDB_CODE_SUCCESS) {
|
|
// (void)printf("failed to create topic, code:%s", taos_errstr(pRes));
|
|
// taos_free_result(pRes);
|
|
// return;
|
|
// }
|
|
|
|
tmq_conf_t* conf = tmq_conf_new();
|
|
(void)tmq_conf_set(conf, "enable.auto.commit", "true");
|
|
(void)tmq_conf_set(conf, "auto.commit.interval.ms", "1000");
|
|
(void)tmq_conf_set(conf, "group.id", "cgrpNamedb");
|
|
(void)tmq_conf_set(conf, "td.connect.user", "root");
|
|
(void)tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
|
(void)tmq_conf_set(conf, "auto.offset.reset", "earliest");
|
|
(void)tmq_conf_set(conf, "experimental.snapshot.enable", "false");
|
|
(void)tmq_conf_set(conf, "msg.with.table.name", "true");
|
|
(void)tmq_conf_set_auto_commit_cb(conf, tmq_commit_cb_print, NULL);
|
|
|
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
|
tmq_conf_destroy(conf);
|
|
|
|
// 创建订阅 topics 列表
|
|
tmq_list_t* topicList = tmq_list_new();
|
|
(void)tmq_list_append(topicList, "topic_t1");
|
|
// tmq_list_append(topicList, "topic_s2");
|
|
|
|
// 启动订阅
|
|
(void)tmq_subscribe(tmq, topicList);
|
|
tmq_list_destroy(topicList);
|
|
|
|
TAOS_FIELD* fields = NULL;
|
|
int32_t numOfFields = 0;
|
|
int32_t precision = 0;
|
|
int32_t totalRows = 0;
|
|
int32_t msgCnt = 0;
|
|
int32_t timeout = 5000;
|
|
|
|
int32_t count = 0;
|
|
|
|
while (1) {
|
|
TAOS_RES* pRes = tmq_consumer_poll(tmq, timeout);
|
|
if (pRes) {
|
|
char buf[1024];
|
|
int32_t rows = 0;
|
|
|
|
const char* topicName = tmq_get_topic_name(pRes);
|
|
const char* dbName = tmq_get_db_name(pRes);
|
|
int32_t vgroupId = tmq_get_vgroup_id(pRes);
|
|
|
|
(void)printf("topic: %s\n", topicName);
|
|
(void)printf("db: %s\n", dbName);
|
|
(void)printf("vgroup id: %d\n", vgroupId);
|
|
|
|
if (count++ > 200) {
|
|
(void)tmq_unsubscribe(tmq);
|
|
break;
|
|
}
|
|
|
|
while (1) {
|
|
TAOS_ROW row = taos_fetch_row(pRes);
|
|
if (row == NULL) break;
|
|
|
|
fields = taos_fetch_fields(pRes);
|
|
ASSERT_NE(fields, nullptr);
|
|
numOfFields = taos_field_count(pRes);
|
|
precision = taos_result_precision(pRes);
|
|
rows++;
|
|
(void)taos_print_row(buf, row, fields, numOfFields);
|
|
(void)printf("precision: %d, row content: %s\n", precision, buf);
|
|
}
|
|
taos_free_result(pRes);
|
|
} else {
|
|
break;
|
|
}
|
|
}
|
|
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
taos_close(pConn);
|
|
}
|
|
|
|
TEST(clientCase, tmq_commit) {
|
|
// taos_options(TSDB_OPTION_CONFIGDIR, "~/first/cfg");
|
|
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
tmq_conf_t* conf = tmq_conf_new();
|
|
|
|
(void)tmq_conf_set(conf, "enable.auto.commit", "false");
|
|
(void)tmq_conf_set(conf, "auto.commit.interval.ms", "2000");
|
|
(void)tmq_conf_set(conf, "group.id", "group_id_2");
|
|
(void)tmq_conf_set(conf, "td.connect.user", "root");
|
|
(void)tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
|
(void)tmq_conf_set(conf, "auto.offset.reset", "earliest");
|
|
(void)tmq_conf_set(conf, "msg.with.table.name", "true");
|
|
|
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
|
tmq_conf_destroy(conf);
|
|
|
|
char topicName[128] = "tp";
|
|
// 创建订阅 topics 列表
|
|
tmq_list_t* topicList = tmq_list_new();
|
|
(void)tmq_list_append(topicList, topicName);
|
|
|
|
// 启动订阅
|
|
(void)tmq_subscribe(tmq, topicList);
|
|
tmq_list_destroy(topicList);
|
|
|
|
int32_t totalRows = 0;
|
|
int32_t msgCnt = 0;
|
|
int32_t timeout = 2000;
|
|
|
|
tmq_topic_assignment* pAssign = NULL;
|
|
int32_t numOfAssign = 0;
|
|
|
|
int32_t code = tmq_get_topic_assignment(tmq, topicName, &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
for(int i = 0; i < numOfAssign; i++){
|
|
tmq_topic_assignment* pa = &pAssign[i];
|
|
std::cout << "assign i:" << i << ", vgId:" << pa->vgId << ", offset:" << pa->currentOffset << ", start:%"
|
|
<< pa->begin << ", end:%" << pa->end << std::endl;
|
|
|
|
int64_t committed = tmq_committed(tmq, topicName, pa->vgId);
|
|
std::cout << "committed vgId:" << pa->vgId << " committed:" << committed << std::endl;
|
|
|
|
int64_t position = tmq_position(tmq, topicName, pa->vgId);
|
|
std::cout << "position vgId:" << pa->vgId << ", position:" << position << std::endl;
|
|
|
|
(void)tmq_offset_seek(tmq, topicName, pa->vgId, 1);
|
|
position = tmq_position(tmq, topicName, pa->vgId);
|
|
std::cout << "after seek 1, position vgId:" << pa->vgId << " position:" << position << std::endl;
|
|
}
|
|
|
|
while (1) {
|
|
(void)printf("start to poll\n");
|
|
TAOS_RES* pRes = tmq_consumer_poll(tmq, timeout);
|
|
if (pRes) {
|
|
printSubResults(pRes, &totalRows);
|
|
} else {
|
|
break;
|
|
}
|
|
|
|
(void)tmq_commit_sync(tmq, pRes);
|
|
for(int i = 0; i < numOfAssign; i++) {
|
|
int64_t committed = tmq_committed(tmq, topicName, pAssign[i].vgId);
|
|
std::cout << "committed vgId:" << pAssign[i].vgId << " , committed:" << committed << std::endl;
|
|
if(committed > 0){
|
|
int32_t code = tmq_commit_offset_sync(tmq, topicName, pAssign[i].vgId, 4);
|
|
(void)printf("tmq_commit_offset_sync vgId:%d, offset:4, code:%d\n", pAssign[i].vgId, code);
|
|
int64_t committed = tmq_committed(tmq, topicName, pAssign[i].vgId);
|
|
|
|
std::cout << "after tmq_commit_offset_sync, committed vgId:" << pAssign[i].vgId << ", committed:" << committed
|
|
<< std::endl;
|
|
}
|
|
}
|
|
if (pRes != NULL) {
|
|
taos_free_result(pRes);
|
|
}
|
|
|
|
// tmq_offset_seek(tmq, "tp", pAssign[0].vgId, pAssign[0].begin);
|
|
}
|
|
|
|
tmq_free_assignment(pAssign);
|
|
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
}
|
|
namespace {
|
|
void doPrintInfo(tmq_topic_assignment* pa, int32_t index) {
|
|
std::cout << "assign i:" << index << ", vgId:" << pa->vgId << ", offset:%" << pa->currentOffset << ", start:%"
|
|
<< pa->begin << ", end:%" << pa->end << std::endl;
|
|
}
|
|
}
|
|
TEST(clientCase, td_25129) {
|
|
// taos_options(TSDB_OPTION_CONFIGDIR, "~/first/cfg");
|
|
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
tmq_conf_t* conf = tmq_conf_new();
|
|
|
|
(void)tmq_conf_set(conf, "enable.auto.commit", "false");
|
|
(void)tmq_conf_set(conf, "auto.commit.interval.ms", "2000");
|
|
(void)tmq_conf_set(conf, "group.id", "group_id_2");
|
|
(void)tmq_conf_set(conf, "td.connect.user", "root");
|
|
(void)tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
|
(void)tmq_conf_set(conf, "auto.offset.reset", "earliest");
|
|
(void)tmq_conf_set(conf, "msg.with.table.name", "true");
|
|
|
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
|
ASSERT_NE(tmq, nullptr);
|
|
tmq_conf_destroy(conf);
|
|
|
|
char topicName[128] = "tp";
|
|
// 创建订阅 topics 列表
|
|
tmq_list_t* topicList = tmq_list_new();
|
|
(void)tmq_list_append(topicList, topicName);
|
|
|
|
// 启动订阅
|
|
(void)tmq_subscribe(tmq, topicList);
|
|
tmq_list_destroy(topicList);
|
|
|
|
TAOS_FIELD* fields = NULL;
|
|
int32_t numOfFields = 0;
|
|
int32_t precision = 0;
|
|
int32_t totalRows = 0;
|
|
int32_t msgCnt = 0;
|
|
int32_t timeout = 2000;
|
|
|
|
int32_t count = 0;
|
|
|
|
tmq_topic_assignment* pAssign = NULL;
|
|
int32_t numOfAssign = 0;
|
|
|
|
int32_t code = tmq_get_topic_assignment(tmq, topicName, &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
for(int i = 0; i < numOfAssign; i++){
|
|
doPrintInfo(&pAssign[i], i);
|
|
}
|
|
|
|
// tmq_offset_seek(tmq, "tp", pAssign[0].vgId, 4);
|
|
tmq_free_assignment(pAssign);
|
|
|
|
code = tmq_get_topic_assignment(tmq, topicName, &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
for(int i = 0; i < numOfAssign; i++){
|
|
doPrintInfo(&pAssign[i], i);
|
|
}
|
|
|
|
tmq_free_assignment(pAssign);
|
|
|
|
code = tmq_get_topic_assignment(tmq, topicName, &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
for(int i = 0; i < numOfAssign; i++){
|
|
int64_t committed = tmq_committed(tmq, topicName, pAssign[i].vgId);
|
|
doPrintInfo(&pAssign[i], i);
|
|
}
|
|
|
|
while (1) {
|
|
(void)printf("start to poll\n");
|
|
TAOS_RES* pRes = tmq_consumer_poll(tmq, timeout);
|
|
if (pRes) {
|
|
char buf[128];
|
|
|
|
const char* topicName = tmq_get_topic_name(pRes);
|
|
// const char* dbName = tmq_get_db_name(pRes);
|
|
// int32_t vgroupId = tmq_get_vgroup_id(pRes);
|
|
//
|
|
// (void)printf("topic: %s\n", topicName);
|
|
// (void)printf("db: %s\n", dbName);
|
|
// (void)printf("vgroup id: %d\n", vgroupId);
|
|
|
|
printSubResults(pRes, &totalRows);
|
|
|
|
code = tmq_get_topic_assignment(tmq, topicName, &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
for(int i = 0; i < numOfAssign; i++){
|
|
doPrintInfo(&pAssign[i], i);
|
|
}
|
|
} else {
|
|
for(int i = 0; i < numOfAssign; i++) {
|
|
(void)tmq_offset_seek(tmq, topicName, pAssign[i].vgId, pAssign[i].currentOffset);
|
|
}
|
|
(void)tmq_commit_sync(tmq, pRes);
|
|
break;
|
|
}
|
|
|
|
// tmq_commit_sync(tmq, pRes);
|
|
if (pRes != NULL) {
|
|
taos_free_result(pRes);
|
|
// if ((++count) > 1) {
|
|
// break;
|
|
// }
|
|
} else {
|
|
break;
|
|
}
|
|
|
|
// tmq_offset_seek(tmq, "tp", pAssign[0].vgId, pAssign[0].begin);
|
|
}
|
|
|
|
tmq_free_assignment(pAssign);
|
|
|
|
code = tmq_get_topic_assignment(tmq, "tp", &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
for(int i = 0; i < numOfAssign; i++){
|
|
doPrintInfo(&pAssign[i], i);
|
|
}
|
|
|
|
tmq_free_assignment(pAssign);
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
}
|
|
|
|
TEST(clientCase, sub_tb_test) {
|
|
(void)taos_options(TSDB_OPTION_CONFIGDIR, "~/first/cfg");
|
|
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
tmq_conf_t* conf = tmq_conf_new();
|
|
|
|
int32_t ts = taosGetTimestampMs()%INT32_MAX;
|
|
char consumerGroupid[128] = {0};
|
|
(void)sprintf(consumerGroupid, "group_id_%d", ts);
|
|
|
|
(void)tmq_conf_set(conf, "enable.auto.commit", "true");
|
|
(void)tmq_conf_set(conf, "auto.commit.interval.ms", "2000");
|
|
(void)tmq_conf_set(conf, "group.id", consumerGroupid);
|
|
(void)tmq_conf_set(conf, "td.connect.user", "root");
|
|
(void)tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
|
(void)tmq_conf_set(conf, "auto.offset.reset", "earliest");
|
|
(void)tmq_conf_set(conf, "experimental.snapshot.enable", "false");
|
|
(void)tmq_conf_set(conf, "msg.with.table.name", "true");
|
|
(void)tmq_conf_set_auto_commit_cb(conf, tmq_commit_cb_print, NULL);
|
|
|
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
|
ASSERT_NE(tmq, nullptr);
|
|
tmq_conf_destroy(conf);
|
|
|
|
// 创建订阅 topics 列表
|
|
tmq_list_t* topicList = tmq_list_new();
|
|
(void)tmq_list_append(topicList, "t1");
|
|
|
|
// 启动订阅
|
|
(void)tmq_subscribe(tmq, topicList);
|
|
tmq_list_destroy(topicList);
|
|
|
|
TAOS_FIELD* fields = NULL;
|
|
int32_t numOfFields = 0;
|
|
int32_t precision = 0;
|
|
int32_t totalRows = 0;
|
|
int32_t msgCnt = 0;
|
|
int32_t timeout = 2500000;
|
|
|
|
int32_t count = 0;
|
|
|
|
tmq_topic_assignment* pAssign = NULL;
|
|
int32_t numOfAssign = 0;
|
|
|
|
int32_t code = tmq_get_topic_assignment(tmq, "t1", &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
(void)tmq_offset_seek(tmq, "t1", pAssign[0].vgId, 4);
|
|
|
|
code = tmq_get_topic_assignment(tmq, "t1", &pAssign, &numOfAssign);
|
|
if (code != 0) {
|
|
(void)printf("error occurs:%s\n", tmq_err2str(code));
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
return;
|
|
}
|
|
|
|
while (1) {
|
|
TAOS_RES* pRes = tmq_consumer_poll(tmq, timeout);
|
|
if (pRes) {
|
|
char buf[128] = {0};
|
|
|
|
const char* topicName = tmq_get_topic_name(pRes);
|
|
// const char* dbName = tmq_get_db_name(pRes);
|
|
// int32_t vgroupId = tmq_get_vgroup_id(pRes);
|
|
//
|
|
// (void)printf("topic: %s\n", topicName);
|
|
// (void)printf("db: %s\n", dbName);
|
|
// (void)printf("vgroup id: %d\n", vgroupId);
|
|
|
|
printSubResults(pRes, &totalRows);
|
|
} else {
|
|
// tmq_offset_seek(tmq, "topic_t1", pAssign[0].vgroupHandle, pAssign[0].begin);
|
|
// break;
|
|
}
|
|
|
|
(void)tmq_commit_sync(tmq, pRes);
|
|
if (pRes != NULL) {
|
|
taos_free_result(pRes);
|
|
// if ((++count) > 1) {
|
|
// break;
|
|
// }
|
|
} else {
|
|
break;
|
|
}
|
|
|
|
(void)tmq_offset_seek(tmq, "topic_t1", pAssign[0].vgId, pAssign[0].begin);
|
|
}
|
|
|
|
(void)tmq_consumer_close(tmq);
|
|
taos_close(pConn);
|
|
(void)fprintf(stderr, "%d msg consumed, include %d rows\n", msgCnt, totalRows);
|
|
}
|
|
|
|
TEST(clientCase, sub_tb_mt_test) {
|
|
char *user = NULL;
|
|
char *auth = NULL;
|
|
char *ip = NULL;
|
|
int port = 0;
|
|
char key[512] = {0};
|
|
(void)snprintf(key, sizeof(key), "%s:%s:%s:%d", user, auth, ip, port);
|
|
|
|
(void)taos_options(TSDB_OPTION_CONFIGDIR, "~/first/cfg");
|
|
TdThread qid[20] = {0};
|
|
|
|
for (int32_t i = 0; i < 1; ++i) {
|
|
(void)taosThreadCreate(&qid[i], NULL, doConsumeData, NULL);
|
|
}
|
|
|
|
for (int32_t i = 0; i < 1; ++i) {
|
|
(void)taosThreadJoin(qid[i], NULL);
|
|
}
|
|
}
|
|
|
|
TEST(clientCase, timezone_Test) {
|
|
{
|
|
// taos_options( TSDB_OPTION_TIMEZONE, "UTC-8");
|
|
int code = taos_options(TSDB_OPTION_TIMEZONE, "UTC-8");
|
|
ASSERT_TRUE(code == 0);
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "drop database if exists db_timezone");
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create database db_timezone");
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "create table db_timezone.t1 (ts timestamp, v int)");
|
|
while (taos_errno(pRes) == TSDB_CODE_MND_DB_IN_CREATING || taos_errno(pRes) == TSDB_CODE_MND_DB_IN_DROPPING) {
|
|
taosMsleep(2000);
|
|
pRes = taos_query(pConn, "create table db_timezone.t1 (ts timestamp, v int)");
|
|
}
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
taos_free_result(pRes);
|
|
|
|
char sql[256] = {0};
|
|
(void)sprintf(sql, "insert into db_timezone.t1 values('2023-09-16 17:00:00', 1)");
|
|
pRes = taos_query(pConn, sql);
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
taos_free_result(pRes);
|
|
|
|
pRes = taos_query(pConn, "select * from db_timezone.t1 where ts == '2023-09-16 17:00:00'");
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
char str[512] = {0};
|
|
int rows = 0;
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
rows++;
|
|
}
|
|
ASSERT_TRUE(rows == 1);
|
|
|
|
taos_free_result(pRes);
|
|
|
|
taos_close(pConn);
|
|
}
|
|
|
|
{
|
|
// taos_options( TSDB_OPTION_TIMEZONE, "UTC+8");
|
|
int code = taos_options(TSDB_OPTION_TIMEZONE, "UTC+8");
|
|
ASSERT_TRUE(code == 0);
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "select * from db_timezone.t1 where ts == '2023-09-16 01:00:00'");
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
int rows = 0;
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
rows++;
|
|
}
|
|
ASSERT_TRUE(rows == 1);
|
|
|
|
taos_free_result(pRes);
|
|
|
|
char sql[256] = {0};
|
|
(void)sprintf(sql, "insert into db_timezone.t1 values('2023-09-16 17:00:01', 1)");
|
|
pRes = taos_query(pConn, sql);
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
|
|
taos_free_result(pRes);
|
|
|
|
taos_close(pConn);
|
|
}
|
|
|
|
{
|
|
// taos_options( TSDB_OPTION_TIMEZONE, "UTC+0");
|
|
int code = taos_options(TSDB_OPTION_TIMEZONE, "UTC+0");
|
|
ASSERT_TRUE(code == 0);
|
|
TAOS* pConn = taos_connect("localhost", "root", "taosdata", NULL, 0);
|
|
ASSERT_NE(pConn, nullptr);
|
|
|
|
TAOS_RES* pRes = taos_query(pConn, "select * from db_timezone.t1 where ts == '2023-09-16 09:00:00'");
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
int rows = 0;
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
rows++;
|
|
}
|
|
ASSERT_TRUE(rows == 1);
|
|
taos_free_result(pRes);
|
|
|
|
{
|
|
TAOS_RES* pRes = taos_query(pConn, "select * from db_timezone.t1 where ts == '2023-09-17 01:00:01'");
|
|
ASSERT_EQ(taos_errno(pRes), TSDB_CODE_SUCCESS);
|
|
|
|
TAOS_ROW pRow = NULL;
|
|
TAOS_FIELD* pFields = taos_fetch_fields(pRes);
|
|
int32_t numOfFields = taos_num_fields(pRes);
|
|
|
|
int rows = 0;
|
|
char str[512] = {0};
|
|
while ((pRow = taos_fetch_row(pRes)) != NULL) {
|
|
rows++;
|
|
}
|
|
ASSERT_TRUE(rows == 1);
|
|
taos_free_result(pRes);
|
|
}
|
|
|
|
taos_close(pConn);
|
|
}
|
|
}
|
|
#pragma GCC diagnostic pop
|