Merge branch '3.0' into 3.0_udfd
This commit is contained in:
commit
eb1b3c4918
12
Jenkinsfile2
12
Jenkinsfile2
|
@ -88,6 +88,12 @@ def pre_test(){
|
||||||
cmake .. > /dev/null
|
cmake .. > /dev/null
|
||||||
make -j4> /dev/null
|
make -j4> /dev/null
|
||||||
'''
|
'''
|
||||||
|
sh'''
|
||||||
|
cd ${WKPY}
|
||||||
|
git reset --hard
|
||||||
|
git pull
|
||||||
|
pip3 install .
|
||||||
|
'''
|
||||||
return 1
|
return 1
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -97,6 +103,7 @@ pipeline {
|
||||||
environment{
|
environment{
|
||||||
WK = '/var/lib/jenkins/workspace/TDinternal'
|
WK = '/var/lib/jenkins/workspace/TDinternal'
|
||||||
WKC= '/var/lib/jenkins/workspace/TDengine'
|
WKC= '/var/lib/jenkins/workspace/TDengine'
|
||||||
|
WKPY= '/var/lib/jenkins/workspace/taos-connector-python'
|
||||||
}
|
}
|
||||||
stages {
|
stages {
|
||||||
stage('pre_build'){
|
stage('pre_build'){
|
||||||
|
@ -117,6 +124,11 @@ pipeline {
|
||||||
./test-all.sh b1fq
|
./test-all.sh b1fq
|
||||||
'''
|
'''
|
||||||
sh'''
|
sh'''
|
||||||
|
export LD_LIBRARY_PATH=${WKC}/debug/build/lib
|
||||||
|
cd ${WKC}/tests/system-test
|
||||||
|
./fulltest.sh
|
||||||
|
'''
|
||||||
|
sh'''
|
||||||
cd ${WKC}/debug
|
cd ${WKC}/debug
|
||||||
ctest
|
ctest
|
||||||
'''
|
'''
|
||||||
|
|
|
@ -141,7 +141,7 @@ int32_t create_topic() {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void tmq_commit_cb_print(tmq_t* tmq, tmq_resp_err_t resp, tmq_topic_vgroup_list_t* offsets, void* param) {
|
void tmq_commit_cb_print(tmq_t* tmq, tmq_resp_err_t resp, tmq_topic_vgroup_list_t* offsets) {
|
||||||
printf("commit %d\n", resp);
|
printf("commit %d\n", resp);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -163,7 +163,7 @@ tmq_t* build_consumer() {
|
||||||
tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
||||||
tmq_conf_set(conf, "td.connect.db", "abc1");
|
tmq_conf_set(conf, "td.connect.db", "abc1");
|
||||||
tmq_conf_set_offset_commit_cb(conf, tmq_commit_cb_print);
|
tmq_conf_set_offset_commit_cb(conf, tmq_commit_cb_print);
|
||||||
tmq_t* tmq = tmq_consumer_new1(conf, NULL, 0);
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
||||||
return tmq;
|
return tmq;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -189,7 +189,7 @@ void basic_consume_loop(tmq_t* tmq, tmq_list_t* topics) {
|
||||||
cnt++;
|
cnt++;
|
||||||
/*printf("get data\n");*/
|
/*printf("get data\n");*/
|
||||||
/*msg_process(tmqmessage);*/
|
/*msg_process(tmqmessage);*/
|
||||||
tmq_message_destroy(tmqmessage);
|
taos_free_result(tmqmessage);
|
||||||
/*} else {*/
|
/*} else {*/
|
||||||
/*break;*/
|
/*break;*/
|
||||||
}
|
}
|
||||||
|
@ -219,7 +219,7 @@ void sync_consume_loop(tmq_t* tmq, tmq_list_t* topics) {
|
||||||
TAOS_RES* tmqmessage = tmq_consumer_poll(tmq, 1000);
|
TAOS_RES* tmqmessage = tmq_consumer_poll(tmq, 1000);
|
||||||
if (tmqmessage) {
|
if (tmqmessage) {
|
||||||
msg_process(tmqmessage);
|
msg_process(tmqmessage);
|
||||||
tmq_message_destroy(tmqmessage);
|
taos_free_result(tmqmessage);
|
||||||
|
|
||||||
/*if ((++msg_count % MIN_COMMIT_COUNT) == 0) tmq_commit(tmq, NULL, 0);*/
|
/*if ((++msg_count % MIN_COMMIT_COUNT) == 0) tmq_commit(tmq, NULL, 0);*/
|
||||||
}
|
}
|
||||||
|
@ -249,7 +249,7 @@ void perf_loop(tmq_t* tmq, tmq_list_t* topics) {
|
||||||
batchCnt++;
|
batchCnt++;
|
||||||
/*skipLogNum += tmqGetSkipLogNum(tmqmessage);*/
|
/*skipLogNum += tmqGetSkipLogNum(tmqmessage);*/
|
||||||
/*msg_process(tmqmessage);*/
|
/*msg_process(tmqmessage);*/
|
||||||
tmq_message_destroy(tmqmessage);
|
taos_free_result(tmqmessage);
|
||||||
} else {
|
} else {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
|
@ -92,38 +92,14 @@ typedef struct taosField {
|
||||||
|
|
||||||
typedef void (*__taos_async_fn_t)(void *param, TAOS_RES *, int code);
|
typedef void (*__taos_async_fn_t)(void *param, TAOS_RES *, int code);
|
||||||
|
|
||||||
typedef struct TAOS_BIND {
|
typedef struct TAOS_BIND_v2 {
|
||||||
int buffer_type;
|
|
||||||
void *buffer;
|
|
||||||
uintptr_t buffer_length; // unused
|
|
||||||
uintptr_t *length;
|
|
||||||
int *is_null;
|
|
||||||
|
|
||||||
int is_unsigned; // unused
|
|
||||||
int *error; // unused
|
|
||||||
union {
|
|
||||||
int64_t ts;
|
|
||||||
int8_t b;
|
|
||||||
int8_t v1;
|
|
||||||
int16_t v2;
|
|
||||||
int32_t v4;
|
|
||||||
int64_t v8;
|
|
||||||
float f4;
|
|
||||||
double f8;
|
|
||||||
unsigned char *bin;
|
|
||||||
char *nchar;
|
|
||||||
} u;
|
|
||||||
unsigned int allocated;
|
|
||||||
} TAOS_BIND;
|
|
||||||
|
|
||||||
typedef struct TAOS_MULTI_BIND {
|
|
||||||
int buffer_type;
|
int buffer_type;
|
||||||
void *buffer;
|
void *buffer;
|
||||||
uintptr_t buffer_length;
|
int32_t buffer_length;
|
||||||
int32_t *length;
|
int32_t *length;
|
||||||
char *is_null;
|
char *is_null;
|
||||||
int num;
|
int num;
|
||||||
} TAOS_MULTI_BIND;
|
} TAOS_BIND_v2;
|
||||||
|
|
||||||
typedef enum {
|
typedef enum {
|
||||||
SET_CONF_RET_SUCC = 0,
|
SET_CONF_RET_SUCC = 0,
|
||||||
|
@ -152,34 +128,34 @@ DLL_EXPORT void taos_close(TAOS *taos);
|
||||||
|
|
||||||
const char *taos_data_type(int type);
|
const char *taos_data_type(int type);
|
||||||
|
|
||||||
DLL_EXPORT TAOS_STMT *taos_stmt_init(TAOS *taos);
|
DLL_EXPORT TAOS_STMT *taos_stmt_init(TAOS *taos);
|
||||||
DLL_EXPORT int taos_stmt_prepare(TAOS_STMT *stmt, const char *sql, unsigned long length);
|
DLL_EXPORT int taos_stmt_prepare(TAOS_STMT *stmt, const char *sql, unsigned long length);
|
||||||
DLL_EXPORT int taos_stmt_set_tbname_tags(TAOS_STMT *stmt, const char *name, TAOS_BIND *tags);
|
DLL_EXPORT int taos_stmt_set_tbname_tags(TAOS_STMT *stmt, const char *name, TAOS_BIND_v2 *tags);
|
||||||
DLL_EXPORT int taos_stmt_set_tbname(TAOS_STMT *stmt, const char *name);
|
DLL_EXPORT int taos_stmt_set_tbname(TAOS_STMT *stmt, const char *name);
|
||||||
DLL_EXPORT int taos_stmt_set_sub_tbname(TAOS_STMT *stmt, const char *name);
|
DLL_EXPORT int taos_stmt_set_sub_tbname(TAOS_STMT *stmt, const char *name);
|
||||||
|
|
||||||
DLL_EXPORT int taos_stmt_is_insert(TAOS_STMT *stmt, int *insert);
|
DLL_EXPORT int taos_stmt_is_insert(TAOS_STMT *stmt, int *insert);
|
||||||
DLL_EXPORT int taos_stmt_num_params(TAOS_STMT *stmt, int *nums);
|
DLL_EXPORT int taos_stmt_num_params(TAOS_STMT *stmt, int *nums);
|
||||||
DLL_EXPORT int taos_stmt_get_param(TAOS_STMT *stmt, int idx, int *type, int *bytes);
|
DLL_EXPORT int taos_stmt_get_param(TAOS_STMT *stmt, int idx, int *type, int *bytes);
|
||||||
DLL_EXPORT int taos_stmt_bind_param(TAOS_STMT *stmt, TAOS_BIND *bind);
|
DLL_EXPORT int taos_stmt_bind_param(TAOS_STMT *stmt, TAOS_BIND_v2 *bind);
|
||||||
DLL_EXPORT int taos_stmt_bind_param_batch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind);
|
DLL_EXPORT int taos_stmt_bind_param_batch(TAOS_STMT *stmt, TAOS_BIND_v2 *bind);
|
||||||
DLL_EXPORT int taos_stmt_bind_single_param_batch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind, int colIdx);
|
DLL_EXPORT int taos_stmt_bind_single_param_batch(TAOS_STMT *stmt, TAOS_BIND_v2 *bind, int colIdx);
|
||||||
DLL_EXPORT int taos_stmt_add_batch(TAOS_STMT *stmt);
|
DLL_EXPORT int taos_stmt_add_batch(TAOS_STMT *stmt);
|
||||||
DLL_EXPORT int taos_stmt_execute(TAOS_STMT *stmt);
|
DLL_EXPORT int taos_stmt_execute(TAOS_STMT *stmt);
|
||||||
DLL_EXPORT TAOS_RES *taos_stmt_use_result(TAOS_STMT *stmt);
|
DLL_EXPORT TAOS_RES *taos_stmt_use_result(TAOS_STMT *stmt);
|
||||||
DLL_EXPORT int taos_stmt_close(TAOS_STMT *stmt);
|
DLL_EXPORT int taos_stmt_close(TAOS_STMT *stmt);
|
||||||
DLL_EXPORT char *taos_stmt_errstr(TAOS_STMT *stmt);
|
DLL_EXPORT char *taos_stmt_errstr(TAOS_STMT *stmt);
|
||||||
DLL_EXPORT int taos_stmt_affected_rows(TAOS_STMT *stmt);
|
DLL_EXPORT int taos_stmt_affected_rows(TAOS_STMT *stmt);
|
||||||
|
|
||||||
DLL_EXPORT TAOS_RES *taos_query(TAOS *taos, const char *sql);
|
DLL_EXPORT TAOS_RES *taos_query(TAOS *taos, const char *sql);
|
||||||
DLL_EXPORT TAOS_RES *taos_query_l(TAOS *taos, const char *sql, int sqlLen);
|
DLL_EXPORT TAOS_RES *taos_query_l(TAOS *taos, const char *sql, int sqlLen);
|
||||||
|
|
||||||
DLL_EXPORT TAOS_ROW taos_fetch_row(TAOS_RES *res);
|
DLL_EXPORT TAOS_ROW taos_fetch_row(TAOS_RES *res);
|
||||||
DLL_EXPORT int taos_result_precision(TAOS_RES *res); // get the time precision of result
|
DLL_EXPORT int taos_result_precision(TAOS_RES *res); // get the time precision of result
|
||||||
DLL_EXPORT void taos_free_result(TAOS_RES *res);
|
DLL_EXPORT void taos_free_result(TAOS_RES *res);
|
||||||
DLL_EXPORT int taos_field_count(TAOS_RES *res);
|
DLL_EXPORT int taos_field_count(TAOS_RES *res);
|
||||||
DLL_EXPORT int taos_num_fields(TAOS_RES *res);
|
DLL_EXPORT int taos_num_fields(TAOS_RES *res);
|
||||||
DLL_EXPORT int taos_affected_rows(TAOS_RES *res);
|
DLL_EXPORT int taos_affected_rows(TAOS_RES *res);
|
||||||
|
|
||||||
DLL_EXPORT TAOS_FIELD *taos_fetch_fields(TAOS_RES *res);
|
DLL_EXPORT TAOS_FIELD *taos_fetch_fields(TAOS_RES *res);
|
||||||
DLL_EXPORT int taos_select_db(TAOS *taos, const char *db);
|
DLL_EXPORT int taos_select_db(TAOS *taos, const char *db);
|
||||||
|
@ -241,17 +217,17 @@ typedef struct tmq_conf_t tmq_conf_t;
|
||||||
typedef struct tmq_list_t tmq_list_t;
|
typedef struct tmq_list_t tmq_list_t;
|
||||||
// typedef struct tmq_message_t tmq_message_t;
|
// typedef struct tmq_message_t tmq_message_t;
|
||||||
|
|
||||||
typedef void(tmq_commit_cb(tmq_t *, tmq_resp_err_t, tmq_topic_vgroup_list_t *, void *param));
|
typedef void(tmq_commit_cb(tmq_t *, tmq_resp_err_t, tmq_topic_vgroup_list_t *));
|
||||||
|
|
||||||
DLL_EXPORT tmq_list_t *tmq_list_new();
|
DLL_EXPORT tmq_list_t *tmq_list_new();
|
||||||
DLL_EXPORT int32_t tmq_list_append(tmq_list_t *, const char *);
|
DLL_EXPORT int32_t tmq_list_append(tmq_list_t *, const char *);
|
||||||
DLL_EXPORT void tmq_list_destroy(tmq_list_t *);
|
DLL_EXPORT void tmq_list_destroy(tmq_list_t *);
|
||||||
|
|
||||||
#if 1
|
#if 0
|
||||||
DLL_EXPORT tmq_t *tmq_consumer_new(void *conn, tmq_conf_t *conf, char *errstr, int32_t errstrLen);
|
DLL_EXPORT tmq_t *tmq_consumer_new(void *conn, tmq_conf_t *conf, char *errstr, int32_t errstrLen);
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
DLL_EXPORT tmq_t *tmq_consumer_new1(tmq_conf_t *conf, char *errstr, int32_t errstrLen);
|
DLL_EXPORT tmq_t *tmq_consumer_new(tmq_conf_t *conf, char *errstr, int32_t errstrLen);
|
||||||
|
|
||||||
DLL_EXPORT const char *tmq_err2str(tmq_resp_err_t);
|
DLL_EXPORT const char *tmq_err2str(tmq_resp_err_t);
|
||||||
|
|
||||||
|
@ -295,14 +271,19 @@ int32_t tmqGetSkipLogNum(tmq_message_t *tmq_message);
|
||||||
|
|
||||||
DLL_EXPORT char *tmq_get_topic_name(TAOS_RES *res);
|
DLL_EXPORT char *tmq_get_topic_name(TAOS_RES *res);
|
||||||
DLL_EXPORT int32_t tmq_get_vgroup_id(TAOS_RES *res);
|
DLL_EXPORT int32_t tmq_get_vgroup_id(TAOS_RES *res);
|
||||||
|
// TODO
|
||||||
|
#if 0
|
||||||
|
DLL_EXPORT char *tmq_get_block_table_name(TAOS_RES *res);
|
||||||
|
#endif
|
||||||
|
|
||||||
#if 0
|
#if 0
|
||||||
DLL_EXPORT TAOS_ROW tmq_get_row(tmq_message_t *message);
|
DLL_EXPORT TAOS_ROW tmq_get_row(tmq_message_t *message);
|
||||||
DLL_EXPORT int64_t tmq_get_request_offset(tmq_message_t *message);
|
DLL_EXPORT int64_t tmq_get_request_offset(tmq_message_t *message);
|
||||||
DLL_EXPORT int64_t tmq_get_response_offset(tmq_message_t *message);
|
DLL_EXPORT int64_t tmq_get_response_offset(tmq_message_t *message);
|
||||||
DLL_EXPORT TAOS_FIELD *tmq_get_fields(tmq_t *tmq, const char *topic);
|
DLL_EXPORT TAOS_FIELD *tmq_get_fields(tmq_t *tmq, const char *topic);
|
||||||
DLL_EXPORT int32_t tmq_field_count(tmq_t *tmq, const char *topic);
|
DLL_EXPORT int32_t tmq_field_count(tmq_t *tmq, const char *topic);
|
||||||
#endif
|
|
||||||
DLL_EXPORT void tmq_message_destroy(TAOS_RES *res);
|
DLL_EXPORT void tmq_message_destroy(TAOS_RES *res);
|
||||||
|
#endif
|
||||||
/* --------------------TMPORARY INTERFACE FOR TESTING--------------------- */
|
/* --------------------TMPORARY INTERFACE FOR TESTING--------------------- */
|
||||||
#if 0
|
#if 0
|
||||||
DLL_EXPORT TAOS_RES *tmq_create_topic(TAOS *taos, const char *name, const char *sql, int sqlLen);
|
DLL_EXPORT TAOS_RES *tmq_create_topic(TAOS *taos, const char *name, const char *sql, int sqlLen);
|
||||||
|
|
|
@ -331,6 +331,8 @@ typedef struct {
|
||||||
int32_t pid;
|
int32_t pid;
|
||||||
char app[TSDB_APP_NAME_LEN];
|
char app[TSDB_APP_NAME_LEN];
|
||||||
char db[TSDB_DB_NAME_LEN];
|
char db[TSDB_DB_NAME_LEN];
|
||||||
|
char user[TSDB_USER_LEN];
|
||||||
|
char passwd[TSDB_PASSWORD_LEN];
|
||||||
int64_t startTime;
|
int64_t startTime;
|
||||||
} SConnectReq;
|
} SConnectReq;
|
||||||
|
|
||||||
|
@ -482,7 +484,7 @@ typedef struct {
|
||||||
char intervalUnit;
|
char intervalUnit;
|
||||||
char slidingUnit;
|
char slidingUnit;
|
||||||
char
|
char
|
||||||
offsetUnit; // TODO Remove it, the offset is the number of precision tickle, and it must be a immutable duration.
|
offsetUnit; // TODO Remove it, the offset is the number of precision tickle, and it must be a immutable duration.
|
||||||
int8_t precision;
|
int8_t precision;
|
||||||
int64_t interval;
|
int64_t interval;
|
||||||
int64_t sliding;
|
int64_t sliding;
|
||||||
|
|
|
@ -684,6 +684,43 @@ static int32_t tdSRowResetBuf(SRowBuilder *pBuilder, void *pBuf) {
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief The invoker is responsible for memory alloc/dealloc.
|
||||||
|
*
|
||||||
|
* @param pBuilder
|
||||||
|
* @param pBuf Output buffer of STSRow
|
||||||
|
*/
|
||||||
|
static int32_t tdSRowGetBuf(SRowBuilder *pBuilder, void *pBuf) {
|
||||||
|
pBuilder->pBuf = (STSRow *)pBuf;
|
||||||
|
if (!pBuilder->pBuf) {
|
||||||
|
TASSERT(0);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
TASSERT(pBuilder->nBitmaps > 0 && pBuilder->flen > 0);
|
||||||
|
|
||||||
|
uint32_t len = 0;
|
||||||
|
switch (pBuilder->rowType) {
|
||||||
|
case TD_ROW_TP:
|
||||||
|
#ifdef TD_SUPPORT_BITMAP
|
||||||
|
pBuilder->pBitmap = tdGetBitmapAddrTp(pBuilder->pBuf, pBuilder->flen);
|
||||||
|
#endif
|
||||||
|
break;
|
||||||
|
case TD_ROW_KV:
|
||||||
|
#ifdef TD_SUPPORT_BITMAP
|
||||||
|
pBuilder->pBitmap = tdGetBitmapAddrKv(pBuilder->pBuf, pBuilder->nBoundCols);
|
||||||
|
#endif
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
TASSERT(0);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* @brief 由调用方管理存储空间的分配及释放,一次输入多个参数
|
* @brief 由调用方管理存储空间的分配及释放,一次输入多个参数
|
||||||
*
|
*
|
||||||
|
|
|
@ -283,12 +283,12 @@ typedef struct SVgDataBlocks {
|
||||||
} SVgDataBlocks;
|
} SVgDataBlocks;
|
||||||
|
|
||||||
typedef struct SVnodeModifOpStmt {
|
typedef struct SVnodeModifOpStmt {
|
||||||
ENodeType nodeType;
|
ENodeType nodeType;
|
||||||
ENodeType sqlNodeType;
|
ENodeType sqlNodeType;
|
||||||
SArray* pDataBlocks; // data block for each vgroup, SArray<SVgDataBlocks*>.
|
SArray* pDataBlocks; // data block for each vgroup, SArray<SVgDataBlocks*>.
|
||||||
uint8_t payloadType; // EPayloadType. 0: K-V payload for non-prepare insert, 1: rawPayload for prepare insert
|
uint8_t payloadType; // EPayloadType. 0: K-V payload for non-prepare insert, 1: rawPayload for prepare insert
|
||||||
uint32_t insertType; // insert data from [file|sql statement| bound statement]
|
uint32_t insertType; // insert data from [file|sql statement| bound statement]
|
||||||
const char* sql; // current sql statement position
|
const char* sql; // current sql statement position
|
||||||
} SVnodeModifOpStmt;
|
} SVnodeModifOpStmt;
|
||||||
|
|
||||||
typedef struct SExplainOptions {
|
typedef struct SExplainOptions {
|
||||||
|
|
|
@ -21,6 +21,15 @@ extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include "querynodes.h"
|
#include "querynodes.h"
|
||||||
|
#include "query.h"
|
||||||
|
|
||||||
|
typedef struct SStmtCallback {
|
||||||
|
TAOS_STMT* pStmt;
|
||||||
|
int32_t (*getTbNameFn)(TAOS_STMT*, char**);
|
||||||
|
int32_t (*setBindInfoFn)(TAOS_STMT*, STableMeta*, void*);
|
||||||
|
int32_t (*setExecInfoFn)(TAOS_STMT*, SHashObj*, SHashObj*);
|
||||||
|
int32_t (*getExecInfoFn)(TAOS_STMT*, SHashObj**, SHashObj**);
|
||||||
|
} SStmtCallback;
|
||||||
|
|
||||||
typedef struct SParseContext {
|
typedef struct SParseContext {
|
||||||
uint64_t requestId;
|
uint64_t requestId;
|
||||||
|
@ -34,6 +43,7 @@ typedef struct SParseContext {
|
||||||
char *pMsg; // extended error message if exists to help identifying the problem in sql statement.
|
char *pMsg; // extended error message if exists to help identifying the problem in sql statement.
|
||||||
int32_t msgLen; // max length of the msg
|
int32_t msgLen; // max length of the msg
|
||||||
struct SCatalog *pCatalog;
|
struct SCatalog *pCatalog;
|
||||||
|
SStmtCallback *pStmtCb;
|
||||||
} SParseContext;
|
} SParseContext;
|
||||||
|
|
||||||
typedef struct SCmdMsgInfo {
|
typedef struct SCmdMsgInfo {
|
||||||
|
@ -66,11 +76,27 @@ typedef struct SQuery {
|
||||||
} SQuery;
|
} SQuery;
|
||||||
|
|
||||||
int32_t qParseQuerySql(SParseContext* pCxt, SQuery** pQuery);
|
int32_t qParseQuerySql(SParseContext* pCxt, SQuery** pQuery);
|
||||||
|
bool isInsertSql(const char* pStr, size_t length);
|
||||||
|
|
||||||
void qDestroyQuery(SQuery* pQueryNode);
|
void qDestroyQuery(SQuery* pQueryNode);
|
||||||
|
|
||||||
int32_t qExtractResultSchema(const SNode* pRoot, int32_t* numOfCols, SSchema** pSchema);
|
int32_t qExtractResultSchema(const SNode* pRoot, int32_t* numOfCols, SSchema** pSchema);
|
||||||
|
|
||||||
|
int32_t qBuildStmtOutput(SQuery* pQuery, SHashObj* pVgHash, SHashObj* pBlockHash);
|
||||||
|
int32_t qResetStmtDataBlock(void* block, bool keepBuf);
|
||||||
|
int32_t qCloneStmtDataBlock(void** pDst, void* pSrc);
|
||||||
|
void qFreeStmtDataBlock(void* pDataBlock);
|
||||||
|
int32_t qRebuildStmtDataBlock(void** pDst, void* pSrc);
|
||||||
|
void qDestroyStmtDataBlock(void* pBlock);
|
||||||
|
int32_t qBindStmtColsValue(void *pBlock, TAOS_BIND_v2 *bind, char *msgBuf, int32_t msgBufLen);
|
||||||
|
int32_t qBindStmtSingleColValue(void *pBlock, TAOS_BIND_v2 *bind, char *msgBuf, int32_t msgBufLen, int32_t colIdx, int32_t rowNum);
|
||||||
|
int32_t qBuildStmtColFields(void *pDataBlock, int32_t *fieldNum, TAOS_FIELD** fields);
|
||||||
|
int32_t qBuildStmtTagFields(void *pBlock, void *boundTags, int32_t *fieldNum, TAOS_FIELD** fields);
|
||||||
|
int32_t qBindStmtTagsValue(void *pBlock, void *boundTags, int64_t suid, SName *pName, TAOS_BIND_v2 *bind, char *msgBuf, int32_t msgBufLen);
|
||||||
|
void destroyBoundColumnInfo(void* pBoundInfo);
|
||||||
|
int32_t qCreateSName(SName* pName, const char* pTableName, int32_t acctId, char* dbName, char *msgBuf, int32_t msgBufLen);
|
||||||
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -30,6 +30,7 @@ typedef struct SPlanContext {
|
||||||
SNode* pAstRoot;
|
SNode* pAstRoot;
|
||||||
bool topicQuery;
|
bool topicQuery;
|
||||||
bool streamQuery;
|
bool streamQuery;
|
||||||
|
bool rSmaQuery;
|
||||||
bool showRewrite;
|
bool showRewrite;
|
||||||
int8_t triggerType;
|
int8_t triggerType;
|
||||||
int64_t watermark;
|
int64_t watermark;
|
||||||
|
@ -45,7 +46,6 @@ int32_t qCreateQueryPlan(SPlanContext* pCxt, SQueryPlan** pPlan, SArray* pExecNo
|
||||||
// @pSource one execution location of this group of datasource subplans
|
// @pSource one execution location of this group of datasource subplans
|
||||||
int32_t qSetSubplanExecutionNode(SSubplan* pSubplan, int32_t groupId, SDownstreamSourceNode* pSource);
|
int32_t qSetSubplanExecutionNode(SSubplan* pSubplan, int32_t groupId, SDownstreamSourceNode* pSource);
|
||||||
|
|
||||||
typedef TAOS_MULTI_BIND TAOS_BIND_v2; // todo remove
|
|
||||||
int32_t qStmtBindParam(SQueryPlan* pPlan, TAOS_BIND_v2* pParams);
|
int32_t qStmtBindParam(SQueryPlan* pPlan, TAOS_BIND_v2* pParams);
|
||||||
|
|
||||||
// Convert to subplan to string for the scheduler to send to the executor
|
// Convert to subplan to string for the scheduler to send to the executor
|
||||||
|
|
|
@ -132,6 +132,9 @@ int32_t* taosGetErrno();
|
||||||
#define TSDB_CODE_TSC_INVALID_JSON_TYPE TAOS_DEF_ERROR_CODE(0, 0x0222)
|
#define TSDB_CODE_TSC_INVALID_JSON_TYPE TAOS_DEF_ERROR_CODE(0, 0x0222)
|
||||||
#define TSDB_CODE_TSC_VALUE_OUT_OF_RANGE TAOS_DEF_ERROR_CODE(0, 0x0223)
|
#define TSDB_CODE_TSC_VALUE_OUT_OF_RANGE TAOS_DEF_ERROR_CODE(0, 0x0223)
|
||||||
#define TSDB_CODE_TSC_INVALID_INPUT TAOS_DEF_ERROR_CODE(0, 0X0224)
|
#define TSDB_CODE_TSC_INVALID_INPUT TAOS_DEF_ERROR_CODE(0, 0X0224)
|
||||||
|
#define TSDB_CODE_TSC_STMT_API_ERROR TAOS_DEF_ERROR_CODE(0, 0X0225)
|
||||||
|
#define TSDB_CODE_TSC_STMT_TBNAME_ERROR TAOS_DEF_ERROR_CODE(0, 0X0226)
|
||||||
|
#define TSDB_CODE_TSC_STMT_CLAUSE_ERROR TAOS_DEF_ERROR_CODE(0, 0X0227)
|
||||||
|
|
||||||
// mnode-common
|
// mnode-common
|
||||||
#define TSDB_CODE_MND_APP_ERROR TAOS_DEF_ERROR_CODE(0, 0x0300)
|
#define TSDB_CODE_MND_APP_ERROR TAOS_DEF_ERROR_CODE(0, 0x0300)
|
||||||
|
|
|
@ -30,6 +30,15 @@ bin_dir="/usr/local/taos/bin"
|
||||||
|
|
||||||
service_config_dir="/etc/systemd/system"
|
service_config_dir="/etc/systemd/system"
|
||||||
|
|
||||||
|
#taos-tools para
|
||||||
|
demoName="taosdemo"
|
||||||
|
benchmarkName="taosBenchmark"
|
||||||
|
dumpName="taosdump"
|
||||||
|
emailName="taosdata.com"
|
||||||
|
taosName="taos"
|
||||||
|
toolsName="taostools"
|
||||||
|
|
||||||
|
|
||||||
# Color setting
|
# Color setting
|
||||||
RED='\033[0;31m'
|
RED='\033[0;31m'
|
||||||
GREEN='\033[1;32m'
|
GREEN='\033[1;32m'
|
||||||
|
@ -230,8 +239,20 @@ function install_header() {
|
||||||
|
|
||||||
# temp install taosBenchmark
|
# temp install taosBenchmark
|
||||||
function install_taosTools() {
|
function install_taosTools() {
|
||||||
cd ${script_dir}/taos-tools/
|
${csudo} rm -f ${bin_link_dir}/${benchmarkName} || :
|
||||||
tar xvf taosTools-1.4.1-Linux-x64.tar.gz && cd taosTools-1.4.1/ && ./install-taostools.sh
|
${csudo} rm -f ${bin_link_dir}/${dumpName} || :
|
||||||
|
${csudo} rm -f ${bin_link_dir}/rm${toolsName} || :
|
||||||
|
|
||||||
|
${csudo} /usr/bin/install -c -m 755 ${script_dir}/bin/${dumpName} ${install_main_dir}/bin/${dumpName}
|
||||||
|
${csudo} /usr/bin/install -c -m 755 ${script_dir}/bin/${benchmarkName} ${install_main_dir}/bin/${benchmarkName}
|
||||||
|
${csudo} ln -sf ${install_main_dir}/bin/${benchmarkName} ${install_main_dir}/bin/${demoName}
|
||||||
|
#Make link
|
||||||
|
[[ -x ${install_main_dir}/bin/${benchmarkName} ]] && \
|
||||||
|
${csudo} ln -s ${install_main_dir}/bin/${benchmarkName} ${bin_link_dir}/${benchmarkName} || :
|
||||||
|
[[ -x ${install_main_dir}/bin/${demoName} ]] && \
|
||||||
|
${csudo} ln -s ${install_main_dir}/bin/${demoName} ${bin_link_dir}/${demoName} || :
|
||||||
|
[[ -x ${install_main_dir}/bin/${dumpName} ]] && \
|
||||||
|
${csudo} ln -s ${install_main_dir}/bin/${dumpName} ${bin_link_dir}/${dumpName} || :
|
||||||
}
|
}
|
||||||
|
|
||||||
function add_newHostname_to_hosts() {
|
function add_newHostname_to_hosts() {
|
||||||
|
|
|
@ -39,7 +39,7 @@ cd ${compile_dir}
|
||||||
|
|
||||||
echo "compile_dir: ${compile_dir}"
|
echo "compile_dir: ${compile_dir}"
|
||||||
|
|
||||||
cmake ..
|
cmake .. -DBUILD_TOOLS=true
|
||||||
make -j32
|
make -j32
|
||||||
|
|
||||||
release_dir="${top_dir}/release"
|
release_dir="${top_dir}/release"
|
||||||
|
@ -55,7 +55,6 @@ mkdir -p ${install_dir}
|
||||||
mkdir -p ${install_dir}/bin
|
mkdir -p ${install_dir}/bin
|
||||||
mkdir -p ${install_dir}/lib
|
mkdir -p ${install_dir}/lib
|
||||||
mkdir -p ${install_dir}/inc
|
mkdir -p ${install_dir}/inc
|
||||||
mkdir -p ${install_dir}/taos-tools
|
|
||||||
|
|
||||||
install_files="${script_dir}/install.sh"
|
install_files="${script_dir}/install.sh"
|
||||||
chmod a+x ${script_dir}/install.sh || :
|
chmod a+x ${script_dir}/install.sh || :
|
||||||
|
@ -64,13 +63,14 @@ cp ${install_files} ${install_dir}
|
||||||
header_files="${top_dir}/include/client/taos.h ${top_dir}/include/util/taoserror.h"
|
header_files="${top_dir}/include/client/taos.h ${top_dir}/include/util/taoserror.h"
|
||||||
cp ${header_files} ${install_dir}/inc
|
cp ${header_files} ${install_dir}/inc
|
||||||
|
|
||||||
bin_files="${compile_dir}/build/bin/taosd ${compile_dir}/build/bin/taos ${compile_dir}/build/bin/create_table ${compile_dir}/build/bin/tmq_sim ${script_dir}/remove.sh"
|
bin_files="${compile_dir}/source/dnode/mgmt/taosd ${compile_dir}/tools/shell/taos ${compile_dir}/tests/test/c/create_table ${compile_dir}/tests/test/c/tmq_sim ${script_dir}/remove.sh ${compile_dir}/build/bin/taosBenchmark ${compile_dir}/build/bin/taosdump"
|
||||||
cp ${bin_files} ${install_dir}/bin && chmod a+x ${install_dir}/bin/* || :
|
cp -rf ${bin_files} ${install_dir}/bin && chmod a+x ${install_dir}/bin/* || :
|
||||||
|
|
||||||
|
cp -rf ${compile_dir}/source/client/libtaos.so ${install_dir}/lib/
|
||||||
|
cp -rf ${compile_dir}/source/libs/tdb/libtdb.so ${install_dir}/lib/
|
||||||
|
cp -rf ${compile_dir}/build/lib/libavro* ${install_dir}/lib/ > /dev/null || echo -e "failed to copy avro libraries"
|
||||||
|
cp -rf ${compile_dir}/build/lib/pkgconfig ${install_dir}/lib/ > /dev/null || echo -e "failed to copy pkgconfig directory"
|
||||||
|
|
||||||
cp ${compile_dir}/build/lib/libtaos.so ${install_dir}/lib/
|
|
||||||
cp ${compile_dir}/build/lib/libtdb.so ${install_dir}/lib/
|
|
||||||
taostoolfile="${top_dir}/tools/taosTools-1.4.1-Linux-x64.tar.gz"
|
|
||||||
cp ${taostoolfile} ${install_dir}/taos-tools
|
|
||||||
|
|
||||||
#cp ${compile_dir}/source/dnode/mnode/impl/libmnode.so ${install_dir}/lib/
|
#cp ${compile_dir}/source/dnode/mnode/impl/libmnode.so ${install_dir}/lib/
|
||||||
#cp ${compile_dir}/source/dnode/qnode/libqnode.so ${install_dir}/lib/
|
#cp ${compile_dir}/source/dnode/qnode/libqnode.so ${install_dir}/lib/
|
||||||
|
|
|
@ -205,7 +205,8 @@ typedef struct SRequestObj {
|
||||||
char* sqlstr; // sql string
|
char* sqlstr; // sql string
|
||||||
int32_t sqlLen;
|
int32_t sqlLen;
|
||||||
int64_t self;
|
int64_t self;
|
||||||
char* msgBuf; // error msg buffer
|
char* msgBuf;
|
||||||
|
int32_t msgBufLen;
|
||||||
int32_t code;
|
int32_t code;
|
||||||
SArray* dbList;
|
SArray* dbList;
|
||||||
SArray* tableList;
|
SArray* tableList;
|
||||||
|
@ -278,7 +279,8 @@ void initMsgHandleFp();
|
||||||
TAOS* taos_connect_internal(const char* ip, const char* user, const char* pass, const char* auth, const char* db,
|
TAOS* taos_connect_internal(const char* ip, const char* user, const char* pass, const char* auth, const char* db,
|
||||||
uint16_t port, int connType);
|
uint16_t port, int connType);
|
||||||
|
|
||||||
int32_t parseSql(SRequestObj* pRequest, bool topicQuery, SQuery** pQuery);
|
int32_t parseSql(SRequestObj* pRequest, bool topicQuery, SQuery** pQuery, SStmtCallback* pStmtCb);
|
||||||
|
|
||||||
int32_t getPlan(SRequestObj* pRequest, SQuery* pQuery, SQueryPlan** pPlan, SArray* pNodeList);
|
int32_t getPlan(SRequestObj* pRequest, SQuery* pQuery, SQueryPlan** pPlan, SArray* pNodeList);
|
||||||
|
|
||||||
int32_t buildRequest(STscObj* pTscObj, const char* sql, int sqlLen, SRequestObj** pRequest);
|
int32_t buildRequest(STscObj* pTscObj, const char* sql, int sqlLen, SRequestObj** pRequest);
|
||||||
|
@ -302,6 +304,8 @@ int hbAddConnInfo(SAppHbMgr* pAppHbMgr, SClientHbKey connKey, void* key, void* v
|
||||||
// --- mq
|
// --- mq
|
||||||
void hbMgrInitMqHbRspHandle();
|
void hbMgrInitMqHbRspHandle();
|
||||||
|
|
||||||
|
SRequestObj* launchQueryImpl(SRequestObj* pRequest, SQuery* pQuery, int32_t code, bool keepQuery);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -19,6 +19,9 @@
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
#include "catalog.h"
|
||||||
|
|
||||||
|
typedef void STableDataBlocks;
|
||||||
|
|
||||||
typedef enum {
|
typedef enum {
|
||||||
STMT_TYPE_INSERT = 1,
|
STMT_TYPE_INSERT = 1,
|
||||||
|
@ -26,17 +29,64 @@ typedef enum {
|
||||||
STMT_TYPE_QUERY,
|
STMT_TYPE_QUERY,
|
||||||
} STMT_TYPE;
|
} STMT_TYPE;
|
||||||
|
|
||||||
typedef struct STscStmt {
|
typedef enum {
|
||||||
STMT_TYPE type;
|
STMT_INIT = 1,
|
||||||
//int16_t last;
|
STMT_PREPARE,
|
||||||
//STscObj* taos;
|
STMT_SETTBNAME,
|
||||||
//SSqlObj* pSql;
|
STMT_SETTAGS,
|
||||||
//SMultiTbStmt mtb;
|
STMT_FETCH_TAG_FIELDS,
|
||||||
//SNormalStmt normal;
|
STMT_FETCH_COL_FIELDS,
|
||||||
|
STMT_BIND,
|
||||||
|
STMT_BIND_COL,
|
||||||
|
STMT_ADD_BATCH,
|
||||||
|
STMT_EXECUTE,
|
||||||
|
} STMT_STATUS;
|
||||||
|
|
||||||
//int numOfRows;
|
typedef struct SStmtTableCache {
|
||||||
|
STableDataBlocks* pDataBlock;
|
||||||
|
void* boundTags;
|
||||||
|
} SStmtTableCache;
|
||||||
|
|
||||||
|
typedef struct SStmtBindInfo {
|
||||||
|
bool needParse;
|
||||||
|
uint64_t tbUid;
|
||||||
|
uint64_t tbSuid;
|
||||||
|
int32_t sBindRowNum;
|
||||||
|
int32_t sBindLastIdx;
|
||||||
|
int8_t tbType;
|
||||||
|
void* boundTags;
|
||||||
|
char* tbName;
|
||||||
|
SName sname;
|
||||||
|
} SStmtBindInfo;
|
||||||
|
|
||||||
|
typedef struct SStmtExecInfo {
|
||||||
|
SRequestObj* pRequest;
|
||||||
|
SHashObj* pVgHash;
|
||||||
|
SHashObj* pBlockHash;
|
||||||
|
} SStmtExecInfo;
|
||||||
|
|
||||||
|
typedef struct SStmtSQLInfo {
|
||||||
|
STMT_TYPE type;
|
||||||
|
STMT_STATUS status;
|
||||||
|
bool autoCreate;
|
||||||
|
uint64_t runTimes;
|
||||||
|
SHashObj* pTableCache; //SHash<SStmtTableCache>
|
||||||
|
SQuery* pQuery;
|
||||||
|
char* sqlStr;
|
||||||
|
int32_t sqlLen;
|
||||||
|
} SStmtSQLInfo;
|
||||||
|
|
||||||
|
typedef struct STscStmt {
|
||||||
|
STscObj* taos;
|
||||||
|
SCatalog* pCatalog;
|
||||||
|
int32_t affectedRows;
|
||||||
|
|
||||||
|
SStmtSQLInfo sql;
|
||||||
|
SStmtExecInfo exec;
|
||||||
|
SStmtBindInfo bInfo;
|
||||||
} STscStmt;
|
} STscStmt;
|
||||||
|
|
||||||
|
|
||||||
#define STMT_ERR_RET(c) do { int32_t _code = c; if (_code != TSDB_CODE_SUCCESS) { terrno = _code; return _code; } } while (0)
|
#define STMT_ERR_RET(c) do { int32_t _code = c; if (_code != TSDB_CODE_SUCCESS) { terrno = _code; return _code; } } while (0)
|
||||||
#define STMT_RET(c) do { int32_t _code = c; if (_code != TSDB_CODE_SUCCESS) { terrno = _code; } return _code; } while (0)
|
#define STMT_RET(c) do { int32_t _code = c; if (_code != TSDB_CODE_SUCCESS) { terrno = _code; } return _code; } while (0)
|
||||||
#define STMT_ERR_JRET(c) do { code = c; if (code != TSDB_CODE_SUCCESS) { terrno = code; goto _return; } } while (0)
|
#define STMT_ERR_JRET(c) do { code = c; if (code != TSDB_CODE_SUCCESS) { terrno = code; goto _return; } } while (0)
|
||||||
|
@ -44,16 +94,16 @@ typedef struct STscStmt {
|
||||||
TAOS_STMT *stmtInit(TAOS *taos);
|
TAOS_STMT *stmtInit(TAOS *taos);
|
||||||
int stmtClose(TAOS_STMT *stmt);
|
int stmtClose(TAOS_STMT *stmt);
|
||||||
int stmtExec(TAOS_STMT *stmt);
|
int stmtExec(TAOS_STMT *stmt);
|
||||||
char *stmtErrstr(TAOS_STMT *stmt);
|
const char *stmtErrstr(TAOS_STMT *stmt);
|
||||||
int stmtAffectedRows(TAOS_STMT *stmt);
|
int stmtAffectedRows(TAOS_STMT *stmt);
|
||||||
int stmtBind(TAOS_STMT *stmt, TAOS_BIND *bind);
|
|
||||||
int stmtPrepare(TAOS_STMT *stmt, const char *sql, unsigned long length);
|
int stmtPrepare(TAOS_STMT *stmt, const char *sql, unsigned long length);
|
||||||
int stmtSetTbNameTags(TAOS_STMT *stmt, const char *name, TAOS_BIND *tags);
|
int stmtSetTbName(TAOS_STMT *stmt, const char *tbName);
|
||||||
|
int stmtSetTbTags(TAOS_STMT *stmt, TAOS_BIND_v2 *tags);
|
||||||
int stmtIsInsert(TAOS_STMT *stmt, int *insert);
|
int stmtIsInsert(TAOS_STMT *stmt, int *insert);
|
||||||
int stmtGetParamNum(TAOS_STMT *stmt, int *nums);
|
int stmtGetParamNum(TAOS_STMT *stmt, int *nums);
|
||||||
int stmtAddBatch(TAOS_STMT *stmt);
|
int stmtAddBatch(TAOS_STMT *stmt);
|
||||||
TAOS_RES *stmtUseResult(TAOS_STMT *stmt);
|
TAOS_RES *stmtUseResult(TAOS_STMT *stmt);
|
||||||
int stmtBindBatch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind);
|
int stmtBindBatch(TAOS_STMT *stmt, TAOS_BIND_v2 *bind, int32_t colIdx);
|
||||||
|
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
|
|
|
@ -186,6 +186,7 @@ void *createRequest(STscObj *pObj, __taos_async_fn_t fp, void *param, int32_t ty
|
||||||
pRequest->pTscObj = pObj;
|
pRequest->pTscObj = pObj;
|
||||||
pRequest->body.fp = fp; // not used it yet
|
pRequest->body.fp = fp; // not used it yet
|
||||||
pRequest->msgBuf = taosMemoryCalloc(1, ERROR_MSG_BUF_DEFAULT_SIZE);
|
pRequest->msgBuf = taosMemoryCalloc(1, ERROR_MSG_BUF_DEFAULT_SIZE);
|
||||||
|
pRequest->msgBufLen = ERROR_MSG_BUF_DEFAULT_SIZE;
|
||||||
tsem_init(&pRequest->body.rspSem, 0, 0);
|
tsem_init(&pRequest->body.rspSem, 0, 0);
|
||||||
|
|
||||||
registerRequest(pRequest);
|
registerRequest(pRequest);
|
||||||
|
|
|
@ -146,7 +146,8 @@ int32_t buildRequest(STscObj* pTscObj, const char* sql, int sqlLen, SRequestObj*
|
||||||
(*pRequest)->sqlstr[sqlLen] = 0;
|
(*pRequest)->sqlstr[sqlLen] = 0;
|
||||||
(*pRequest)->sqlLen = sqlLen;
|
(*pRequest)->sqlLen = sqlLen;
|
||||||
|
|
||||||
if (taosHashPut(pTscObj->pRequests, &(*pRequest)->self, sizeof((*pRequest)->self), &(*pRequest)->self, sizeof((*pRequest)->self))) {
|
if (taosHashPut(pTscObj->pRequests, &(*pRequest)->self, sizeof((*pRequest)->self), &(*pRequest)->self,
|
||||||
|
sizeof((*pRequest)->self))) {
|
||||||
destroyRequest(*pRequest);
|
destroyRequest(*pRequest);
|
||||||
*pRequest = NULL;
|
*pRequest = NULL;
|
||||||
tscError("put request to request hash failed");
|
tscError("put request to request hash failed");
|
||||||
|
@ -157,7 +158,7 @@ int32_t buildRequest(STscObj* pTscObj, const char* sql, int sqlLen, SRequestObj*
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t parseSql(SRequestObj* pRequest, bool topicQuery, SQuery** pQuery) {
|
int32_t parseSql(SRequestObj* pRequest, bool topicQuery, SQuery** pQuery, SStmtCallback* pStmtCb) {
|
||||||
STscObj* pTscObj = pRequest->pTscObj;
|
STscObj* pTscObj = pRequest->pTscObj;
|
||||||
|
|
||||||
SParseContext cxt = {
|
SParseContext cxt = {
|
||||||
|
@ -170,6 +171,7 @@ int32_t parseSql(SRequestObj* pRequest, bool topicQuery, SQuery** pQuery) {
|
||||||
.pMsg = pRequest->msgBuf,
|
.pMsg = pRequest->msgBuf,
|
||||||
.msgLen = ERROR_MSG_BUF_DEFAULT_SIZE,
|
.msgLen = ERROR_MSG_BUF_DEFAULT_SIZE,
|
||||||
.pTransporter = pTscObj->pAppInfo->pTransporter,
|
.pTransporter = pTscObj->pAppInfo->pTransporter,
|
||||||
|
.pStmtCb = pStmtCb,
|
||||||
};
|
};
|
||||||
|
|
||||||
cxt.mgmtEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
|
cxt.mgmtEpSet = getEpSet_s(&pTscObj->pAppInfo->mgmtEp);
|
||||||
|
@ -262,7 +264,8 @@ void setResSchemaInfo(SReqResultInfo* pResInfo, const SSchema* pSchema, int32_t
|
||||||
}
|
}
|
||||||
|
|
||||||
void setResPrecision(SReqResultInfo* pResInfo, int32_t precision) {
|
void setResPrecision(SReqResultInfo* pResInfo, int32_t precision) {
|
||||||
if (precision != TSDB_TIME_PRECISION_MILLI && precision != TSDB_TIME_PRECISION_MICRO && precision != TSDB_TIME_PRECISION_NANO) {
|
if (precision != TSDB_TIME_PRECISION_MILLI && precision != TSDB_TIME_PRECISION_MICRO &&
|
||||||
|
precision != TSDB_TIME_PRECISION_NANO) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -274,7 +277,7 @@ int32_t scheduleQuery(SRequestObj* pRequest, SQueryPlan* pDag, SArray* pNodeList
|
||||||
|
|
||||||
SQueryResult res = {.code = 0, .numOfRows = 0, .msgSize = ERROR_MSG_BUF_DEFAULT_SIZE, .msg = pRequest->msgBuf};
|
SQueryResult res = {.code = 0, .numOfRows = 0, .msgSize = ERROR_MSG_BUF_DEFAULT_SIZE, .msg = pRequest->msgBuf};
|
||||||
int32_t code = schedulerExecJob(pTransporter, pNodeList, pDag, &pRequest->body.queryJob, pRequest->sqlstr,
|
int32_t code = schedulerExecJob(pTransporter, pNodeList, pDag, &pRequest->body.queryJob, pRequest->sqlstr,
|
||||||
pRequest->metric.start, &res);
|
pRequest->metric.start, &res);
|
||||||
if (code != TSDB_CODE_SUCCESS) {
|
if (code != TSDB_CODE_SUCCESS) {
|
||||||
if (pRequest->body.queryJob != 0) {
|
if (pRequest->body.queryJob != 0) {
|
||||||
schedulerFreeJob(pRequest->body.queryJob);
|
schedulerFreeJob(pRequest->body.queryJob);
|
||||||
|
@ -298,15 +301,8 @@ int32_t scheduleQuery(SRequestObj* pRequest, SQueryPlan* pDag, SArray* pNodeList
|
||||||
return pRequest->code;
|
return pRequest->code;
|
||||||
}
|
}
|
||||||
|
|
||||||
SRequestObj* execQueryImpl(STscObj* pTscObj, const char* sql, int sqlLen) {
|
SRequestObj* launchQueryImpl(SRequestObj* pRequest, SQuery* pQuery, int32_t code, bool keepQuery) {
|
||||||
SRequestObj* pRequest = NULL;
|
SArray* pNodeList = taosArrayInit(4, sizeof(struct SQueryNodeAddr));
|
||||||
SQuery* pQuery = NULL;
|
|
||||||
SArray* pNodeList = taosArrayInit(4, sizeof(struct SQueryNodeAddr));
|
|
||||||
|
|
||||||
int32_t code = buildRequest(pTscObj, sql, sqlLen, &pRequest);
|
|
||||||
if (TSDB_CODE_SUCCESS == code) {
|
|
||||||
code = parseSql(pRequest, false, &pQuery);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (TSDB_CODE_SUCCESS == code) {
|
if (TSDB_CODE_SUCCESS == code) {
|
||||||
switch (pQuery->execMode) {
|
switch (pQuery->execMode) {
|
||||||
|
@ -331,7 +327,10 @@ SRequestObj* execQueryImpl(STscObj* pTscObj, const char* sql, int sqlLen) {
|
||||||
}
|
}
|
||||||
|
|
||||||
taosArrayDestroy(pNodeList);
|
taosArrayDestroy(pNodeList);
|
||||||
qDestroyQuery(pQuery);
|
if (!keepQuery) {
|
||||||
|
qDestroyQuery(pQuery);
|
||||||
|
}
|
||||||
|
|
||||||
if (NULL != pRequest && TSDB_CODE_SUCCESS != code) {
|
if (NULL != pRequest && TSDB_CODE_SUCCESS != code) {
|
||||||
pRequest->code = terrno;
|
pRequest->code = terrno;
|
||||||
}
|
}
|
||||||
|
@ -339,6 +338,18 @@ SRequestObj* execQueryImpl(STscObj* pTscObj, const char* sql, int sqlLen) {
|
||||||
return pRequest;
|
return pRequest;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
SRequestObj* launchQuery(STscObj* pTscObj, const char* sql, int sqlLen) {
|
||||||
|
SRequestObj* pRequest = NULL;
|
||||||
|
SQuery* pQuery = NULL;
|
||||||
|
|
||||||
|
int32_t code = buildRequest(pTscObj, sql, sqlLen, &pRequest);
|
||||||
|
if (TSDB_CODE_SUCCESS == code) {
|
||||||
|
code = parseSql(pRequest, false, &pQuery, NULL);
|
||||||
|
}
|
||||||
|
|
||||||
|
return launchQueryImpl(pRequest, pQuery, code, false);
|
||||||
|
}
|
||||||
|
|
||||||
int32_t refreshMeta(STscObj* pTscObj, SRequestObj* pRequest) {
|
int32_t refreshMeta(STscObj* pTscObj, SRequestObj* pRequest) {
|
||||||
SCatalog* pCatalog = NULL;
|
SCatalog* pCatalog = NULL;
|
||||||
int32_t code = 0;
|
int32_t code = 0;
|
||||||
|
@ -383,7 +394,7 @@ SRequestObj* execQuery(STscObj* pTscObj, const char* sql, int sqlLen) {
|
||||||
int32_t code = 0;
|
int32_t code = 0;
|
||||||
|
|
||||||
while (retryNum++ < REQUEST_MAX_TRY_TIMES) {
|
while (retryNum++ < REQUEST_MAX_TRY_TIMES) {
|
||||||
pRequest = execQueryImpl(pTscObj, sql, sqlLen);
|
pRequest = launchQuery(pTscObj, sql, sqlLen);
|
||||||
if (TSDB_CODE_SUCCESS == pRequest->code || !NEED_CLIENT_HANDLE_ERROR(pRequest->code)) {
|
if (TSDB_CODE_SUCCESS == pRequest->code || !NEED_CLIENT_HANDLE_ERROR(pRequest->code)) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
@ -512,6 +523,8 @@ static SMsgSendInfo* buildConnectMsg(SRequestObj* pRequest, int8_t connType) {
|
||||||
connectReq.pid = htonl(appInfo.pid);
|
connectReq.pid = htonl(appInfo.pid);
|
||||||
connectReq.startTime = htobe64(appInfo.startTime);
|
connectReq.startTime = htobe64(appInfo.startTime);
|
||||||
tstrncpy(connectReq.app, appInfo.appName, sizeof(connectReq.app));
|
tstrncpy(connectReq.app, appInfo.appName, sizeof(connectReq.app));
|
||||||
|
tstrncpy(connectReq.user, pObj->user, sizeof(connectReq.user));
|
||||||
|
tstrncpy(connectReq.passwd, pObj->pass, sizeof(connectReq.passwd));
|
||||||
|
|
||||||
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
||||||
void* pReq = taosMemoryMalloc(contLen);
|
void* pReq = taosMemoryMalloc(contLen);
|
||||||
|
@ -715,6 +728,7 @@ static int32_t doConvertUCS4(SReqResultInfo* pResultInfo, int32_t numOfRows, int
|
||||||
|
|
||||||
int32_t len = taosUcs4ToMbs((TdUcs4*)varDataVal(pStart), varDataLen(pStart), varDataVal(p));
|
int32_t len = taosUcs4ToMbs((TdUcs4*)varDataVal(pStart), varDataLen(pStart), varDataVal(p));
|
||||||
ASSERT(len <= bytes);
|
ASSERT(len <= bytes);
|
||||||
|
ASSERT((p + len) < (pResultInfo->convertBuf[i] + colLength[i]));
|
||||||
|
|
||||||
varDataSetLen(p, len);
|
varDataSetLen(p, len);
|
||||||
pCol->offset[j] = (p - pResultInfo->convertBuf[i]);
|
pCol->offset[j] = (p - pResultInfo->convertBuf[i]);
|
||||||
|
@ -733,51 +747,55 @@ static int32_t doConvertUCS4(SReqResultInfo* pResultInfo, int32_t numOfRows, int
|
||||||
}
|
}
|
||||||
|
|
||||||
pResultInfo->convertBuf[i] = p;
|
pResultInfo->convertBuf[i] = p;
|
||||||
int32_t len = 0;
|
int32_t len = 0;
|
||||||
SResultColumn* pCol = &pResultInfo->pCol[i];
|
SResultColumn* pCol = &pResultInfo->pCol[i];
|
||||||
for (int32_t j = 0; j < numOfRows; ++j) {
|
for (int32_t j = 0; j < numOfRows; ++j) {
|
||||||
if (pCol->offset[j] != -1) {
|
if (pCol->offset[j] != -1) {
|
||||||
char* pStart = pCol->offset[j] + pCol->pData;
|
char* pStart = pCol->offset[j] + pCol->pData;
|
||||||
|
|
||||||
int32_t jsonInnerType = *pStart;
|
int32_t jsonInnerType = *pStart;
|
||||||
char *jsonInnerData = pStart + CHAR_BYTES;
|
char* jsonInnerData = pStart + CHAR_BYTES;
|
||||||
char dst[TSDB_MAX_JSON_TAG_LEN] = {0};
|
char dst[TSDB_MAX_JSON_TAG_LEN] = {0};
|
||||||
if(jsonInnerType == TSDB_DATA_TYPE_NULL){
|
if (jsonInnerType == TSDB_DATA_TYPE_NULL) {
|
||||||
sprintf(varDataVal(dst), "%s", TSDB_DATA_NULL_STR_L);
|
sprintf(varDataVal(dst), "%s", TSDB_DATA_NULL_STR_L);
|
||||||
varDataSetLen(dst, strlen(varDataVal(dst)));
|
varDataSetLen(dst, strlen(varDataVal(dst)));
|
||||||
}else if(jsonInnerType == TSDB_DATA_TYPE_JSON){
|
} else if (jsonInnerType == TSDB_DATA_TYPE_JSON) {
|
||||||
int32_t length = taosUcs4ToMbs((TdUcs4 *)varDataVal(jsonInnerData), varDataLen(jsonInnerData), varDataVal(dst));
|
int32_t length =
|
||||||
|
taosUcs4ToMbs((TdUcs4*)varDataVal(jsonInnerData), varDataLen(jsonInnerData), varDataVal(dst));
|
||||||
|
|
||||||
if (length <= 0) {
|
if (length <= 0) {
|
||||||
tscError("charset:%s to %s. val:%s convert failed.", DEFAULT_UNICODE_ENCODEC, tsCharset, varDataVal(jsonInnerData));
|
tscError("charset:%s to %s. val:%s convert failed.", DEFAULT_UNICODE_ENCODEC, tsCharset,
|
||||||
|
varDataVal(jsonInnerData));
|
||||||
length = 0;
|
length = 0;
|
||||||
}
|
}
|
||||||
varDataSetLen(dst, length);
|
varDataSetLen(dst, length);
|
||||||
}else if (jsonInnerType == TSDB_DATA_TYPE_NCHAR) { // value -> "value"
|
} else if (jsonInnerType == TSDB_DATA_TYPE_NCHAR) { // value -> "value"
|
||||||
*(char*)varDataVal(dst) = '\"';
|
*(char*)varDataVal(dst) = '\"';
|
||||||
int32_t length = taosUcs4ToMbs((TdUcs4 *)varDataVal(jsonInnerData), varDataLen(jsonInnerData), varDataVal(dst) + CHAR_BYTES);
|
int32_t length = taosUcs4ToMbs((TdUcs4*)varDataVal(jsonInnerData), varDataLen(jsonInnerData),
|
||||||
|
varDataVal(dst) + CHAR_BYTES);
|
||||||
if (length <= 0) {
|
if (length <= 0) {
|
||||||
tscError("charset:%s to %s. val:%s convert failed.", DEFAULT_UNICODE_ENCODEC, tsCharset, varDataVal(jsonInnerData));
|
tscError("charset:%s to %s. val:%s convert failed.", DEFAULT_UNICODE_ENCODEC, tsCharset,
|
||||||
|
varDataVal(jsonInnerData));
|
||||||
length = 0;
|
length = 0;
|
||||||
}
|
}
|
||||||
varDataSetLen(dst, length + CHAR_BYTES*2);
|
varDataSetLen(dst, length + CHAR_BYTES * 2);
|
||||||
*(char*)(varDataVal(dst), length + CHAR_BYTES) = '\"';
|
*(char*)(varDataVal(dst), length + CHAR_BYTES) = '\"';
|
||||||
}else if(jsonInnerType == TSDB_DATA_TYPE_DOUBLE){
|
} else if (jsonInnerType == TSDB_DATA_TYPE_DOUBLE) {
|
||||||
double jsonVd = *(double*)(jsonInnerData);
|
double jsonVd = *(double*)(jsonInnerData);
|
||||||
sprintf(varDataVal(dst), "%.9lf", jsonVd);
|
sprintf(varDataVal(dst), "%.9lf", jsonVd);
|
||||||
varDataSetLen(dst, strlen(varDataVal(dst)));
|
varDataSetLen(dst, strlen(varDataVal(dst)));
|
||||||
}else if(jsonInnerType == TSDB_DATA_TYPE_BIGINT){
|
} else if (jsonInnerType == TSDB_DATA_TYPE_BIGINT) {
|
||||||
int64_t jsonVd = *(int64_t*)(jsonInnerData);
|
int64_t jsonVd = *(int64_t*)(jsonInnerData);
|
||||||
sprintf(varDataVal(dst), "%" PRId64, jsonVd);
|
sprintf(varDataVal(dst), "%" PRId64, jsonVd);
|
||||||
varDataSetLen(dst, strlen(varDataVal(dst)));
|
varDataSetLen(dst, strlen(varDataVal(dst)));
|
||||||
}else if(jsonInnerType == TSDB_DATA_TYPE_BOOL){
|
} else if (jsonInnerType == TSDB_DATA_TYPE_BOOL) {
|
||||||
sprintf(varDataVal(dst), "%s", (*((char *)jsonInnerData) == 1) ? "true" : "false");
|
sprintf(varDataVal(dst), "%s", (*((char*)jsonInnerData) == 1) ? "true" : "false");
|
||||||
varDataSetLen(dst, strlen(varDataVal(dst)));
|
varDataSetLen(dst, strlen(varDataVal(dst)));
|
||||||
}else {
|
} else {
|
||||||
ASSERT(0);
|
ASSERT(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
if(len + varDataTLen(dst) > colLength[i]){
|
if (len + varDataTLen(dst) > colLength[i]) {
|
||||||
p = taosMemoryRealloc(pResultInfo->convertBuf[i], len + varDataTLen(dst));
|
p = taosMemoryRealloc(pResultInfo->convertBuf[i], len + varDataTLen(dst));
|
||||||
if (p == NULL) {
|
if (p == NULL) {
|
||||||
return TSDB_CODE_OUT_OF_MEMORY;
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
|
|
@ -128,6 +128,10 @@ const char *taos_errstr(TAOS_RES *res) {
|
||||||
}
|
}
|
||||||
|
|
||||||
void taos_free_result(TAOS_RES *res) {
|
void taos_free_result(TAOS_RES *res) {
|
||||||
|
if (NULL == res) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
if (TD_RES_QUERY(res)) {
|
if (TD_RES_QUERY(res)) {
|
||||||
SRequestObj *pRequest = (SRequestObj *)res;
|
SRequestObj *pRequest = (SRequestObj *)res;
|
||||||
destroyRequest(pRequest);
|
destroyRequest(pRequest);
|
||||||
|
@ -579,56 +583,6 @@ TAOS_STMT *taos_stmt_init(TAOS *taos) {
|
||||||
return stmtInit(taos);
|
return stmtInit(taos);
|
||||||
}
|
}
|
||||||
|
|
||||||
int taos_stmt_close(TAOS_STMT *stmt) {
|
|
||||||
if (stmt == NULL) {
|
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
|
||||||
return terrno;
|
|
||||||
}
|
|
||||||
|
|
||||||
return stmtClose(stmt);
|
|
||||||
}
|
|
||||||
|
|
||||||
int taos_stmt_execute(TAOS_STMT *stmt) {
|
|
||||||
if (stmt == NULL) {
|
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
|
||||||
return terrno;
|
|
||||||
}
|
|
||||||
|
|
||||||
return stmtExec(stmt);
|
|
||||||
}
|
|
||||||
|
|
||||||
char *taos_stmt_errstr(TAOS_STMT *stmt) {
|
|
||||||
if (stmt == NULL) {
|
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
return stmtErrstr(stmt);
|
|
||||||
}
|
|
||||||
|
|
||||||
int taos_stmt_affected_rows(TAOS_STMT *stmt) {
|
|
||||||
if (stmt == NULL) {
|
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
return stmtAffectedRows(stmt);
|
|
||||||
}
|
|
||||||
|
|
||||||
int taos_stmt_bind_param(TAOS_STMT *stmt, TAOS_BIND *bind) {
|
|
||||||
if (stmt == NULL || bind == NULL) {
|
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
|
||||||
return terrno;
|
|
||||||
}
|
|
||||||
|
|
||||||
return stmtBind(stmt, bind);
|
|
||||||
}
|
|
||||||
|
|
||||||
int taos_stmt_prepare(TAOS_STMT *stmt, const char *sql, unsigned long length) {
|
int taos_stmt_prepare(TAOS_STMT *stmt, const char *sql, unsigned long length) {
|
||||||
if (stmt == NULL || sql == NULL) {
|
if (stmt == NULL || sql == NULL) {
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
@ -639,14 +593,23 @@ int taos_stmt_prepare(TAOS_STMT *stmt, const char *sql, unsigned long length) {
|
||||||
return stmtPrepare(stmt, sql, length);
|
return stmtPrepare(stmt, sql, length);
|
||||||
}
|
}
|
||||||
|
|
||||||
int taos_stmt_set_tbname_tags(TAOS_STMT *stmt, const char *name, TAOS_BIND *tags) {
|
int taos_stmt_set_tbname_tags(TAOS_STMT *stmt, const char *name, TAOS_BIND_v2 *tags) {
|
||||||
if (stmt == NULL || name == NULL || tags == NULL) {
|
if (stmt == NULL || name == NULL) {
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
return terrno;
|
return terrno;
|
||||||
}
|
}
|
||||||
|
|
||||||
return stmtSetTbNameTags(stmt, name, tags);
|
int32_t code = stmtSetTbName(stmt, name);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (tags) {
|
||||||
|
return stmtSetTbTags(stmt, tags);
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
int taos_stmt_set_tbname(TAOS_STMT *stmt, const char *name) {
|
int taos_stmt_set_tbname(TAOS_STMT *stmt, const char *name) {
|
||||||
|
@ -656,7 +619,75 @@ int taos_stmt_set_tbname(TAOS_STMT *stmt, const char *name) {
|
||||||
return terrno;
|
return terrno;
|
||||||
}
|
}
|
||||||
|
|
||||||
return stmtSetTbNameTags(stmt, name, NULL);
|
return stmtSetTbName(stmt, name);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_bind_param(TAOS_STMT *stmt, TAOS_BIND_v2 *bind) {
|
||||||
|
if (stmt == NULL || bind == NULL) {
|
||||||
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bind->num > 1) {
|
||||||
|
tscError("invalid bind number %d for %s", bind->num, __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stmtBindBatch(stmt, bind, -1);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_bind_param_batch(TAOS_STMT *stmt, TAOS_BIND_v2 *bind) {
|
||||||
|
if (stmt == NULL || bind == NULL) {
|
||||||
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bind->num <= 0 || bind->num > INT16_MAX) {
|
||||||
|
tscError("invalid bind num %d", bind->num);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stmtBindBatch(stmt, bind, -1);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_bind_single_param_batch(TAOS_STMT *stmt, TAOS_BIND_v2 *bind, int colIdx) {
|
||||||
|
if (stmt == NULL || bind == NULL) {
|
||||||
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (colIdx < 0) {
|
||||||
|
tscError("invalid bind column idx %d", colIdx);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stmtBindBatch(stmt, bind, colIdx);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_add_batch(TAOS_STMT *stmt) {
|
||||||
|
if (stmt == NULL) {
|
||||||
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stmtAddBatch(stmt);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_execute(TAOS_STMT *stmt) {
|
||||||
|
if (stmt == NULL) {
|
||||||
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stmtExec(stmt);
|
||||||
}
|
}
|
||||||
|
|
||||||
int taos_stmt_is_insert(TAOS_STMT *stmt, int *insert) {
|
int taos_stmt_is_insert(TAOS_STMT *stmt, int *insert) {
|
||||||
|
@ -679,16 +710,6 @@ int taos_stmt_num_params(TAOS_STMT *stmt, int *nums) {
|
||||||
return stmtGetParamNum(stmt, nums);
|
return stmtGetParamNum(stmt, nums);
|
||||||
}
|
}
|
||||||
|
|
||||||
int taos_stmt_add_batch(TAOS_STMT *stmt) {
|
|
||||||
if (stmt == NULL) {
|
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
|
||||||
return terrno;
|
|
||||||
}
|
|
||||||
|
|
||||||
return stmtAddBatch(stmt);
|
|
||||||
}
|
|
||||||
|
|
||||||
TAOS_RES *taos_stmt_use_result(TAOS_STMT *stmt) {
|
TAOS_RES *taos_stmt_use_result(TAOS_STMT *stmt) {
|
||||||
if (stmt == NULL) {
|
if (stmt == NULL) {
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
@ -699,14 +720,28 @@ TAOS_RES *taos_stmt_use_result(TAOS_STMT *stmt) {
|
||||||
return stmtUseResult(stmt);
|
return stmtUseResult(stmt);
|
||||||
}
|
}
|
||||||
|
|
||||||
int taos_stmt_bind_param_batch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind) {
|
char *taos_stmt_errstr(TAOS_STMT *stmt) {
|
||||||
if (stmt == NULL || bind == NULL) {
|
return (char *)stmtErrstr(stmt);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_affected_rows(TAOS_STMT *stmt) {
|
||||||
|
if (stmt == NULL) {
|
||||||
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return stmtAffectedRows(stmt);
|
||||||
|
}
|
||||||
|
|
||||||
|
int taos_stmt_close(TAOS_STMT *stmt) {
|
||||||
|
if (stmt == NULL) {
|
||||||
tscError("NULL parameter for %s", __FUNCTION__);
|
tscError("NULL parameter for %s", __FUNCTION__);
|
||||||
terrno = TSDB_CODE_INVALID_PARA;
|
terrno = TSDB_CODE_INVALID_PARA;
|
||||||
return terrno;
|
return terrno;
|
||||||
}
|
}
|
||||||
|
|
||||||
return stmtBindBatch(stmt, bind);
|
return stmtClose(stmt);
|
||||||
}
|
}
|
||||||
|
|
||||||
TAOS_RES *taos_schemaless_insert(TAOS *taos, char *lines[], int numLines, int protocol, int precision) {
|
TAOS_RES *taos_schemaless_insert(TAOS *taos, char *lines[], int numLines, int protocol, int precision) {
|
||||||
|
|
|
@ -4,86 +4,539 @@
|
||||||
#include "clientStmt.h"
|
#include "clientStmt.h"
|
||||||
#include "tdef.h"
|
#include "tdef.h"
|
||||||
|
|
||||||
|
int32_t stmtSwitchStatus(STscStmt* pStmt, STMT_STATUS newStatus) {
|
||||||
|
switch (newStatus) {
|
||||||
|
case STMT_SETTBNAME:
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
//STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
|
||||||
|
|
||||||
|
pStmt->sql.status = newStatus;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t stmtGetTbName(TAOS_STMT *stmt, char **tbName) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
pStmt->sql.type = STMT_TYPE_MULTI_INSERT;
|
||||||
|
|
||||||
|
if (NULL == pStmt->bInfo.tbName) {
|
||||||
|
tscError("no table name set");
|
||||||
|
STMT_ERR_RET(TSDB_CODE_TSC_STMT_TBNAME_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
*tbName = pStmt->bInfo.tbName;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtSetBindInfo(TAOS_STMT* stmt, STableMeta* pTableMeta, void* tags) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
pStmt->bInfo.tbUid = pTableMeta->uid;
|
||||||
|
pStmt->bInfo.tbSuid = pTableMeta->suid;
|
||||||
|
pStmt->bInfo.tbType = pTableMeta->tableType;
|
||||||
|
pStmt->bInfo.boundTags = tags;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtSetExecInfo(TAOS_STMT* stmt, SHashObj* pVgHash, SHashObj* pBlockHash) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
pStmt->exec.pVgHash = pVgHash;
|
||||||
|
pStmt->exec.pBlockHash = pBlockHash;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtGetExecInfo(TAOS_STMT* stmt, SHashObj** pVgHash, SHashObj** pBlockHash) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
*pVgHash = pStmt->exec.pVgHash;
|
||||||
|
*pBlockHash = pStmt->exec.pBlockHash;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtCacheBlock(STscStmt *pStmt) {
|
||||||
|
if (pStmt->sql.type != STMT_TYPE_MULTI_INSERT) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
uint64_t uid;
|
||||||
|
if (TSDB_CHILD_TABLE == pStmt->bInfo.tbType) {
|
||||||
|
uid = pStmt->bInfo.tbSuid;
|
||||||
|
} else {
|
||||||
|
ASSERT(TSDB_NORMAL_TABLE == pStmt->bInfo.tbType);
|
||||||
|
uid = pStmt->bInfo.tbUid;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (taosHashGet(pStmt->sql.pTableCache, &uid, sizeof(uid))) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks** pSrc = taosHashGet(pStmt->exec.pBlockHash, &uid, sizeof(uid));
|
||||||
|
STableDataBlocks* pDst = NULL;
|
||||||
|
|
||||||
|
STMT_ERR_RET(qCloneStmtDataBlock(&pDst, *pSrc));
|
||||||
|
|
||||||
|
SStmtTableCache cache = {
|
||||||
|
.pDataBlock = pDst,
|
||||||
|
.boundTags = pStmt->bInfo.boundTags,
|
||||||
|
};
|
||||||
|
|
||||||
|
if (taosHashPut(pStmt->sql.pTableCache, &uid, sizeof(uid), &cache, sizeof(cache))) {
|
||||||
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
|
||||||
|
pStmt->bInfo.boundTags = NULL;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtParseSql(STscStmt* pStmt) {
|
||||||
|
SStmtCallback stmtCb = {
|
||||||
|
.pStmt = pStmt,
|
||||||
|
.getTbNameFn = stmtGetTbName,
|
||||||
|
.setBindInfoFn = stmtSetBindInfo,
|
||||||
|
.setExecInfoFn = stmtSetExecInfo,
|
||||||
|
.getExecInfoFn = stmtGetExecInfo,
|
||||||
|
};
|
||||||
|
|
||||||
|
if (NULL == pStmt->exec.pRequest) {
|
||||||
|
STMT_ERR_RET(buildRequest(pStmt->taos, pStmt->sql.sqlStr, pStmt->sql.sqlLen, &pStmt->exec.pRequest));
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(parseSql(pStmt->exec.pRequest, false, &pStmt->sql.pQuery, &stmtCb));
|
||||||
|
|
||||||
|
pStmt->bInfo.needParse = false;
|
||||||
|
|
||||||
|
switch (nodeType(pStmt->sql.pQuery->pRoot)) {
|
||||||
|
case QUERY_NODE_VNODE_MODIF_STMT:
|
||||||
|
if (0 == pStmt->sql.type) {
|
||||||
|
pStmt->sql.type = STMT_TYPE_INSERT;
|
||||||
|
}
|
||||||
|
break;
|
||||||
|
case QUERY_NODE_SELECT_STMT:
|
||||||
|
pStmt->sql.type = STMT_TYPE_QUERY;
|
||||||
|
break;
|
||||||
|
default:
|
||||||
|
tscError("not supported stmt type %d", nodeType(pStmt->sql.pQuery->pRoot));
|
||||||
|
STMT_ERR_RET(TSDB_CODE_TSC_STMT_CLAUSE_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtCacheBlock(pStmt));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtCleanBindInfo(STscStmt* pStmt) {
|
||||||
|
pStmt->bInfo.tbUid = 0;
|
||||||
|
pStmt->bInfo.tbSuid = 0;
|
||||||
|
pStmt->bInfo.tbType = 0;
|
||||||
|
pStmt->bInfo.needParse = true;
|
||||||
|
|
||||||
|
taosMemoryFreeClear(pStmt->bInfo.tbName);
|
||||||
|
destroyBoundColumnInfo(pStmt->bInfo.boundTags);
|
||||||
|
taosMemoryFreeClear(pStmt->bInfo.boundTags);
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtCleanExecInfo(STscStmt* pStmt, bool keepTable) {
|
||||||
|
taos_free_result(pStmt->exec.pRequest);
|
||||||
|
pStmt->exec.pRequest = NULL;
|
||||||
|
|
||||||
|
void *pIter = taosHashIterate(pStmt->exec.pBlockHash, NULL);
|
||||||
|
while (pIter) {
|
||||||
|
STableDataBlocks* pBlocks = *(STableDataBlocks**)pIter;
|
||||||
|
uint64_t *key = taosHashGetKey(pIter, NULL);
|
||||||
|
|
||||||
|
if (keepTable && (*key == pStmt->bInfo.tbUid)) {
|
||||||
|
STMT_ERR_RET(qResetStmtDataBlock(pBlocks, true));
|
||||||
|
|
||||||
|
pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
qFreeStmtDataBlock(pBlocks);
|
||||||
|
taosHashRemove(pStmt->exec.pBlockHash, key, sizeof(*key));
|
||||||
|
|
||||||
|
pIter = taosHashIterate(pStmt->exec.pBlockHash, pIter);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (keepTable) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosHashCleanup(pStmt->exec.pBlockHash);
|
||||||
|
pStmt->exec.pBlockHash = NULL;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtCleanBindInfo(pStmt));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtCleanSQLInfo(STscStmt* pStmt) {
|
||||||
|
taosMemoryFree(pStmt->sql.sqlStr);
|
||||||
|
qDestroyQuery(pStmt->sql.pQuery);
|
||||||
|
|
||||||
|
void *pIter = taosHashIterate(pStmt->sql.pTableCache, NULL);
|
||||||
|
while (pIter) {
|
||||||
|
SStmtTableCache* pCache = (SStmtTableCache*)pIter;
|
||||||
|
|
||||||
|
qDestroyStmtDataBlock(pCache->pDataBlock);
|
||||||
|
destroyBoundColumnInfo(pCache->boundTags);
|
||||||
|
|
||||||
|
pIter = taosHashIterate(pStmt->sql.pTableCache, pIter);
|
||||||
|
}
|
||||||
|
taosHashCleanup(pStmt->sql.pTableCache);
|
||||||
|
pStmt->sql.pTableCache = NULL;
|
||||||
|
|
||||||
|
memset(&pStmt->sql, 0, sizeof(pStmt->sql));
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtCleanExecInfo(pStmt, false));
|
||||||
|
STMT_ERR_RET(stmtCleanBindInfo(pStmt));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtGetFromCache(STscStmt* pStmt) {
|
||||||
|
pStmt->bInfo.needParse = true;
|
||||||
|
|
||||||
|
if (NULL == pStmt->sql.pTableCache || taosHashGetSize(pStmt->sql.pTableCache) <= 0) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (NULL == pStmt->pCatalog) {
|
||||||
|
STMT_ERR_RET(catalogGetHandle(pStmt->taos->pAppInfo->clusterId, &pStmt->pCatalog));
|
||||||
|
}
|
||||||
|
|
||||||
|
STableMeta *pTableMeta = NULL;
|
||||||
|
SEpSet ep = getEpSet_s(&pStmt->taos->pAppInfo->mgmtEp);
|
||||||
|
STMT_ERR_RET(catalogGetTableMeta(pStmt->pCatalog, pStmt->taos->pAppInfo->pTransporter, &ep, &pStmt->bInfo.sname, &pTableMeta));
|
||||||
|
|
||||||
|
if (pTableMeta->uid == pStmt->bInfo.tbUid) {
|
||||||
|
pStmt->bInfo.needParse = false;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (taosHashGet(pStmt->exec.pBlockHash, &pTableMeta->uid, sizeof(pTableMeta->uid))) {
|
||||||
|
SStmtTableCache* pCache = taosHashGet(pStmt->sql.pTableCache, &pTableMeta->uid, sizeof(pTableMeta->uid));
|
||||||
|
if (NULL == pCache) {
|
||||||
|
tscError("table uid %" PRIx64 "found in exec blockHash, but not in sql blockHash", pTableMeta->uid);
|
||||||
|
STMT_ERR_RET(TSDB_CODE_TSC_APP_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
pStmt->bInfo.needParse = false;
|
||||||
|
|
||||||
|
pStmt->bInfo.tbUid = pTableMeta->uid;
|
||||||
|
pStmt->bInfo.tbSuid = pTableMeta->suid;
|
||||||
|
pStmt->bInfo.tbType = pTableMeta->tableType;
|
||||||
|
pStmt->bInfo.boundTags = pCache->boundTags;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
SStmtTableCache* pCache = taosHashGet(pStmt->sql.pTableCache, &pTableMeta->uid, sizeof(pTableMeta->uid));
|
||||||
|
if (pCache) {
|
||||||
|
pStmt->bInfo.needParse = false;
|
||||||
|
|
||||||
|
pStmt->bInfo.tbUid = pTableMeta->uid;
|
||||||
|
pStmt->bInfo.tbSuid = pTableMeta->suid;
|
||||||
|
pStmt->bInfo.tbType = pTableMeta->tableType;
|
||||||
|
pStmt->bInfo.boundTags = pCache->boundTags;
|
||||||
|
|
||||||
|
STableDataBlocks* pNewBlock = NULL;
|
||||||
|
STMT_ERR_RET(qRebuildStmtDataBlock(&pNewBlock, pCache->pDataBlock));
|
||||||
|
|
||||||
|
if (taosHashPut(pStmt->exec.pBlockHash, &pStmt->bInfo.tbUid, sizeof(pStmt->bInfo.tbUid), &pNewBlock, POINTER_BYTES)) {
|
||||||
|
STMT_ERR_RET(TSDB_CODE_OUT_OF_MEMORY);
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtCleanBindInfo(pStmt));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtResetStmt(STscStmt* pStmt) {
|
||||||
|
STMT_ERR_RET(stmtCleanSQLInfo(pStmt));
|
||||||
|
|
||||||
|
pStmt->sql.pTableCache = taosHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
|
||||||
|
if (NULL == pStmt->sql.pTableCache) {
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
STMT_ERR_RET(terrno);
|
||||||
|
}
|
||||||
|
|
||||||
|
pStmt->sql.status = STMT_INIT;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
TAOS_STMT *stmtInit(TAOS *taos) {
|
TAOS_STMT *stmtInit(TAOS *taos) {
|
||||||
STscObj* pObj = (STscObj*)taos;
|
STscObj* pObj = (STscObj*)taos;
|
||||||
STscStmt* pStmt = NULL;
|
STscStmt* pStmt = NULL;
|
||||||
|
|
||||||
#if 0
|
|
||||||
pStmt = taosMemoryCalloc(1, sizeof(STscStmt));
|
pStmt = taosMemoryCalloc(1, sizeof(STscStmt));
|
||||||
if (pStmt == NULL) {
|
if (NULL == pStmt) {
|
||||||
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
tscError("failed to allocate memory for statement");
|
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pStmt->sql.pTableCache = taosHashInit(100, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), false, HASH_NO_LOCK);
|
||||||
|
if (NULL == pStmt->sql.pTableCache) {
|
||||||
|
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
taosMemoryFree(pStmt);
|
||||||
|
return NULL;
|
||||||
|
}
|
||||||
|
|
||||||
pStmt->taos = pObj;
|
pStmt->taos = pObj;
|
||||||
|
pStmt->bInfo.needParse = true;
|
||||||
SSqlObj* pSql = calloc(1, sizeof(SSqlObj));
|
pStmt->sql.status = STMT_INIT;
|
||||||
|
|
||||||
if (pSql == NULL) {
|
|
||||||
free(pStmt);
|
|
||||||
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
|
||||||
tscError("failed to allocate memory for statement");
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (TSDB_CODE_SUCCESS != tscAllocPayload(&pSql->cmd, TSDB_DEFAULT_PAYLOAD_SIZE)) {
|
|
||||||
free(pSql);
|
|
||||||
free(pStmt);
|
|
||||||
terrno = TSDB_CODE_TSC_OUT_OF_MEMORY;
|
|
||||||
tscError("failed to malloc payload buffer");
|
|
||||||
return NULL;
|
|
||||||
}
|
|
||||||
|
|
||||||
tsem_init(&pSql->rspSem, 0, 0);
|
|
||||||
pSql->signature = pSql;
|
|
||||||
pSql->pTscObj = pObj;
|
|
||||||
pSql->maxRetry = TSDB_MAX_REPLICA;
|
|
||||||
pStmt->pSql = pSql;
|
|
||||||
pStmt->last = STMT_INIT;
|
|
||||||
pStmt->numOfRows = 0;
|
|
||||||
registerSqlObj(pSql);
|
|
||||||
#endif
|
|
||||||
|
|
||||||
return pStmt;
|
return pStmt;
|
||||||
}
|
}
|
||||||
|
|
||||||
int stmtClose(TAOS_STMT *stmt) {
|
int stmtPrepare(TAOS_STMT *stmt, const char *sql, unsigned long length) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
if (pStmt->sql.status >= STMT_PREPARE) {
|
||||||
|
STMT_ERR_RET(stmtResetStmt(pStmt));
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_PREPARE));
|
||||||
|
|
||||||
|
if (length <= 0) {
|
||||||
|
length = strlen(sql);
|
||||||
|
}
|
||||||
|
|
||||||
|
pStmt->sql.sqlStr = strndup(sql, length);
|
||||||
|
pStmt->sql.sqlLen = length;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int stmtSetTbName(TAOS_STMT *stmt, const char *tbName) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTBNAME));
|
||||||
|
|
||||||
|
if (NULL == pStmt->exec.pRequest) {
|
||||||
|
STMT_ERR_RET(buildRequest(pStmt->taos, pStmt->sql.sqlStr, pStmt->sql.sqlLen, &pStmt->exec.pRequest));
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(qCreateSName(&pStmt->bInfo.sname, tbName, pStmt->taos->acctId, pStmt->exec.pRequest->pDb, pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen));
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtGetFromCache(pStmt));
|
||||||
|
|
||||||
|
if (pStmt->bInfo.needParse) {
|
||||||
|
taosMemoryFree(pStmt->bInfo.tbName);
|
||||||
|
pStmt->bInfo.tbName = strdup(tbName);
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int stmtSetTbTags(TAOS_STMT *stmt, TAOS_BIND_v2 *tags) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_SETTAGS));
|
||||||
|
|
||||||
|
if (pStmt->bInfo.needParse) {
|
||||||
|
STMT_ERR_RET(stmtParseSql(pStmt));
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks **pDataBlock = (STableDataBlocks**)taosHashGet(pStmt->exec.pBlockHash, (const char*)&pStmt->bInfo.tbUid, sizeof(pStmt->bInfo.tbUid));
|
||||||
|
if (NULL == pDataBlock) {
|
||||||
|
tscError("table uid %" PRIx64 "not found in exec blockHash", pStmt->bInfo.tbUid);
|
||||||
|
STMT_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(qBindStmtTagsValue(*pDataBlock, pStmt->bInfo.boundTags, pStmt->bInfo.tbSuid, &pStmt->bInfo.sname, tags, pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t stmtFetchTagFields(TAOS_STMT *stmt, int32_t *fieldNum, TAOS_FIELD** fields) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_FETCH_TAG_FIELDS));
|
||||||
|
|
||||||
|
if (pStmt->bInfo.needParse) {
|
||||||
|
STMT_ERR_RET(stmtParseSql(pStmt));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (STMT_TYPE_QUERY == pStmt->sql.type) {
|
||||||
|
tscError("invalid operation to get query tag fileds");
|
||||||
|
STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks **pDataBlock = (STableDataBlocks**)taosHashGet(pStmt->exec.pBlockHash, (const char*)&pStmt->bInfo.tbUid, sizeof(pStmt->bInfo.tbUid));
|
||||||
|
if (NULL == pDataBlock) {
|
||||||
|
tscError("table uid %" PRIx64 "not found in exec blockHash", pStmt->bInfo.tbUid);
|
||||||
|
STMT_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(qBuildStmtTagFields(*pDataBlock, pStmt->bInfo.boundTags, fieldNum, fields));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t stmtFetchColFields(TAOS_STMT *stmt, int32_t *fieldNum, TAOS_FIELD** fields) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_FETCH_COL_FIELDS));
|
||||||
|
|
||||||
|
if (pStmt->bInfo.needParse) {
|
||||||
|
STMT_ERR_RET(stmtParseSql(pStmt));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (STMT_TYPE_QUERY == pStmt->sql.type) {
|
||||||
|
tscError("invalid operation to get query column fileds");
|
||||||
|
STMT_ERR_RET(TSDB_CODE_TSC_STMT_API_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks **pDataBlock = (STableDataBlocks**)taosHashGet(pStmt->exec.pBlockHash, (const char*)&pStmt->bInfo.tbUid, sizeof(pStmt->bInfo.tbUid));
|
||||||
|
if (NULL == pDataBlock) {
|
||||||
|
tscError("table uid %" PRIx64 "not found in exec blockHash", pStmt->bInfo.tbUid);
|
||||||
|
STMT_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
STMT_ERR_RET(qBuildStmtColFields(*pDataBlock, fieldNum, fields));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int stmtBindBatch(TAOS_STMT *stmt, TAOS_BIND_v2 *bind, int32_t colIdx) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_BIND));
|
||||||
|
|
||||||
|
if (pStmt->bInfo.needParse && pStmt->sql.runTimes && pStmt->sql.type > 0 && STMT_TYPE_MULTI_INSERT != pStmt->sql.type) {
|
||||||
|
pStmt->bInfo.needParse = false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (NULL == pStmt->exec.pRequest) {
|
||||||
|
STMT_ERR_RET(buildRequest(pStmt->taos, pStmt->sql.sqlStr, pStmt->sql.sqlLen, &pStmt->exec.pRequest));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pStmt->bInfo.needParse) {
|
||||||
|
STMT_ERR_RET(stmtParseSql(pStmt));
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks **pDataBlock = (STableDataBlocks**)taosHashGet(pStmt->exec.pBlockHash, (const char*)&pStmt->bInfo.tbUid, sizeof(pStmt->bInfo.tbUid));
|
||||||
|
if (NULL == pDataBlock) {
|
||||||
|
tscError("table uid %" PRIx64 "not found in exec blockHash", pStmt->bInfo.tbUid);
|
||||||
|
STMT_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (colIdx < 0) {
|
||||||
|
qBindStmtColsValue(*pDataBlock, bind, pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen);
|
||||||
|
} else {
|
||||||
|
if (colIdx != (pStmt->bInfo.sBindLastIdx + 1) && colIdx != 0) {
|
||||||
|
tscError("bind column index not in sequence");
|
||||||
|
STMT_ERR_RET(TSDB_CODE_QRY_APP_ERROR);
|
||||||
|
}
|
||||||
|
|
||||||
|
pStmt->bInfo.sBindLastIdx = colIdx;
|
||||||
|
|
||||||
|
if (0 == colIdx) {
|
||||||
|
pStmt->bInfo.sBindRowNum = bind->num;
|
||||||
|
}
|
||||||
|
|
||||||
|
qBindStmtSingleColValue(*pDataBlock, bind, pStmt->exec.pRequest->msgBuf, pStmt->exec.pRequest->msgBufLen, colIdx, pStmt->bInfo.sBindRowNum);
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int stmtAddBatch(TAOS_STMT *stmt) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_ADD_BATCH));
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtCacheBlock(pStmt));
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
int stmtExec(TAOS_STMT *stmt) {
|
int stmtExec(TAOS_STMT *stmt) {
|
||||||
return TSDB_CODE_SUCCESS;
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
int32_t code = 0;
|
||||||
|
|
||||||
|
STMT_ERR_RET(stmtSwitchStatus(pStmt, STMT_EXECUTE));
|
||||||
|
|
||||||
|
STMT_ERR_RET(qBuildStmtOutput(pStmt->sql.pQuery, pStmt->exec.pVgHash, pStmt->exec.pBlockHash));
|
||||||
|
|
||||||
|
launchQueryImpl(pStmt->exec.pRequest, pStmt->sql.pQuery, TSDB_CODE_SUCCESS, true);
|
||||||
|
|
||||||
|
STMT_ERR_JRET(pStmt->exec.pRequest->code);
|
||||||
|
|
||||||
|
pStmt->affectedRows += taos_affected_rows(pStmt->exec.pRequest);
|
||||||
|
|
||||||
|
_return:
|
||||||
|
|
||||||
|
stmtCleanExecInfo(pStmt, (code ? false : true));
|
||||||
|
|
||||||
|
++pStmt->sql.runTimes;
|
||||||
|
|
||||||
|
STMT_RET(code);
|
||||||
}
|
}
|
||||||
|
|
||||||
char *stmtErrstr(TAOS_STMT *stmt) {
|
|
||||||
return NULL;
|
int stmtClose(TAOS_STMT *stmt) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
STMT_RET(stmtCleanSQLInfo(pStmt));
|
||||||
|
}
|
||||||
|
|
||||||
|
const char *stmtErrstr(TAOS_STMT *stmt) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
if (stmt == NULL) {
|
||||||
|
return (char*) tstrerror(terrno);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pStmt->exec.pRequest) {
|
||||||
|
pStmt->exec.pRequest->code = terrno;
|
||||||
|
}
|
||||||
|
|
||||||
|
return taos_errstr(pStmt->exec.pRequest);
|
||||||
}
|
}
|
||||||
|
|
||||||
int stmtAffectedRows(TAOS_STMT *stmt) {
|
int stmtAffectedRows(TAOS_STMT *stmt) {
|
||||||
return TSDB_CODE_SUCCESS;
|
return ((STscStmt*)stmt)->affectedRows;
|
||||||
}
|
|
||||||
|
|
||||||
int stmtBind(TAOS_STMT *stmt, TAOS_BIND *bind) {
|
|
||||||
return TSDB_CODE_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
int stmtPrepare(TAOS_STMT *stmt, const char *sql, unsigned long length) {
|
|
||||||
return TSDB_CODE_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
int stmtSetTbNameTags(TAOS_STMT *stmt, const char *name, TAOS_BIND *tags) {
|
|
||||||
return TSDB_CODE_SUCCESS;
|
|
||||||
}
|
}
|
||||||
|
|
||||||
int stmtIsInsert(TAOS_STMT *stmt, int *insert) {
|
int stmtIsInsert(TAOS_STMT *stmt, int *insert) {
|
||||||
|
STscStmt* pStmt = (STscStmt*)stmt;
|
||||||
|
|
||||||
|
if (pStmt->sql.type) {
|
||||||
|
*insert = (STMT_TYPE_INSERT == pStmt->sql.type || STMT_TYPE_MULTI_INSERT == pStmt->sql.type);
|
||||||
|
} else {
|
||||||
|
*insert = isInsertSql(pStmt->sql.sqlStr, 0);
|
||||||
|
}
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
int stmtGetParamNum(TAOS_STMT *stmt, int *nums) {
|
int stmtGetParamNum(TAOS_STMT *stmt, int *nums) {
|
||||||
return TSDB_CODE_SUCCESS;
|
STMT_ERR_RET(stmtFetchColFields(stmt, nums, NULL));
|
||||||
}
|
|
||||||
|
|
||||||
int stmtAddBatch(TAOS_STMT *stmt) {
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -91,9 +544,5 @@ TAOS_RES *stmtUseResult(TAOS_STMT *stmt) {
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
int stmtBindBatch(TAOS_STMT *stmt, TAOS_MULTI_BIND *bind) {
|
|
||||||
return TSDB_CODE_SUCCESS;
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -24,6 +24,7 @@
|
||||||
#include "tqueue.h"
|
#include "tqueue.h"
|
||||||
#include "tref.h"
|
#include "tref.h"
|
||||||
|
|
||||||
|
#if 0
|
||||||
struct tmq_message_t {
|
struct tmq_message_t {
|
||||||
SMqPollRsp msg;
|
SMqPollRsp msg;
|
||||||
char* topic;
|
char* topic;
|
||||||
|
@ -31,6 +32,7 @@ struct tmq_message_t {
|
||||||
int32_t vgId;
|
int32_t vgId;
|
||||||
int32_t resIter;
|
int32_t resIter;
|
||||||
};
|
};
|
||||||
|
#endif
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
int8_t tmqRspType;
|
int8_t tmqRspType;
|
||||||
|
@ -52,9 +54,7 @@ struct tmq_topic_vgroup_t {
|
||||||
};
|
};
|
||||||
|
|
||||||
struct tmq_topic_vgroup_list_t {
|
struct tmq_topic_vgroup_list_t {
|
||||||
int32_t cnt;
|
SArray container; // SArray<tmq_topic_vgroup_t*>
|
||||||
int32_t size;
|
|
||||||
tmq_topic_vgroup_t* elems;
|
|
||||||
};
|
};
|
||||||
|
|
||||||
struct tmq_conf_t {
|
struct tmq_conf_t {
|
||||||
|
@ -63,6 +63,7 @@ struct tmq_conf_t {
|
||||||
int8_t autoCommit;
|
int8_t autoCommit;
|
||||||
int8_t resetOffset;
|
int8_t resetOffset;
|
||||||
uint16_t port;
|
uint16_t port;
|
||||||
|
uint16_t autoCommitInterval;
|
||||||
char* ip;
|
char* ip;
|
||||||
char* user;
|
char* user;
|
||||||
char* pass;
|
char* pass;
|
||||||
|
@ -202,6 +203,11 @@ tmq_conf_res_t tmq_conf_set(tmq_conf_t* conf, const char* key, const char* value
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (strcmp(key, "auto.commit.interval.ms") == 0) {
|
||||||
|
conf->autoCommitInterval = atoi(value);
|
||||||
|
return TMQ_CONF_OK;
|
||||||
|
}
|
||||||
|
|
||||||
if (strcmp(key, "auto.offset.reset") == 0) {
|
if (strcmp(key, "auto.offset.reset") == 0) {
|
||||||
if (strcmp(value, "none") == 0) {
|
if (strcmp(value, "none") == 0) {
|
||||||
conf->resetOffset = TMQ_CONF__RESET_OFFSET__NONE;
|
conf->resetOffset = TMQ_CONF__RESET_OFFSET__NONE;
|
||||||
|
@ -300,7 +306,7 @@ int32_t tmqCommitCb(void* param, const SDataBuf* pMsg, int32_t code) {
|
||||||
SMqCommitCbParam* pParam = (SMqCommitCbParam*)param;
|
SMqCommitCbParam* pParam = (SMqCommitCbParam*)param;
|
||||||
pParam->rspErr = code == 0 ? TMQ_RESP_ERR__SUCCESS : TMQ_RESP_ERR__FAIL;
|
pParam->rspErr = code == 0 ? TMQ_RESP_ERR__SUCCESS : TMQ_RESP_ERR__FAIL;
|
||||||
if (pParam->tmq->commit_cb) {
|
if (pParam->tmq->commit_cb) {
|
||||||
pParam->tmq->commit_cb(pParam->tmq, pParam->rspErr, NULL, NULL);
|
pParam->tmq->commit_cb(pParam->tmq, pParam->rspErr, NULL);
|
||||||
}
|
}
|
||||||
if (!pParam->async) tsem_post(&pParam->rspSem);
|
if (!pParam->async) tsem_post(&pParam->rspSem);
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -322,6 +328,7 @@ tmq_resp_err_t tmq_unsubscribe(tmq_t* tmq) {
|
||||||
return tmq_subscribe(tmq, lst);
|
return tmq_subscribe(tmq, lst);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#if 0
|
||||||
tmq_t* tmq_consumer_new(void* conn, tmq_conf_t* conf, char* errstr, int32_t errstrLen) {
|
tmq_t* tmq_consumer_new(void* conn, tmq_conf_t* conf, char* errstr, int32_t errstrLen) {
|
||||||
tmq_t* pTmq = taosMemoryCalloc(sizeof(tmq_t), 1);
|
tmq_t* pTmq = taosMemoryCalloc(sizeof(tmq_t), 1);
|
||||||
if (pTmq == NULL) {
|
if (pTmq == NULL) {
|
||||||
|
@ -357,8 +364,9 @@ tmq_t* tmq_consumer_new(void* conn, tmq_conf_t* conf, char* errstr, int32_t errs
|
||||||
|
|
||||||
return pTmq;
|
return pTmq;
|
||||||
}
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
tmq_t* tmq_consumer_new1(tmq_conf_t* conf, char* errstr, int32_t errstrLen) {
|
tmq_t* tmq_consumer_new(tmq_conf_t* conf, char* errstr, int32_t errstrLen) {
|
||||||
tmq_t* pTmq = taosMemoryCalloc(1, sizeof(tmq_t));
|
tmq_t* pTmq = taosMemoryCalloc(1, sizeof(tmq_t));
|
||||||
if (pTmq == NULL) {
|
if (pTmq == NULL) {
|
||||||
return NULL;
|
return NULL;
|
||||||
|
@ -369,6 +377,7 @@ tmq_t* tmq_consumer_new1(tmq_conf_t* conf, char* errstr, int32_t errstrLen) {
|
||||||
ASSERT(user);
|
ASSERT(user);
|
||||||
ASSERT(pass);
|
ASSERT(pass);
|
||||||
ASSERT(conf->db);
|
ASSERT(conf->db);
|
||||||
|
ASSERT(conf->groupId[0]);
|
||||||
|
|
||||||
pTmq->pTscObj = taos_connect_internal(conf->ip, user, pass, NULL, conf->db, conf->port, CONN_TYPE__TMQ);
|
pTmq->pTscObj = taos_connect_internal(conf->ip, user, pass, NULL, conf->db, conf->port, CONN_TYPE__TMQ);
|
||||||
if (pTmq->pTscObj == NULL) return NULL;
|
if (pTmq->pTscObj == NULL) return NULL;
|
||||||
|
@ -429,8 +438,8 @@ tmq_resp_err_t tmq_commit(tmq_t* tmq, const tmq_topic_vgroup_list_t* offsets, in
|
||||||
req.num = pArray->size;
|
req.num = pArray->size;
|
||||||
req.offsets = pArray->pData;
|
req.offsets = pArray->pData;
|
||||||
} else {
|
} else {
|
||||||
req.num = offsets->cnt;
|
req.num = taosArrayGetSize(&offsets->container);
|
||||||
req.offsets = (SMqOffset*)offsets->elems;
|
req.offsets = (SMqOffset*)offsets->container.pData;
|
||||||
}
|
}
|
||||||
|
|
||||||
SCoder encoder;
|
SCoder encoder;
|
||||||
|
@ -615,7 +624,7 @@ TAOS_RES* tmq_create_stream(TAOS* taos, const char* streamName, const char* tbNa
|
||||||
|
|
||||||
int32_t code = 0;
|
int32_t code = 0;
|
||||||
CHECK_CODE_GOTO(buildRequest(pTscObj, sql, sqlLen, &pRequest), _return);
|
CHECK_CODE_GOTO(buildRequest(pTscObj, sql, sqlLen, &pRequest), _return);
|
||||||
CHECK_CODE_GOTO(parseSql(pRequest, false, &pQueryNode), _return);
|
CHECK_CODE_GOTO(parseSql(pRequest, false, &pQueryNode, NULL), _return);
|
||||||
|
|
||||||
// todo check for invalid sql statement and return with error code
|
// todo check for invalid sql statement and return with error code
|
||||||
|
|
||||||
|
@ -1538,16 +1547,6 @@ tmq_resp_err_t tmq_commit(tmq_t* tmq, const tmq_topic_vgroup_list_t* tmq_topic_v
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if 0
|
|
||||||
void tmq_message_destroy(tmq_message_t* tmq_message) {
|
|
||||||
if (tmq_message == NULL) return;
|
|
||||||
SMqPollRsp* pRsp = &tmq_message->msg;
|
|
||||||
tDeleteSMqConsumeRsp(pRsp);
|
|
||||||
/*taosMemoryFree(tmq_message);*/
|
|
||||||
taosFreeQitem(tmq_message);
|
|
||||||
}
|
|
||||||
#endif
|
|
||||||
|
|
||||||
tmq_resp_err_t tmq_consumer_close(tmq_t* tmq) { return TMQ_RESP_ERR__SUCCESS; }
|
tmq_resp_err_t tmq_consumer_close(tmq_t* tmq) { return TMQ_RESP_ERR__SUCCESS; }
|
||||||
|
|
||||||
const char* tmq_err2str(tmq_resp_err_t err) {
|
const char* tmq_err2str(tmq_resp_err_t err) {
|
||||||
|
|
|
@ -108,7 +108,7 @@ TEST(testCase, tmq_subscribe_ctb_Test) {
|
||||||
|
|
||||||
while (1) {
|
while (1) {
|
||||||
tmq_message_t* msg = tmq_consumer_poll(tmq, 1000);
|
tmq_message_t* msg = tmq_consumer_poll(tmq, 1000);
|
||||||
tmq_message_destroy(msg);
|
taos_free_result(msg);
|
||||||
//printf("get msg\n");
|
//printf("get msg\n");
|
||||||
//if (msg == NULL) break;
|
//if (msg == NULL) break;
|
||||||
}
|
}
|
||||||
|
@ -141,7 +141,7 @@ TEST(testCase, tmq_subscribe_stb_Test) {
|
||||||
tmq_commit(tmq, NULL, 0);
|
tmq_commit(tmq, NULL, 0);
|
||||||
}
|
}
|
||||||
//tmq_commit(tmq, NULL, 0);
|
//tmq_commit(tmq, NULL, 0);
|
||||||
tmq_message_destroy(msg);
|
taos_free_result(msg);
|
||||||
//printf("get msg\n");
|
//printf("get msg\n");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
|
@ -2756,6 +2756,8 @@ int32_t tSerializeSConnectReq(void *buf, int32_t bufLen, SConnectReq *pReq) {
|
||||||
if (tEncodeI32(&encoder, pReq->pid) < 0) return -1;
|
if (tEncodeI32(&encoder, pReq->pid) < 0) return -1;
|
||||||
if (tEncodeCStr(&encoder, pReq->app) < 0) return -1;
|
if (tEncodeCStr(&encoder, pReq->app) < 0) return -1;
|
||||||
if (tEncodeCStr(&encoder, pReq->db) < 0) return -1;
|
if (tEncodeCStr(&encoder, pReq->db) < 0) return -1;
|
||||||
|
if (tEncodeCStr(&encoder, pReq->user) < 0) return -1;
|
||||||
|
if (tEncodeCStr(&encoder, pReq->passwd) < 0) return -1;
|
||||||
if (tEncodeI64(&encoder, pReq->startTime) < 0) return -1;
|
if (tEncodeI64(&encoder, pReq->startTime) < 0) return -1;
|
||||||
tEndEncode(&encoder);
|
tEndEncode(&encoder);
|
||||||
|
|
||||||
|
@ -2773,6 +2775,8 @@ int32_t tDeserializeSConnectReq(void *buf, int32_t bufLen, SConnectReq *pReq) {
|
||||||
if (tDecodeI32(&decoder, &pReq->pid) < 0) return -1;
|
if (tDecodeI32(&decoder, &pReq->pid) < 0) return -1;
|
||||||
if (tDecodeCStrTo(&decoder, pReq->app) < 0) return -1;
|
if (tDecodeCStrTo(&decoder, pReq->app) < 0) return -1;
|
||||||
if (tDecodeCStrTo(&decoder, pReq->db) < 0) return -1;
|
if (tDecodeCStrTo(&decoder, pReq->db) < 0) return -1;
|
||||||
|
if (tDecodeCStrTo(&decoder, pReq->user) < 0) return -1;
|
||||||
|
if (tDecodeCStrTo(&decoder, pReq->passwd) < 0) return -1;
|
||||||
if (tDecodeI64(&decoder, &pReq->startTime) < 0) return -1;
|
if (tDecodeI64(&decoder, &pReq->startTime) < 0) return -1;
|
||||||
tEndDecode(&decoder);
|
tEndDecode(&decoder);
|
||||||
|
|
||||||
|
@ -3032,7 +3036,6 @@ int32_t tDeserializeSCompactVnodeReq(void *buf, int32_t bufLen, SCompactVnodeReq
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
int32_t tSerializeSAlterVnodeReq(void *buf, int32_t bufLen, SAlterVnodeReq *pReq) {
|
int32_t tSerializeSAlterVnodeReq(void *buf, int32_t bufLen, SAlterVnodeReq *pReq) {
|
||||||
SCoder encoder = {0};
|
SCoder encoder = {0};
|
||||||
tCoderInit(&encoder, TD_LITTLE_ENDIAN, buf, bufLen, TD_ENCODER);
|
tCoderInit(&encoder, TD_LITTLE_ENDIAN, buf, bufLen, TD_ENCODER);
|
||||||
|
@ -3052,7 +3055,7 @@ int32_t tSerializeSAlterVnodeReq(void *buf, int32_t bufLen, SAlterVnodeReq *pReq
|
||||||
SReplica *pReplica = &pReq->replicas[i];
|
SReplica *pReplica = &pReq->replicas[i];
|
||||||
if (tEncodeSReplica(&encoder, pReplica) < 0) return -1;
|
if (tEncodeSReplica(&encoder, pReplica) < 0) return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
tEndEncode(&encoder);
|
tEndEncode(&encoder);
|
||||||
|
|
||||||
int32_t tlen = encoder.pos;
|
int32_t tlen = encoder.pos;
|
||||||
|
@ -3085,7 +3088,6 @@ int32_t tDeserializeSAlterVnodeReq(void *buf, int32_t bufLen, SAlterVnodeReq *pR
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
int32_t tSerializeSKillQueryReq(void *buf, int32_t bufLen, SKillQueryReq *pReq) {
|
int32_t tSerializeSKillQueryReq(void *buf, int32_t bufLen, SKillQueryReq *pReq) {
|
||||||
SCoder encoder = {0};
|
SCoder encoder = {0};
|
||||||
tCoderInit(&encoder, TD_LITTLE_ENDIAN, buf, bufLen, TD_ENCODER);
|
tCoderInit(&encoder, TD_LITTLE_ENDIAN, buf, bufLen, TD_ENCODER);
|
||||||
|
|
|
@ -519,8 +519,9 @@ static int32_t tdAppendKvRowToDataCol(STSRow *pRow, STSchema *pSchema, SDataCols
|
||||||
SKvRowIdx *pIdx = tdKvRowColIdxAt(pRow, rcol);
|
SKvRowIdx *pIdx = tdKvRowColIdxAt(pRow, rcol);
|
||||||
int16_t colIdx = -1;
|
int16_t colIdx = -1;
|
||||||
if (pIdx) {
|
if (pIdx) {
|
||||||
colIdx = POINTER_DISTANCE(pRow->data, pIdx) / sizeof(SKvRowIdx);
|
colIdx = POINTER_DISTANCE(pIdx, TD_ROW_COL_IDX(pRow)) / sizeof(SKvRowIdx);
|
||||||
}
|
}
|
||||||
|
TASSERT(colIdx >= 0);
|
||||||
SCellVal sVal = {0};
|
SCellVal sVal = {0};
|
||||||
if (pIdx->colId == pDataCol->colId) {
|
if (pIdx->colId == pDataCol->colId) {
|
||||||
if (tdGetKvRowValOfCol(&sVal, pRow, pBitmap, pIdx->offset, colIdx) < 0) {
|
if (tdGetKvRowValOfCol(&sVal, pRow, pBitmap, pIdx->offset, colIdx) < 0) {
|
||||||
|
|
|
@ -119,10 +119,10 @@ _OVER:
|
||||||
}
|
}
|
||||||
|
|
||||||
static void dmProcessMsg(SDnode *pDnode, SRpcMsg *pMsg, SEpSet *pEpSet) {
|
static void dmProcessMsg(SDnode *pDnode, SRpcMsg *pMsg, SEpSet *pEpSet) {
|
||||||
SDnodeTrans *pTrans = &pDnode->trans;
|
SDnodeTrans * pTrans = &pDnode->trans;
|
||||||
tmsg_t msgType = pMsg->msgType;
|
tmsg_t msgType = pMsg->msgType;
|
||||||
bool isReq = msgType & 1u;
|
bool isReq = msgType & 1u;
|
||||||
SMsgHandle *pHandle = &pTrans->msgHandles[TMSG_INDEX(msgType)];
|
SMsgHandle * pHandle = &pTrans->msgHandles[TMSG_INDEX(msgType)];
|
||||||
SMgmtWrapper *pWrapper = pHandle->pNdWrapper;
|
SMgmtWrapper *pWrapper = pHandle->pNdWrapper;
|
||||||
|
|
||||||
if (msgType == TDMT_DND_SERVER_STATUS) {
|
if (msgType == TDMT_DND_SERVER_STATUS) {
|
||||||
|
@ -443,7 +443,7 @@ static inline int32_t dmRetrieveUserAuthInfo(SDnode *pDnode, char *user, char *s
|
||||||
SAuthReq authReq = {0};
|
SAuthReq authReq = {0};
|
||||||
tstrncpy(authReq.user, user, TSDB_USER_LEN);
|
tstrncpy(authReq.user, user, TSDB_USER_LEN);
|
||||||
int32_t contLen = tSerializeSAuthReq(NULL, 0, &authReq);
|
int32_t contLen = tSerializeSAuthReq(NULL, 0, &authReq);
|
||||||
void *pReq = rpcMallocCont(contLen);
|
void * pReq = rpcMallocCont(contLen);
|
||||||
tSerializeSAuthReq(pReq, contLen, &authReq);
|
tSerializeSAuthReq(pReq, contLen, &authReq);
|
||||||
|
|
||||||
SRpcMsg rpcMsg = {.pCont = pReq, .contLen = contLen, .msgType = TDMT_MND_AUTH, .ahandle = (void *)9528};
|
SRpcMsg rpcMsg = {.pCont = pReq, .contLen = contLen, .msgType = TDMT_MND_AUTH, .ahandle = (void *)9528};
|
||||||
|
@ -523,4 +523,4 @@ SMsgCb dmGetMsgcb(SMgmtWrapper *pWrapper) {
|
||||||
.pWrapper = pWrapper,
|
.pWrapper = pWrapper,
|
||||||
};
|
};
|
||||||
return msgCb;
|
return msgCb;
|
||||||
}
|
}
|
||||||
|
|
|
@ -29,6 +29,8 @@ int32_t mndSchedInitSubEp(SMnode* pMnode, const SMqTopicObj* pTopic, SMqSubscrib
|
||||||
|
|
||||||
int32_t mndScheduleStream(SMnode* pMnode, STrans* pTrans, SStreamObj* pStream);
|
int32_t mndScheduleStream(SMnode* pMnode, STrans* pTrans, SStreamObj* pStream);
|
||||||
|
|
||||||
|
int32_t mndConvertRSmaTask(const char* ast, int8_t triggerType, int64_t watermark, char** pStr, int32_t* pLen);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -24,20 +24,20 @@
|
||||||
#include "version.h"
|
#include "version.h"
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
uint32_t id;
|
uint32_t id;
|
||||||
int8_t connType;
|
int8_t connType;
|
||||||
char user[TSDB_USER_LEN];
|
char user[TSDB_USER_LEN];
|
||||||
char app[TSDB_APP_NAME_LEN]; // app name that invokes taosc
|
char app[TSDB_APP_NAME_LEN]; // app name that invokes taosc
|
||||||
int64_t appStartTimeMs; // app start time
|
int64_t appStartTimeMs; // app start time
|
||||||
int32_t pid; // pid of app that invokes taosc
|
int32_t pid; // pid of app that invokes taosc
|
||||||
uint32_t ip;
|
uint32_t ip;
|
||||||
uint16_t port;
|
uint16_t port;
|
||||||
int8_t killed;
|
int8_t killed;
|
||||||
int64_t loginTimeMs;
|
int64_t loginTimeMs;
|
||||||
int64_t lastAccessTimeMs;
|
int64_t lastAccessTimeMs;
|
||||||
uint64_t killId;
|
uint64_t killId;
|
||||||
int32_t numOfQueries;
|
int32_t numOfQueries;
|
||||||
SArray *pQueries; //SArray<SQueryDesc>
|
SArray * pQueries; // SArray<SQueryDesc>
|
||||||
} SConnObj;
|
} SConnObj;
|
||||||
|
|
||||||
static SConnObj *mndCreateConn(SMnode *pMnode, const char *user, int8_t connType, uint32_t ip, uint16_t port,
|
static SConnObj *mndCreateConn(SMnode *pMnode, const char *user, int8_t connType, uint32_t ip, uint16_t port,
|
||||||
|
@ -45,7 +45,7 @@ static SConnObj *mndCreateConn(SMnode *pMnode, const char *user, int8_t connType
|
||||||
static void mndFreeConn(SConnObj *pConn);
|
static void mndFreeConn(SConnObj *pConn);
|
||||||
static SConnObj *mndAcquireConn(SMnode *pMnode, uint32_t connId);
|
static SConnObj *mndAcquireConn(SMnode *pMnode, uint32_t connId);
|
||||||
static void mndReleaseConn(SMnode *pMnode, SConnObj *pConn);
|
static void mndReleaseConn(SMnode *pMnode, SConnObj *pConn);
|
||||||
static void *mndGetNextConn(SMnode *pMnode, SCacheIter *pIter);
|
static void * mndGetNextConn(SMnode *pMnode, SCacheIter *pIter);
|
||||||
static void mndCancelGetNextConn(SMnode *pMnode, void *pIter);
|
static void mndCancelGetNextConn(SMnode *pMnode, void *pIter);
|
||||||
static int32_t mndProcessHeartBeatReq(SNodeMsg *pReq);
|
static int32_t mndProcessHeartBeatReq(SNodeMsg *pReq);
|
||||||
static int32_t mndProcessConnectReq(SNodeMsg *pReq);
|
static int32_t mndProcessConnectReq(SNodeMsg *pReq);
|
||||||
|
@ -71,9 +71,9 @@ int32_t mndInitProfile(SMnode *pMnode) {
|
||||||
mndSetMsgHandle(pMnode, TDMT_MND_KILL_QUERY, mndProcessKillQueryReq);
|
mndSetMsgHandle(pMnode, TDMT_MND_KILL_QUERY, mndProcessKillQueryReq);
|
||||||
mndSetMsgHandle(pMnode, TDMT_MND_KILL_CONN, mndProcessKillConnReq);
|
mndSetMsgHandle(pMnode, TDMT_MND_KILL_CONN, mndProcessKillConnReq);
|
||||||
|
|
||||||
// mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_CONNS, mndRetrieveConns);
|
// mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_CONNS, mndRetrieveConns);
|
||||||
mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_CONNS, mndCancelGetNextConn);
|
mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_CONNS, mndCancelGetNextConn);
|
||||||
// mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_QUERIES, mndRetrieveQueries);
|
// mndAddShowRetrieveHandle(pMnode, TSDB_MGMT_TABLE_QUERIES, mndRetrieveQueries);
|
||||||
mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_QUERIES, mndCancelGetNextQuery);
|
mndAddShowFreeIterHandle(pMnode, TSDB_MGMT_TABLE_QUERIES, mndCancelGetNextQuery);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
|
@ -91,7 +91,7 @@ static SConnObj *mndCreateConn(SMnode *pMnode, const char *user, int8_t connType
|
||||||
int32_t pid, const char *app, int64_t startTime) {
|
int32_t pid, const char *app, int64_t startTime) {
|
||||||
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
||||||
|
|
||||||
char connStr[255] = {0};
|
char connStr[255] = {0};
|
||||||
int32_t len = snprintf(connStr, sizeof(connStr), "%s%d%d%d%s", user, ip, port, pid, app);
|
int32_t len = snprintf(connStr, sizeof(connStr), "%s%d%d%d%s", user, ip, port, pid, app);
|
||||||
int32_t connId = mndGenerateUid(connStr, len);
|
int32_t connId = mndGenerateUid(connStr, len);
|
||||||
if (startTime == 0) startTime = taosGetTimestampMs();
|
if (startTime == 0) startTime = taosGetTimestampMs();
|
||||||
|
@ -174,10 +174,10 @@ static void mndCancelGetNextConn(SMnode *pMnode, void *pIter) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t mndProcessConnectReq(SNodeMsg *pReq) {
|
static int32_t mndProcessConnectReq(SNodeMsg *pReq) {
|
||||||
SMnode *pMnode = pReq->pNode;
|
SMnode * pMnode = pReq->pNode;
|
||||||
SUserObj *pUser = NULL;
|
SUserObj * pUser = NULL;
|
||||||
SDbObj *pDb = NULL;
|
SDbObj * pDb = NULL;
|
||||||
SConnObj *pConn = NULL;
|
SConnObj * pConn = NULL;
|
||||||
int32_t code = -1;
|
int32_t code = -1;
|
||||||
SConnectReq connReq = {0};
|
SConnectReq connReq = {0};
|
||||||
char ip[30] = {0};
|
char ip[30] = {0};
|
||||||
|
@ -194,6 +194,11 @@ static int32_t mndProcessConnectReq(SNodeMsg *pReq) {
|
||||||
mError("user:%s, failed to login while acquire user since %s", pReq->user, terrstr());
|
mError("user:%s, failed to login while acquire user since %s", pReq->user, terrstr());
|
||||||
goto CONN_OVER;
|
goto CONN_OVER;
|
||||||
}
|
}
|
||||||
|
if (0 != strncmp(connReq.passwd, pUser->pass, TSDB_PASSWORD_LEN - 1)) {
|
||||||
|
mError("user:%s, failed to auth while acquire user\n %s \r\n %s", pReq->user, connReq.passwd, pUser->pass);
|
||||||
|
code = TSDB_CODE_RPC_AUTH_FAILURE;
|
||||||
|
goto CONN_OVER;
|
||||||
|
}
|
||||||
|
|
||||||
if (connReq.db[0]) {
|
if (connReq.db[0]) {
|
||||||
char db[TSDB_DB_FNAME_LEN];
|
char db[TSDB_DB_FNAME_LEN];
|
||||||
|
@ -253,8 +258,8 @@ static int32_t mndSaveQueryList(SConnObj *pConn, SQueryHbReqBasic *pBasic) {
|
||||||
|
|
||||||
pConn->pQueries = pBasic->queryDesc;
|
pConn->pQueries = pBasic->queryDesc;
|
||||||
pBasic->queryDesc = NULL;
|
pBasic->queryDesc = NULL;
|
||||||
|
|
||||||
pConn->numOfQueries = pBasic->queryDesc ? taosArrayGetSize(pBasic->queryDesc) : 0;
|
pConn->numOfQueries = pBasic->queryDesc ? taosArrayGetSize(pBasic->queryDesc) : 0;
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -324,9 +329,10 @@ static SClientHbRsp *mndMqHbBuildRsp(SMnode *pMnode, SClientHbReq *pReq) {
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHbReq *pHbReq, SClientHbBatchRsp *pBatchRsp) {
|
static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHbReq *pHbReq,
|
||||||
|
SClientHbBatchRsp *pBatchRsp) {
|
||||||
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
||||||
SClientHbRsp hbRsp = {.connKey = pHbReq->connKey, .status = 0, .info = NULL, .query = NULL};
|
SClientHbRsp hbRsp = {.connKey = pHbReq->connKey, .status = 0, .info = NULL, .query = NULL};
|
||||||
|
|
||||||
if (pHbReq->query) {
|
if (pHbReq->query) {
|
||||||
SQueryHbReqBasic *pBasic = pHbReq->query;
|
SQueryHbReqBasic *pBasic = pHbReq->query;
|
||||||
|
@ -335,8 +341,9 @@ static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHb
|
||||||
rpcGetConnInfo(pMsg->handle, &connInfo);
|
rpcGetConnInfo(pMsg->handle, &connInfo);
|
||||||
|
|
||||||
SConnObj *pConn = mndAcquireConn(pMnode, pBasic->connId);
|
SConnObj *pConn = mndAcquireConn(pMnode, pBasic->connId);
|
||||||
if (pConn == NULL) {
|
if (pConn == NULL) {
|
||||||
pConn = mndCreateConn(pMnode, connInfo.user, CONN_TYPE__QUERY, connInfo.clientIp, connInfo.clientPort, pBasic->pid, pBasic->app, 0);
|
pConn = mndCreateConn(pMnode, connInfo.user, CONN_TYPE__QUERY, connInfo.clientIp, connInfo.clientPort,
|
||||||
|
pBasic->pid, pBasic->app, 0);
|
||||||
if (pConn == NULL) {
|
if (pConn == NULL) {
|
||||||
mError("user:%s, conn:%u is freed and failed to create new since %s", connInfo.user, pBasic->connId, terrstr());
|
mError("user:%s, conn:%u is freed and failed to create new since %s", connInfo.user, pBasic->connId, terrstr());
|
||||||
return -1;
|
return -1;
|
||||||
|
@ -345,7 +352,7 @@ static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHb
|
||||||
}
|
}
|
||||||
} else if (pConn->killed) {
|
} else if (pConn->killed) {
|
||||||
mError("user:%s, conn:%u is already killed", connInfo.user, pConn->id);
|
mError("user:%s, conn:%u is already killed", connInfo.user, pConn->id);
|
||||||
mndReleaseConn(pMnode, pConn);
|
mndReleaseConn(pMnode, pConn);
|
||||||
terrno = TSDB_CODE_MND_INVALID_CONNECTION;
|
terrno = TSDB_CODE_MND_INVALID_CONNECTION;
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
@ -369,8 +376,8 @@ static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHb
|
||||||
}
|
}
|
||||||
|
|
||||||
rspBasic->connId = pConn->id;
|
rspBasic->connId = pConn->id;
|
||||||
rspBasic->totalDnodes = 1; //TODO
|
rspBasic->totalDnodes = 1; // TODO
|
||||||
rspBasic->onlineDnodes = 1; //TODO
|
rspBasic->onlineDnodes = 1; // TODO
|
||||||
mndGetMnodeEpSet(pMnode, &rspBasic->epSet);
|
mndGetMnodeEpSet(pMnode, &rspBasic->epSet);
|
||||||
mndReleaseConn(pMnode, pConn);
|
mndReleaseConn(pMnode, pConn);
|
||||||
|
|
||||||
|
@ -379,7 +386,7 @@ static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHb
|
||||||
|
|
||||||
int32_t kvNum = taosHashGetSize(pHbReq->info);
|
int32_t kvNum = taosHashGetSize(pHbReq->info);
|
||||||
if (NULL == pHbReq->info || kvNum <= 0) {
|
if (NULL == pHbReq->info || kvNum <= 0) {
|
||||||
taosArrayPush(pBatchRsp->rsps, &hbRsp);
|
taosArrayPush(pBatchRsp->rsps, &hbRsp);
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -396,7 +403,7 @@ static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHb
|
||||||
|
|
||||||
switch (kv->key) {
|
switch (kv->key) {
|
||||||
case HEARTBEAT_KEY_DBINFO: {
|
case HEARTBEAT_KEY_DBINFO: {
|
||||||
void *rspMsg = NULL;
|
void * rspMsg = NULL;
|
||||||
int32_t rspLen = 0;
|
int32_t rspLen = 0;
|
||||||
mndValidateDbInfo(pMnode, kv->value, kv->valueLen / sizeof(SDbVgVersion), &rspMsg, &rspLen);
|
mndValidateDbInfo(pMnode, kv->value, kv->valueLen / sizeof(SDbVgVersion), &rspMsg, &rspLen);
|
||||||
if (rspMsg && rspLen > 0) {
|
if (rspMsg && rspLen > 0) {
|
||||||
|
@ -406,7 +413,7 @@ static int32_t mndProcessQueryHeartBeat(SMnode *pMnode, SRpcMsg *pMsg, SClientHb
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
case HEARTBEAT_KEY_STBINFO: {
|
case HEARTBEAT_KEY_STBINFO: {
|
||||||
void *rspMsg = NULL;
|
void * rspMsg = NULL;
|
||||||
int32_t rspLen = 0;
|
int32_t rspLen = 0;
|
||||||
mndValidateStbInfo(pMnode, kv->value, kv->valueLen / sizeof(SSTableMetaVersion), &rspMsg, &rspLen);
|
mndValidateStbInfo(pMnode, kv->value, kv->valueLen / sizeof(SSTableMetaVersion), &rspMsg, &rspLen);
|
||||||
if (rspMsg && rspLen > 0) {
|
if (rspMsg && rspLen > 0) {
|
||||||
|
@ -457,7 +464,7 @@ static int32_t mndProcessHeartBeatReq(SNodeMsg *pReq) {
|
||||||
taosArrayDestroyEx(batchReq.reqs, tFreeClientHbReq);
|
taosArrayDestroyEx(batchReq.reqs, tFreeClientHbReq);
|
||||||
|
|
||||||
int32_t tlen = tSerializeSClientHbBatchRsp(NULL, 0, &batchRsp);
|
int32_t tlen = tSerializeSClientHbBatchRsp(NULL, 0, &batchRsp);
|
||||||
void *buf = rpcMallocCont(tlen);
|
void * buf = rpcMallocCont(tlen);
|
||||||
tSerializeSClientHbBatchRsp(buf, tlen, &batchRsp);
|
tSerializeSClientHbBatchRsp(buf, tlen, &batchRsp);
|
||||||
|
|
||||||
int32_t rspNum = (int32_t)taosArrayGetSize(batchRsp.rsps);
|
int32_t rspNum = (int32_t)taosArrayGetSize(batchRsp.rsps);
|
||||||
|
@ -479,7 +486,7 @@ static int32_t mndProcessHeartBeatReq(SNodeMsg *pReq) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t mndProcessKillQueryReq(SNodeMsg *pReq) {
|
static int32_t mndProcessKillQueryReq(SNodeMsg *pReq) {
|
||||||
SMnode *pMnode = pReq->pNode;
|
SMnode * pMnode = pReq->pNode;
|
||||||
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pReq->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pReq->user);
|
||||||
|
@ -513,7 +520,7 @@ static int32_t mndProcessKillQueryReq(SNodeMsg *pReq) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t mndProcessKillConnReq(SNodeMsg *pReq) {
|
static int32_t mndProcessKillConnReq(SNodeMsg *pReq) {
|
||||||
SMnode *pMnode = pReq->pNode;
|
SMnode * pMnode = pReq->pNode;
|
||||||
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
SProfileMgmt *pMgmt = &pMnode->profileMgmt;
|
||||||
|
|
||||||
SUserObj *pUser = mndAcquireUser(pMnode, pReq->user);
|
SUserObj *pUser = mndAcquireUser(pMnode, pReq->user);
|
||||||
|
@ -545,11 +552,11 @@ static int32_t mndProcessKillConnReq(SNodeMsg *pReq) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t mndRetrieveConns(SNodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows) {
|
static int32_t mndRetrieveConns(SNodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows) {
|
||||||
SMnode *pMnode = pReq->pNode;
|
SMnode * pMnode = pReq->pNode;
|
||||||
int32_t numOfRows = 0;
|
int32_t numOfRows = 0;
|
||||||
SConnObj *pConn = NULL;
|
SConnObj *pConn = NULL;
|
||||||
int32_t cols = 0;
|
int32_t cols = 0;
|
||||||
char *pWrite;
|
char * pWrite;
|
||||||
char ipStr[TSDB_IPv4ADDR_LEN + 6];
|
char ipStr[TSDB_IPv4ADDR_LEN + 6];
|
||||||
|
|
||||||
if (pShow->pIter == NULL) {
|
if (pShow->pIter == NULL) {
|
||||||
|
@ -604,8 +611,8 @@ static int32_t mndRetrieveConns(SNodeMsg *pReq, SShowObj *pShow, char *data, int
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t mndRetrieveQueries(SNodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows) {
|
static int32_t mndRetrieveQueries(SNodeMsg *pReq, SShowObj *pShow, char *data, int32_t rows) {
|
||||||
SMnode *pMnode = pReq->pNode;
|
SMnode *pMnode = pReq->pNode;
|
||||||
int32_t numOfRows = 0;
|
int32_t numOfRows = 0;
|
||||||
#if 0
|
#if 0
|
||||||
SConnObj *pConn = NULL;
|
SConnObj *pConn = NULL;
|
||||||
int32_t cols = 0;
|
int32_t cols = 0;
|
||||||
|
@ -703,7 +710,7 @@ static int32_t mndRetrieveQueries(SNodeMsg *pReq, SShowObj *pShow, char *data, i
|
||||||
}
|
}
|
||||||
|
|
||||||
pShow->numOfRows += numOfRows;
|
pShow->numOfRows += numOfRows;
|
||||||
#endif
|
#endif
|
||||||
return numOfRows;
|
return numOfRows;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -34,6 +34,54 @@
|
||||||
|
|
||||||
extern bool tsStreamSchedV;
|
extern bool tsStreamSchedV;
|
||||||
|
|
||||||
|
int32_t mndConvertRSmaTask(const char* ast, int8_t triggerType, int64_t watermark, char** pStr, int32_t* pLen) {
|
||||||
|
SNode* pAst = NULL;
|
||||||
|
SQueryPlan* pPlan = NULL;
|
||||||
|
terrno = TSDB_CODE_SUCCESS;
|
||||||
|
|
||||||
|
if (nodesStringToNode(ast, &pAst) < 0) {
|
||||||
|
terrno = TSDB_CODE_QRY_INVALID_INPUT;
|
||||||
|
goto END;
|
||||||
|
}
|
||||||
|
|
||||||
|
SPlanContext cxt = {
|
||||||
|
.pAstRoot = pAst,
|
||||||
|
.topicQuery = false,
|
||||||
|
.streamQuery = true,
|
||||||
|
.rSmaQuery = true,
|
||||||
|
.triggerType = triggerType,
|
||||||
|
.watermark = watermark,
|
||||||
|
};
|
||||||
|
if (qCreateQueryPlan(&cxt, &pPlan, NULL) < 0) {
|
||||||
|
terrno = TSDB_CODE_QRY_INVALID_INPUT;
|
||||||
|
goto END;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t levelNum = LIST_LENGTH(pPlan->pSubplans);
|
||||||
|
if (levelNum != 1) {
|
||||||
|
terrno = TSDB_CODE_QRY_INVALID_INPUT;
|
||||||
|
goto END;
|
||||||
|
}
|
||||||
|
SNodeListNode* inner = nodesListGetNode(pPlan->pSubplans, 0);
|
||||||
|
|
||||||
|
int32_t opNum = LIST_LENGTH(inner->pNodeList);
|
||||||
|
if (opNum != 1) {
|
||||||
|
terrno = TSDB_CODE_QRY_INVALID_INPUT;
|
||||||
|
goto END;
|
||||||
|
}
|
||||||
|
|
||||||
|
SSubplan* plan = nodesListGetNode(inner->pNodeList, 0);
|
||||||
|
if (qSubPlanToString(plan, pStr, pLen) < 0) {
|
||||||
|
terrno = TSDB_CODE_QRY_INVALID_INPUT;
|
||||||
|
goto END;
|
||||||
|
}
|
||||||
|
|
||||||
|
END:
|
||||||
|
if (pAst) nodesDestroyNode(pAst);
|
||||||
|
if (pPlan) nodesDestroyNode(pPlan);
|
||||||
|
return terrno;
|
||||||
|
}
|
||||||
|
|
||||||
int32_t mndPersistTaskDeployReq(STrans* pTrans, SStreamTask* pTask, const SEpSet* pEpSet, tmsg_t type, int32_t nodeId) {
|
int32_t mndPersistTaskDeployReq(STrans* pTrans, SStreamTask* pTask, const SEpSet* pEpSet, tmsg_t type, int32_t nodeId) {
|
||||||
SCoder encoder;
|
SCoder encoder;
|
||||||
tCoderInit(&encoder, TD_LITTLE_ENDIAN, NULL, 0, TD_ENCODER);
|
tCoderInit(&encoder, TD_LITTLE_ENDIAN, NULL, 0, TD_ENCODER);
|
||||||
|
|
|
@ -30,8 +30,15 @@ int32_t MndTestProfile::connId;
|
||||||
TEST_F(MndTestProfile, 01_ConnectMsg) {
|
TEST_F(MndTestProfile, 01_ConnectMsg) {
|
||||||
SConnectReq connectReq = {0};
|
SConnectReq connectReq = {0};
|
||||||
connectReq.pid = 1234;
|
connectReq.pid = 1234;
|
||||||
|
|
||||||
|
char passwd[] = "taosdata";
|
||||||
|
char secretEncrypt[TSDB_PASSWORD_LEN] = {0};
|
||||||
|
taosEncryptPass_c((uint8_t*)passwd, strlen(passwd), secretEncrypt);
|
||||||
|
|
||||||
strcpy(connectReq.app, "mnode_test_profile");
|
strcpy(connectReq.app, "mnode_test_profile");
|
||||||
strcpy(connectReq.db, "");
|
strcpy(connectReq.db, "");
|
||||||
|
strcpy(connectReq.user, "root");
|
||||||
|
strcpy(connectReq.passwd, secretEncrypt);
|
||||||
|
|
||||||
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
||||||
void* pReq = rpcMallocCont(contLen);
|
void* pReq = rpcMallocCont(contLen);
|
||||||
|
@ -58,10 +65,16 @@ TEST_F(MndTestProfile, 01_ConnectMsg) {
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F(MndTestProfile, 02_ConnectMsg_InvalidDB) {
|
TEST_F(MndTestProfile, 02_ConnectMsg_InvalidDB) {
|
||||||
|
char passwd[] = "taosdata";
|
||||||
|
char secretEncrypt[TSDB_PASSWORD_LEN] = {0};
|
||||||
|
taosEncryptPass_c((uint8_t*)passwd, strlen(passwd), secretEncrypt);
|
||||||
|
|
||||||
SConnectReq connectReq = {0};
|
SConnectReq connectReq = {0};
|
||||||
connectReq.pid = 1234;
|
connectReq.pid = 1234;
|
||||||
strcpy(connectReq.app, "mnode_test_profile");
|
strcpy(connectReq.app, "mnode_test_profile");
|
||||||
strcpy(connectReq.db, "invalid_db");
|
strcpy(connectReq.db, "invalid_db");
|
||||||
|
strcpy(connectReq.user, "root");
|
||||||
|
strcpy(connectReq.passwd, secretEncrypt);
|
||||||
|
|
||||||
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
||||||
void* pReq = rpcMallocCont(contLen);
|
void* pReq = rpcMallocCont(contLen);
|
||||||
|
|
|
@ -54,10 +54,16 @@ TEST_F(MndTestShow, 02_ShowMsg_InvalidMsgStart) {
|
||||||
}
|
}
|
||||||
|
|
||||||
TEST_F(MndTestShow, 03_ShowMsg_Conn) {
|
TEST_F(MndTestShow, 03_ShowMsg_Conn) {
|
||||||
|
char passwd[] = "taosdata";
|
||||||
|
char secretEncrypt[TSDB_PASSWORD_LEN] = {0};
|
||||||
|
taosEncryptPass_c((uint8_t*)passwd, strlen(passwd), secretEncrypt);
|
||||||
|
|
||||||
SConnectReq connectReq = {0};
|
SConnectReq connectReq = {0};
|
||||||
connectReq.pid = 1234;
|
connectReq.pid = 1234;
|
||||||
strcpy(connectReq.app, "mnode_test_show");
|
strcpy(connectReq.app, "mnode_test_show");
|
||||||
strcpy(connectReq.db, "");
|
strcpy(connectReq.db, "");
|
||||||
|
strcpy(connectReq.user, "root");
|
||||||
|
strcpy(connectReq.passwd, secretEncrypt);
|
||||||
|
|
||||||
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
int32_t contLen = tSerializeSConnectReq(NULL, 0, &connectReq);
|
||||||
void* pReq = rpcMallocCont(contLen);
|
void* pReq = rpcMallocCont(contLen);
|
||||||
|
@ -74,4 +80,4 @@ TEST_F(MndTestShow, 03_ShowMsg_Conn) {
|
||||||
TEST_F(MndTestShow, 04_ShowMsg_Cluster) {
|
TEST_F(MndTestShow, 04_ShowMsg_Cluster) {
|
||||||
test.SendShowReq(TSDB_MGMT_TABLE_CLUSTER, "cluster", "");
|
test.SendShowReq(TSDB_MGMT_TABLE_CLUSTER, "cluster", "");
|
||||||
EXPECT_EQ(test.GetShowRows(), 1);
|
EXPECT_EQ(test.GetShowRows(), 1);
|
||||||
}
|
}
|
||||||
|
|
|
@ -109,8 +109,7 @@ int tqReadHandleSetTbUidList(STqReadHandle *pHandle, const SArray *tbUidList
|
||||||
int tqReadHandleAddTbUidList(STqReadHandle *pHandle, const SArray *tbUidList);
|
int tqReadHandleAddTbUidList(STqReadHandle *pHandle, const SArray *tbUidList);
|
||||||
int32_t tqReadHandleSetMsg(STqReadHandle *pHandle, SSubmitReq *pMsg, int64_t ver);
|
int32_t tqReadHandleSetMsg(STqReadHandle *pHandle, SSubmitReq *pMsg, int64_t ver);
|
||||||
bool tqNextDataBlock(STqReadHandle *pHandle);
|
bool tqNextDataBlock(STqReadHandle *pHandle);
|
||||||
int tqRetrieveDataBlockInfo(STqReadHandle *pHandle, SDataBlockInfo *pBlockInfo);
|
int32_t tqRetrieveDataBlock(SArray **ppCols, STqReadHandle *pHandle, uint64_t *pGroupId, int32_t *pNumOfRows);
|
||||||
SArray *tqRetrieveDataBlock(STqReadHandle *pHandle);
|
|
||||||
|
|
||||||
// need to reposition
|
// need to reposition
|
||||||
|
|
||||||
|
|
|
@ -82,16 +82,7 @@ bool tqNextDataBlock(STqReadHandle* pHandle) {
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
int tqRetrieveDataBlockInfo(STqReadHandle* pHandle, SDataBlockInfo* pBlockInfo) {
|
int32_t tqRetrieveDataBlock(SArray** ppCols, STqReadHandle* pHandle, uint64_t* pGroupId, int32_t* pNumOfRows) {
|
||||||
// currently only rows are used
|
|
||||||
|
|
||||||
pBlockInfo->numOfCols = taosArrayGetSize(pHandle->pColIdList);
|
|
||||||
pBlockInfo->rows = pHandle->pBlock->numOfRows;
|
|
||||||
// pBlockInfo->uid = pHandle->pBlock->uid; // the uid can not be assigned to pBlockData.
|
|
||||||
return 0;
|
|
||||||
}
|
|
||||||
|
|
||||||
SArray* tqRetrieveDataBlock(STqReadHandle* pHandle) {
|
|
||||||
/*int32_t sversion = pHandle->pBlock->sversion;*/
|
/*int32_t sversion = pHandle->pBlock->sversion;*/
|
||||||
// TODO set to real sversion
|
// TODO set to real sversion
|
||||||
int32_t sversion = 0;
|
int32_t sversion = 0;
|
||||||
|
@ -112,7 +103,7 @@ SArray* tqRetrieveDataBlock(STqReadHandle* pHandle) {
|
||||||
STSchema* pTschema = pHandle->pSchema;
|
STSchema* pTschema = pHandle->pSchema;
|
||||||
SSchemaWrapper* pSchemaWrapper = pHandle->pSchemaWrapper;
|
SSchemaWrapper* pSchemaWrapper = pHandle->pSchemaWrapper;
|
||||||
|
|
||||||
int32_t numOfRows = pHandle->pBlock->numOfRows;
|
*pNumOfRows = pHandle->pBlock->numOfRows;
|
||||||
/*int32_t numOfCols = pHandle->pSchema->numOfCols;*/
|
/*int32_t numOfCols = pHandle->pSchema->numOfCols;*/
|
||||||
int32_t colNumNeed = taosArrayGetSize(pHandle->pColIdList);
|
int32_t colNumNeed = taosArrayGetSize(pHandle->pColIdList);
|
||||||
|
|
||||||
|
@ -120,10 +111,11 @@ SArray* tqRetrieveDataBlock(STqReadHandle* pHandle) {
|
||||||
colNumNeed = pSchemaWrapper->nCols;
|
colNumNeed = pSchemaWrapper->nCols;
|
||||||
}
|
}
|
||||||
|
|
||||||
SArray* pArray = taosArrayInit(colNumNeed, sizeof(SColumnInfoData));
|
*ppCols = taosArrayInit(colNumNeed, sizeof(SColumnInfoData));
|
||||||
if (pArray == NULL) {
|
if (*ppCols == NULL) {
|
||||||
return NULL;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t colMeta = 0;
|
int32_t colMeta = 0;
|
||||||
int32_t colNeed = 0;
|
int32_t colNeed = 0;
|
||||||
while (colMeta < pSchemaWrapper->nCols && colNeed < colNumNeed) {
|
while (colMeta < pSchemaWrapper->nCols && colNeed < colNumNeed) {
|
||||||
|
@ -136,21 +128,24 @@ SArray* tqRetrieveDataBlock(STqReadHandle* pHandle) {
|
||||||
colNeed++;
|
colNeed++;
|
||||||
} else {
|
} else {
|
||||||
SColumnInfoData colInfo = {0};
|
SColumnInfoData colInfo = {0};
|
||||||
/*int sz = numOfRows * pColSchema->bytes;*/
|
|
||||||
colInfo.info.bytes = pColSchema->bytes;
|
colInfo.info.bytes = pColSchema->bytes;
|
||||||
colInfo.info.colId = pColSchema->colId;
|
colInfo.info.colId = pColSchema->colId;
|
||||||
colInfo.info.type = pColSchema->type;
|
colInfo.info.type = pColSchema->type;
|
||||||
|
|
||||||
if (colInfoDataEnsureCapacity(&colInfo, 0, numOfRows) < 0) {
|
if (colInfoDataEnsureCapacity(&colInfo, 0, *pNumOfRows) < 0) {
|
||||||
taosArrayDestroyEx(pArray, (void (*)(void*))tDeleteSSDataBlock);
|
goto FAIL;
|
||||||
return NULL;
|
|
||||||
}
|
}
|
||||||
taosArrayPush(pArray, &colInfo);
|
taosArrayPush(*ppCols, &colInfo);
|
||||||
colMeta++;
|
colMeta++;
|
||||||
colNeed++;
|
colNeed++;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int32_t colActual = taosArrayGetSize(*ppCols);
|
||||||
|
|
||||||
|
// TODO in stream shuffle case, fetch groupId
|
||||||
|
*pGroupId = 0;
|
||||||
|
|
||||||
STSRowIter iter = {0};
|
STSRowIter iter = {0};
|
||||||
tdSTSRowIterInit(&iter, pTschema);
|
tdSTSRowIterInit(&iter, pTschema);
|
||||||
STSRow* row;
|
STSRow* row;
|
||||||
|
@ -159,22 +154,22 @@ SArray* tqRetrieveDataBlock(STqReadHandle* pHandle) {
|
||||||
while ((row = tGetSubmitBlkNext(&pHandle->blkIter)) != NULL) {
|
while ((row = tGetSubmitBlkNext(&pHandle->blkIter)) != NULL) {
|
||||||
tdSTSRowIterReset(&iter, row);
|
tdSTSRowIterReset(&iter, row);
|
||||||
// get all wanted col of that block
|
// get all wanted col of that block
|
||||||
int32_t colTot = taosArrayGetSize(pArray);
|
for (int32_t i = 0; i < colActual; i++) {
|
||||||
for (int32_t i = 0; i < colTot; i++) {
|
SColumnInfoData* pColData = taosArrayGet(*ppCols, i);
|
||||||
SColumnInfoData* pColData = taosArrayGet(pArray, i);
|
|
||||||
SCellVal sVal = {0};
|
SCellVal sVal = {0};
|
||||||
if (!tdSTSRowIterNext(&iter, pColData->info.colId, pColData->info.type, &sVal)) {
|
if (!tdSTSRowIterNext(&iter, pColData->info.colId, pColData->info.type, &sVal)) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
/*if (colDataAppend(pColData, curRow, sVal.val, false) < 0) {*/
|
|
||||||
if (colDataAppend(pColData, curRow, sVal.val, sVal.valType == TD_VTYPE_NULL) < 0) {
|
if (colDataAppend(pColData, curRow, sVal.val, sVal.valType == TD_VTYPE_NULL) < 0) {
|
||||||
taosArrayDestroyEx(pArray, (void (*)(void*))tDeleteSSDataBlock);
|
goto FAIL;
|
||||||
return NULL;
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
curRow++;
|
curRow++;
|
||||||
}
|
}
|
||||||
return pArray;
|
return 0;
|
||||||
|
FAIL:
|
||||||
|
taosArrayDestroy(*ppCols);
|
||||||
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
void tqReadHandleSetColIdList(STqReadHandle* pReadHandle, SArray* pColIdList) { pReadHandle->pColIdList = pColIdList; }
|
void tqReadHandleSetColIdList(STqReadHandle* pReadHandle, SArray* pColIdList) { pReadHandle->pColIdList = pColIdList; }
|
||||||
|
|
|
@ -30,12 +30,11 @@
|
||||||
#include "query.h"
|
#include "query.h"
|
||||||
#include "tcompare.h"
|
#include "tcompare.h"
|
||||||
#include "thash.h"
|
#include "thash.h"
|
||||||
#include "vnode.h"
|
|
||||||
#include "ttypes.h"
|
#include "ttypes.h"
|
||||||
|
#include "vnode.h"
|
||||||
|
|
||||||
#define SET_REVERSE_SCAN_FLAG(_info) ((_info)->scanFlag = REVERSE_SCAN)
|
#define SET_REVERSE_SCAN_FLAG(_info) ((_info)->scanFlag = REVERSE_SCAN)
|
||||||
#define SWITCH_ORDER(n) (((n) = ((n) == TSDB_ORDER_ASC) ? TSDB_ORDER_DESC : TSDB_ORDER_ASC))
|
#define SWITCH_ORDER(n) (((n) = ((n) == TSDB_ORDER_ASC) ? TSDB_ORDER_DESC : TSDB_ORDER_ASC))
|
||||||
|
|
||||||
|
|
||||||
void switchCtxOrder(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
|
void switchCtxOrder(SqlFunctionCtx* pCtx, int32_t numOfOutput) {
|
||||||
for (int32_t i = 0; i < numOfOutput; ++i) {
|
for (int32_t i = 0; i < numOfOutput; ++i) {
|
||||||
|
@ -90,7 +89,7 @@ static void getNextTimeWindow(SInterval* pInterval, STimeWindow* tw, int32_t ord
|
||||||
}
|
}
|
||||||
|
|
||||||
int64_t key = tw->skey, interval = pInterval->interval;
|
int64_t key = tw->skey, interval = pInterval->interval;
|
||||||
//convert key to second
|
// convert key to second
|
||||||
key = convertTimePrecision(key, pInterval->precision, TSDB_TIME_PRECISION_MILLI) / 1000;
|
key = convertTimePrecision(key, pInterval->precision, TSDB_TIME_PRECISION_MILLI) / 1000;
|
||||||
|
|
||||||
if (pInterval->intervalUnit == 'y') {
|
if (pInterval->intervalUnit == 'y') {
|
||||||
|
@ -98,7 +97,7 @@ static void getNextTimeWindow(SInterval* pInterval, STimeWindow* tw, int32_t ord
|
||||||
}
|
}
|
||||||
|
|
||||||
struct tm tm;
|
struct tm tm;
|
||||||
time_t t = (time_t)key;
|
time_t t = (time_t)key;
|
||||||
taosLocalTime(&t, &tm);
|
taosLocalTime(&t, &tm);
|
||||||
|
|
||||||
int mon = (int)(tm.tm_year * 12 + tm.tm_mon + interval * factor);
|
int mon = (int)(tm.tm_year * 12 + tm.tm_mon + interval * factor);
|
||||||
|
@ -125,8 +124,8 @@ static bool overlapWithTimeWindow(SInterval* pInterval, SDataBlockInfo* pBlockIn
|
||||||
// todo handle the time range case
|
// todo handle the time range case
|
||||||
TSKEY sk = INT64_MIN;
|
TSKEY sk = INT64_MIN;
|
||||||
TSKEY ek = INT64_MAX;
|
TSKEY ek = INT64_MAX;
|
||||||
// TSKEY sk = MIN(pQueryAttr->window.skey, pQueryAttr->window.ekey);
|
// TSKEY sk = MIN(pQueryAttr->window.skey, pQueryAttr->window.ekey);
|
||||||
// TSKEY ek = MAX(pQueryAttr->window.skey, pQueryAttr->window.ekey);
|
// TSKEY ek = MAX(pQueryAttr->window.skey, pQueryAttr->window.ekey);
|
||||||
|
|
||||||
if (true) {
|
if (true) {
|
||||||
getAlignQueryTimeWindow(pInterval, pInterval->precision, pBlockInfo->window.skey, &w);
|
getAlignQueryTimeWindow(pInterval, pInterval->precision, pBlockInfo->window.skey, &w);
|
||||||
|
@ -136,7 +135,7 @@ static bool overlapWithTimeWindow(SInterval* pInterval, SDataBlockInfo* pBlockIn
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
while(1) { // todo handle the desc order scan case
|
while (1) { // todo handle the desc order scan case
|
||||||
getNextTimeWindow(pInterval, &w, TSDB_ORDER_ASC);
|
getNextTimeWindow(pInterval, &w, TSDB_ORDER_ASC);
|
||||||
if (w.skey > pBlockInfo->window.ekey) {
|
if (w.skey > pBlockInfo->window.ekey) {
|
||||||
break;
|
break;
|
||||||
|
@ -148,31 +147,31 @@ static bool overlapWithTimeWindow(SInterval* pInterval, SDataBlockInfo* pBlockIn
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.ekey, sk, ek, &w);
|
// getAlignQueryTimeWindow(pQueryAttr, pBlockInfo->window.ekey, sk, ek, &w);
|
||||||
// assert(w.skey <= pBlockInfo->window.ekey);
|
// assert(w.skey <= pBlockInfo->window.ekey);
|
||||||
//
|
//
|
||||||
// if (w.skey > pBlockInfo->window.skey) {
|
// if (w.skey > pBlockInfo->window.skey) {
|
||||||
// return true;
|
// return true;
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
// while(1) {
|
// while(1) {
|
||||||
// getNextTimeWindow(pQueryAttr, &w);
|
// getNextTimeWindow(pQueryAttr, &w);
|
||||||
// if (w.ekey < pBlockInfo->window.skey) {
|
// if (w.ekey < pBlockInfo->window.skey) {
|
||||||
// break;
|
// break;
|
||||||
// }
|
// }
|
||||||
//
|
//
|
||||||
// assert(w.skey < pBlockInfo->window.skey);
|
// assert(w.skey < pBlockInfo->window.skey);
|
||||||
// if (w.ekey < pBlockInfo->window.ekey && w.ekey >= pBlockInfo->window.skey) {
|
// if (w.ekey < pBlockInfo->window.ekey && w.ekey >= pBlockInfo->window.skey) {
|
||||||
// return true;
|
// return true;
|
||||||
// }
|
// }
|
||||||
// }
|
// }
|
||||||
}
|
}
|
||||||
|
|
||||||
return false;
|
return false;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock, uint32_t* status) {
|
int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanInfo* pTableScanInfo, SSDataBlock* pBlock, uint32_t* status) {
|
||||||
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
|
SExecTaskInfo* pTaskInfo = pOperator->pTaskInfo;
|
||||||
STableScanInfo* pInfo = pOperator->info;
|
STableScanInfo* pInfo = pOperator->info;
|
||||||
|
|
||||||
STaskCostInfo* pCost = &pTaskInfo->cost;
|
STaskCostInfo* pCost = &pTaskInfo->cost;
|
||||||
|
@ -189,13 +188,13 @@ int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanInfo* pTableScanInfo,
|
||||||
taosMemoryFreeClear(pBlock->pBlockAgg);
|
taosMemoryFreeClear(pBlock->pBlockAgg);
|
||||||
|
|
||||||
if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
|
if (*status == FUNC_DATA_REQUIRED_FILTEROUT) {
|
||||||
qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo), pBlockInfo->window.skey,
|
qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
|
||||||
pBlockInfo->window.ekey, pBlockInfo->rows);
|
pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
|
||||||
pCost->filterOutBlocks += 1;
|
pCost->filterOutBlocks += 1;
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
} else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
|
} else if (*status == FUNC_DATA_REQUIRED_NOT_LOAD) {
|
||||||
qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo), pBlockInfo->window.skey,
|
qDebug("%s data block skipped, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
|
||||||
pBlockInfo->window.ekey, pBlockInfo->rows);
|
pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
|
||||||
pCost->skipBlocks += 1;
|
pCost->skipBlocks += 1;
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
} else if (*status == FUNC_DATA_REQUIRED_STATIS_LOAD) {
|
} else if (*status == FUNC_DATA_REQUIRED_STATIS_LOAD) {
|
||||||
|
@ -218,12 +217,12 @@ int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanInfo* pTableScanInfo,
|
||||||
}
|
}
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
} else { // failed to load the block sma data, data block statistics does not exist, load data block instead
|
} else { // failed to load the block sma data, data block statistics does not exist, load data block instead
|
||||||
*status = FUNC_DATA_REQUIRED_DATA_LOAD;
|
*status = FUNC_DATA_REQUIRED_DATA_LOAD;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
ASSERT (*status == FUNC_DATA_REQUIRED_DATA_LOAD);
|
ASSERT(*status == FUNC_DATA_REQUIRED_DATA_LOAD);
|
||||||
|
|
||||||
// todo filter data block according to the block sma data firstly
|
// todo filter data block according to the block sma data firstly
|
||||||
#if 0
|
#if 0
|
||||||
|
@ -249,8 +248,8 @@ int32_t loadDataBlock(SOperatorInfo* pOperator, STableScanInfo* pTableScanInfo,
|
||||||
doFilter(pTableScanInfo->pFilterNode, pBlock);
|
doFilter(pTableScanInfo->pFilterNode, pBlock);
|
||||||
if (pBlock->info.rows == 0) {
|
if (pBlock->info.rows == 0) {
|
||||||
pCost->filterOutBlocks += 1;
|
pCost->filterOutBlocks += 1;
|
||||||
qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo), pBlockInfo->window.skey,
|
qDebug("%s data block filter out, brange:%" PRId64 "-%" PRId64 ", rows:%d", GET_TASKID(pTaskInfo),
|
||||||
pBlockInfo->window.ekey, pBlockInfo->rows);
|
pBlockInfo->window.skey, pBlockInfo->window.ekey, pBlockInfo->rows);
|
||||||
}
|
}
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
|
@ -348,9 +347,9 @@ static SSDataBlock* doTableScan(SOperatorInfo* pOperator, bool* newgroup) {
|
||||||
setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
|
setTaskStatus(pTaskInfo, TASK_NOT_COMPLETED);
|
||||||
pTableScanInfo->scanFlag = REPEAT_SCAN;
|
pTableScanInfo->scanFlag = REPEAT_SCAN;
|
||||||
|
|
||||||
// if (pResultRowInfo->size > 0) {
|
// if (pResultRowInfo->size > 0) {
|
||||||
// pResultRowInfo->curPos = 0;
|
// pResultRowInfo->curPos = 0;
|
||||||
// }
|
// }
|
||||||
|
|
||||||
qDebug("%s start to repeat scan data blocks due to query func required, qrange:%" PRId64 "-%" PRId64,
|
qDebug("%s start to repeat scan data blocks due to query func required, qrange:%" PRId64 "-%" PRId64,
|
||||||
GET_TASKID(pTaskInfo), pTaskInfo->window.skey, pTaskInfo->window.ekey);
|
GET_TASKID(pTaskInfo), pTaskInfo->window.skey, pTaskInfo->window.ekey);
|
||||||
|
@ -367,7 +366,7 @@ static SSDataBlock* doTableScan(SOperatorInfo* pOperator, bool* newgroup) {
|
||||||
GET_TASKID(pTaskInfo), pTaskInfo->window.skey, pTaskInfo->window.ekey);
|
GET_TASKID(pTaskInfo), pTaskInfo->window.skey, pTaskInfo->window.ekey);
|
||||||
|
|
||||||
if (pResultRowInfo->size > 0) {
|
if (pResultRowInfo->size > 0) {
|
||||||
// pResultRowInfo->curPos = pResultRowInfo->size - 1;
|
// pResultRowInfo->curPos = pResultRowInfo->size - 1;
|
||||||
}
|
}
|
||||||
|
|
||||||
p = doTableScanImpl(pOperator, newgroup);
|
p = doTableScanImpl(pOperator, newgroup);
|
||||||
|
@ -376,9 +375,10 @@ static SSDataBlock* doTableScan(SOperatorInfo* pOperator, bool* newgroup) {
|
||||||
return p;
|
return p;
|
||||||
}
|
}
|
||||||
|
|
||||||
SOperatorInfo* createTableScanOperatorInfo(void* pTsdbReadHandle, int32_t order, int32_t numOfOutput, int32_t dataLoadFlag,
|
SOperatorInfo* createTableScanOperatorInfo(void* pTsdbReadHandle, int32_t order, int32_t numOfOutput,
|
||||||
int32_t repeatTime, int32_t reverseTime, SArray* pColMatchInfo, SSDataBlock* pResBlock,
|
int32_t dataLoadFlag, int32_t repeatTime, int32_t reverseTime,
|
||||||
SNode* pCondition, SInterval* pInterval, double sampleRatio, SExecTaskInfo* pTaskInfo) {
|
SArray* pColMatchInfo, SSDataBlock* pResBlock, SNode* pCondition,
|
||||||
|
SInterval* pInterval, double sampleRatio, SExecTaskInfo* pTaskInfo) {
|
||||||
assert(repeatTime > 0);
|
assert(repeatTime > 0);
|
||||||
|
|
||||||
STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
|
STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
|
||||||
|
@ -391,26 +391,26 @@ SOperatorInfo* createTableScanOperatorInfo(void* pTsdbReadHandle, int32_t order,
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
pInfo->interval = *pInterval;
|
pInfo->interval = *pInterval;
|
||||||
pInfo->sampleRatio = sampleRatio;
|
pInfo->sampleRatio = sampleRatio;
|
||||||
pInfo->dataBlockLoadFlag= dataLoadFlag;
|
pInfo->dataBlockLoadFlag = dataLoadFlag;
|
||||||
pInfo->pResBlock = pResBlock;
|
pInfo->pResBlock = pResBlock;
|
||||||
pInfo->pFilterNode = pCondition;
|
pInfo->pFilterNode = pCondition;
|
||||||
pInfo->dataReader = pTsdbReadHandle;
|
pInfo->dataReader = pTsdbReadHandle;
|
||||||
pInfo->times = repeatTime;
|
pInfo->times = repeatTime;
|
||||||
pInfo->reverseTimes = reverseTime;
|
pInfo->reverseTimes = reverseTime;
|
||||||
pInfo->order = order;
|
pInfo->order = order;
|
||||||
pInfo->current = 0;
|
pInfo->current = 0;
|
||||||
pInfo->scanFlag = MAIN_SCAN;
|
pInfo->scanFlag = MAIN_SCAN;
|
||||||
pInfo->pColMatchInfo = pColMatchInfo;
|
pInfo->pColMatchInfo = pColMatchInfo;
|
||||||
pOperator->name = "TableScanOperator";
|
pOperator->name = "TableScanOperator";
|
||||||
pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN;
|
pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_TABLE_SCAN;
|
||||||
pOperator->blockingOptr = false;
|
pOperator->blockingOptr = false;
|
||||||
pOperator->status = OP_NOT_OPENED;
|
pOperator->status = OP_NOT_OPENED;
|
||||||
pOperator->info = pInfo;
|
pOperator->info = pInfo;
|
||||||
pOperator->numOfOutput = numOfOutput;
|
pOperator->numOfOutput = numOfOutput;
|
||||||
pOperator->getNextFn = doTableScan;
|
pOperator->getNextFn = doTableScan;
|
||||||
pOperator->pTaskInfo = pTaskInfo;
|
pOperator->pTaskInfo = pTaskInfo;
|
||||||
|
|
||||||
static int32_t cost = 0;
|
static int32_t cost = 0;
|
||||||
pOperator->cost.openCost = ++cost;
|
pOperator->cost.openCost = ++cost;
|
||||||
|
@ -456,67 +456,67 @@ static SSDataBlock* doBlockInfoScan(SOperatorInfo* pOperator, bool* newgroup) {
|
||||||
++numRowSteps;
|
++numRowSteps;
|
||||||
}
|
}
|
||||||
|
|
||||||
tableBlockDist.dataBlockInfos = taosArrayInit(numRowSteps, sizeof(SFileBlockInfo));
|
tableBlockDist.dataBlockInfos = taosArrayInit(numRowSteps, sizeof(SFileBlockInfo));
|
||||||
taosArraySetSize(tableBlockDist.dataBlockInfos, numRowSteps);
|
taosArraySetSize(tableBlockDist.dataBlockInfos, numRowSteps);
|
||||||
|
|
||||||
tableBlockDist.maxRows = INT_MIN;
|
tableBlockDist.maxRows = INT_MIN;
|
||||||
tableBlockDist.minRows = INT_MAX;
|
tableBlockDist.minRows = INT_MAX;
|
||||||
|
|
||||||
tsdbGetFileBlocksDistInfo(pTableScanInfo->dataReader, &tableBlockDist);
|
tsdbGetFileBlocksDistInfo(pTableScanInfo->dataReader, &tableBlockDist);
|
||||||
tableBlockDist.numOfRowsInMemTable = (int32_t) tsdbGetNumOfRowsInMemTable(pTableScanInfo->dataReader);
|
tableBlockDist.numOfRowsInMemTable = (int32_t)tsdbGetNumOfRowsInMemTable(pTableScanInfo->dataReader);
|
||||||
|
|
||||||
SSDataBlock* pBlock = pTableScanInfo->pResBlock;
|
SSDataBlock* pBlock = pTableScanInfo->pResBlock;
|
||||||
pBlock->info.rows = 1;
|
pBlock->info.rows = 1;
|
||||||
pBlock->info.numOfCols = 1;
|
pBlock->info.numOfCols = 1;
|
||||||
|
|
||||||
// SBufferWriter bw = tbufInitWriter(NULL, false);
|
// SBufferWriter bw = tbufInitWriter(NULL, false);
|
||||||
// blockDistInfoToBinary(&tableBlockDist, &bw);
|
// blockDistInfoToBinary(&tableBlockDist, &bw);
|
||||||
SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, 0);
|
SColumnInfoData* pColInfo = taosArrayGet(pBlock->pDataBlock, 0);
|
||||||
|
|
||||||
// int32_t len = (int32_t) tbufTell(&bw);
|
// int32_t len = (int32_t) tbufTell(&bw);
|
||||||
// pColInfo->pData = taosMemoryMalloc(len + sizeof(int32_t));
|
// pColInfo->pData = taosMemoryMalloc(len + sizeof(int32_t));
|
||||||
// *(int32_t*) pColInfo->pData = len;
|
// *(int32_t*) pColInfo->pData = len;
|
||||||
// memcpy(pColInfo->pData + sizeof(int32_t), tbufGetData(&bw, false), len);
|
// memcpy(pColInfo->pData + sizeof(int32_t), tbufGetData(&bw, false), len);
|
||||||
//
|
//
|
||||||
// tbufCloseWriter(&bw);
|
// tbufCloseWriter(&bw);
|
||||||
|
|
||||||
// SArray* g = GET_TABLEGROUP(pOperator->, 0);
|
// SArray* g = GET_TABLEGROUP(pOperator->, 0);
|
||||||
// pOperator->pRuntimeEnv->current = taosArrayGetP(g, 0);
|
// pOperator->pRuntimeEnv->current = taosArrayGetP(g, 0);
|
||||||
|
|
||||||
pOperator->status = OP_EXEC_DONE;
|
pOperator->status = OP_EXEC_DONE;
|
||||||
return pBlock;
|
return pBlock;
|
||||||
}
|
}
|
||||||
|
|
||||||
SOperatorInfo* createDataBlockInfoScanOperator(void* dataReader, SExecTaskInfo* pTaskInfo) {
|
SOperatorInfo* createDataBlockInfoScanOperator(void* dataReader, SExecTaskInfo* pTaskInfo) {
|
||||||
STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
|
STableScanInfo* pInfo = taosMemoryCalloc(1, sizeof(STableScanInfo));
|
||||||
SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
|
SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
|
||||||
if (pInfo == NULL || pOperator == NULL) {
|
if (pInfo == NULL || pOperator == NULL) {
|
||||||
pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
|
pTaskInfo->code = TSDB_CODE_OUT_OF_MEMORY;
|
||||||
goto _error;
|
goto _error;
|
||||||
}
|
}
|
||||||
|
|
||||||
pInfo->dataReader = dataReader;
|
pInfo->dataReader = dataReader;
|
||||||
// pInfo->block.pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
|
// pInfo->block.pDataBlock = taosArrayInit(1, sizeof(SColumnInfoData));
|
||||||
|
|
||||||
SColumnInfoData infoData = {0};
|
SColumnInfoData infoData = {0};
|
||||||
infoData.info.type = TSDB_DATA_TYPE_BINARY;
|
infoData.info.type = TSDB_DATA_TYPE_BINARY;
|
||||||
infoData.info.bytes = 1024;
|
infoData.info.bytes = 1024;
|
||||||
infoData.info.colId = 0;
|
infoData.info.colId = 0;
|
||||||
// taosArrayPush(pInfo->block.pDataBlock, &infoData);
|
// taosArrayPush(pInfo->block.pDataBlock, &infoData);
|
||||||
|
|
||||||
pOperator->name = "DataBlockInfoScanOperator";
|
pOperator->name = "DataBlockInfoScanOperator";
|
||||||
// pOperator->operatorType = OP_TableBlockInfoScan;
|
// pOperator->operatorType = OP_TableBlockInfoScan;
|
||||||
pOperator->blockingOptr = false;
|
pOperator->blockingOptr = false;
|
||||||
pOperator->status = OP_NOT_OPENED;
|
pOperator->status = OP_NOT_OPENED;
|
||||||
pOperator->_openFn = operatorDummyOpenFn;
|
pOperator->_openFn = operatorDummyOpenFn;
|
||||||
pOperator->getNextFn = doBlockInfoScan;
|
pOperator->getNextFn = doBlockInfoScan;
|
||||||
|
|
||||||
pOperator->info = pInfo;
|
pOperator->info = pInfo;
|
||||||
pOperator->pTaskInfo = pTaskInfo;
|
pOperator->pTaskInfo = pTaskInfo;
|
||||||
|
|
||||||
return pOperator;
|
return pOperator;
|
||||||
|
|
||||||
_error:
|
_error:
|
||||||
taosMemoryFreeClear(pInfo);
|
taosMemoryFreeClear(pInfo);
|
||||||
taosMemoryFreeClear(pOperator);
|
taosMemoryFreeClear(pOperator);
|
||||||
return NULL;
|
return NULL;
|
||||||
|
@ -558,29 +558,42 @@ static SSDataBlock* doStreamBlockScan(SOperatorInfo* pOperator, bool* newgroup)
|
||||||
blockDataCleanup(pInfo->pRes);
|
blockDataCleanup(pInfo->pRes);
|
||||||
|
|
||||||
while (tqNextDataBlock(pInfo->readerHandle)) {
|
while (tqNextDataBlock(pInfo->readerHandle)) {
|
||||||
pTaskInfo->code = tqRetrieveDataBlockInfo(pInfo->readerHandle, pBlockInfo);
|
SArray* pCols = NULL;
|
||||||
if (pTaskInfo->code != TSDB_CODE_SUCCESS) {
|
uint64_t groupId;
|
||||||
terrno = pTaskInfo->code;
|
int32_t numOfRows;
|
||||||
pOperator->status = OP_EXEC_DONE;
|
int32_t code = tqRetrieveDataBlock(&pCols, pInfo->readerHandle, &groupId, &numOfRows);
|
||||||
|
|
||||||
|
if (code != TSDB_CODE_SUCCESS || numOfRows == 0) {
|
||||||
|
pTaskInfo->code = code;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
if (pBlockInfo->rows == 0) {
|
pInfo->pRes->info.groupId = groupId;
|
||||||
break;
|
pInfo->pRes->info.rows = numOfRows;
|
||||||
}
|
|
||||||
|
|
||||||
SArray* pCols = tqRetrieveDataBlock(pInfo->readerHandle);
|
|
||||||
|
|
||||||
int32_t numOfCols = pInfo->pRes->info.numOfCols;
|
int32_t numOfCols = pInfo->pRes->info.numOfCols;
|
||||||
for (int32_t i = 0; i < numOfCols; ++i) {
|
for (int32_t i = 0; i < numOfCols; ++i) {
|
||||||
SColumnInfoData* p = taosArrayGet(pCols, i);
|
SColMatchInfo* pColMatchInfo = taosArrayGet(pInfo->pColMatchInfo, i);
|
||||||
SColMatchInfo* pColMatchInfo = taosArrayGet(pInfo->pColMatchInfo, i);
|
|
||||||
if (!pColMatchInfo->output) {
|
if (!pColMatchInfo->output) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
ASSERT(pColMatchInfo->colId == p->info.colId);
|
bool colExists = false;
|
||||||
taosArraySet(pInfo->pRes->pDataBlock, pColMatchInfo->targetSlotId, p);
|
for (int32_t j = 0; j < taosArrayGetSize(pCols); ++j) {
|
||||||
|
SColumnInfoData* pResCol = taosArrayGet(pCols, j);
|
||||||
|
if (pResCol->info.colId == pColMatchInfo->colId) {
|
||||||
|
taosArraySet(pInfo->pRes->pDataBlock, pColMatchInfo->targetSlotId, pResCol);
|
||||||
|
colExists = true;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// the required column does not exists in submit block, let's set it to be all null value
|
||||||
|
if (!colExists) {
|
||||||
|
SColumnInfoData* pDst = taosArrayGet(pInfo->pRes->pDataBlock, pColMatchInfo->targetSlotId);
|
||||||
|
colInfoDataEnsureCapacity(pDst, 0, pBlockInfo->rows);
|
||||||
|
colDataAppendNNULL(pDst, 0, pBlockInfo->rows);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (pInfo->pRes->pDataBlock == NULL) {
|
if (pInfo->pRes->pDataBlock == NULL) {
|
||||||
|
@ -605,7 +618,8 @@ static SSDataBlock* doStreamBlockScan(SOperatorInfo* pOperator, bool* newgroup)
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
SOperatorInfo* createStreamScanOperatorInfo(void* streamReadHandle, SSDataBlock* pResBlock, SArray* pColList, SArray* pTableIdList, SExecTaskInfo* pTaskInfo) {
|
SOperatorInfo* createStreamScanOperatorInfo(void* streamReadHandle, SSDataBlock* pResBlock, SArray* pColList,
|
||||||
|
SArray* pTableIdList, SExecTaskInfo* pTaskInfo) {
|
||||||
SStreamBlockScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamBlockScanInfo));
|
SStreamBlockScanInfo* pInfo = taosMemoryCalloc(1, sizeof(SStreamBlockScanInfo));
|
||||||
SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
|
SOperatorInfo* pOperator = taosMemoryCalloc(1, sizeof(SOperatorInfo));
|
||||||
if (pInfo == NULL || pOperator == NULL) {
|
if (pInfo == NULL || pOperator == NULL) {
|
||||||
|
@ -618,7 +632,7 @@ SOperatorInfo* createStreamScanOperatorInfo(void* streamReadHandle, SSDataBlock*
|
||||||
int32_t numOfOutput = taosArrayGetSize(pColList);
|
int32_t numOfOutput = taosArrayGetSize(pColList);
|
||||||
|
|
||||||
SArray* pColIds = taosArrayInit(4, sizeof(int16_t));
|
SArray* pColIds = taosArrayInit(4, sizeof(int16_t));
|
||||||
for(int32_t i = 0; i < numOfOutput; ++i) {
|
for (int32_t i = 0; i < numOfOutput; ++i) {
|
||||||
int16_t* id = taosArrayGet(pColList, i);
|
int16_t* id = taosArrayGet(pColList, i);
|
||||||
taosArrayPush(pColIds, id);
|
taosArrayPush(pColIds, id);
|
||||||
}
|
}
|
||||||
|
@ -644,16 +658,16 @@ SOperatorInfo* createStreamScanOperatorInfo(void* streamReadHandle, SSDataBlock*
|
||||||
pInfo->readerHandle = streamReadHandle;
|
pInfo->readerHandle = streamReadHandle;
|
||||||
pInfo->pRes = pResBlock;
|
pInfo->pRes = pResBlock;
|
||||||
|
|
||||||
pOperator->name = "StreamBlockScanOperator";
|
pOperator->name = "StreamBlockScanOperator";
|
||||||
pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN;
|
pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_STREAM_SCAN;
|
||||||
pOperator->blockingOptr = false;
|
pOperator->blockingOptr = false;
|
||||||
pOperator->status = OP_NOT_OPENED;
|
pOperator->status = OP_NOT_OPENED;
|
||||||
pOperator->info = pInfo;
|
pOperator->info = pInfo;
|
||||||
pOperator->numOfOutput = pResBlock->info.numOfCols;
|
pOperator->numOfOutput = pResBlock->info.numOfCols;
|
||||||
pOperator->_openFn = operatorDummyOpenFn;
|
pOperator->_openFn = operatorDummyOpenFn;
|
||||||
pOperator->getNextFn = doStreamBlockScan;
|
pOperator->getNextFn = doStreamBlockScan;
|
||||||
pOperator->closeFn = operatorDummyCloseFn;
|
pOperator->closeFn = operatorDummyCloseFn;
|
||||||
pOperator->pTaskInfo = pTaskInfo;
|
pOperator->pTaskInfo = pTaskInfo;
|
||||||
|
|
||||||
return pOperator;
|
return pOperator;
|
||||||
}
|
}
|
||||||
|
@ -733,9 +747,9 @@ static int32_t loadSysTableContentCb(void* param, const SDataBuf* pMsg, int32_t
|
||||||
|
|
||||||
SRetrieveMetaTableRsp* pRsp = pScanResInfo->pRsp;
|
SRetrieveMetaTableRsp* pRsp = pScanResInfo->pRsp;
|
||||||
pRsp->numOfRows = htonl(pRsp->numOfRows);
|
pRsp->numOfRows = htonl(pRsp->numOfRows);
|
||||||
pRsp->useconds = htobe64(pRsp->useconds);
|
pRsp->useconds = htobe64(pRsp->useconds);
|
||||||
pRsp->handle = htobe64(pRsp->handle);
|
pRsp->handle = htobe64(pRsp->handle);
|
||||||
pRsp->compLen = htonl(pRsp->compLen);
|
pRsp->compLen = htonl(pRsp->compLen);
|
||||||
} else {
|
} else {
|
||||||
operator->pTaskInfo->code = code;
|
operator->pTaskInfo->code = code;
|
||||||
}
|
}
|
||||||
|
@ -777,7 +791,7 @@ static SSDataBlock* doFilterResult(SSysTableScanInfo* pInfo) {
|
||||||
if (rowRes[j] == 0) {
|
if (rowRes[j] == 0) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
colDataAppend(pDest, numOfRow, colDataGetData(pSrc, j), false);
|
colDataAppend(pDest, numOfRow, colDataGetData(pSrc, j), false);
|
||||||
numOfRow += 1;
|
numOfRow += 1;
|
||||||
}
|
}
|
||||||
|
@ -828,7 +842,7 @@ static SSDataBlock* doSysTableScan(SOperatorInfo* pOperator, bool* newgroup) {
|
||||||
SColumnInfoData* pColInfoData = taosArrayGet(pInfo->pRes->pDataBlock, i);
|
SColumnInfoData* pColInfoData = taosArrayGet(pInfo->pRes->pDataBlock, i);
|
||||||
int64_t tmp = 0;
|
int64_t tmp = 0;
|
||||||
char t[10] = {0};
|
char t[10] = {0};
|
||||||
STR_TO_VARSTR(t, "_"); //TODO
|
STR_TO_VARSTR(t, "_"); // TODO
|
||||||
if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
|
if (IS_VAR_DATA_TYPE(pColInfoData->info.type)) {
|
||||||
colDataAppend(pColInfoData, numOfRows, t, false);
|
colDataAppend(pColInfoData, numOfRows, t, false);
|
||||||
} else {
|
} else {
|
||||||
|
@ -926,12 +940,12 @@ SOperatorInfo* createSysTableScanOperatorInfo(void* pSysTableReadHandle, SSDataB
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
pInfo->accountId = accountId;
|
pInfo->accountId = accountId;
|
||||||
pInfo->showRewrite = showRewrite;
|
pInfo->showRewrite = showRewrite;
|
||||||
pInfo->pRes = pResBlock;
|
pInfo->pRes = pResBlock;
|
||||||
pInfo->capacity = 4096;
|
pInfo->capacity = 4096;
|
||||||
pInfo->pCondition = pCondition;
|
pInfo->pCondition = pCondition;
|
||||||
pInfo->scanCols = colList;
|
pInfo->scanCols = colList;
|
||||||
|
|
||||||
// TODO remove it
|
// TODO remove it
|
||||||
int32_t tableType = 0;
|
int32_t tableType = 0;
|
||||||
|
@ -986,9 +1000,9 @@ SOperatorInfo* createSysTableScanOperatorInfo(void* pSysTableReadHandle, SSDataB
|
||||||
tableType = TSDB_MGMT_TABLE_CONNS;
|
tableType = TSDB_MGMT_TABLE_CONNS;
|
||||||
} else if (strncasecmp(name, TSDB_INS_TABLE_QUERIES, tListLen(pName->tname)) == 0) {
|
} else if (strncasecmp(name, TSDB_INS_TABLE_QUERIES, tListLen(pName->tname)) == 0) {
|
||||||
tableType = TSDB_MGMT_TABLE_QUERIES;
|
tableType = TSDB_MGMT_TABLE_QUERIES;
|
||||||
} else if (strncasecmp(name, TSDB_INS_TABLE_VNODES, tListLen(pName->tname)) == 0) {
|
} else if (strncasecmp(name, TSDB_INS_TABLE_VNODES, tListLen(pName->tname)) == 0) {
|
||||||
tableType = TSDB_MGMT_TABLE_VNODES;
|
tableType = TSDB_MGMT_TABLE_VNODES;
|
||||||
}else {
|
} else {
|
||||||
ASSERT(0);
|
ASSERT(0);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1025,15 +1039,15 @@ SOperatorInfo* createSysTableScanOperatorInfo(void* pSysTableReadHandle, SSDataB
|
||||||
#endif
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
pOperator->name = "SysTableScanOperator";
|
pOperator->name = "SysTableScanOperator";
|
||||||
pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN;
|
pOperator->operatorType = QUERY_NODE_PHYSICAL_PLAN_SYSTABLE_SCAN;
|
||||||
pOperator->blockingOptr = false;
|
pOperator->blockingOptr = false;
|
||||||
pOperator->status = OP_NOT_OPENED;
|
pOperator->status = OP_NOT_OPENED;
|
||||||
pOperator->info = pInfo;
|
pOperator->info = pInfo;
|
||||||
pOperator->numOfOutput = pResBlock->info.numOfCols;
|
pOperator->numOfOutput = pResBlock->info.numOfCols;
|
||||||
pOperator->getNextFn = doSysTableScan;
|
pOperator->getNextFn = doSysTableScan;
|
||||||
pOperator->closeFn = destroySysScanOperator;
|
pOperator->closeFn = destroySysScanOperator;
|
||||||
pOperator->pTaskInfo = pTaskInfo;
|
pOperator->pTaskInfo = pTaskInfo;
|
||||||
|
|
||||||
return pOperator;
|
return pOperator;
|
||||||
}
|
}
|
||||||
|
|
|
@ -89,7 +89,7 @@ static FORCE_INLINE int32_t getExtendedRowSize(STableDataBlocks *pBlock) {
|
||||||
(int32_t)TD_BITMAP_BYTES(pTableInfo->numOfColumns - 1);
|
(int32_t)TD_BITMAP_BYTES(pTableInfo->numOfColumns - 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
static FORCE_INLINE void getSTSRowAppendInfo(SSchema *pSchema, uint8_t rowType, SParsedDataColInfo *spd, col_id_t idx,
|
static FORCE_INLINE void getSTSRowAppendInfo(uint8_t rowType, SParsedDataColInfo *spd, col_id_t idx,
|
||||||
int32_t *toffset, col_id_t *colIdx) {
|
int32_t *toffset, col_id_t *colIdx) {
|
||||||
col_id_t schemaIdx = 0;
|
col_id_t schemaIdx = 0;
|
||||||
if (IS_DATA_COL_ORDERED(spd)) {
|
if (IS_DATA_COL_ORDERED(spd)) {
|
||||||
|
@ -131,7 +131,6 @@ static FORCE_INLINE int32_t setBlockInfo(SSubmitBlk *pBlocks, STableDataBlocks*
|
||||||
int32_t schemaIdxCompar(const void *lhs, const void *rhs);
|
int32_t schemaIdxCompar(const void *lhs, const void *rhs);
|
||||||
int32_t boundIdxCompar(const void *lhs, const void *rhs);
|
int32_t boundIdxCompar(const void *lhs, const void *rhs);
|
||||||
void setBoundColumnInfo(SParsedDataColInfo *pColList, SSchema *pSchema, col_id_t numOfCols);
|
void setBoundColumnInfo(SParsedDataColInfo *pColList, SSchema *pSchema, col_id_t numOfCols);
|
||||||
void destroyBoundColumnInfo(SParsedDataColInfo* pColList);
|
|
||||||
void destroyBlockArrayList(SArray* pDataBlockList);
|
void destroyBlockArrayList(SArray* pDataBlockList);
|
||||||
void destroyBlockHashmap(SHashObj* pDataBlockHash);
|
void destroyBlockHashmap(SHashObj* pDataBlockHash);
|
||||||
int initRowBuilder(SRowBuilder *pBuilder, int16_t schemaVer, SParsedDataColInfo *pColInfo);
|
int initRowBuilder(SRowBuilder *pBuilder, int16_t schemaVer, SParsedDataColInfo *pColInfo);
|
||||||
|
@ -139,5 +138,7 @@ int32_t allocateMemIfNeed(STableDataBlocks *pDataBlock, int32_t rowSize, int32_t
|
||||||
int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int32_t startOffset, int32_t rowSize,
|
int32_t getDataBlockFromList(SHashObj* pHashList, int64_t id, int32_t size, int32_t startOffset, int32_t rowSize,
|
||||||
const STableMeta* pTableMeta, STableDataBlocks** dataBlocks, SArray* pBlockList, SVCreateTbReq* pCreateTbReq);
|
const STableMeta* pTableMeta, STableDataBlocks** dataBlocks, SArray* pBlockList, SVCreateTbReq* pCreateTbReq);
|
||||||
int32_t mergeTableDataBlocks(SHashObj* pHashObj, uint8_t payloadType, SArray** pVgDataBlocks);
|
int32_t mergeTableDataBlocks(SHashObj* pHashObj, uint8_t payloadType, SArray** pVgDataBlocks);
|
||||||
|
int32_t buildCreateTbMsg(STableDataBlocks* pBlocks, SVCreateTbReq* pCreateTbReq);
|
||||||
|
int32_t allocateMemForSize(STableDataBlocks *pDataBlock, int32_t allSize);
|
||||||
|
|
||||||
#endif // TDENGINE_DATABLOCKMGT_H
|
#endif // TDENGINE_DATABLOCKMGT_H
|
||||||
|
|
|
@ -21,6 +21,8 @@ extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#include "parser.h"
|
#include "parser.h"
|
||||||
|
#include "parToken.h"
|
||||||
|
#include "parUtil.h"
|
||||||
|
|
||||||
int32_t parseInsertSql(SParseContext* pContext, SQuery** pQuery);
|
int32_t parseInsertSql(SParseContext* pContext, SQuery** pQuery);
|
||||||
int32_t parse(SParseContext* pParseCxt, SQuery** pQuery);
|
int32_t parse(SParseContext* pParseCxt, SQuery** pQuery);
|
||||||
|
|
|
@ -40,29 +40,21 @@
|
||||||
sToken = tStrGetToken(pSql, &index, false); \
|
sToken = tStrGetToken(pSql, &index, false); \
|
||||||
} while (0)
|
} while (0)
|
||||||
|
|
||||||
#define CHECK_CODE(expr) \
|
|
||||||
do { \
|
|
||||||
int32_t code = expr; \
|
|
||||||
if (TSDB_CODE_SUCCESS != code) { \
|
|
||||||
return code; \
|
|
||||||
} \
|
|
||||||
} while (0)
|
|
||||||
|
|
||||||
typedef struct SInsertParseContext {
|
typedef struct SInsertParseContext {
|
||||||
SParseContext* pComCxt; // input
|
SParseContext* pComCxt; // input
|
||||||
char* pSql; // input
|
char *pSql; // input
|
||||||
SMsgBuf msg; // input
|
SMsgBuf msg; // input
|
||||||
STableMeta* pTableMeta; // each table
|
STableMeta* pTableMeta; // each table
|
||||||
SParsedDataColInfo tags; // each table
|
SParsedDataColInfo tags; // each table
|
||||||
SKVRowBuilder tagsBuilder; // each table
|
SKVRowBuilder tagsBuilder; // each table
|
||||||
SVCreateTbReq createTblReq; // each table
|
SVCreateTbReq createTblReq; // each table
|
||||||
SHashObj* pVgroupsHashObj; // global
|
SHashObj* pVgroupsHashObj; // global
|
||||||
SHashObj* pTableBlockHashObj; // global
|
SHashObj* pTableBlockHashObj; // global
|
||||||
SHashObj* pSubTableHashObj; // global
|
SHashObj* pSubTableHashObj; // global
|
||||||
SArray* pTableDataBlocks; // global
|
SArray* pVgDataBlocks; // global
|
||||||
SArray* pVgDataBlocks; // global
|
int32_t totalNum;
|
||||||
int32_t totalNum;
|
|
||||||
SVnodeModifOpStmt* pOutput;
|
SVnodeModifOpStmt* pOutput;
|
||||||
|
SStmtCallback* pStmtCb;
|
||||||
} SInsertParseContext;
|
} SInsertParseContext;
|
||||||
|
|
||||||
typedef int32_t (*_row_append_fn_t)(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param);
|
typedef int32_t (*_row_append_fn_t)(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param);
|
||||||
|
@ -70,6 +62,29 @@ typedef int32_t (*_row_append_fn_t)(SMsgBuf* pMsgBuf, const void* value, int32_t
|
||||||
static uint8_t TRUE_VALUE = (uint8_t)TSDB_TRUE;
|
static uint8_t TRUE_VALUE = (uint8_t)TSDB_TRUE;
|
||||||
static uint8_t FALSE_VALUE = (uint8_t)TSDB_FALSE;
|
static uint8_t FALSE_VALUE = (uint8_t)TSDB_FALSE;
|
||||||
|
|
||||||
|
typedef struct SKvParam {
|
||||||
|
SKVRowBuilder *builder;
|
||||||
|
SSchema *schema;
|
||||||
|
char buf[TSDB_MAX_TAGS_LEN];
|
||||||
|
} SKvParam;
|
||||||
|
|
||||||
|
typedef struct SMemParam {
|
||||||
|
SRowBuilder* rb;
|
||||||
|
SSchema* schema;
|
||||||
|
int32_t toffset;
|
||||||
|
col_id_t colIdx;
|
||||||
|
} SMemParam;
|
||||||
|
|
||||||
|
|
||||||
|
#define CHECK_CODE(expr) \
|
||||||
|
do { \
|
||||||
|
int32_t code = expr; \
|
||||||
|
if (TSDB_CODE_SUCCESS != code) { \
|
||||||
|
return code; \
|
||||||
|
} \
|
||||||
|
} while (0)
|
||||||
|
|
||||||
|
|
||||||
static int32_t skipInsertInto(SInsertParseContext* pCxt) {
|
static int32_t skipInsertInto(SInsertParseContext* pCxt) {
|
||||||
SToken sToken;
|
SToken sToken;
|
||||||
NEXT_TOKEN(pCxt->pSql, sToken);
|
NEXT_TOKEN(pCxt->pSql, sToken);
|
||||||
|
@ -163,7 +178,8 @@ static int32_t buildName(SInsertParseContext* pCxt, SToken* pStname, char* fullD
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t createSName(SName* pName, SToken* pTableName, SParseContext* pParseCtx, SMsgBuf* pMsgBuf) {
|
|
||||||
|
static int32_t createSName(SName* pName, SToken* pTableName, int32_t acctId, const char* dbName, SMsgBuf* pMsgBuf) {
|
||||||
const char* msg1 = "name too long";
|
const char* msg1 = "name too long";
|
||||||
const char* msg2 = "invalid database name";
|
const char* msg2 = "invalid database name";
|
||||||
const char* msg3 = "db is not specified";
|
const char* msg3 = "db is not specified";
|
||||||
|
@ -179,7 +195,7 @@ static int32_t createSName(SName* pName, SToken* pTableName, SParseContext* pPar
|
||||||
strncpy(name, pTableName->z, dbLen);
|
strncpy(name, pTableName->z, dbLen);
|
||||||
dbLen = strdequote(name);
|
dbLen = strdequote(name);
|
||||||
|
|
||||||
code = tNameSetDbName(pName, pParseCtx->acctId, name, dbLen);
|
code = tNameSetDbName(pName, acctId, name, dbLen);
|
||||||
if (code != TSDB_CODE_SUCCESS) {
|
if (code != TSDB_CODE_SUCCESS) {
|
||||||
return buildInvalidOperationMsg(pMsgBuf, msg1);
|
return buildInvalidOperationMsg(pMsgBuf, msg1);
|
||||||
}
|
}
|
||||||
|
@ -204,11 +220,11 @@ static int32_t createSName(SName* pName, SToken* pTableName, SParseContext* pPar
|
||||||
strncpy(name, pTableName->z, pTableName->n);
|
strncpy(name, pTableName->z, pTableName->n);
|
||||||
strdequote(name);
|
strdequote(name);
|
||||||
|
|
||||||
if (pParseCtx->db == NULL) {
|
if (dbName == NULL) {
|
||||||
return buildInvalidOperationMsg(pMsgBuf, msg3);
|
return buildInvalidOperationMsg(pMsgBuf, msg3);
|
||||||
}
|
}
|
||||||
|
|
||||||
code = tNameSetDbName(pName, pParseCtx->acctId, pParseCtx->db, strlen(pParseCtx->db));
|
code = tNameSetDbName(pName, acctId, dbName, strlen(dbName));
|
||||||
if (code != TSDB_CODE_SUCCESS) {
|
if (code != TSDB_CODE_SUCCESS) {
|
||||||
code = buildInvalidOperationMsg(pMsgBuf, msg2);
|
code = buildInvalidOperationMsg(pMsgBuf, msg2);
|
||||||
return code;
|
return code;
|
||||||
|
@ -225,8 +241,8 @@ static int32_t createSName(SName* pName, SToken* pTableName, SParseContext* pPar
|
||||||
|
|
||||||
static int32_t getTableMetaImpl(SInsertParseContext* pCxt, SToken* pTname, bool isStb) {
|
static int32_t getTableMetaImpl(SInsertParseContext* pCxt, SToken* pTname, bool isStb) {
|
||||||
SParseContext* pBasicCtx = pCxt->pComCxt;
|
SParseContext* pBasicCtx = pCxt->pComCxt;
|
||||||
SName name = {0};
|
SName name = {0};
|
||||||
createSName(&name, pTname, pBasicCtx, &pCxt->msg);
|
createSName(&name, pTname, pBasicCtx->acctId, pBasicCtx->db, &pCxt->msg);
|
||||||
if (isStb) {
|
if (isStb) {
|
||||||
CHECK_CODE(catalogGetSTableMeta(pBasicCtx->pCatalog, pBasicCtx->pTransporter, &pBasicCtx->mgmtEpSet, &name,
|
CHECK_CODE(catalogGetSTableMeta(pBasicCtx->pCatalog, pBasicCtx->pTransporter, &pBasicCtx->mgmtEpSet, &name,
|
||||||
&pCxt->pTableMeta));
|
&pCxt->pTableMeta));
|
||||||
|
@ -299,7 +315,7 @@ static int32_t buildOutput(SInsertParseContext* pCxt) {
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t checkTimestamp(STableDataBlocks* pDataBlocks, const char* start) {
|
int32_t checkTimestamp(STableDataBlocks *pDataBlocks, const char *start) {
|
||||||
// once the data block is disordered, we do NOT keep previous timestamp any more
|
// once the data block is disordered, we do NOT keep previous timestamp any more
|
||||||
if (!pDataBlocks->ordered) {
|
if (!pDataBlocks->ordered) {
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
|
@ -441,7 +457,7 @@ static int32_t parseValueToken(char** end, SToken* pToken, SSchema* pSchema, int
|
||||||
|
|
||||||
if (isNullStr(pToken)) {
|
if (isNullStr(pToken)) {
|
||||||
if (TSDB_DATA_TYPE_TIMESTAMP == pSchema->type && PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
|
if (TSDB_DATA_TYPE_TIMESTAMP == pSchema->type && PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
|
||||||
return buildSyntaxErrMsg(pMsgBuf, "primary timestamp can not be null", pToken->z);
|
return buildSyntaxErrMsg(pMsgBuf, "primary timestamp should not be null", pToken->z);
|
||||||
}
|
}
|
||||||
|
|
||||||
return func(pMsgBuf, NULL, 0, param);
|
return func(pMsgBuf, NULL, 0, param);
|
||||||
|
@ -603,13 +619,6 @@ static int32_t parseValueToken(char** end, SToken* pToken, SSchema* pSchema, int
|
||||||
return TSDB_CODE_FAILED;
|
return TSDB_CODE_FAILED;
|
||||||
}
|
}
|
||||||
|
|
||||||
typedef struct SMemParam {
|
|
||||||
SRowBuilder* rb;
|
|
||||||
SSchema* schema;
|
|
||||||
int32_t toffset;
|
|
||||||
col_id_t colIdx;
|
|
||||||
} SMemParam;
|
|
||||||
|
|
||||||
static FORCE_INLINE int32_t MemRowAppend(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param) {
|
static FORCE_INLINE int32_t MemRowAppend(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param) {
|
||||||
SMemParam* pa = (SMemParam*)param;
|
SMemParam* pa = (SMemParam*)param;
|
||||||
SRowBuilder* rb = pa->rb;
|
SRowBuilder* rb = pa->rb;
|
||||||
|
@ -718,14 +727,8 @@ static int32_t parseBoundColumns(SInsertParseContext* pCxt, SParsedDataColInfo*
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
typedef struct SKvParam {
|
static int32_t KvRowAppend(SMsgBuf* pMsgBuf, const void *value, int32_t len, void *param) {
|
||||||
SKVRowBuilder* builder;
|
SKvParam* pa = (SKvParam*) param;
|
||||||
SSchema* schema;
|
|
||||||
char buf[TSDB_MAX_TAGS_LEN];
|
|
||||||
} SKvParam;
|
|
||||||
|
|
||||||
static int32_t KvRowAppend(SMsgBuf* pMsgBuf, const void* value, int32_t len, void* param) {
|
|
||||||
SKvParam* pa = (SKvParam*)param;
|
|
||||||
|
|
||||||
int8_t type = pa->schema->type;
|
int8_t type = pa->schema->type;
|
||||||
int16_t colId = pa->schema->colId;
|
int16_t colId = pa->schema->colId;
|
||||||
|
@ -760,13 +763,13 @@ static int32_t KvRowAppend(SMsgBuf* pMsgBuf, const void* value, int32_t len, voi
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t buildCreateTbReq(SInsertParseContext* pCxt, const SName* pName, SKVRow row) {
|
static int32_t buildCreateTbReq(SVCreateTbReq *pTbReq, const SName* pName, SKVRow row, int64_t suid) {
|
||||||
char dbFName[TSDB_DB_FNAME_LEN] = {0};
|
char dbFName[TSDB_DB_FNAME_LEN] = {0};
|
||||||
tNameGetFullDbName(pName, dbFName);
|
tNameGetFullDbName(pName, dbFName);
|
||||||
pCxt->createTblReq.type = TD_CHILD_TABLE;
|
pTbReq->type = TD_CHILD_TABLE;
|
||||||
pCxt->createTblReq.name = strdup(pName->tname);
|
pTbReq->name = strdup(pName->tname);
|
||||||
pCxt->createTblReq.ctbCfg.suid = pCxt->pTableMeta->suid;
|
pTbReq->ctbCfg.suid = suid;
|
||||||
pCxt->createTblReq.ctbCfg.pTag = row;
|
pTbReq->ctbCfg.pTag = row;
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -778,23 +781,42 @@ static int32_t parseTagsClause(SInsertParseContext* pCxt, SSchema* pSchema, uint
|
||||||
}
|
}
|
||||||
|
|
||||||
SKvParam param = {.builder = &pCxt->tagsBuilder};
|
SKvParam param = {.builder = &pCxt->tagsBuilder};
|
||||||
SToken sToken;
|
SToken sToken;
|
||||||
char tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW] = {0}; // used for deleting Escape character: \\, \', \"
|
bool isParseBindParam = false;
|
||||||
|
char tmpTokenBuf[TSDB_MAX_BYTES_PER_ROW] = {0}; // used for deleting Escape character: \\, \', \"
|
||||||
for (int i = 0; i < pCxt->tags.numOfBound; ++i) {
|
for (int i = 0; i < pCxt->tags.numOfBound; ++i) {
|
||||||
NEXT_TOKEN_WITH_PREV(pCxt->pSql, sToken);
|
NEXT_TOKEN_WITH_PREV(pCxt->pSql, sToken);
|
||||||
SSchema* pTagSchema = &pSchema[pCxt->tags.boundColumns[i] - 1]; // colId starts with 1
|
|
||||||
|
if (sToken.type == TK_NK_QUESTION) {
|
||||||
|
isParseBindParam = true;
|
||||||
|
if (NULL == pCxt->pStmtCb) {
|
||||||
|
return buildSyntaxErrMsg(&pCxt->msg, "? only used in stmt", sToken.z);
|
||||||
|
}
|
||||||
|
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isParseBindParam) {
|
||||||
|
return buildInvalidOperationMsg(&pCxt->msg, "no mix usage for ? and tag values");
|
||||||
|
}
|
||||||
|
|
||||||
|
SSchema* pTagSchema = &pSchema[pCxt->tags.boundColumns[i] - 1]; // colId starts with 1
|
||||||
param.schema = pTagSchema;
|
param.schema = pTagSchema;
|
||||||
CHECK_CODE(
|
CHECK_CODE(
|
||||||
parseValueToken(&pCxt->pSql, &sToken, pTagSchema, precision, tmpTokenBuf, KvRowAppend, ¶m, &pCxt->msg));
|
parseValueToken(&pCxt->pSql, &sToken, pTagSchema, precision, tmpTokenBuf, KvRowAppend, ¶m, &pCxt->msg));
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (isParseBindParam) {
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
SKVRow row = tdGetKVRowFromBuilder(&pCxt->tagsBuilder);
|
SKVRow row = tdGetKVRowFromBuilder(&pCxt->tagsBuilder);
|
||||||
if (NULL == row) {
|
if (NULL == row) {
|
||||||
return buildInvalidOperationMsg(&pCxt->msg, "tag value expected");
|
return buildInvalidOperationMsg(&pCxt->msg, "tag value expected");
|
||||||
}
|
}
|
||||||
tdSortKVRowByColIdx(row);
|
tdSortKVRowByColIdx(row);
|
||||||
|
|
||||||
return buildCreateTbReq(pCxt, pName, row);
|
return buildCreateTbReq(&pCxt->createTblReq, pName, row, pCxt->pTableMeta->suid);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t cloneTableMeta(STableMeta* pSrc, STableMeta** pDst) {
|
static int32_t cloneTableMeta(STableMeta* pSrc, STableMeta** pDst) {
|
||||||
|
@ -818,7 +840,7 @@ static int32_t storeTableMeta(SHashObj* pHash, const char* pName, int32_t len, S
|
||||||
// pSql -> stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)
|
// pSql -> stb_name [(tag1_name, ...)] TAGS (tag1_value, ...)
|
||||||
static int32_t parseUsingClause(SInsertParseContext* pCxt, SToken* pTbnameToken) {
|
static int32_t parseUsingClause(SInsertParseContext* pCxt, SToken* pTbnameToken) {
|
||||||
SName name;
|
SName name;
|
||||||
createSName(&name, pTbnameToken, pCxt->pComCxt, &pCxt->msg);
|
createSName(&name, pTbnameToken, pCxt->pComCxt->acctId, pCxt->pComCxt->db, &pCxt->msg);
|
||||||
char tbFName[TSDB_TABLE_FNAME_LEN];
|
char tbFName[TSDB_TABLE_FNAME_LEN];
|
||||||
tNameExtractFullName(&name, tbFName);
|
tNameExtractFullName(&name, tbFName);
|
||||||
int32_t len = strlen(tbFName);
|
int32_t len = strlen(tbFName);
|
||||||
|
@ -863,8 +885,7 @@ static int32_t parseUsingClause(SInsertParseContext* pCxt, SToken* pTbnameToken)
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int parseOneRow(SInsertParseContext* pCxt, STableDataBlocks* pDataBlocks, int16_t timePrec, int32_t* len,
|
static int parseOneRow(SInsertParseContext* pCxt, STableDataBlocks* pDataBlocks, int16_t timePrec, bool* gotRow, char* tmpTokenBuf) {
|
||||||
char* tmpTokenBuf) {
|
|
||||||
SParsedDataColInfo* spd = &pDataBlocks->boundColumnInfo;
|
SParsedDataColInfo* spd = &pDataBlocks->boundColumnInfo;
|
||||||
SRowBuilder* pBuilder = &pDataBlocks->rowBuilder;
|
SRowBuilder* pBuilder = &pDataBlocks->rowBuilder;
|
||||||
STSRow* row = (STSRow*)(pDataBlocks->pData + pDataBlocks->size); // skip the SSubmitBlk header
|
STSRow* row = (STSRow*)(pDataBlocks->pData + pDataBlocks->size); // skip the SSubmitBlk header
|
||||||
|
@ -879,8 +900,22 @@ static int parseOneRow(SInsertParseContext* pCxt, STableDataBlocks* pDataBlocks,
|
||||||
for (int i = 0; i < spd->numOfBound; ++i) {
|
for (int i = 0; i < spd->numOfBound; ++i) {
|
||||||
NEXT_TOKEN_WITH_PREV(pCxt->pSql, sToken);
|
NEXT_TOKEN_WITH_PREV(pCxt->pSql, sToken);
|
||||||
SSchema* pSchema = &schema[spd->boundColumns[i] - 1];
|
SSchema* pSchema = &schema[spd->boundColumns[i] - 1];
|
||||||
|
|
||||||
|
if (sToken.type == TK_NK_QUESTION) {
|
||||||
|
isParseBindParam = true;
|
||||||
|
if (NULL == pCxt->pStmtCb) {
|
||||||
|
return buildSyntaxErrMsg(&pCxt->msg, "? only used in stmt", sToken.z);
|
||||||
|
}
|
||||||
|
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (isParseBindParam) {
|
||||||
|
return buildInvalidOperationMsg(&pCxt->msg, "no mix usage for ? and values");
|
||||||
|
}
|
||||||
|
|
||||||
param.schema = pSchema;
|
param.schema = pSchema;
|
||||||
getSTSRowAppendInfo(schema, pBuilder->rowType, spd, i, ¶m.toffset, ¶m.colIdx);
|
getSTSRowAppendInfo(pBuilder->rowType, spd, i, ¶m.toffset, ¶m.colIdx);
|
||||||
CHECK_CODE(parseValueToken(&pCxt->pSql, &sToken, pSchema, timePrec, tmpTokenBuf, MemRowAppend, ¶m, &pCxt->msg));
|
CHECK_CODE(parseValueToken(&pCxt->pSql, &sToken, pSchema, timePrec, tmpTokenBuf, MemRowAppend, ¶m, &pCxt->msg));
|
||||||
|
|
||||||
if (PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
|
if (PRIMARYKEY_TIMESTAMP_COL_ID == pSchema->colId) {
|
||||||
|
@ -899,6 +934,8 @@ static int parseOneRow(SInsertParseContext* pCxt, STableDataBlocks* pDataBlocks,
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
*gotRow = true;
|
||||||
}
|
}
|
||||||
|
|
||||||
// *len = pBuilder->extendedRowSize;
|
// *len = pBuilder->extendedRowSize;
|
||||||
|
@ -929,20 +966,24 @@ static int32_t parseValues(SInsertParseContext* pCxt, STableDataBlocks* pDataBlo
|
||||||
maxRows = tSize;
|
maxRows = tSize;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t len = 0;
|
bool gotRow = false;
|
||||||
CHECK_CODE(parseOneRow(pCxt, pDataBlock, tinfo.precision, &len, tmpTokenBuf));
|
CHECK_CODE(parseOneRow(pCxt, pDataBlock, tinfo.precision, &gotRow, tmpTokenBuf));
|
||||||
pDataBlock->size += extendedRowSize; // len;
|
if (gotRow) {
|
||||||
|
pDataBlock->size += extendedRowSize; //len;
|
||||||
|
}
|
||||||
|
|
||||||
NEXT_TOKEN(pCxt->pSql, sToken);
|
NEXT_TOKEN(pCxt->pSql, sToken);
|
||||||
if (TK_NK_RP != sToken.type) {
|
if (TK_NK_RP != sToken.type) {
|
||||||
return buildSyntaxErrMsg(&pCxt->msg, ") expected", sToken.z);
|
return buildSyntaxErrMsg(&pCxt->msg, ") expected", sToken.z);
|
||||||
}
|
}
|
||||||
|
|
||||||
(*numOfRows)++;
|
if (gotRow) {
|
||||||
|
(*numOfRows)++;
|
||||||
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (0 == (*numOfRows)) {
|
if (0 == (*numOfRows) && (!TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT))) {
|
||||||
return buildSyntaxErrMsg(&pCxt->msg, "no any data points", NULL);
|
return buildSyntaxErrMsg(&pCxt->msg, "no any data points", NULL);
|
||||||
}
|
}
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
@ -964,7 +1005,7 @@ static int32_t parseValuesClause(SInsertParseContext* pCxt, STableDataBlocks* da
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static void destroyCreateSubTbReq(SVCreateTbReq* pReq) {
|
void destroyCreateSubTbReq(SVCreateTbReq* pReq) {
|
||||||
taosMemoryFreeClear(pReq->name);
|
taosMemoryFreeClear(pReq->name);
|
||||||
taosMemoryFreeClear(pReq->ctbCfg.pTag);
|
taosMemoryFreeClear(pReq->ctbCfg.pTag);
|
||||||
}
|
}
|
||||||
|
@ -998,7 +1039,6 @@ static void destroyInsertParseContext(SInsertParseContext* pCxt) {
|
||||||
taosHashCleanup(pCxt->pVgroupsHashObj);
|
taosHashCleanup(pCxt->pVgroupsHashObj);
|
||||||
|
|
||||||
destroyBlockHashmap(pCxt->pTableBlockHashObj);
|
destroyBlockHashmap(pCxt->pTableBlockHashObj);
|
||||||
destroyBlockArrayList(pCxt->pTableDataBlocks);
|
|
||||||
destroyBlockArrayList(pCxt->pVgDataBlocks);
|
destroyBlockArrayList(pCxt->pVgDataBlocks);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1008,23 +1048,41 @@ static void destroyInsertParseContext(SInsertParseContext* pCxt) {
|
||||||
// VALUES (field1_value, ...) [(field1_value2, ...) ...] | FILE csv_file_path
|
// VALUES (field1_value, ...) [(field1_value2, ...) ...] | FILE csv_file_path
|
||||||
// [...];
|
// [...];
|
||||||
static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
|
int32_t tbNum = 0;
|
||||||
|
|
||||||
// for each table
|
// for each table
|
||||||
while (1) {
|
while (1) {
|
||||||
destroyInsertParseContextForTable(pCxt);
|
|
||||||
|
|
||||||
SToken sToken;
|
SToken sToken;
|
||||||
|
char *tbName = NULL;
|
||||||
|
|
||||||
// pSql -> tb_name ...
|
// pSql -> tb_name ...
|
||||||
NEXT_TOKEN(pCxt->pSql, sToken);
|
NEXT_TOKEN(pCxt->pSql, sToken);
|
||||||
|
|
||||||
// no data in the sql string anymore.
|
// no data in the sql string anymore.
|
||||||
if (sToken.n == 0) {
|
if (sToken.n == 0) {
|
||||||
if (0 == pCxt->totalNum) {
|
if (0 == pCxt->totalNum && (!TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT))) {
|
||||||
return buildInvalidOperationMsg(&pCxt->msg, "no data in sql");
|
return buildInvalidOperationMsg(&pCxt->msg, "no data in sql");
|
||||||
;
|
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT) && tbNum > 0) {
|
||||||
|
return buildInvalidOperationMsg(&pCxt->msg, "single table allowed in one stmt");;
|
||||||
|
}
|
||||||
|
|
||||||
|
destroyInsertParseContextForTable(pCxt);
|
||||||
|
|
||||||
|
if (TK_NK_QUESTION == sToken.type) {
|
||||||
|
if (pCxt->pStmtCb) {
|
||||||
|
CHECK_CODE((*pCxt->pStmtCb->getTbNameFn)(pCxt->pStmtCb->pStmt, &tbName));
|
||||||
|
|
||||||
|
sToken.z = tbName;
|
||||||
|
sToken.n = strlen(tbName);
|
||||||
|
} else {
|
||||||
|
return buildSyntaxErrMsg(&pCxt->msg, "? only used in stmt", sToken.z);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
SToken tbnameToken = sToken;
|
SToken tbnameToken = sToken;
|
||||||
NEXT_TOKEN(pCxt->pSql, sToken);
|
NEXT_TOKEN(pCxt->pSql, sToken);
|
||||||
|
|
||||||
|
@ -1050,7 +1108,9 @@ static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
if (TK_VALUES == sToken.type) {
|
if (TK_VALUES == sToken.type) {
|
||||||
// pSql -> (field1_value, ...) [(field1_value2, ...) ...]
|
// pSql -> (field1_value, ...) [(field1_value2, ...) ...]
|
||||||
CHECK_CODE(parseValuesClause(pCxt, dataBuf));
|
CHECK_CODE(parseValuesClause(pCxt, dataBuf));
|
||||||
pCxt->pOutput->insertType = TSDB_QUERY_TYPE_INSERT;
|
TSDB_QUERY_SET_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_INSERT);
|
||||||
|
|
||||||
|
tbNum++;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1063,14 +1123,32 @@ static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
}
|
}
|
||||||
// todo
|
// todo
|
||||||
pCxt->pOutput->insertType = TSDB_QUERY_TYPE_FILE_INSERT;
|
pCxt->pOutput->insertType = TSDB_QUERY_TYPE_FILE_INSERT;
|
||||||
|
|
||||||
|
tbNum++;
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
return buildSyntaxErrMsg(&pCxt->msg, "keyword VALUES or FILE is expected", sToken.z);
|
return buildSyntaxErrMsg(&pCxt->msg, "keyword VALUES or FILE is expected", sToken.z);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if (TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT)) {
|
||||||
|
SParsedDataColInfo *tags = taosMemoryMalloc(sizeof(pCxt->tags));
|
||||||
|
if (NULL == tags) {
|
||||||
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
memcpy(tags, &pCxt->tags, sizeof(pCxt->tags));
|
||||||
|
(*pCxt->pStmtCb->setBindInfoFn)(pCxt->pStmtCb->pStmt, pCxt->pTableMeta, tags);
|
||||||
|
memset(&pCxt->tags, 0, sizeof(pCxt->tags));
|
||||||
|
|
||||||
|
(*pCxt->pStmtCb->setExecInfoFn)(pCxt->pStmtCb->pStmt, pCxt->pVgroupsHashObj, pCxt->pTableBlockHashObj);
|
||||||
|
pCxt->pVgroupsHashObj = NULL;
|
||||||
|
pCxt->pTableBlockHashObj = NULL;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
// merge according to vgId
|
// merge according to vgId
|
||||||
if (!TSDB_QUERY_HAS_TYPE(pCxt->pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT) &&
|
if (taosHashGetSize(pCxt->pTableBlockHashObj) > 0) {
|
||||||
taosHashGetSize(pCxt->pTableBlockHashObj) > 0) {
|
|
||||||
CHECK_CODE(mergeTableDataBlocks(pCxt->pTableBlockHashObj, pCxt->pOutput->payloadType, &pCxt->pVgDataBlocks));
|
CHECK_CODE(mergeTableDataBlocks(pCxt->pTableBlockHashObj, pCxt->pOutput->payloadType, &pCxt->pVgDataBlocks));
|
||||||
}
|
}
|
||||||
return buildOutput(pCxt);
|
return buildOutput(pCxt);
|
||||||
|
@ -1084,29 +1162,43 @@ static int32_t parseInsertBody(SInsertParseContext* pCxt) {
|
||||||
// [...];
|
// [...];
|
||||||
int32_t parseInsertSql(SParseContext* pContext, SQuery** pQuery) {
|
int32_t parseInsertSql(SParseContext* pContext, SQuery** pQuery) {
|
||||||
SInsertParseContext context = {
|
SInsertParseContext context = {
|
||||||
.pComCxt = pContext,
|
.pComCxt = pContext,
|
||||||
.pSql = (char*)pContext->pSql,
|
.pSql = (char*) pContext->pSql,
|
||||||
.msg = {.buf = pContext->pMsg, .len = pContext->msgLen},
|
.msg = {.buf = pContext->pMsg, .len = pContext->msgLen},
|
||||||
.pTableMeta = NULL,
|
.pTableMeta = NULL,
|
||||||
.pVgroupsHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, false),
|
.pSubTableHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), true, false),
|
||||||
.pTableBlockHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, false),
|
.totalNum = 0,
|
||||||
.pSubTableHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_VARCHAR), true, false),
|
.pOutput = (SVnodeModifOpStmt*)nodesMakeNode(QUERY_NODE_VNODE_MODIF_STMT),
|
||||||
.totalNum = 0,
|
.pStmtCb = pContext->pStmtCb
|
||||||
.pOutput = (SVnodeModifOpStmt*)nodesMakeNode(QUERY_NODE_VNODE_MODIF_STMT)};
|
};
|
||||||
|
|
||||||
if (NULL == context.pVgroupsHashObj || NULL == context.pTableBlockHashObj || NULL == context.pSubTableHashObj ||
|
if (pContext->pStmtCb && *pQuery) {
|
||||||
NULL == context.pOutput) {
|
(*pContext->pStmtCb->getExecInfoFn)(pContext->pStmtCb->pStmt, &context.pVgroupsHashObj, &context.pTableBlockHashObj);
|
||||||
|
} else {
|
||||||
|
context.pVgroupsHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_INT), true, false);
|
||||||
|
context.pTableBlockHashObj = taosHashInit(128, taosGetDefaultHashFunction(TSDB_DATA_TYPE_BIGINT), true, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (NULL == context.pVgroupsHashObj || NULL == context.pTableBlockHashObj ||
|
||||||
|
NULL == context.pSubTableHashObj || NULL == context.pOutput) {
|
||||||
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
}
|
}
|
||||||
|
|
||||||
*pQuery = taosMemoryCalloc(1, sizeof(SQuery));
|
if (pContext->pStmtCb) {
|
||||||
if (NULL == *pQuery) {
|
TSDB_QUERY_SET_TYPE(context.pOutput->insertType, TSDB_QUERY_TYPE_STMT_INSERT);
|
||||||
return TSDB_CODE_OUT_OF_MEMORY;
|
|
||||||
}
|
}
|
||||||
(*pQuery)->execMode = QUERY_EXEC_MODE_SCHEDULE;
|
|
||||||
(*pQuery)->haveResultSet = false;
|
if (NULL == *pQuery) {
|
||||||
(*pQuery)->msgType = TDMT_VND_SUBMIT;
|
*pQuery = taosMemoryCalloc(1, sizeof(SQuery));
|
||||||
(*pQuery)->pRoot = (SNode*)context.pOutput;
|
if (NULL == *pQuery) {
|
||||||
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
(*pQuery)->execMode = QUERY_EXEC_MODE_SCHEDULE;
|
||||||
|
(*pQuery)->haveResultSet = false;
|
||||||
|
(*pQuery)->msgType = TDMT_VND_SUBMIT;
|
||||||
|
(*pQuery)->pRoot = (SNode*)context.pOutput;
|
||||||
|
}
|
||||||
|
|
||||||
context.pOutput->payloadType = PAYLOAD_TYPE_KV;
|
context.pOutput->payloadType = PAYLOAD_TYPE_KV;
|
||||||
|
|
||||||
int32_t code = skipInsertInto(&context);
|
int32_t code = skipInsertInto(&context);
|
||||||
|
@ -1116,3 +1208,318 @@ int32_t parseInsertSql(SParseContext* pContext, SQuery** pQuery) {
|
||||||
destroyInsertParseContext(&context);
|
destroyInsertParseContext(&context);
|
||||||
return code;
|
return code;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t qCreateSName(SName* pName, const char* pTableName, int32_t acctId, char* dbName, char *msgBuf, int32_t msgBufLen) {
|
||||||
|
SMsgBuf msg = {.buf = msgBuf, .len =msgBufLen};
|
||||||
|
SToken sToken;
|
||||||
|
int32_t code = 0;
|
||||||
|
char *tbName = NULL;
|
||||||
|
|
||||||
|
NEXT_TOKEN(pTableName, sToken);
|
||||||
|
|
||||||
|
if (sToken.n == 0) {
|
||||||
|
return buildInvalidOperationMsg(&msg, "empty table name");
|
||||||
|
}
|
||||||
|
|
||||||
|
code = createSName(pName, &sToken, acctId, dbName, &msg);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
NEXT_TOKEN(pTableName, sToken);
|
||||||
|
|
||||||
|
if (sToken.n > 0) {
|
||||||
|
return buildInvalidOperationMsg(&msg, "table name format is wrong");
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t qBuildStmtOutput(SQuery* pQuery, SHashObj* pVgHash, SHashObj* pBlockHash) {
|
||||||
|
SVnodeModifOpStmt *modifyNode = (SVnodeModifOpStmt *)pQuery->pRoot;
|
||||||
|
int32_t code = 0;
|
||||||
|
SInsertParseContext insertCtx = {
|
||||||
|
.pVgroupsHashObj = pVgHash,
|
||||||
|
.pTableBlockHashObj = pBlockHash,
|
||||||
|
.pOutput = (SVnodeModifOpStmt*)pQuery->pRoot,
|
||||||
|
};
|
||||||
|
|
||||||
|
// merge according to vgId
|
||||||
|
if (taosHashGetSize(insertCtx.pTableBlockHashObj) > 0) {
|
||||||
|
CHECK_CODE(mergeTableDataBlocks(insertCtx.pTableBlockHashObj, modifyNode->payloadType, &insertCtx.pVgDataBlocks));
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(buildOutput(&insertCtx));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t qBindStmtTagsValue(void *pBlock, void *boundTags, int64_t suid, SName *pName, TAOS_BIND_v2 *bind, char *msgBuf, int32_t msgBufLen){
|
||||||
|
STableDataBlocks *pDataBlock = (STableDataBlocks *)pBlock;
|
||||||
|
SMsgBuf pBuf = {.buf = msgBuf, .len = msgBufLen};
|
||||||
|
SParsedDataColInfo* tags = (SParsedDataColInfo*)boundTags;
|
||||||
|
if (NULL == tags) {
|
||||||
|
return TSDB_CODE_QRY_APP_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
SKVRowBuilder tagBuilder;
|
||||||
|
if (tdInitKVRowBuilder(&tagBuilder) < 0) {
|
||||||
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
|
||||||
|
SSchema* pSchema = getTableTagSchema(pDataBlock->pTableMeta);
|
||||||
|
SKvParam param = {.builder = &tagBuilder};
|
||||||
|
|
||||||
|
for (int c = 0; c < tags->numOfBound; ++c) {
|
||||||
|
if (bind[c].is_null && bind[c].is_null[0]) {
|
||||||
|
KvRowAppend(&pBuf, NULL, 0, ¶m);
|
||||||
|
continue;
|
||||||
|
}
|
||||||
|
|
||||||
|
SSchema* pTagSchema = &pSchema[tags->boundColumns[c] - 1]; // colId starts with 1
|
||||||
|
param.schema = pTagSchema;
|
||||||
|
|
||||||
|
int32_t colLen = pTagSchema->bytes;
|
||||||
|
if (IS_VAR_DATA_TYPE(pTagSchema->type)) {
|
||||||
|
colLen = bind[c].length[0];
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(KvRowAppend(&pBuf, (char *)bind[c].buffer, colLen, ¶m));
|
||||||
|
}
|
||||||
|
|
||||||
|
SKVRow row = tdGetKVRowFromBuilder(&tagBuilder);
|
||||||
|
if (NULL == row) {
|
||||||
|
tdDestroyKVRowBuilder(&tagBuilder);
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "tag value expected");
|
||||||
|
}
|
||||||
|
tdSortKVRowByColIdx(row);
|
||||||
|
|
||||||
|
SVCreateTbReq tbReq = {0};
|
||||||
|
CHECK_CODE(buildCreateTbReq(&tbReq, pName, row, suid));
|
||||||
|
CHECK_CODE(buildCreateTbMsg(pDataBlock, &tbReq));
|
||||||
|
|
||||||
|
destroyCreateSubTbReq(&tbReq);
|
||||||
|
tdDestroyKVRowBuilder(&tagBuilder);
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t qBindStmtColsValue(void *pBlock, TAOS_BIND_v2 *bind, char *msgBuf, int32_t msgBufLen) {
|
||||||
|
STableDataBlocks *pDataBlock = (STableDataBlocks *)pBlock;
|
||||||
|
SSchema* pSchema = getTableColumnSchema(pDataBlock->pTableMeta);
|
||||||
|
int32_t extendedRowSize = getExtendedRowSize(pDataBlock);
|
||||||
|
SParsedDataColInfo* spd = &pDataBlock->boundColumnInfo;
|
||||||
|
SRowBuilder* pBuilder = &pDataBlock->rowBuilder;
|
||||||
|
SMemParam param = {.rb = pBuilder};
|
||||||
|
SMsgBuf pBuf = {.buf = msgBuf, .len = msgBufLen};
|
||||||
|
int32_t rowNum = bind->num;
|
||||||
|
|
||||||
|
CHECK_CODE(initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo));
|
||||||
|
|
||||||
|
CHECK_CODE(allocateMemForSize(pDataBlock, extendedRowSize * bind->num));
|
||||||
|
|
||||||
|
for (int32_t r = 0; r < bind->num; ++r) {
|
||||||
|
STSRow* row = (STSRow*)(pDataBlock->pData + pDataBlock->size); // skip the SSubmitBlk header
|
||||||
|
tdSRowResetBuf(pBuilder, row);
|
||||||
|
|
||||||
|
// 1. set the parsed value from sql string
|
||||||
|
for (int c = 0; c < spd->numOfBound; ++c) {
|
||||||
|
SSchema* pColSchema = &pSchema[spd->boundColumns[c] - 1];
|
||||||
|
|
||||||
|
if (bind[c].buffer_type != pColSchema->type) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "column type mis-match with buffer type");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bind[c].num != rowNum) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "row number in each bind param should be the same");
|
||||||
|
}
|
||||||
|
|
||||||
|
param.schema = pColSchema;
|
||||||
|
getSTSRowAppendInfo(pBuilder->rowType, spd, c, ¶m.toffset, ¶m.colIdx);
|
||||||
|
|
||||||
|
if (bind[c].is_null && bind[c].is_null[r]) {
|
||||||
|
if (pColSchema->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "primary timestamp should not be NULL");
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(MemRowAppend(&pBuf, NULL, 0, ¶m));
|
||||||
|
} else {
|
||||||
|
int32_t colLen = pColSchema->bytes;
|
||||||
|
if (IS_VAR_DATA_TYPE(pColSchema->type)) {
|
||||||
|
colLen = bind[c].length[r];
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(MemRowAppend(&pBuf, (char *)bind[c].buffer + bind[c].buffer_length * r, colLen, ¶m));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (PRIMARYKEY_TIMESTAMP_COL_ID == pColSchema->colId) {
|
||||||
|
TSKEY tsKey = TD_ROW_KEY(row);
|
||||||
|
checkTimestamp(pDataBlock, (const char *)&tsKey);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// set the null value for the columns that do not assign values
|
||||||
|
if ((spd->numOfBound < spd->numOfCols) && TD_IS_TP_ROW(row)) {
|
||||||
|
for (int32_t i = 0; i < spd->numOfCols; ++i) {
|
||||||
|
if (spd->cols[i].valStat == VAL_STAT_NONE) { // the primary TS key is not VAL_STAT_NONE
|
||||||
|
tdAppendColValToTpRow(pBuilder, TD_VTYPE_NONE, getNullValue(pSchema[i].type), true, pSchema[i].type, i,
|
||||||
|
spd->cols[i].toffset);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pDataBlock->size += extendedRowSize;
|
||||||
|
}
|
||||||
|
|
||||||
|
SSubmitBlk *pBlocks = (SSubmitBlk *)(pDataBlock->pData);
|
||||||
|
if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, pDataBlock, bind->num)) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "too many rows in sql, total number of rows should be less than 32767");
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t qBindStmtSingleColValue(void *pBlock, TAOS_BIND_v2 *bind, char *msgBuf, int32_t msgBufLen, int32_t colIdx, int32_t rowNum) {
|
||||||
|
STableDataBlocks *pDataBlock = (STableDataBlocks *)pBlock;
|
||||||
|
SSchema* pSchema = getTableColumnSchema(pDataBlock->pTableMeta);
|
||||||
|
int32_t extendedRowSize = getExtendedRowSize(pDataBlock);
|
||||||
|
SParsedDataColInfo* spd = &pDataBlock->boundColumnInfo;
|
||||||
|
SRowBuilder* pBuilder = &pDataBlock->rowBuilder;
|
||||||
|
SMemParam param = {.rb = pBuilder};
|
||||||
|
SMsgBuf pBuf = {.buf = msgBuf, .len = msgBufLen};
|
||||||
|
bool rowStart = (0 == colIdx);
|
||||||
|
bool rowEnd = ((colIdx + 1) == spd->numOfBound);
|
||||||
|
|
||||||
|
if (rowStart) {
|
||||||
|
CHECK_CODE(initRowBuilder(&pDataBlock->rowBuilder, pDataBlock->pTableMeta->sversion, &pDataBlock->boundColumnInfo));
|
||||||
|
CHECK_CODE(allocateMemForSize(pDataBlock, extendedRowSize * bind->num));
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int32_t r = 0; r < bind->num; ++r) {
|
||||||
|
STSRow* row = (STSRow*)(pDataBlock->pData + pDataBlock->size + extendedRowSize * r); // skip the SSubmitBlk header
|
||||||
|
if (rowStart) {
|
||||||
|
tdSRowResetBuf(pBuilder, row);
|
||||||
|
} else {
|
||||||
|
tdSRowGetBuf(pBuilder, row);
|
||||||
|
}
|
||||||
|
|
||||||
|
SSchema* pColSchema = &pSchema[spd->boundColumns[colIdx] - 1];
|
||||||
|
|
||||||
|
if (bind->buffer_type != pColSchema->type) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "column type mis-match with buffer type");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (bind->num != rowNum) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "row number in each bind param should be the same");
|
||||||
|
}
|
||||||
|
|
||||||
|
param.schema = pColSchema;
|
||||||
|
getSTSRowAppendInfo(pBuilder->rowType, spd, colIdx, ¶m.toffset, ¶m.colIdx);
|
||||||
|
|
||||||
|
if (bind->is_null && bind->is_null[r]) {
|
||||||
|
if (pColSchema->colId == PRIMARYKEY_TIMESTAMP_COL_ID) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "primary timestamp should not be NULL");
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(MemRowAppend(&pBuf, NULL, 0, ¶m));
|
||||||
|
} else {
|
||||||
|
int32_t colLen = pColSchema->bytes;
|
||||||
|
if (IS_VAR_DATA_TYPE(pColSchema->type)) {
|
||||||
|
colLen = bind->length[r];
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(MemRowAppend(&pBuf, (char *)bind->buffer + bind->buffer_length * r, colLen, ¶m));
|
||||||
|
}
|
||||||
|
|
||||||
|
if (PRIMARYKEY_TIMESTAMP_COL_ID == pColSchema->colId) {
|
||||||
|
TSKEY tsKey = TD_ROW_KEY(row);
|
||||||
|
checkTimestamp(pDataBlock, (const char *)&tsKey);
|
||||||
|
}
|
||||||
|
|
||||||
|
// set the null value for the columns that do not assign values
|
||||||
|
if (rowEnd && (spd->numOfBound < spd->numOfCols) && TD_IS_TP_ROW(row)) {
|
||||||
|
for (int32_t i = 0; i < spd->numOfCols; ++i) {
|
||||||
|
if (spd->cols[i].valStat == VAL_STAT_NONE) { // the primary TS key is not VAL_STAT_NONE
|
||||||
|
tdAppendColValToTpRow(pBuilder, TD_VTYPE_NONE, getNullValue(pSchema[i].type), true, pSchema[i].type, i,
|
||||||
|
spd->cols[i].toffset);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (rowEnd) {
|
||||||
|
pDataBlock->size += extendedRowSize * bind->num;
|
||||||
|
|
||||||
|
SSubmitBlk *pBlocks = (SSubmitBlk *)(pDataBlock->pData);
|
||||||
|
if (TSDB_CODE_SUCCESS != setBlockInfo(pBlocks, pDataBlock, bind->num)) {
|
||||||
|
return buildInvalidOperationMsg(&pBuf, "too many rows in sql, total number of rows should be less than 32767");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t buildBoundFields(SParsedDataColInfo *boundInfo, SSchema *pSchema, int32_t *fieldNum, TAOS_FIELD** fields) {
|
||||||
|
if (fields) {
|
||||||
|
*fields = taosMemoryCalloc(boundInfo->numOfBound, sizeof(TAOS_FIELD));
|
||||||
|
if (NULL == *fields) {
|
||||||
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
|
||||||
|
for (int32_t i = 0; i < boundInfo->numOfBound; ++i) {
|
||||||
|
SSchema* pTagSchema = &pSchema[boundInfo->boundColumns[i] - 1];
|
||||||
|
strcpy((*fields)[i].name, pTagSchema->name);
|
||||||
|
(*fields)[i].type = pTagSchema->type;
|
||||||
|
(*fields)[i].bytes = pTagSchema->bytes;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
*fieldNum = boundInfo->numOfBound;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t qBuildStmtTagFields(void *pBlock, void *boundTags, int32_t *fieldNum, TAOS_FIELD** fields) {
|
||||||
|
STableDataBlocks *pDataBlock = (STableDataBlocks *)pBlock;
|
||||||
|
SParsedDataColInfo* tags = (SParsedDataColInfo*)boundTags;
|
||||||
|
if (NULL == tags) {
|
||||||
|
return TSDB_CODE_QRY_APP_ERROR;
|
||||||
|
}
|
||||||
|
|
||||||
|
SSchema* pSchema = getTableTagSchema(pDataBlock->pTableMeta);
|
||||||
|
if (tags->numOfBound <= 0) {
|
||||||
|
*fieldNum = 0;
|
||||||
|
*fields = NULL;
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(buildBoundFields(tags, pSchema, fieldNum, fields));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t qBuildStmtColFields(void *pBlock, int32_t *fieldNum, TAOS_FIELD** fields) {
|
||||||
|
STableDataBlocks *pDataBlock = (STableDataBlocks *)pBlock;
|
||||||
|
SSchema* pSchema = getTableColumnSchema(pDataBlock->pTableMeta);
|
||||||
|
if (pDataBlock->boundColumnInfo.numOfBound <= 0) {
|
||||||
|
*fieldNum = 0;
|
||||||
|
if (fields) {
|
||||||
|
*fields = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
CHECK_CODE(buildBoundFields(&pDataBlock->boundColumnInfo, pSchema, fieldNum, fields));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -18,6 +18,7 @@
|
||||||
#include "catalog.h"
|
#include "catalog.h"
|
||||||
#include "parUtil.h"
|
#include "parUtil.h"
|
||||||
#include "querynodes.h"
|
#include "querynodes.h"
|
||||||
|
#include "parInt.h"
|
||||||
|
|
||||||
#define IS_RAW_PAYLOAD(t) \
|
#define IS_RAW_PAYLOAD(t) \
|
||||||
(((int)(t)) == PAYLOAD_TYPE_RAW) // 0: K-V payload for non-prepare insert, 1: rawPayload for prepare insert
|
(((int)(t)) == PAYLOAD_TYPE_RAW) // 0: K-V payload for non-prepare insert, 1: rawPayload for prepare insert
|
||||||
|
@ -102,7 +103,13 @@ int32_t boundIdxCompar(const void *lhs, const void *rhs) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
void destroyBoundColumnInfo(SParsedDataColInfo* pColList) {
|
void destroyBoundColumnInfo(void* pBoundInfo) {
|
||||||
|
if (NULL == pBoundInfo) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
SParsedDataColInfo* pColList = (SParsedDataColInfo*)pBoundInfo;
|
||||||
|
|
||||||
taosMemoryFreeClear(pColList->boundColumns);
|
taosMemoryFreeClear(pColList->boundColumns);
|
||||||
taosMemoryFreeClear(pColList->cols);
|
taosMemoryFreeClear(pColList->cols);
|
||||||
taosMemoryFreeClear(pColList->colIdxInfo);
|
taosMemoryFreeClear(pColList->colIdxInfo);
|
||||||
|
@ -149,7 +156,7 @@ static int32_t createDataBlock(size_t defaultSize, int32_t rowSize, int32_t star
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t buildCreateTbMsg(STableDataBlocks* pBlocks, SVCreateTbReq* pCreateTbReq) {
|
int32_t buildCreateTbMsg(STableDataBlocks* pBlocks, SVCreateTbReq* pCreateTbReq) {
|
||||||
int32_t len = tSerializeSVCreateTbReq(NULL, pCreateTbReq);
|
int32_t len = tSerializeSVCreateTbReq(NULL, pCreateTbReq);
|
||||||
if (pBlocks->nAllocSize - pBlocks->size < len) {
|
if (pBlocks->nAllocSize - pBlocks->size < len) {
|
||||||
pBlocks->nAllocSize += len + pBlocks->rowSize;
|
pBlocks->nAllocSize += len + pBlocks->rowSize;
|
||||||
|
@ -506,6 +513,28 @@ int32_t mergeTableDataBlocks(SHashObj* pHashObj, uint8_t payloadType, SArray** p
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int32_t allocateMemForSize(STableDataBlocks *pDataBlock, int32_t allSize) {
|
||||||
|
size_t remain = pDataBlock->nAllocSize - pDataBlock->size;
|
||||||
|
uint32_t nAllocSizeOld = pDataBlock->nAllocSize;
|
||||||
|
|
||||||
|
// expand the allocated size
|
||||||
|
if (remain < allSize) {
|
||||||
|
pDataBlock->nAllocSize = (pDataBlock->size + allSize) * 1.5;
|
||||||
|
|
||||||
|
char *tmp = taosMemoryRealloc(pDataBlock->pData, (size_t)pDataBlock->nAllocSize);
|
||||||
|
if (tmp != NULL) {
|
||||||
|
pDataBlock->pData = tmp;
|
||||||
|
memset(pDataBlock->pData + pDataBlock->size, 0, pDataBlock->nAllocSize - pDataBlock->size);
|
||||||
|
} else {
|
||||||
|
// do nothing, if allocate more memory failed
|
||||||
|
pDataBlock->nAllocSize = nAllocSizeOld;
|
||||||
|
return TSDB_CODE_TSC_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
int32_t allocateMemIfNeed(STableDataBlocks *pDataBlock, int32_t rowSize, int32_t * numOfRows) {
|
int32_t allocateMemIfNeed(STableDataBlocks *pDataBlock, int32_t rowSize, int32_t * numOfRows) {
|
||||||
size_t remain = pDataBlock->nAllocSize - pDataBlock->size;
|
size_t remain = pDataBlock->nAllocSize - pDataBlock->size;
|
||||||
const int factor = 5;
|
const int factor = 5;
|
||||||
|
@ -541,3 +570,84 @@ int initRowBuilder(SRowBuilder *pBuilder, int16_t schemaVer, SParsedDataColInfo
|
||||||
pColInfo->boundNullLen);
|
pColInfo->boundNullLen);
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t qResetStmtDataBlock(void* block, bool keepBuf) {
|
||||||
|
STableDataBlocks* pBlock = (STableDataBlocks*)block;
|
||||||
|
|
||||||
|
if (keepBuf) {
|
||||||
|
taosMemoryFreeClear(pBlock->pData);
|
||||||
|
pBlock->pData = taosMemoryMalloc(TSDB_PAYLOAD_SIZE);
|
||||||
|
if (NULL == pBlock->pData) {
|
||||||
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
memset(pBlock->pData, 0, sizeof(SSubmitBlk));
|
||||||
|
} else {
|
||||||
|
pBlock->pData = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
pBlock->ordered = true;
|
||||||
|
pBlock->prevTS = INT64_MIN;
|
||||||
|
pBlock->size = sizeof(SSubmitBlk);
|
||||||
|
pBlock->tsSource = -1;
|
||||||
|
pBlock->numOfTables = 1;
|
||||||
|
pBlock->nAllocSize = TSDB_PAYLOAD_SIZE;
|
||||||
|
pBlock->headerSize = pBlock->size;
|
||||||
|
|
||||||
|
memset(&pBlock->rowBuilder, 0, sizeof(pBlock->rowBuilder));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
int32_t qCloneStmtDataBlock(void** pDst, void* pSrc) {
|
||||||
|
*pDst = taosMemoryMalloc(sizeof(STableDataBlocks));
|
||||||
|
if (NULL == *pDst) {
|
||||||
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
|
||||||
|
memcpy(*pDst, pSrc, sizeof(STableDataBlocks));
|
||||||
|
((STableDataBlocks*)(*pDst))->cloned = true;
|
||||||
|
|
||||||
|
return qResetStmtDataBlock(*pDst, false);
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t qRebuildStmtDataBlock(void** pDst, void* pSrc) {
|
||||||
|
int32_t code = qCloneStmtDataBlock(pDst, pSrc);
|
||||||
|
if (code) {
|
||||||
|
return code;
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks *pBlock = (STableDataBlocks*)*pDst;
|
||||||
|
pBlock->pData = taosMemoryMalloc(pBlock->nAllocSize);
|
||||||
|
if (NULL == pBlock->pData) {
|
||||||
|
qFreeStmtDataBlock(pBlock);
|
||||||
|
return TSDB_CODE_OUT_OF_MEMORY;
|
||||||
|
}
|
||||||
|
|
||||||
|
memset(pBlock->pData, 0, sizeof(SSubmitBlk));
|
||||||
|
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
void qFreeStmtDataBlock(void* pDataBlock) {
|
||||||
|
if (pDataBlock == NULL) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
taosMemoryFreeClear(((STableDataBlocks*)pDataBlock)->pData);
|
||||||
|
taosMemoryFreeClear(pDataBlock);
|
||||||
|
}
|
||||||
|
|
||||||
|
void qDestroyStmtDataBlock(void* pBlock) {
|
||||||
|
if (pBlock == NULL) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
STableDataBlocks* pDataBlock = (STableDataBlocks*)pBlock;
|
||||||
|
|
||||||
|
pDataBlock->cloned = false;
|
||||||
|
destroyDataBlock(pDataBlock);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
|
@ -18,7 +18,11 @@
|
||||||
#include "parInt.h"
|
#include "parInt.h"
|
||||||
#include "parToken.h"
|
#include "parToken.h"
|
||||||
|
|
||||||
static bool isInsertSql(const char* pStr, size_t length) {
|
bool isInsertSql(const char* pStr, size_t length) {
|
||||||
|
if (NULL == pStr) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
int32_t index = 0;
|
int32_t index = 0;
|
||||||
|
|
||||||
do {
|
do {
|
||||||
|
@ -68,4 +72,4 @@ void qDestroyQuery(SQuery* pQueryNode) {
|
||||||
|
|
||||||
int32_t qExtractResultSchema(const SNode* pRoot, int32_t* numOfCols, SSchema** pSchema) {
|
int32_t qExtractResultSchema(const SNode* pRoot, int32_t* numOfCols, SSchema** pSchema) {
|
||||||
return extractResultSchema(pRoot, numOfCols, pSchema);
|
return extractResultSchema(pRoot, numOfCols, pSchema);
|
||||||
}
|
}
|
||||||
|
|
|
@ -716,7 +716,8 @@ int32_t castFunction(SScalarParam *pInput, int32_t inputNum, SScalarParam *pOutp
|
||||||
int32_t len = sprintf(varDataVal(output), "%.*s", (int32_t)(outputLen - VARSTR_HEADER_SIZE), *(int8_t *)input ? "true" : "false");
|
int32_t len = sprintf(varDataVal(output), "%.*s", (int32_t)(outputLen - VARSTR_HEADER_SIZE), *(int8_t *)input ? "true" : "false");
|
||||||
varDataSetLen(output, len);
|
varDataSetLen(output, len);
|
||||||
} else if (inputType == TSDB_DATA_TYPE_BINARY) {
|
} else if (inputType == TSDB_DATA_TYPE_BINARY) {
|
||||||
int32_t len = sprintf(varDataVal(output), "%.*s", (int32_t)(outputLen - VARSTR_HEADER_SIZE), varDataVal(input));
|
int32_t len = MIN(varDataLen(input), outputLen - VARSTR_HEADER_SIZE);
|
||||||
|
len = sprintf(varDataVal(output), "%.*s", len, varDataVal(input));
|
||||||
varDataSetLen(output, len);
|
varDataSetLen(output, len);
|
||||||
} else if (inputType == TSDB_DATA_TYPE_NCHAR) {
|
} else if (inputType == TSDB_DATA_TYPE_NCHAR) {
|
||||||
//not support
|
//not support
|
||||||
|
|
|
@ -158,6 +158,7 @@ typedef struct {
|
||||||
char secured : 2;
|
char secured : 2;
|
||||||
char spi : 2;
|
char spi : 2;
|
||||||
|
|
||||||
|
char user[TSDB_UNI_LEN];
|
||||||
uint64_t ahandle; // ahandle assigned by client
|
uint64_t ahandle; // ahandle assigned by client
|
||||||
uint32_t code; // del later
|
uint32_t code; // del later
|
||||||
uint32_t msgType;
|
uint32_t msgType;
|
||||||
|
@ -186,23 +187,23 @@ typedef enum { Normal, Quit, Release, Register } STransMsgType;
|
||||||
typedef enum { ConnNormal, ConnAcquire, ConnRelease, ConnBroken, ConnInPool } ConnStatus;
|
typedef enum { ConnNormal, ConnAcquire, ConnRelease, ConnBroken, ConnInPool } ConnStatus;
|
||||||
|
|
||||||
#define container_of(ptr, type, member) ((type*)((char*)(ptr)-offsetof(type, member)))
|
#define container_of(ptr, type, member) ((type*)((char*)(ptr)-offsetof(type, member)))
|
||||||
#define RPC_RESERVE_SIZE (sizeof(STranConnCtx))
|
#define RPC_RESERVE_SIZE (sizeof(STranConnCtx))
|
||||||
|
|
||||||
#define RPC_MSG_OVERHEAD (sizeof(SRpcHead) + sizeof(SRpcDigest))
|
#define RPC_MSG_OVERHEAD (sizeof(SRpcHead) + sizeof(SRpcDigest))
|
||||||
#define rpcHeadFromCont(cont) ((SRpcHead*)((char*)cont - sizeof(SRpcHead)))
|
#define rpcHeadFromCont(cont) ((SRpcHead*)((char*)cont - sizeof(SRpcHead)))
|
||||||
#define rpcContFromHead(msg) (msg + sizeof(SRpcHead))
|
#define rpcContFromHead(msg) (msg + sizeof(SRpcHead))
|
||||||
#define rpcMsgLenFromCont(contLen) (contLen + sizeof(SRpcHead))
|
#define rpcMsgLenFromCont(contLen) (contLen + sizeof(SRpcHead))
|
||||||
#define rpcContLenFromMsg(msgLen) (msgLen - sizeof(SRpcHead))
|
#define rpcContLenFromMsg(msgLen) (msgLen - sizeof(SRpcHead))
|
||||||
#define rpcIsReq(type) (type & 1U)
|
#define rpcIsReq(type) (type & 1U)
|
||||||
|
|
||||||
#define TRANS_RESERVE_SIZE (sizeof(STranConnCtx))
|
#define TRANS_RESERVE_SIZE (sizeof(STranConnCtx))
|
||||||
|
|
||||||
#define TRANS_MSG_OVERHEAD (sizeof(STransMsgHead))
|
#define TRANS_MSG_OVERHEAD (sizeof(STransMsgHead))
|
||||||
#define transHeadFromCont(cont) ((STransMsgHead*)((char*)cont - sizeof(STransMsgHead)))
|
#define transHeadFromCont(cont) ((STransMsgHead*)((char*)cont - sizeof(STransMsgHead)))
|
||||||
#define transContFromHead(msg) (msg + sizeof(STransMsgHead))
|
#define transContFromHead(msg) (msg + sizeof(STransMsgHead))
|
||||||
#define transMsgLenFromCont(contLen) (contLen + sizeof(STransMsgHead))
|
#define transMsgLenFromCont(contLen) (contLen + sizeof(STransMsgHead))
|
||||||
#define transContLenFromMsg(msgLen) (msgLen - sizeof(STransMsgHead));
|
#define transContLenFromMsg(msgLen) (msgLen - sizeof(STransMsgHead));
|
||||||
#define transIsReq(type) (type & 1U)
|
#define transIsReq(type) (type & 1U)
|
||||||
|
|
||||||
int rpcAuthenticateMsg(void* pMsg, int msgLen, void* pAuth, void* pKey);
|
int rpcAuthenticateMsg(void* pMsg, int msgLen, void* pAuth, void* pKey);
|
||||||
void rpcBuildAuthHead(void* pMsg, int msgLen, void* pAuth, void* pKey);
|
void rpcBuildAuthHead(void* pMsg, int msgLen, void* pAuth, void* pKey);
|
||||||
|
|
|
@ -614,35 +614,16 @@ void cliSend(SCliConn* pConn) {
|
||||||
pMsg->pCont = (void*)rpcMallocCont(0);
|
pMsg->pCont = (void*)rpcMallocCont(0);
|
||||||
pMsg->contLen = 0;
|
pMsg->contLen = 0;
|
||||||
}
|
}
|
||||||
STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
|
|
||||||
pHead->ahandle = pCtx != NULL ? (uint64_t)pCtx->ahandle : 0;
|
|
||||||
|
|
||||||
int msgLen = transMsgLenFromCont(pMsg->contLen);
|
int msgLen = transMsgLenFromCont(pMsg->contLen);
|
||||||
|
|
||||||
if (!pConn->secured) {
|
STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
|
||||||
char* buf = taosMemoryCalloc(1, msgLen + sizeof(STransUserMsg));
|
pHead->ahandle = pCtx != NULL ? (uint64_t)pCtx->ahandle : 0;
|
||||||
memcpy(buf, (char*)pHead, msgLen);
|
|
||||||
|
|
||||||
STransUserMsg* uMsg = (STransUserMsg*)(buf + msgLen);
|
|
||||||
memcpy(uMsg->user, pTransInst->user, tListLen(uMsg->user));
|
|
||||||
memcpy(uMsg->secret, pTransInst->secret, tListLen(uMsg->secret));
|
|
||||||
|
|
||||||
// to avoid mem leak
|
|
||||||
destroyUserdata(pMsg);
|
|
||||||
|
|
||||||
pMsg->pCont = (char*)buf + sizeof(STransMsgHead);
|
|
||||||
pMsg->contLen = msgLen + sizeof(STransUserMsg) - sizeof(STransMsgHead);
|
|
||||||
|
|
||||||
pHead = (STransMsgHead*)buf;
|
|
||||||
pHead->secured = 1;
|
|
||||||
msgLen += sizeof(STransUserMsg);
|
|
||||||
}
|
|
||||||
|
|
||||||
pHead->noResp = REQUEST_NO_RESP(pMsg) ? 1 : 0;
|
pHead->noResp = REQUEST_NO_RESP(pMsg) ? 1 : 0;
|
||||||
pHead->persist = REQUEST_PERSIS_HANDLE(pMsg) ? 1 : 0;
|
pHead->persist = REQUEST_PERSIS_HANDLE(pMsg) ? 1 : 0;
|
||||||
pHead->msgType = pMsg->msgType;
|
pHead->msgType = pMsg->msgType;
|
||||||
pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
|
pHead->msgLen = (int32_t)htonl((uint32_t)msgLen);
|
||||||
pHead->release = REQUEST_RELEASE_HANDLE(pCliMsg) ? 1 : 0;
|
pHead->release = REQUEST_RELEASE_HANDLE(pCliMsg) ? 1 : 0;
|
||||||
|
memcpy(pHead->user, pTransInst->user, strlen(pTransInst->user));
|
||||||
|
|
||||||
uv_buf_t wb = uv_buf_init((char*)pHead, msgLen);
|
uv_buf_t wb = uv_buf_init((char*)pHead, msgLen);
|
||||||
tDebug("%s cli conn %p %s is send to %s:%d, local info %s:%d", CONN_GET_INST_LABEL(pConn), pConn,
|
tDebug("%s cli conn %p %s is send to %s:%d, local info %s:%d", CONN_GET_INST_LABEL(pConn), pConn,
|
||||||
|
|
|
@ -46,7 +46,6 @@ typedef struct SSrvConn {
|
||||||
struct sockaddr_in addr;
|
struct sockaddr_in addr;
|
||||||
struct sockaddr_in locaddr;
|
struct sockaddr_in locaddr;
|
||||||
|
|
||||||
char secured;
|
|
||||||
int spi;
|
int spi;
|
||||||
char info[64];
|
char info[64];
|
||||||
char user[TSDB_UNI_LEN]; // user ID for the link
|
char user[TSDB_UNI_LEN]; // user ID for the link
|
||||||
|
@ -104,6 +103,13 @@ static void uvOnConnectionCb(uv_stream_t* q, ssize_t nread, const uv_buf_t* buf)
|
||||||
static void uvWorkerAsyncCb(uv_async_t* handle);
|
static void uvWorkerAsyncCb(uv_async_t* handle);
|
||||||
static void uvAcceptAsyncCb(uv_async_t* handle);
|
static void uvAcceptAsyncCb(uv_async_t* handle);
|
||||||
static void uvShutDownCb(uv_shutdown_t* req, int status);
|
static void uvShutDownCb(uv_shutdown_t* req, int status);
|
||||||
|
|
||||||
|
/*
|
||||||
|
* time-consuming task throwed into BG work thread
|
||||||
|
*/
|
||||||
|
static void uvWorkDoTask(uv_work_t* req);
|
||||||
|
static void uvWorkAfterTask(uv_work_t* req, int status);
|
||||||
|
|
||||||
static void uvWalkCb(uv_handle_t* handle, void* arg);
|
static void uvWalkCb(uv_handle_t* handle, void* arg);
|
||||||
static void uvFreeCb(uv_handle_t* handle);
|
static void uvFreeCb(uv_handle_t* handle);
|
||||||
|
|
||||||
|
@ -181,16 +187,16 @@ static void uvHandleReq(SSrvConn* pConn) {
|
||||||
uint32_t msgLen = pBuf->len;
|
uint32_t msgLen = pBuf->len;
|
||||||
|
|
||||||
STransMsgHead* pHead = (STransMsgHead*)msg;
|
STransMsgHead* pHead = (STransMsgHead*)msg;
|
||||||
if (pHead->secured == 1) {
|
|
||||||
STransUserMsg* uMsg = (STransUserMsg*)((char*)msg + msgLen - sizeof(STransUserMsg));
|
|
||||||
memcpy(pConn->user, uMsg->user, tListLen(uMsg->user));
|
|
||||||
memcpy(pConn->secret, uMsg->secret, tListLen(uMsg->secret));
|
|
||||||
}
|
|
||||||
pHead->code = htonl(pHead->code);
|
pHead->code = htonl(pHead->code);
|
||||||
pHead->msgLen = htonl(pHead->msgLen);
|
pHead->msgLen = htonl(pHead->msgLen);
|
||||||
if (pHead->secured == 1) {
|
memcpy(pConn->user, pHead->user, strlen(pHead->user));
|
||||||
pHead->msgLen -= sizeof(STransUserMsg);
|
|
||||||
}
|
// TODO(dengyihao): time-consuming task throwed into BG Thread
|
||||||
|
// uv_work_t* wreq = taosMemoryMalloc(sizeof(uv_work_t));
|
||||||
|
// wreq->data = pConn;
|
||||||
|
// uv_read_stop((uv_stream_t*)pConn->pTcp);
|
||||||
|
// transRefSrvHandle(pConn);
|
||||||
|
// uv_queue_work(((SWorkThrdObj*)pConn->hostThrd)->loop, wreq, uvWorkDoTask, uvWorkAfterTask);
|
||||||
|
|
||||||
CONN_SHOULD_RELEASE(pConn, pHead);
|
CONN_SHOULD_RELEASE(pConn, pHead);
|
||||||
|
|
||||||
|
@ -344,12 +350,6 @@ static void uvPrepareSendData(SSrvMsg* smsg, uv_buf_t* wb) {
|
||||||
STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
|
STransMsgHead* pHead = transHeadFromCont(pMsg->pCont);
|
||||||
pHead->ahandle = (uint64_t)pMsg->ahandle;
|
pHead->ahandle = (uint64_t)pMsg->ahandle;
|
||||||
|
|
||||||
// pHead->secured = pMsg->code == 0 ? 1 : 0; //
|
|
||||||
if (!pConn->secured) {
|
|
||||||
pConn->secured = pMsg->code == 0 ? 1 : 0;
|
|
||||||
}
|
|
||||||
pHead->secured = pConn->secured;
|
|
||||||
|
|
||||||
if (pConn->status == ConnNormal) {
|
if (pConn->status == ConnNormal) {
|
||||||
pHead->msgType = pConn->inType + 1;
|
pHead->msgType = pConn->inType + 1;
|
||||||
} else {
|
} else {
|
||||||
|
@ -464,6 +464,24 @@ static void uvShutDownCb(uv_shutdown_t* req, int status) {
|
||||||
taosMemoryFree(req);
|
taosMemoryFree(req);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
static void uvWorkDoTask(uv_work_t* req) {
|
||||||
|
// doing time-consumeing task
|
||||||
|
// only auth conn currently, add more func later
|
||||||
|
tTrace("server conn %p start to be processed in BG Thread", req->data);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void uvWorkAfterTask(uv_work_t* req, int status) {
|
||||||
|
if (status != 0) {
|
||||||
|
tTrace("server conn %p failed to processed ", req->data);
|
||||||
|
}
|
||||||
|
// Done time-consumeing task
|
||||||
|
// add more func later
|
||||||
|
// this func called in main loop
|
||||||
|
tTrace("server conn %p already processed ", req->data);
|
||||||
|
taosMemoryFree(req);
|
||||||
|
}
|
||||||
|
|
||||||
void uvOnAcceptCb(uv_stream_t* stream, int status) {
|
void uvOnAcceptCb(uv_stream_t* stream, int status) {
|
||||||
if (status == -1) {
|
if (status == -1) {
|
||||||
return;
|
return;
|
||||||
|
|
|
@ -138,6 +138,9 @@ TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON, "Invalid JSON format")
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON_TYPE, "Invalid JSON data type")
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_JSON_TYPE, "Invalid JSON data type")
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_VALUE_OUT_OF_RANGE, "Value out of range")
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_VALUE_OUT_OF_RANGE, "Value out of range")
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_INPUT, "Invalid tsc input")
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_INVALID_INPUT, "Invalid tsc input")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_STMT_API_ERROR, "Stmt API usage error")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_STMT_TBNAME_ERROR, "Stmt table name not set")
|
||||||
|
TAOS_DEFINE_ERROR(TSDB_CODE_TSC_STMT_CLAUSE_ERROR, "not supported stmt clause")
|
||||||
|
|
||||||
// mnode-common
|
// mnode-common
|
||||||
TAOS_DEFINE_ERROR(TSDB_CODE_MND_APP_ERROR, "Mnode internal error")
|
TAOS_DEFINE_ERROR(TSDB_CODE_MND_APP_ERROR, "Mnode internal error")
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
|
@ -4,18 +4,14 @@
|
||||||
ROOT=./
|
ROOT=./
|
||||||
TARGET=exe
|
TARGET=exe
|
||||||
LFLAGS = '-Wl,-rpath,/usr/local/taos/driver/' -ltaos -lpthread -lm -lrt
|
LFLAGS = '-Wl,-rpath,/usr/local/taos/driver/' -ltaos -lpthread -lm -lrt
|
||||||
CFLAGS = -O0 -g -Wall -Wno-deprecated -fPIC -Wno-unused-result -Wconversion \
|
CFLAGS = -O0 -g -Wno-deprecated -fPIC -Wno-unused-result -Wconversion \
|
||||||
-Wno-char-subscripts -D_REENTRANT -Wno-format -D_REENTRANT -DLINUX \
|
-Wno-char-subscripts -D_REENTRANT -Wno-format -D_REENTRANT -DLINUX \
|
||||||
-Wno-unused-function -D_M_X64 -I/usr/local/taos/include -std=gnu99 \
|
-Wno-unused-function -D_M_X64 -I/usr/local/taos/include -std=gnu99 -Wno-sign-conversion
|
||||||
-fsanitize=address
|
|
||||||
|
|
||||||
all: $(TARGET)
|
all: $(TARGET)
|
||||||
|
|
||||||
exe:
|
exe:
|
||||||
gcc $(CFLAGS) ./batchprepare.c -o $(ROOT)batchprepare $(LFLAGS)
|
gcc $(CFLAGS) ./batchprepare.c -o $(ROOT)batchprepare $(LFLAGS)
|
||||||
gcc $(CFLAGS) ./stmtBatchTest.c -o $(ROOT)stmtBatchTest $(LFLAGS)
|
|
||||||
gcc $(CFLAGS) ./stmtTest.c -o $(ROOT)stmtTest $(LFLAGS)
|
|
||||||
gcc $(CFLAGS) ./stmt_function.c -o $(ROOT)stmt_function $(LFLAGS)
|
|
||||||
|
|
||||||
clean:
|
clean:
|
||||||
rm $(ROOT)batchprepare
|
rm $(ROOT)batchprepare
|
||||||
|
|
|
@ -1,4 +1,4 @@
|
||||||
!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
##################################################
|
##################################################
|
||||||
#
|
#
|
||||||
|
|
|
@ -1,5 +1,6 @@
|
||||||
import taos
|
import taos
|
||||||
import sys
|
import sys
|
||||||
|
import datetime
|
||||||
|
|
||||||
from util.log import *
|
from util.log import *
|
||||||
from util.sql import *
|
from util.sql import *
|
||||||
|
@ -65,9 +66,9 @@ class TDTestCase:
|
||||||
)
|
)
|
||||||
|
|
||||||
tdSql.query("select c1 from ct4")
|
tdSql.query("select c1 from ct4")
|
||||||
data_ct4 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
data_ct4_c1 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
tdSql.query("select c1 from t1")
|
tdSql.query("select c1 from t1")
|
||||||
data_t1 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
data_t1_c1 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
# tdLog.printNoPrefix("==========step1: cast int to int, expect no changes")
|
# tdLog.printNoPrefix("==========step1: cast int to int, expect no changes")
|
||||||
|
|
||||||
|
@ -82,11 +83,11 @@ class TDTestCase:
|
||||||
tdLog.printNoPrefix("==========step2: cast int to bigint, expect no changes")
|
tdLog.printNoPrefix("==========step2: cast int to bigint, expect no changes")
|
||||||
|
|
||||||
tdSql.query("select cast(c1 as bigint) as b from ct4")
|
tdSql.query("select cast(c1 as bigint) as b from ct4")
|
||||||
for i in range(len(data_ct4)):
|
for i in range(len(data_ct4_c1)):
|
||||||
tdSql.checkData( i, 0, data_ct4[i])
|
tdSql.checkData( i, 0, data_ct4_c1[i])
|
||||||
tdSql.query("select cast(c1 as bigint) as b from t1")
|
tdSql.query("select cast(c1 as bigint) as b from t1")
|
||||||
for i in range(len(data_t1)):
|
for i in range(len(data_t1_c1)):
|
||||||
tdSql.checkData( i, 0, data_t1[i])
|
tdSql.checkData( i, 0, data_t1_c1[i])
|
||||||
|
|
||||||
# tdLog.printNoPrefix("==========step3: cast int to float, expect no changes")
|
# tdLog.printNoPrefix("==========step3: cast int to float, expect no changes")
|
||||||
|
|
||||||
|
@ -106,32 +107,447 @@ class TDTestCase:
|
||||||
# for i in range(len(data_t1)):
|
# for i in range(len(data_t1)):
|
||||||
# tdSql.checkData( i, 0, data_t1[i])
|
# tdSql.checkData( i, 0, data_t1[i])
|
||||||
|
|
||||||
tdLog.printNoPrefix("==========step5: cast int to binary, expect no changes")
|
tdLog.printNoPrefix("==========step5: cast int to binary, expect changes to str(int) ")
|
||||||
|
|
||||||
tdSql.query("select cast(c1 as binary(32)) as b from ct4")
|
tdSql.query("select cast(c1 as binary(32)) as b from ct4")
|
||||||
for i in range(len(data_ct4)):
|
for i in range(len(data_ct4_c1)):
|
||||||
tdSql.checkData( i, 0, str(data_ct4[i]) )
|
tdSql.checkData( i, 0, str(data_ct4_c1[i]) )
|
||||||
tdSql.query("select cast(c1 as binary(32)) as b from t1")
|
tdSql.query("select cast(c1 as binary(32)) as b from t1")
|
||||||
for i in range(len(data_t1)):
|
for i in range(len(data_t1_c1)):
|
||||||
tdSql.checkData( i, 0, str(data_t1[i]) )
|
tdSql.checkData( i, 0, str(data_t1_c1[i]) )
|
||||||
|
|
||||||
tdLog.printNoPrefix("==========step6: cast int to nchar, expect no changes")
|
tdLog.printNoPrefix("==========step6: cast int to nchar, expect changes to str(int) ")
|
||||||
|
|
||||||
tdSql.query("select cast(c1 as nchar(32)) as b from ct4")
|
tdSql.query("select cast(c1 as nchar(32)) as b from ct4")
|
||||||
for i in range(len(data_ct4)):
|
for i in range(len(data_ct4_c1)):
|
||||||
tdSql.checkData( i, 0, str(data_ct4[i]) )
|
tdSql.checkData( i, 0, str(data_ct4_c1[i]) )
|
||||||
tdSql.query("select cast(c1 as nchar(32)) as b from t1")
|
tdSql.query("select cast(c1 as nchar(32)) as b from t1")
|
||||||
for i in range(len(data_t1)):
|
for i in range(len(data_t1_c1)):
|
||||||
tdSql.checkData( i, 0, str(data_t1[i]) )
|
tdSql.checkData( i, 0, str(data_t1_c1[i]) )
|
||||||
|
|
||||||
tdLog.printNoPrefix("==========step7: cast int to timestamp, expect no changes")
|
tdLog.printNoPrefix("==========step7: cast int to timestamp, expect changes to timestamp ")
|
||||||
|
|
||||||
tdSql.query("select cast(c1 as timestamp) as b from ct4")
|
tdSql.query("select cast(c1 as timestamp) as b from ct4")
|
||||||
for i in range(len(data_ct4)):
|
for i in range(len(data_ct4_c1)):
|
||||||
tdSql.checkData( i, 0, data_ct4[i])
|
if data_ct4_c1[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c1[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
tdSql.query("select cast(c1 as timestamp) as b from t1")
|
tdSql.query("select cast(c1 as timestamp) as b from t1")
|
||||||
for i in range(len(data_t1)):
|
for i in range(len(data_t1_c1)):
|
||||||
tdSql.checkData( i, 0, data_t1[i])
|
if data_ct4_c1[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c1[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step8: cast bigint to bigint, expect no changes")
|
||||||
|
tdSql.query("select c2 from ct4")
|
||||||
|
data_ct4_c2 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c2 from t1")
|
||||||
|
data_t1_c2 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdSql.query("select cast(c2 as bigint) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c2)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c2[i])
|
||||||
|
tdSql.query("select cast(c2 as bigint) as b from t1")
|
||||||
|
for i in range(len(data_t1_c2)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c2[i])
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step9: cast bigint to binary, expect changes to str(int) ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c2 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c2)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c2[i]) )
|
||||||
|
tdSql.query("select cast(c2 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c2)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c2[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step10: cast bigint to nchar, expect changes to str(int) ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c2 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c2)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c2[i]) )
|
||||||
|
tdSql.query("select cast(c2 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c2)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c2[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step11: cast bigint to timestamp, expect changes to timestamp ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c2 as timestamp) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c2)):
|
||||||
|
if data_ct4_c2[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c2[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
tdSql.query("select cast(c2 as timestamp) as b from t1")
|
||||||
|
for i in range(len(data_t1_c2)):
|
||||||
|
if data_t1_c2[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_t1_c2[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step12: cast smallint to bigint, expect no changes")
|
||||||
|
tdSql.query("select c3 from ct4")
|
||||||
|
data_ct4_c3 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c3 from t1")
|
||||||
|
data_t1_c3 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdSql.query("select cast(c3 as bigint) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c3)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c3[i])
|
||||||
|
tdSql.query("select cast(c3 as bigint) as b from t1")
|
||||||
|
for i in range(len(data_t1_c3)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c3[i])
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step13: cast smallint to binary, expect changes to str(int) ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c3 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c3)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c3[i]) )
|
||||||
|
tdSql.query("select cast(c3 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c3)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c3[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step14: cast smallint to nchar, expect changes to str(int) ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c3 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c3)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c3[i]) )
|
||||||
|
tdSql.query("select cast(c3 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c3)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c3[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step15: cast smallint to timestamp, expect changes to timestamp ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c3 as timestamp) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c3)):
|
||||||
|
if data_ct4_c3[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c3[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
tdSql.query("select cast(c3 as timestamp) as b from t1")
|
||||||
|
for i in range(len(data_t1_c3)):
|
||||||
|
if data_ct4_c3[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c3[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step16: cast smallint to bigint, expect no changes")
|
||||||
|
tdSql.query("select c4 from ct4")
|
||||||
|
data_ct4_c4 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c4 from t1")
|
||||||
|
data_t1_c4 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdSql.query("select cast(c4 as bigint) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c4)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c4[i])
|
||||||
|
tdSql.query("select cast(c4 as bigint) as b from t1")
|
||||||
|
for i in range(len(data_t1_c4)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c4[i])
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step17: cast smallint to binary, expect changes to str(int) ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c4 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c4)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c4[i]) )
|
||||||
|
tdSql.query("select cast(c4 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c4)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c4[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step18: cast smallint to nchar, expect changes to str(int) ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c4 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c4)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c4[i]) )
|
||||||
|
tdSql.query("select cast(c4 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c4)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c4[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step19: cast smallint to timestamp, expect changes to timestamp ")
|
||||||
|
|
||||||
|
tdSql.query("select cast(c4 as timestamp) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c4)):
|
||||||
|
if data_ct4_c4[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c4[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
tdSql.query("select cast(c4 as timestamp) as b from t1")
|
||||||
|
for i in range(len(data_t1_c4)):
|
||||||
|
if data_ct4_c4[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(data_ct4_c4[i]/1000)
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step20: cast float to bigint, expect no changes")
|
||||||
|
tdSql.query("select c5 from ct4")
|
||||||
|
data_ct4_c5 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c5 from t1")
|
||||||
|
data_t1_c5 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdSql.query("select cast(c5 as bigint) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c5)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c5[i] ) if data_ct4_c5[i] is None else tdSql.checkData( i, 0, int(data_ct4_c5[i]) )
|
||||||
|
tdSql.query("select cast(c5 as bigint) as b from t1")
|
||||||
|
for i in range(len(data_t1_c5)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c5[i] ) if data_t1_c5[i] is None else tdSql.checkData( i, 0, int(data_t1_c5[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step21: cast float to binary, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c5 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c5)):
|
||||||
|
# tdSql.checkData( i, 0, str(data_ct4_c5[i]) ) if data_ct4_c5[i] is None else tdSql.checkData( i, 0, str(round(data_ct4_c5[i], 6)) )
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c5[i]) ) if data_ct4_c5[i] is None else tdSql.checkData( i, 0, f'{data_ct4_c5[i]:.6f}' )
|
||||||
|
tdSql.query("select cast(c5 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c5)):
|
||||||
|
# tdSql.checkData( i, 0, str(data_t1_c5[i]) ) if data_t1_c5[i] is None else tdSql.checkData( i, 0, str(round(data_t1_c5[i], 6)) )
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c5[i]) ) if data_t1_c5[i] is None else tdSql.checkData( i, 0, f'{data_t1_c5[i]:.6f}' )
|
||||||
|
# tdSql.checkData( i, 0, str(data_t1_c5[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step22: cast float to nchar, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c5 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c5)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c5[i]) ) if data_ct4_c5[i] is None else tdSql.checkData( i, 0, f'{data_ct4_c5[i]:.6f}' )
|
||||||
|
tdSql.query("select cast(c5 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c5)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c5[i]) ) if data_t1_c5[i] is None else tdSql.checkData( i, 0, f'{data_t1_c5[i]:.6f}' )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step23: cast float to timestamp, expect changes to timestamp ")
|
||||||
|
tdSql.query("select cast(c5 as timestamp) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c5)):
|
||||||
|
if data_ct4_c5[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(int(data_ct4_c5[i]/1000))
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
tdSql.query("select cast(c5 as timestamp) as b from t1")
|
||||||
|
for i in range(len(data_t1_c5)):
|
||||||
|
if data_t1_c5[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(int(data_t1_c5[i]/1000))
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step24: cast double to bigint, expect no changes")
|
||||||
|
tdSql.query("select c6 from ct4")
|
||||||
|
data_ct4_c6 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c6 from t1")
|
||||||
|
data_t1_c6 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdSql.query("select cast(c6 as bigint) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c6)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c6[i] ) if data_ct4_c6[i] is None else tdSql.checkData( i, 0, int(data_ct4_c6[i]) )
|
||||||
|
tdSql.query("select cast(c6 as bigint) as b from t1")
|
||||||
|
for i in range(len(data_t1_c6)):
|
||||||
|
if data_t1_c6[i] is None:
|
||||||
|
tdSql.checkData( i, 0, None )
|
||||||
|
elif data_t1_c6[i] > 99999999 or data_t1_c6[i] < -999999:
|
||||||
|
continue
|
||||||
|
else:
|
||||||
|
tdSql.checkData( i, 0, int(data_t1_c6[i]) )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step25: cast double to binary, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c6 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c6)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c6[i]) ) if data_ct4_c6[i] is None else tdSql.checkData( i, 0, f'{data_ct4_c6[i]:.6f}' )
|
||||||
|
tdSql.query("select cast(c6 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c6)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c6[i]) ) if data_t1_c6[i] is None else tdSql.checkData( i, 0, f'{data_t1_c6[i]:.6f}' )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step26: cast double to nchar, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c6 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c6)):
|
||||||
|
tdSql.checkData( i, 0, str(data_ct4_c6[i]) ) if data_ct4_c6[i] is None else tdSql.checkData( i, 0, f'{data_ct4_c6[i]:.6f}' )
|
||||||
|
tdSql.query("select cast(c6 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c6)):
|
||||||
|
tdSql.checkData( i, 0, str(data_t1_c6[i]) ) if data_t1_c6[i] is None else tdSql.checkData( i, 0, f'{data_t1_c6[i]:.6f}' )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step27: cast double to timestamp, expect changes to timestamp ")
|
||||||
|
tdSql.query("select cast(c6 as timestamp) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c6)):
|
||||||
|
if data_ct4_c6[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(int(data_ct4_c6[i]/1000))
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
# tdSql.query("select cast(c6 as timestamp) as b from t1")
|
||||||
|
# for i in range(len(data_t1_c6)):
|
||||||
|
# if data_t1_c6[i] is None:
|
||||||
|
# tdSql.checkData( i, 0 , None )
|
||||||
|
# else:
|
||||||
|
# utc_zone = datetime.timezone.utc
|
||||||
|
# utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
# date_init_stamp = datetime.datetime.utcfromtimestamp(int(data_t1_c6[i]/1000))
|
||||||
|
# date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
# tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step28: cast bool to bigint, expect no changes")
|
||||||
|
tdSql.query("select c7 from ct4")
|
||||||
|
data_ct4_c7 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c7 from t1")
|
||||||
|
data_t1_c7 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdSql.query("select cast(c7 as bigint) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c7)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c7[i])
|
||||||
|
tdSql.query("select cast(c7 as bigint) as b from t1")
|
||||||
|
for i in range(len(data_t1_c7)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c7[i])
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step29: cast bool to binary, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c7 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c7)):
|
||||||
|
tdSql.checkData( i, 0, None ) if data_ct4_c7[i] is None else tdSql.checkData( i, 0, str(data_ct4_c7[i]).lower() )
|
||||||
|
tdSql.query("select cast(c7 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c7)):
|
||||||
|
tdSql.checkData( i, 0, None ) if data_t1_c7[i] is None else tdSql.checkData( i, 0, str(data_t1_c7[i]).lower() )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step30: cast bool to nchar, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c7 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c7)):
|
||||||
|
tdSql.checkData( i, 0, None ) if data_ct4_c7[i] is None else tdSql.checkData( i, 0, str(data_ct4_c7[i]).lower() )
|
||||||
|
tdSql.query("select cast(c7 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c7)):
|
||||||
|
tdSql.checkData( i, 0, None ) if data_t1_c7[i] is None else tdSql.checkData( i, 0, str(data_t1_c7[i]).lower() )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step31: cast bool to timestamp, expect changes to timestamp ")
|
||||||
|
tdSql.query("select cast(c7 as timestamp) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c7)):
|
||||||
|
if data_ct4_c7[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(int(data_ct4_c7[i]/1000))
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
tdSql.query("select cast(c7 as timestamp) as b from t1")
|
||||||
|
for i in range(len(data_t1_c7)):
|
||||||
|
if data_t1_c7[i] is None:
|
||||||
|
tdSql.checkData( i, 0 , None )
|
||||||
|
else:
|
||||||
|
utc_zone = datetime.timezone.utc
|
||||||
|
utc_8 = datetime.timezone(datetime.timedelta(hours=8))
|
||||||
|
date_init_stamp = datetime.datetime.utcfromtimestamp(int(data_t1_c7[i]/1000))
|
||||||
|
date_data = date_init_stamp.replace(tzinfo=utc_zone).astimezone(utc_8).strftime("%Y-%m-%d %H:%M:%S.%f")
|
||||||
|
tdSql.checkData( i, 0, date_data)
|
||||||
|
|
||||||
|
|
||||||
|
tdSql.query("select c8 from ct4")
|
||||||
|
data_ct4_c8 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c8 from t1")
|
||||||
|
data_t1_c8 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step32: cast binary to binary, expect no changes ")
|
||||||
|
tdSql.query("select cast(c8 as binary(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c8)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c8[i] )
|
||||||
|
tdSql.query("select cast(c8 as binary(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c8)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c8[i] )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step33: cast binary to binary, expect truncate ")
|
||||||
|
tdSql.query("select cast(c8 as binary(2)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c8)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c8[i][:2] )
|
||||||
|
tdSql.query("select cast(c8 as binary(2)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c8)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c8[i][:2] )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step34: cast binary to nchar, expect changes to str(int) ")
|
||||||
|
tdSql.query("select cast(c8 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c8)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c8[i] )
|
||||||
|
tdSql.query("select cast(c8 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c8)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c8[i] )
|
||||||
|
|
||||||
|
|
||||||
|
tdSql.query("select c9 from ct4")
|
||||||
|
data_ct4_c9 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c9 from t1")
|
||||||
|
data_t1_c9 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
"c10 timestamp"
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step35: cast nchar to nchar, expect no changes ")
|
||||||
|
tdSql.query("select cast(c9 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c9)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c9[i])
|
||||||
|
tdSql.query("select cast(c9 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c9)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c9[i] )
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step36: cast nchar to nchar, expect truncate ")
|
||||||
|
tdSql.query("select cast(c9 as nchar(2)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c9)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c9[i][:2] )
|
||||||
|
tdSql.query("select cast(c9 as nchar(2)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c9)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c9[i][:2] )
|
||||||
|
|
||||||
|
tdSql.query("select c9 from ct4")
|
||||||
|
data_ct4_c10 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
tdSql.query("select c9 from t1")
|
||||||
|
data_t1_c10 = [tdSql.getData(i,0) for i in range(tdSql.queryRows)]
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step37: cast timestamp to nchar, expect no changes ")
|
||||||
|
tdSql.query("select cast(c9 as nchar(32)) as b from ct4")
|
||||||
|
for i in range(len(data_ct4_c10)):
|
||||||
|
tdSql.checkData( i, 0, data_ct4_c10[i])
|
||||||
|
tdSql.query("select cast(c9 as nchar(32)) as b from t1")
|
||||||
|
for i in range(len(data_t1_c10)):
|
||||||
|
tdSql.checkData( i, 0, data_t1_c10[i] )
|
||||||
|
|
||||||
|
|
||||||
tdSql.error("select cast(c1 as int) as b from ct4")
|
tdSql.error("select cast(c1 as int) as b from ct4")
|
||||||
tdSql.error("select cast(c1 as bool) as b from ct4")
|
tdSql.error("select cast(c1 as bool) as b from ct4")
|
||||||
|
@ -151,10 +567,10 @@ class TDTestCase:
|
||||||
tdSql.error("select cast(c7 as double) as b from ct4")
|
tdSql.error("select cast(c7 as double) as b from ct4")
|
||||||
tdSql.error("select cast(c8 as tinyint unsigned) as b from ct4")
|
tdSql.error("select cast(c8 as tinyint unsigned) as b from ct4")
|
||||||
|
|
||||||
tdSql.error("select cast(c8 as timestamp ) as b from ct4")
|
tdSql.query("select cast(c8 as timestamp ) as b from ct4")
|
||||||
|
tdSql.query("select cast(c9 as timestamp ) as b from ct4")
|
||||||
|
|
||||||
tdSql.error("select cast(c9 as binary(64) ) as b from ct4")
|
tdSql.error("select cast(c9 as binary(64) ) as b from ct4")
|
||||||
tdSql.error("select cast(c9 as timestamp ) as b from ct4")
|
|
||||||
tdSql.error("select cast(c10 as binary(64) ) as b from ct4")
|
tdSql.error("select cast(c10 as binary(64) ) as b from ct4")
|
||||||
tdSql.error("select cast(c10 as nchar(64) ) as b from ct4")
|
tdSql.error("select cast(c10 as nchar(64) ) as b from ct4")
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,262 @@
|
||||||
|
import taos
|
||||||
|
import sys
|
||||||
|
|
||||||
|
from util.log import *
|
||||||
|
from util.sql import *
|
||||||
|
from util.cases import *
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class TDTestCase:
|
||||||
|
|
||||||
|
def init(self, conn, logSql):
|
||||||
|
tdLog.debug(f"start to excute {__file__}")
|
||||||
|
tdSql.init(conn.cursor())
|
||||||
|
|
||||||
|
def run(self): # sourcery skip: extract-duplicate-method, remove-redundant-fstring
|
||||||
|
tdSql.prepare()
|
||||||
|
|
||||||
|
tdLog.printNoPrefix("==========step1:create table")
|
||||||
|
tdSql.execute("create stable db.stb1 (ts timestamp, c1 int, c2 int) tags(t0 tinyint, t1 int, t2 int)")
|
||||||
|
tdSql.execute("create stable db.stb2 (ts timestamp, c2 int, c3 binary(16)) tags(t2 binary(16), t3 binary(16), t4 int)")
|
||||||
|
maxRemainderNum=7
|
||||||
|
tbnum=101
|
||||||
|
for i in range(tbnum-1):
|
||||||
|
sql = f"create table db.t{i} using db.stb1 tags({i%maxRemainderNum}, {(i-1)%maxRemainderNum}, {i%2})"
|
||||||
|
tdSql.execute(sql)
|
||||||
|
tdSql.execute(f"insert into db.t{i} values (now-10d, {i}, {i%3})")
|
||||||
|
tdSql.execute(f"insert into db.t{i} values (now-9d, {i}, {(i-1)%3})")
|
||||||
|
tdSql.execute(f"insert into db.t{i} values (now-8d, {i}, {(i-2)%3})")
|
||||||
|
tdSql.execute(f"insert into db.t{i} (ts )values (now-7d)")
|
||||||
|
|
||||||
|
tdSql.execute(f"create table db.t0{i} using db.stb2 tags('{i%maxRemainderNum}', '{(i-1)%maxRemainderNum}', {i%3})")
|
||||||
|
tdSql.execute(f"insert into db.t0{i} values (now-10d, {i}, '{(i+1)%3}')")
|
||||||
|
tdSql.execute(f"insert into db.t0{i} values (now-9d, {i}, '{(i+2)%3}')")
|
||||||
|
tdSql.execute(f"insert into db.t0{i} values (now-8d, {i}, '{(i)%3}')")
|
||||||
|
tdSql.execute(f"insert into db.t0{i} (ts )values (now-7d)")
|
||||||
|
# tdSql.execute("create table db.t100num using db.stb1 tags(null, null, null)")
|
||||||
|
# tdSql.execute("create table db.t0100num using db.stb2 tags(null, null, null)")
|
||||||
|
# tdSql.execute(f"insert into db.t100num values (now-10d, {tbnum-1}, 1)")
|
||||||
|
# tdSql.execute(f"insert into db.t100num values (now-9d, {tbnum-1}, 0)")
|
||||||
|
# tdSql.execute(f"insert into db.t100num values (now-8d, {tbnum-1}, 2)")
|
||||||
|
# tdSql.execute(f"insert into db.t100num (ts )values (now-7d)")
|
||||||
|
# tdSql.execute(f"insert into db.t0100num values (now-10d, {tbnum-1}, 1)")
|
||||||
|
# tdSql.execute(f"insert into db.t0100num values (now-9d, {tbnum-1}, 0)")
|
||||||
|
# tdSql.execute(f"insert into db.t0100num values (now-8d, {tbnum-1}, 2)")
|
||||||
|
# tdSql.execute(f"insert into db.t0100num (ts )values (now-7d)")
|
||||||
|
|
||||||
|
#========== distinct multi-data-coloumn ==========
|
||||||
|
# tdSql.query(f"select distinct c1 from stb1 where c1 <{tbnum}")
|
||||||
|
# tdSql.checkRows(tbnum)
|
||||||
|
# tdSql.query(f"select distinct c2 from stb1")
|
||||||
|
# tdSql.checkRows(4)
|
||||||
|
# tdSql.query(f"select distinct c1,c2 from stb1 where c1 <{tbnum}")
|
||||||
|
# tdSql.checkRows(tbnum*3)
|
||||||
|
# tdSql.query(f"select distinct c1,c1 from stb1 where c1 <{tbnum}")
|
||||||
|
# tdSql.checkRows(tbnum)
|
||||||
|
# tdSql.query(f"select distinct c1,c2 from stb1 where c1 <{tbnum} limit 3")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.query(f"select distinct c1,c2 from stb1 where c1 <{tbnum} limit 3 offset {tbnum*3-2}")
|
||||||
|
# tdSql.checkRows(2)
|
||||||
|
|
||||||
|
tdSql.query(f"select distinct c1 from t1 where c1 <{tbnum}")
|
||||||
|
tdSql.checkRows(1)
|
||||||
|
tdSql.query(f"select distinct c2 from t1")
|
||||||
|
tdSql.checkRows(4)
|
||||||
|
tdSql.query(f"select distinct c1,c2 from t1 where c1 <{tbnum}")
|
||||||
|
tdSql.checkRows(3)
|
||||||
|
tdSql.query(f"select distinct c1,c1 from t1 ")
|
||||||
|
tdSql.checkRows(2)
|
||||||
|
tdSql.query(f"select distinct c1,c1 from t1 where c1 <{tbnum}")
|
||||||
|
tdSql.checkRows(1)
|
||||||
|
tdSql.query(f"select distinct c1,c2 from t1 where c1 <{tbnum} limit 3")
|
||||||
|
tdSql.checkRows(3)
|
||||||
|
tdSql.query(f"select distinct c1,c2 from t1 where c1 <{tbnum} limit 3 offset 2")
|
||||||
|
tdSql.checkRows(1)
|
||||||
|
|
||||||
|
# tdSql.query(f"select distinct c3 from stb2 where c2 <{tbnum} ")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.query(f"select distinct c3, c2 from stb2 where c2 <{tbnum} limit 2")
|
||||||
|
# tdSql.checkRows(2)
|
||||||
|
|
||||||
|
# tdSql.error("select distinct c5 from stb1")
|
||||||
|
tdSql.error("select distinct c5 from t1")
|
||||||
|
tdSql.error("select distinct c1 from db.*")
|
||||||
|
tdSql.error("select c2, distinct c1 from stb1")
|
||||||
|
tdSql.error("select c2, distinct c1 from t1")
|
||||||
|
tdSql.error("select distinct c2 from ")
|
||||||
|
tdSql.error("distinct c2 from stb1")
|
||||||
|
tdSql.error("distinct c2 from t1")
|
||||||
|
tdSql.error("select distinct c1, c2, c3 from stb1")
|
||||||
|
tdSql.error("select distinct c1, c2, c3 from t1")
|
||||||
|
tdSql.error("select distinct stb1.c1, stb1.c2, stb2.c2, stb2.c3 from stb1")
|
||||||
|
tdSql.error("select distinct stb1.c1, stb1.c2, stb2.c2, stb2.c3 from t1")
|
||||||
|
tdSql.error("select distinct t1.c1, t1.c2, t2.c1, t2.c2 from t1")
|
||||||
|
# tdSql.query(f"select distinct c1 c2, c2 c3 from stb1 where c1 <{tbnum}")
|
||||||
|
# tdSql.checkRows(tbnum*3)
|
||||||
|
tdSql.query(f"select distinct c1 c2, c2 c3 from t1 where c1 <{tbnum}")
|
||||||
|
tdSql.checkRows(3)
|
||||||
|
# tdSql.error("select distinct c1, c2 from stb1 order by ts")
|
||||||
|
tdSql.error("select distinct c1, c2 from t1 order by ts")
|
||||||
|
# tdSql.error("select distinct c1, ts from stb1 group by c2")
|
||||||
|
tdSql.error("select distinct c1, ts from t1 group by c2")
|
||||||
|
# tdSql.error("select distinct c1, max(c2) from stb1 ")
|
||||||
|
tdSql.error("select distinct c1, max(c2) from t1 ")
|
||||||
|
# tdSql.error("select max(c2), distinct c1 from stb1 ")
|
||||||
|
tdSql.error("select max(c2), distinct c1 from t1 ")
|
||||||
|
# tdSql.error("select distinct c1, c2 from stb1 where c1 > 3 group by t0")
|
||||||
|
tdSql.error("select distinct c1, c2 from t1 where c1 > 3 group by t0")
|
||||||
|
# tdSql.error("select distinct c1, c2 from stb1 where c1 > 3 interval(1d) ")
|
||||||
|
tdSql.error("select distinct c1, c2 from t1 where c1 > 3 interval(1d) ")
|
||||||
|
# tdSql.error("select distinct c1, c2 from stb1 where c1 > 3 interval(1d) fill(next)")
|
||||||
|
tdSql.error("select distinct c1, c2 from t1 where c1 > 3 interval(1d) fill(next)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from stb1 where ts > now-10d and ts < now interval(1d) fill(next)")
|
||||||
|
tdSql.error("select distinct c1, c2 from t1 where ts > now-10d and ts < now interval(1d) fill(next)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from stb1 where c1 > 3 slimit 1")
|
||||||
|
# tdSql.error("select distinct c1, c2 from t1 where c1 > 3 slimit 1")
|
||||||
|
# tdSql.query(f"select distinct c1, c2 from stb1 where c1 between {tbnum-2} and {tbnum} ")
|
||||||
|
# tdSql.checkRows(6)
|
||||||
|
tdSql.query(f"select distinct c1, c2 from t1 where c1 between {tbnum-2} and {tbnum} ")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.query("select distinct c1, c2 from stb1 where c1 in (1,2,3,4,5)")
|
||||||
|
# tdSql.checkRows(15)
|
||||||
|
tdSql.query("select distinct c1, c2 from t1 where c1 in (1,2,3,4,5)")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.query("select distinct c1, c2 from stb1 where c1 in (100,1000,10000)")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
tdSql.query("select distinct c1, c2 from t1 where c1 in (100,1000,10000)")
|
||||||
|
# tdSql.checkRows(0)
|
||||||
|
|
||||||
|
# tdSql.query(f"select distinct c1,c2 from (select * from stb1 where c1 > {tbnum-2}) ")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.query(f"select distinct c1,c2 from (select * from t1 where c1 < {tbnum}) ")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.query(f"select distinct c1,c2 from (select * from stb1 where t2 !=0 and t2 != 1) ")
|
||||||
|
# tdSql.checkRows(0)
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select distinct c1, c2 from stb1 where t0 > 2 and t1 < 3) ")
|
||||||
|
# tdSql.error("select c1, c2 from (select distinct c1, c2 from stb1 where t0 > 2 and t1 < 3) ")
|
||||||
|
# tdSql.query("select distinct c1, c2 from (select c2, c1 from stb1 where c1 > 2 ) where c1 < 4")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select c1 from stb1 where t0 > 2 ) where t1 < 3")
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select c2, c1 from stb1 where c1 > 2 order by ts)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select c2, c1 from t1 where c1 > 2 order by ts)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select c2, c1 from stb1 where c1 > 2 group by c1)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select max(c1) c1, max(c2) c2 from stb1 group by c1)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select max(c1) c1, max(c2) c2 from t1 group by c1)")
|
||||||
|
# tdSql.query("select distinct c1, c2 from (select max(c1) c1, max(c2) c2 from stb1 )")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.query("select distinct c1, c2 from (select max(c1) c1, max(c2) c2 from t1 )")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.error("select distinct stb1.c1, stb1.c2 from stb1 , stb2 where stb1.ts=stb2.ts and stb1.t2=stb2.t4")
|
||||||
|
# tdSql.error("select distinct t1.c1, t1.c2 from t1 , t2 where t1.ts=t2.ts ")
|
||||||
|
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select count(c1) c1, count(c2) c2 from stb1 group by ts)")
|
||||||
|
# tdSql.error("select distinct c1, c2 from (select count(c1) c1, count(c2) c2 from t1 group by ts)")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
# #========== suport distinct multi-tags-coloumn ==========
|
||||||
|
# tdSql.query("select distinct t1 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t1, t0 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t1, t2 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum*2+1)
|
||||||
|
# tdSql.query("select distinct t0, t1, t2 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum*2+1)
|
||||||
|
# tdSql.query("select distinct t0 t1, t1 t2 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t0, t0, t0 from stb1")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t0, t1 from t1")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.query("select distinct t0, t1 from t100num")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
|
||||||
|
# tdSql.query("select distinct t3 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t2, t3 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t3, t2 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t4, t2 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum*3+1)
|
||||||
|
# tdSql.query("select distinct t2, t3, t4 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum*3+1)
|
||||||
|
# tdSql.query("select distinct t2 t1, t3 t2 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t3, t3, t3 from stb2")
|
||||||
|
# tdSql.checkRows(maxRemainderNum+1)
|
||||||
|
# tdSql.query("select distinct t2, t3 from t01")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.query("select distinct t3, t4 from t0100num")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
|
||||||
|
|
||||||
|
# ########## should be error #########
|
||||||
|
# tdSql.error("select distinct from stb1")
|
||||||
|
# tdSql.error("select distinct t3 from stb1")
|
||||||
|
# tdSql.error("select distinct t1 from db.*")
|
||||||
|
# tdSql.error("select distinct t2 from ")
|
||||||
|
# tdSql.error("distinct t2 from stb1")
|
||||||
|
# tdSql.error("select distinct stb1")
|
||||||
|
# tdSql.error("select distinct t0, t1, t2, t3 from stb1")
|
||||||
|
# tdSql.error("select distinct stb1.t0, stb1.t1, stb2.t2, stb2.t3 from stb1")
|
||||||
|
|
||||||
|
# tdSql.error("select dist t0 from stb1")
|
||||||
|
# tdSql.error("select distinct stb2.t2, stb2.t3 from stb1")
|
||||||
|
# tdSql.error("select distinct stb2.t2 t1, stb2.t3 t2 from stb1")
|
||||||
|
|
||||||
|
# tdSql.error("select distinct t0, t1 from t1 where t0 < 7")
|
||||||
|
|
||||||
|
# ########## add where condition ##########
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1 where t1 > 3")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1 where t1 > 3 limit 2")
|
||||||
|
# tdSql.checkRows(2)
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1 where t1 > 3 limit 2 offset 2")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1 where t1 > 3 slimit 2")
|
||||||
|
# tdSql.checkRows(3)
|
||||||
|
# tdSql.error("select distinct t0, t1 from stb1 where c1 > 2")
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1 where t1 > 3 and t1 < 5")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.error("select distinct stb1.t0, stb1.t1 from stb1, stb2 where stb1.t2=stb2.t4")
|
||||||
|
# tdSql.error("select distinct t0, t1 from stb1 where stb2.t4 > 2")
|
||||||
|
# tdSql.error("select distinct t0, t1 from stb1 where t1 > 3 group by t0")
|
||||||
|
# tdSql.error("select distinct t0, t1 from stb1 where t1 > 3 interval(1d) ")
|
||||||
|
# tdSql.error("select distinct t0, t1 from stb1 where t1 > 3 interval(1d) fill(next)")
|
||||||
|
# tdSql.error("select distinct t0, t1 from stb1 where ts > now-10d and ts < now interval(1d) fill(next)")
|
||||||
|
|
||||||
|
# tdSql.error("select max(c1), distinct t0 from stb1 where t0 > 2")
|
||||||
|
# tdSql.error("select distinct t0, max(c1) from stb1 where t0 > 2")
|
||||||
|
# tdSql.error("select distinct t0 from stb1 where t0 in (select t0 from stb1 where t0 > 2)")
|
||||||
|
# tdSql.query("select distinct t0, t1 from stb1 where t0 in (1,2,3,4,5)")
|
||||||
|
# tdSql.checkRows(5)
|
||||||
|
# tdSql.query("select distinct t1 from (select t0, t1 from stb1 where t0 > 2) ")
|
||||||
|
# tdSql.checkRows(4)
|
||||||
|
# tdSql.error("select distinct t1 from (select distinct t0, t1 from stb1 where t0 > 2 and t1 < 3) ")
|
||||||
|
# tdSql.error("select distinct t1 from (select distinct t0, t1 from stb1 where t0 > 2 ) where t1 < 3")
|
||||||
|
# tdSql.query("select distinct t1 from (select t0, t1 from stb1 where t0 > 2 ) where t1 < 3")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.error("select distinct t1, t0 from (select t1 from stb1 where t0 > 2 ) where t1 < 3")
|
||||||
|
# tdSql.error("select distinct t1, t0 from (select max(t1) t1, max(t0) t0 from stb1 group by t1)")
|
||||||
|
# tdSql.error("select distinct t1, t0 from (select max(t1) t1, max(t0) t0 from stb1)")
|
||||||
|
# tdSql.query("select distinct t1, t0 from (select t1,t0 from stb1 where t0 > 2 ) where t1 < 3")
|
||||||
|
# tdSql.checkRows(1)
|
||||||
|
# tdSql.error(" select distinct t1, t0 from (select t1,t0 from stb1 where t0 > 2 order by ts) where t1 < 3")
|
||||||
|
# tdSql.error("select t1, t0 from (select distinct t1,t0 from stb1 where t0 > 2 ) where t1 < 3")
|
||||||
|
# tdSql.error(" select distinct t1, t0 from (select t1,t0 from stb1 where t0 > 2 group by ts) where t1 < 3")
|
||||||
|
# tdSql.error("select distinct stb1.t1, stb1.t2 from stb1 , stb2 where stb1.ts=stb2.ts and stb1.t2=stb2.t4")
|
||||||
|
# tdSql.error("select distinct t1.t1, t1.t2 from t1 , t2 where t1.ts=t2.ts ")
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
tdSql.close()
|
||||||
|
tdLog.success(f"{__file__} successfully executed")
|
||||||
|
|
||||||
|
tdCases.addLinux(__file__, TDTestCase())
|
||||||
|
tdCases.addWindows(__file__, TDTestCase())
|
|
@ -1,17 +1,3 @@
|
||||||
python3 ./test.py -f 2-query/between.py
|
python3 ./test.py -f 2-query/between.py
|
||||||
|
python3 ./test.py -f 2-query/distinct.py
|
||||||
|
python3 ./test.py -f 2-query/varchar.py
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
|
@ -340,7 +340,7 @@ tmq_t* build_consumer() {
|
||||||
tmq_conf_set(conf, "td.connect.user", "root");
|
tmq_conf_set(conf, "td.connect.user", "root");
|
||||||
tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
tmq_conf_set(conf, "td.connect.pass", "taosdata");
|
||||||
tmq_conf_set(conf, "td.connect.db", g_stConfInfo.dbName);
|
tmq_conf_set(conf, "td.connect.db", g_stConfInfo.dbName);
|
||||||
tmq_t* tmq = tmq_consumer_new1(conf, NULL, 0);
|
tmq_t* tmq = tmq_consumer_new(conf, NULL, 0);
|
||||||
assert(tmq);
|
assert(tmq);
|
||||||
tmq_conf_destroy(conf);
|
tmq_conf_destroy(conf);
|
||||||
return tmq;
|
return tmq;
|
||||||
|
@ -367,7 +367,7 @@ void sync_consume_loop(tmq_t* tmq, tmq_list_t* topics) {
|
||||||
TAOS_RES* tmqmessage = tmq_consumer_poll(tmq, 1);
|
TAOS_RES* tmqmessage = tmq_consumer_poll(tmq, 1);
|
||||||
if (tmqmessage) {
|
if (tmqmessage) {
|
||||||
/*msg_process(tmqmessage);*/
|
/*msg_process(tmqmessage);*/
|
||||||
tmq_message_destroy(tmqmessage);
|
taos_free_result(tmqmessage);
|
||||||
|
|
||||||
if ((++msg_count % MIN_COMMIT_COUNT) == 0) tmq_commit(tmq, NULL, 0);
|
if ((++msg_count % MIN_COMMIT_COUNT) == 0) tmq_commit(tmq, NULL, 0);
|
||||||
}
|
}
|
||||||
|
@ -400,7 +400,7 @@ void perf_loop(tmq_t* tmq, tmq_list_t* topics, int32_t totalMsgs, int64_t walLog
|
||||||
if (0 != g_stConfInfo.showMsgFlag) {
|
if (0 != g_stConfInfo.showMsgFlag) {
|
||||||
/*msg_process(tmqmessage);*/
|
/*msg_process(tmqmessage);*/
|
||||||
}
|
}
|
||||||
tmq_message_destroy(tmqmessage);
|
taos_free_result(tmqmessage);
|
||||||
} else {
|
} else {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
|
@ -31,46 +31,46 @@
|
||||||
#define NC "\033[0m"
|
#define NC "\033[0m"
|
||||||
#define min(a, b) (((a) < (b)) ? (a) : (b))
|
#define min(a, b) (((a) < (b)) ? (a) : (b))
|
||||||
|
|
||||||
#define MAX_SQL_STR_LEN (1024 * 1024)
|
#define MAX_SQL_STR_LEN (1024 * 1024)
|
||||||
#define MAX_ROW_STR_LEN (16 * 1024)
|
#define MAX_ROW_STR_LEN (16 * 1024)
|
||||||
#define MAX_CONSUMER_THREAD_CNT (16)
|
#define MAX_CONSUMER_THREAD_CNT (16)
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
TdThread thread;
|
TdThread thread;
|
||||||
int32_t consumerId;
|
int32_t consumerId;
|
||||||
|
|
||||||
int32_t ifCheckData;
|
int32_t ifCheckData;
|
||||||
int64_t expectMsgCnt;
|
int64_t expectMsgCnt;
|
||||||
|
|
||||||
int64_t consumeMsgCnt;
|
|
||||||
int32_t checkresult;
|
|
||||||
|
|
||||||
char topicString[1024];
|
int64_t consumeMsgCnt;
|
||||||
char keyString[1024];
|
int32_t checkresult;
|
||||||
|
|
||||||
int32_t numOfTopic;
|
char topicString[1024];
|
||||||
char topics[32][64];
|
char keyString[1024];
|
||||||
|
|
||||||
int32_t numOfKey;
|
int32_t numOfTopic;
|
||||||
char key[32][64];
|
char topics[32][64];
|
||||||
char value[32][64];
|
|
||||||
|
int32_t numOfKey;
|
||||||
|
char key[32][64];
|
||||||
|
char value[32][64];
|
||||||
|
|
||||||
tmq_t* tmq;
|
tmq_t* tmq;
|
||||||
tmq_list_t* topicList;
|
tmq_list_t* topicList;
|
||||||
|
|
||||||
} SThreadInfo;
|
} SThreadInfo;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
// input from argvs
|
// input from argvs
|
||||||
char dbName[32];
|
char dbName[32];
|
||||||
int32_t showMsgFlag;
|
int32_t showMsgFlag;
|
||||||
int32_t consumeDelay; // unit s
|
int32_t consumeDelay; // unit s
|
||||||
int32_t numOfThread;
|
int32_t numOfThread;
|
||||||
SThreadInfo stThreads[MAX_CONSUMER_THREAD_CNT];
|
SThreadInfo stThreads[MAX_CONSUMER_THREAD_CNT];
|
||||||
} SConfInfo;
|
} SConfInfo;
|
||||||
|
|
||||||
static SConfInfo g_stConfInfo;
|
static SConfInfo g_stConfInfo;
|
||||||
TdFilePtr g_fp = NULL;
|
TdFilePtr g_fp = NULL;
|
||||||
|
|
||||||
// char* g_pRowValue = NULL;
|
// char* g_pRowValue = NULL;
|
||||||
// TdFilePtr g_fp = NULL;
|
// TdFilePtr g_fp = NULL;
|
||||||
|
@ -95,7 +95,7 @@ void initLogFile() {
|
||||||
TdFilePtr pFile = taosOpenFile("./tmqlog.txt", TD_FILE_CREATE | TD_FILE_WRITE | TD_FILE_APPEND | TD_FILE_STREAM);
|
TdFilePtr pFile = taosOpenFile("./tmqlog.txt", TD_FILE_CREATE | TD_FILE_WRITE | TD_FILE_APPEND | TD_FILE_STREAM);
|
||||||
if (NULL == pFile) {
|
if (NULL == pFile) {
|
||||||
fprintf(stderr, "Failed to open %s for save result\n", "./tmqlog.txt");
|
fprintf(stderr, "Failed to open %s for save result\n", "./tmqlog.txt");
|
||||||
exit -1;
|
exit - 1;
|
||||||
};
|
};
|
||||||
g_fp = pFile;
|
g_fp = pFile;
|
||||||
|
|
||||||
|
@ -103,27 +103,27 @@ void initLogFile() {
|
||||||
struct tm tm = *taosLocalTime(&tTime, NULL);
|
struct tm tm = *taosLocalTime(&tTime, NULL);
|
||||||
|
|
||||||
taosFprintfFile(pFile, "###################################################################\n");
|
taosFprintfFile(pFile, "###################################################################\n");
|
||||||
taosFprintfFile(pFile, "# configDir: %s\n", configDir);
|
taosFprintfFile(pFile, "# configDir: %s\n", configDir);
|
||||||
taosFprintfFile(pFile, "# dbName: %s\n", g_stConfInfo.dbName);
|
taosFprintfFile(pFile, "# dbName: %s\n", g_stConfInfo.dbName);
|
||||||
taosFprintfFile(pFile, "# showMsgFlag: %d\n", g_stConfInfo.showMsgFlag);
|
taosFprintfFile(pFile, "# showMsgFlag: %d\n", g_stConfInfo.showMsgFlag);
|
||||||
taosFprintfFile(pFile, "# consumeDelay: %d\n", g_stConfInfo.consumeDelay);
|
taosFprintfFile(pFile, "# consumeDelay: %d\n", g_stConfInfo.consumeDelay);
|
||||||
|
|
||||||
for (int32_t i = 0; i < g_stConfInfo.numOfThread; i++) {
|
for (int32_t i = 0; i < g_stConfInfo.numOfThread; i++) {
|
||||||
taosFprintfFile(pFile, "# consumer %d info:\n", g_stConfInfo.stThreads[i].consumerId);
|
taosFprintfFile(pFile, "# consumer %d info:\n", g_stConfInfo.stThreads[i].consumerId);
|
||||||
taosFprintfFile(pFile, " Topics: ");
|
taosFprintfFile(pFile, " Topics: ");
|
||||||
for (int i = 0 ; i < g_stConfInfo.stThreads[i].numOfTopic; i++) {
|
for (int i = 0; i < g_stConfInfo.stThreads[i].numOfTopic; i++) {
|
||||||
taosFprintfFile(pFile, "%s, ", g_stConfInfo.stThreads[i].topics[i]);
|
taosFprintfFile(pFile, "%s, ", g_stConfInfo.stThreads[i].topics[i]);
|
||||||
}
|
}
|
||||||
taosFprintfFile(pFile, "\n");
|
taosFprintfFile(pFile, "\n");
|
||||||
taosFprintfFile(pFile, " Key: ");
|
taosFprintfFile(pFile, " Key: ");
|
||||||
for (int i = 0 ; i < g_stConfInfo.stThreads[i].numOfKey; i++) {
|
for (int i = 0; i < g_stConfInfo.stThreads[i].numOfKey; i++) {
|
||||||
taosFprintfFile(pFile, "%s:%s, ", g_stConfInfo.stThreads[i].key[i], g_stConfInfo.stThreads[i].value[i]);
|
taosFprintfFile(pFile, "%s:%s, ", g_stConfInfo.stThreads[i].key[i], g_stConfInfo.stThreads[i].value[i]);
|
||||||
}
|
}
|
||||||
taosFprintfFile(pFile, "\n");
|
taosFprintfFile(pFile, "\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
taosFprintfFile(pFile, "# Test time: %d-%02d-%02d %02d:%02d:%02d\n", tm.tm_year + 1900, tm.tm_mon + 1,
|
taosFprintfFile(pFile, "# Test time: %d-%02d-%02d %02d:%02d:%02d\n", tm.tm_year + 1900, tm.tm_mon + 1,
|
||||||
tm.tm_mday, tm.tm_hour, tm.tm_min, tm.tm_sec);
|
tm.tm_mday, tm.tm_hour, tm.tm_min, tm.tm_sec);
|
||||||
taosFprintfFile(pFile, "###################################################################\n");
|
taosFprintfFile(pFile, "###################################################################\n");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -180,23 +180,23 @@ void ltrim(char* str) {
|
||||||
// return str;
|
// return str;
|
||||||
}
|
}
|
||||||
|
|
||||||
static int running = 1;
|
static int running = 1;
|
||||||
static void msg_process(TAOS_RES* msg, int32_t msgIndex, int32_t threadLable) {
|
static void msg_process(TAOS_RES* msg, int32_t msgIndex, int32_t threadLable) {
|
||||||
char buf[1024];
|
char buf[1024];
|
||||||
|
|
||||||
//printf("topic: %s\n", tmq_get_topic_name(msg));
|
// printf("topic: %s\n", tmq_get_topic_name(msg));
|
||||||
//printf("vg:%d\n", tmq_get_vgroup_id(msg));
|
// printf("vg:%d\n", tmq_get_vgroup_id(msg));
|
||||||
taosFprintfFile(g_fp, "msg index:%d, threadLable: %d\n", msgIndex, threadLable);
|
taosFprintfFile(g_fp, "msg index:%d, threadLable: %d\n", msgIndex, threadLable);
|
||||||
taosFprintfFile(g_fp, "topic: %s, vgroupId: %d\n", tmq_get_topic_name(msg), tmq_get_vgroup_id(msg));
|
taosFprintfFile(g_fp, "topic: %s, vgroupId: %d\n", tmq_get_topic_name(msg), tmq_get_vgroup_id(msg));
|
||||||
|
|
||||||
while (1) {
|
while (1) {
|
||||||
TAOS_ROW row = taos_fetch_row(msg);
|
TAOS_ROW row = taos_fetch_row(msg);
|
||||||
if (row == NULL) break;
|
if (row == NULL) break;
|
||||||
TAOS_FIELD* fields = taos_fetch_fields(msg);
|
TAOS_FIELD* fields = taos_fetch_fields(msg);
|
||||||
int32_t numOfFields = taos_field_count(msg);
|
int32_t numOfFields = taos_field_count(msg);
|
||||||
//taos_print_row(buf, row, fields, numOfFields);
|
// taos_print_row(buf, row, fields, numOfFields);
|
||||||
//printf("%s\n", buf);
|
// printf("%s\n", buf);
|
||||||
//taosFprintfFile(g_fp, "%s\n", buf);
|
// taosFprintfFile(g_fp, "%s\n", buf);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -213,7 +213,7 @@ int queryDB(TAOS* taos, char* command) {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void build_consumer(SThreadInfo *pInfo) {
|
void build_consumer(SThreadInfo* pInfo) {
|
||||||
char sqlStr[1024] = {0};
|
char sqlStr[1024] = {0};
|
||||||
|
|
||||||
TAOS* pConn = taos_connect(NULL, "root", "taosdata", NULL, 0);
|
TAOS* pConn = taos_connect(NULL, "root", "taosdata", NULL, 0);
|
||||||
|
@ -233,11 +233,11 @@ void build_consumer(SThreadInfo *pInfo) {
|
||||||
for (int32_t i = 0; i < pInfo->numOfKey; i++) {
|
for (int32_t i = 0; i < pInfo->numOfKey; i++) {
|
||||||
tmq_conf_set(conf, pInfo->key[i], pInfo->value[i]);
|
tmq_conf_set(conf, pInfo->key[i], pInfo->value[i]);
|
||||||
}
|
}
|
||||||
pInfo->tmq = tmq_consumer_new(pConn, conf, NULL, 0);
|
pInfo->tmq = tmq_consumer_new(conf, NULL, 0);
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
void build_topic_list(SThreadInfo *pInfo) {
|
void build_topic_list(SThreadInfo* pInfo) {
|
||||||
pInfo->topicList = tmq_list_new();
|
pInfo->topicList = tmq_list_new();
|
||||||
// tmq_list_append(topic_list, "test_stb_topic_1");
|
// tmq_list_append(topic_list, "test_stb_topic_1");
|
||||||
for (int32_t i = 0; i < pInfo->numOfTopic; i++) {
|
for (int32_t i = 0; i < pInfo->numOfTopic; i++) {
|
||||||
|
@ -246,48 +246,45 @@ void build_topic_list(SThreadInfo *pInfo) {
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t saveConsumeResult(SThreadInfo *pInfo) {
|
int32_t saveConsumeResult(SThreadInfo* pInfo) {
|
||||||
char sqlStr[1024] = {0};
|
char sqlStr[1024] = {0};
|
||||||
|
|
||||||
TAOS* pConn = taos_connect(NULL, "root", "taosdata", NULL, 0);
|
TAOS* pConn = taos_connect(NULL, "root", "taosdata", NULL, 0);
|
||||||
assert(pConn != NULL);
|
assert(pConn != NULL);
|
||||||
|
|
||||||
// schema: ts timestamp, consumerid int, consummsgcnt bigint, checkresult int
|
// schema: ts timestamp, consumerid int, consummsgcnt bigint, checkresult int
|
||||||
sprintf(sqlStr, "insert into %s.consumeresult values (now, %d, %" PRId64 ", %d)",
|
sprintf(sqlStr, "insert into %s.consumeresult values (now, %d, %" PRId64 ", %d)", g_stConfInfo.dbName,
|
||||||
g_stConfInfo.dbName,
|
pInfo->consumerId, pInfo->consumeMsgCnt, pInfo->checkresult);
|
||||||
pInfo->consumerId,
|
|
||||||
pInfo->consumeMsgCnt,
|
|
||||||
pInfo->checkresult);
|
|
||||||
|
|
||||||
TAOS_RES* pRes = taos_query(pConn, sqlStr);
|
TAOS_RES* pRes = taos_query(pConn, sqlStr);
|
||||||
if (taos_errno(pRes) != 0) {
|
if (taos_errno(pRes) != 0) {
|
||||||
printf("error in save consumeinfo, reason:%s\n", taos_errstr(pRes));
|
printf("error in save consumeinfo, reason:%s\n", taos_errstr(pRes));
|
||||||
taos_free_result(pRes);
|
taos_free_result(pRes);
|
||||||
exit(-1);
|
exit(-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
taos_free_result(pRes);
|
taos_free_result(pRes);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
void loop_consume(SThreadInfo *pInfo) {
|
void loop_consume(SThreadInfo* pInfo) {
|
||||||
tmq_resp_err_t err;
|
tmq_resp_err_t err;
|
||||||
|
|
||||||
int64_t totalMsgs = 0;
|
int64_t totalMsgs = 0;
|
||||||
//int64_t totalRows = 0;
|
// int64_t totalRows = 0;
|
||||||
|
|
||||||
while (running) {
|
while (running) {
|
||||||
TAOS_RES* tmqMsg = tmq_consumer_poll(pInfo->tmq, g_stConfInfo.consumeDelay * 1000);
|
TAOS_RES* tmqMsg = tmq_consumer_poll(pInfo->tmq, g_stConfInfo.consumeDelay * 1000);
|
||||||
if (tmqMsg) {
|
if (tmqMsg) {
|
||||||
if (0 != g_stConfInfo.showMsgFlag) {
|
if (0 != g_stConfInfo.showMsgFlag) {
|
||||||
msg_process(tmqMsg, totalMsgs, 0);
|
msg_process(tmqMsg, totalMsgs, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
tmq_message_destroy(tmqMsg);
|
taos_free_result(tmqMsg);
|
||||||
|
|
||||||
totalMsgs++;
|
totalMsgs++;
|
||||||
|
|
||||||
if (totalMsgs >= pInfo->expectMsgCnt) {
|
if (totalMsgs >= pInfo->expectMsgCnt) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
@ -295,7 +292,7 @@ void loop_consume(SThreadInfo *pInfo) {
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
err = tmq_consumer_close(pInfo->tmq);
|
err = tmq_consumer_close(pInfo->tmq);
|
||||||
if (err) {
|
if (err) {
|
||||||
printf("tmq_consumer_close() fail, reason: %s\n", tmq_err2str(err));
|
printf("tmq_consumer_close() fail, reason: %s\n", tmq_err2str(err));
|
||||||
|
@ -303,35 +300,34 @@ void loop_consume(SThreadInfo *pInfo) {
|
||||||
}
|
}
|
||||||
|
|
||||||
pInfo->consumeMsgCnt = totalMsgs;
|
pInfo->consumeMsgCnt = totalMsgs;
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void *consumeThreadFunc(void *param) {
|
void* consumeThreadFunc(void* param) {
|
||||||
int32_t totalMsgs = 0;
|
int32_t totalMsgs = 0;
|
||||||
|
|
||||||
SThreadInfo *pInfo = (SThreadInfo *)param;
|
SThreadInfo* pInfo = (SThreadInfo*)param;
|
||||||
|
|
||||||
build_consumer(pInfo);
|
build_consumer(pInfo);
|
||||||
build_topic_list(pInfo);
|
build_topic_list(pInfo);
|
||||||
if ((NULL == pInfo->tmq) || (NULL == pInfo->topicList)){
|
if ((NULL == pInfo->tmq) || (NULL == pInfo->topicList)) {
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
tmq_resp_err_t err = tmq_subscribe(pInfo->tmq, pInfo->topicList);
|
tmq_resp_err_t err = tmq_subscribe(pInfo->tmq, pInfo->topicList);
|
||||||
if (err) {
|
if (err) {
|
||||||
printf("tmq_subscribe() fail, reason: %s\n", tmq_err2str(err));
|
printf("tmq_subscribe() fail, reason: %s\n", tmq_err2str(err));
|
||||||
exit(-1);
|
exit(-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
loop_consume(pInfo);
|
loop_consume(pInfo);
|
||||||
|
|
||||||
err = tmq_unsubscribe(pInfo->tmq);
|
err = tmq_unsubscribe(pInfo->tmq);
|
||||||
if (err) {
|
if (err) {
|
||||||
printf("tmq_unsubscribe() fail, reason: %s\n", tmq_err2str(err));
|
printf("tmq_unsubscribe() fail, reason: %s\n", tmq_err2str(err));
|
||||||
pInfo->consumeMsgCnt = -1;
|
pInfo->consumeMsgCnt = -1;
|
||||||
return NULL;
|
return NULL;
|
||||||
}
|
}
|
||||||
|
|
||||||
// save consume result into consumeresult table
|
// save consume result into consumeresult table
|
||||||
saveConsumeResult(pInfo);
|
saveConsumeResult(pInfo);
|
||||||
|
|
||||||
|
@ -343,7 +339,7 @@ void parseConsumeInfo() {
|
||||||
const char delim[2] = ",";
|
const char delim[2] = ",";
|
||||||
const char ch = ':';
|
const char ch = ':';
|
||||||
|
|
||||||
for (int32_t i = 0; i < g_stConfInfo.numOfThread; i++) {
|
for (int32_t i = 0; i < g_stConfInfo.numOfThread; i++) {
|
||||||
token = strtok(g_stConfInfo.stThreads[i].topicString, delim);
|
token = strtok(g_stConfInfo.stThreads[i].topicString, delim);
|
||||||
while (token != NULL) {
|
while (token != NULL) {
|
||||||
// printf("%s\n", token );
|
// printf("%s\n", token );
|
||||||
|
@ -351,10 +347,10 @@ void parseConsumeInfo() {
|
||||||
ltrim(g_stConfInfo.stThreads[i].topics[g_stConfInfo.stThreads[i].numOfTopic]);
|
ltrim(g_stConfInfo.stThreads[i].topics[g_stConfInfo.stThreads[i].numOfTopic]);
|
||||||
// printf("%s\n", g_stConfInfo.topics[g_stConfInfo.numOfTopic]);
|
// printf("%s\n", g_stConfInfo.topics[g_stConfInfo.numOfTopic]);
|
||||||
g_stConfInfo.stThreads[i].numOfTopic++;
|
g_stConfInfo.stThreads[i].numOfTopic++;
|
||||||
|
|
||||||
token = strtok(NULL, delim);
|
token = strtok(NULL, delim);
|
||||||
}
|
}
|
||||||
|
|
||||||
token = strtok(g_stConfInfo.stThreads[i].keyString, delim);
|
token = strtok(g_stConfInfo.stThreads[i].keyString, delim);
|
||||||
while (token != NULL) {
|
while (token != NULL) {
|
||||||
// printf("%s\n", token );
|
// printf("%s\n", token );
|
||||||
|
@ -368,7 +364,7 @@ void parseConsumeInfo() {
|
||||||
// g_stConfInfo.value[g_stConfInfo.numOfKey]);
|
// g_stConfInfo.value[g_stConfInfo.numOfKey]);
|
||||||
g_stConfInfo.stThreads[i].numOfKey++;
|
g_stConfInfo.stThreads[i].numOfKey++;
|
||||||
}
|
}
|
||||||
|
|
||||||
token = strtok(NULL, delim);
|
token = strtok(NULL, delim);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -376,46 +372,47 @@ void parseConsumeInfo() {
|
||||||
|
|
||||||
int32_t getConsumeInfo() {
|
int32_t getConsumeInfo() {
|
||||||
char sqlStr[1024] = {0};
|
char sqlStr[1024] = {0};
|
||||||
|
|
||||||
TAOS* pConn = taos_connect(NULL, "root", "taosdata", NULL, 0);
|
TAOS* pConn = taos_connect(NULL, "root", "taosdata", NULL, 0);
|
||||||
assert(pConn != NULL);
|
assert(pConn != NULL);
|
||||||
|
|
||||||
sprintf(sqlStr, "select * from %s.consumeinfo", g_stConfInfo.dbName);
|
sprintf(sqlStr, "select * from %s.consumeinfo", g_stConfInfo.dbName);
|
||||||
TAOS_RES* pRes = taos_query(pConn, sqlStr);
|
TAOS_RES* pRes = taos_query(pConn, sqlStr);
|
||||||
if (taos_errno(pRes) != 0) {
|
if (taos_errno(pRes) != 0) {
|
||||||
printf("error in get consumeinfo, reason:%s\n", taos_errstr(pRes));
|
printf("error in get consumeinfo, reason:%s\n", taos_errstr(pRes));
|
||||||
taos_free_result(pRes);
|
taos_free_result(pRes);
|
||||||
exit(-1);
|
exit(-1);
|
||||||
}
|
}
|
||||||
|
|
||||||
TAOS_ROW row = NULL;
|
TAOS_ROW row = NULL;
|
||||||
int num_fields = taos_num_fields(pRes);
|
int num_fields = taos_num_fields(pRes);
|
||||||
TAOS_FIELD* fields = taos_fetch_fields(pRes);
|
TAOS_FIELD* fields = taos_fetch_fields(pRes);
|
||||||
|
|
||||||
// schema: ts timestamp, consumerid int, topiclist binary(1024), keylist binary(1024), expectmsgcnt bigint, ifcheckdata int
|
// schema: ts timestamp, consumerid int, topiclist binary(1024), keylist binary(1024), expectmsgcnt bigint,
|
||||||
|
// ifcheckdata int
|
||||||
|
|
||||||
int32_t numOfThread = 0;
|
int32_t numOfThread = 0;
|
||||||
while ((row = taos_fetch_row(pRes))) {
|
while ((row = taos_fetch_row(pRes))) {
|
||||||
int32_t* lengths = taos_fetch_lengths(pRes);
|
int32_t* lengths = taos_fetch_lengths(pRes);
|
||||||
|
|
||||||
for (int i = 0; i < num_fields; ++i) {
|
for (int i = 0; i < num_fields; ++i) {
|
||||||
if (row[i] == NULL || 0 == i) {
|
if (row[i] == NULL || 0 == i) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
if ((1 == i) && (fields[i].type == TSDB_DATA_TYPE_INT)) {
|
if ((1 == i) && (fields[i].type == TSDB_DATA_TYPE_INT)) {
|
||||||
g_stConfInfo.stThreads[numOfThread].consumerId = *((int32_t *)row[i]);
|
g_stConfInfo.stThreads[numOfThread].consumerId = *((int32_t*)row[i]);
|
||||||
} else if ((2 == i) && (fields[i].type == TSDB_DATA_TYPE_BINARY)) {
|
} else if ((2 == i) && (fields[i].type == TSDB_DATA_TYPE_BINARY)) {
|
||||||
memcpy(g_stConfInfo.stThreads[numOfThread].topicString, row[i], lengths[i]);
|
memcpy(g_stConfInfo.stThreads[numOfThread].topicString, row[i], lengths[i]);
|
||||||
} else if ((3 == i) && (fields[i].type == TSDB_DATA_TYPE_BINARY)) {
|
} else if ((3 == i) && (fields[i].type == TSDB_DATA_TYPE_BINARY)) {
|
||||||
memcpy(g_stConfInfo.stThreads[numOfThread].keyString, row[i], lengths[i]);
|
memcpy(g_stConfInfo.stThreads[numOfThread].keyString, row[i], lengths[i]);
|
||||||
} else if ((4 == i) && (fields[i].type == TSDB_DATA_TYPE_BIGINT)) {
|
} else if ((4 == i) && (fields[i].type == TSDB_DATA_TYPE_BIGINT)) {
|
||||||
g_stConfInfo.stThreads[numOfThread].expectMsgCnt = *((int64_t *)row[i]);
|
g_stConfInfo.stThreads[numOfThread].expectMsgCnt = *((int64_t*)row[i]);
|
||||||
} else if ((5 == i) && (fields[i].type == TSDB_DATA_TYPE_INT)) {
|
} else if ((5 == i) && (fields[i].type == TSDB_DATA_TYPE_INT)) {
|
||||||
g_stConfInfo.stThreads[numOfThread].ifCheckData = *((int32_t *)row[i]);
|
g_stConfInfo.stThreads[numOfThread].ifCheckData = *((int32_t*)row[i]);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
numOfThread ++;
|
numOfThread++;
|
||||||
}
|
}
|
||||||
g_stConfInfo.numOfThread = numOfThread;
|
g_stConfInfo.numOfThread = numOfThread;
|
||||||
|
|
||||||
|
@ -426,7 +423,6 @@ int32_t getConsumeInfo() {
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
int main(int32_t argc, char* argv[]) {
|
int main(int32_t argc, char* argv[]) {
|
||||||
parseArgument(argc, argv);
|
parseArgument(argc, argv);
|
||||||
getConsumeInfo();
|
getConsumeInfo();
|
||||||
|
@ -438,18 +434,19 @@ int main(int32_t argc, char* argv[]) {
|
||||||
|
|
||||||
// pthread_create one thread to consume
|
// pthread_create one thread to consume
|
||||||
for (int32_t i = 0; i < g_stConfInfo.numOfThread; ++i) {
|
for (int32_t i = 0; i < g_stConfInfo.numOfThread; ++i) {
|
||||||
taosThreadCreate(&(g_stConfInfo.stThreads[i].thread), &thattr, consumeThreadFunc, (void *)(&(g_stConfInfo.stThreads[i])));
|
taosThreadCreate(&(g_stConfInfo.stThreads[i].thread), &thattr, consumeThreadFunc,
|
||||||
|
(void*)(&(g_stConfInfo.stThreads[i])));
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int32_t i = 0; i < g_stConfInfo.numOfThread; i++) {
|
for (int32_t i = 0; i < g_stConfInfo.numOfThread; i++) {
|
||||||
taosThreadJoin(g_stConfInfo.stThreads[i].thread, NULL);
|
taosThreadJoin(g_stConfInfo.stThreads[i].thread, NULL);
|
||||||
}
|
}
|
||||||
|
|
||||||
//printf("consumer: %d, cosumer1: %d\n", totalMsgs, pInfo->consumeMsgCnt);
|
// printf("consumer: %d, cosumer1: %d\n", totalMsgs, pInfo->consumeMsgCnt);
|
||||||
|
|
||||||
taosFprintfFile(g_fp, "\n");
|
taosFprintfFile(g_fp, "\n");
|
||||||
taosCloseFile(&g_fp);
|
taosCloseFile(&g_fp);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue