Merge remote-tracking branch 'origin/develop' into feature/wal
This commit is contained in:
commit
98d254d521
|
@ -4,6 +4,9 @@
|
||||||
[submodule "src/connector/grafanaplugin"]
|
[submodule "src/connector/grafanaplugin"]
|
||||||
path = src/connector/grafanaplugin
|
path = src/connector/grafanaplugin
|
||||||
url = https://github.com/taosdata/grafanaplugin
|
url = https://github.com/taosdata/grafanaplugin
|
||||||
|
[submodule "tests/examples/rust"]
|
||||||
|
path = tests/examples/rust
|
||||||
|
url = https://github.com/songtianyi/tdengine-rust-bindings.git
|
||||||
[submodule "src/connector/hivemq-tdengine-extension"]
|
[submodule "src/connector/hivemq-tdengine-extension"]
|
||||||
path = src/connector/hivemq-tdengine-extension
|
path = src/connector/hivemq-tdengine-extension
|
||||||
url = https://github.com/huskar-t/hivemq-tdengine-extension.git
|
url = https://github.com/huskar-t/hivemq-tdengine-extension.git
|
|
@ -62,6 +62,7 @@ typedef struct SLocalReducer {
|
||||||
bool hasUnprocessedRow;
|
bool hasUnprocessedRow;
|
||||||
tOrderDescriptor * pDesc;
|
tOrderDescriptor * pDesc;
|
||||||
SColumnModel * resColModel;
|
SColumnModel * resColModel;
|
||||||
|
SColumnModel* finalModel;
|
||||||
tExtMemBuffer ** pExtMemBuffer; // disk-based buffer
|
tExtMemBuffer ** pExtMemBuffer; // disk-based buffer
|
||||||
SFillInfo* pFillInfo; // interpolation support structure
|
SFillInfo* pFillInfo; // interpolation support structure
|
||||||
char* pFinalRes; // result data after interpo
|
char* pFinalRes; // result data after interpo
|
||||||
|
@ -74,7 +75,8 @@ typedef struct SLocalReducer {
|
||||||
typedef struct SRetrieveSupport {
|
typedef struct SRetrieveSupport {
|
||||||
tExtMemBuffer ** pExtMemBuffer; // for build loser tree
|
tExtMemBuffer ** pExtMemBuffer; // for build loser tree
|
||||||
tOrderDescriptor *pOrderDescriptor;
|
tOrderDescriptor *pOrderDescriptor;
|
||||||
SColumnModel * pFinalColModel; // colModel for final result
|
SColumnModel* pFinalColModel; // colModel for final result
|
||||||
|
SColumnModel* pFFColModel;
|
||||||
int32_t subqueryIndex; // index of current vnode in vnode list
|
int32_t subqueryIndex; // index of current vnode in vnode list
|
||||||
SSqlObj * pParentSql;
|
SSqlObj * pParentSql;
|
||||||
tFilePage * localBuffer; // temp buffer, there is a buffer for each vnode to
|
tFilePage * localBuffer; // temp buffer, there is a buffer for each vnode to
|
||||||
|
@ -96,7 +98,7 @@ int32_t tscFlushTmpBuffer(tExtMemBuffer *pMemoryBuf, tOrderDescriptor *pDesc, tF
|
||||||
* create local reducer to launch the second-stage reduce process at client site
|
* create local reducer to launch the second-stage reduce process at client site
|
||||||
*/
|
*/
|
||||||
void tscCreateLocalReducer(tExtMemBuffer **pMemBuffer, int32_t numOfBuffer, tOrderDescriptor *pDesc,
|
void tscCreateLocalReducer(tExtMemBuffer **pMemBuffer, int32_t numOfBuffer, tOrderDescriptor *pDesc,
|
||||||
SColumnModel *finalModel, SSqlObj* pSql);
|
SColumnModel *finalModel, SColumnModel *pFFModel, SSqlObj* pSql);
|
||||||
|
|
||||||
void tscDestroyLocalReducer(SSqlObj *pSql);
|
void tscDestroyLocalReducer(SSqlObj *pSql);
|
||||||
|
|
||||||
|
|
|
@ -41,6 +41,8 @@ int32_t tscHandleInsertRetry(SSqlObj* pSql);
|
||||||
void tscBuildResFromSubqueries(SSqlObj *pSql);
|
void tscBuildResFromSubqueries(SSqlObj *pSql);
|
||||||
TAOS_ROW doSetResultRowData(SSqlObj *pSql, bool finalResult);
|
TAOS_ROW doSetResultRowData(SSqlObj *pSql, bool finalResult);
|
||||||
|
|
||||||
|
char *getArithemicInputSrc(void *param, const char *name, int32_t colId);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
|
@ -125,6 +125,7 @@ int32_t tscGetDataBlockFromList(void* pHashList, SArray* pDataBlockList, int64_t
|
||||||
*/
|
*/
|
||||||
bool tscIsPointInterpQuery(SQueryInfo* pQueryInfo);
|
bool tscIsPointInterpQuery(SQueryInfo* pQueryInfo);
|
||||||
bool tscIsTWAQuery(SQueryInfo* pQueryInfo);
|
bool tscIsTWAQuery(SQueryInfo* pQueryInfo);
|
||||||
|
bool tscIsSecondStageQuery(SQueryInfo* pQueryInfo);
|
||||||
|
|
||||||
bool tscNonOrderedProjectionQueryOnSTable(SQueryInfo *pQueryInfo, int32_t tableIndex);
|
bool tscNonOrderedProjectionQueryOnSTable(SQueryInfo *pQueryInfo, int32_t tableIndex);
|
||||||
bool tscOrderedProjectionQueryOnSTable(SQueryInfo* pQueryInfo, int32_t tableIndex);
|
bool tscOrderedProjectionQueryOnSTable(SQueryInfo* pQueryInfo, int32_t tableIndex);
|
||||||
|
@ -158,7 +159,7 @@ SInternalField* tscFieldInfoGetInternalField(SFieldInfo* pFieldInfo, int32_t ind
|
||||||
TAOS_FIELD* tscFieldInfoGetField(SFieldInfo* pFieldInfo, int32_t index);
|
TAOS_FIELD* tscFieldInfoGetField(SFieldInfo* pFieldInfo, int32_t index);
|
||||||
|
|
||||||
void tscFieldInfoUpdateOffset(SQueryInfo* pQueryInfo);
|
void tscFieldInfoUpdateOffset(SQueryInfo* pQueryInfo);
|
||||||
void tscFieldInfoUpdateOffsetForInterResult(SQueryInfo* pQueryInfo);
|
void tscFieldInfoUpdateOffset(SQueryInfo* pQueryInfo);
|
||||||
|
|
||||||
int16_t tscFieldInfoGetOffset(SQueryInfo* pQueryInfo, int32_t index);
|
int16_t tscFieldInfoGetOffset(SQueryInfo* pQueryInfo, int32_t index);
|
||||||
void tscFieldInfoClear(SFieldInfo* pFieldInfo);
|
void tscFieldInfoClear(SFieldInfo* pFieldInfo);
|
||||||
|
@ -167,15 +168,15 @@ static FORCE_INLINE int32_t tscNumOfFields(SQueryInfo* pQueryInfo) { return pQue
|
||||||
|
|
||||||
int32_t tscFieldInfoCompare(const SFieldInfo* pFieldInfo1, const SFieldInfo* pFieldInfo2);
|
int32_t tscFieldInfoCompare(const SFieldInfo* pFieldInfo1, const SFieldInfo* pFieldInfo2);
|
||||||
|
|
||||||
void addExprParams(SSqlExpr* pExpr, char* argument, int32_t type, int32_t bytes, int16_t tableIndex);
|
void addExprParams(SSqlExpr* pExpr, char* argument, int32_t type, int32_t bytes);
|
||||||
|
|
||||||
int32_t tscGetResRowLength(SArray* pExprList);
|
int32_t tscGetResRowLength(SArray* pExprList);
|
||||||
|
|
||||||
SSqlExpr* tscSqlExprInsert(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
SSqlExpr* tscSqlExprInsert(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
||||||
int16_t size, int16_t interSize, bool isTagCol);
|
int16_t size, int16_t resColId, int16_t interSize, bool isTagCol);
|
||||||
|
|
||||||
SSqlExpr* tscSqlExprAppend(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
SSqlExpr* tscSqlExprAppend(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
||||||
int16_t size, int16_t interSize, bool isTagCol);
|
int16_t size, int16_t resColId, int16_t interSize, bool isTagCol);
|
||||||
|
|
||||||
SSqlExpr* tscSqlExprUpdate(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, int16_t srcColumnIndex, int16_t type,
|
SSqlExpr* tscSqlExprUpdate(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, int16_t srcColumnIndex, int16_t type,
|
||||||
int16_t size);
|
int16_t size);
|
||||||
|
|
|
@ -136,6 +136,7 @@ typedef struct SSqlExpr {
|
||||||
int16_t numOfParams; // argument value of each function
|
int16_t numOfParams; // argument value of each function
|
||||||
tVariant param[3]; // parameters are not more than 3
|
tVariant param[3]; // parameters are not more than 3
|
||||||
int32_t offset; // sub result column value of arithmetic expression.
|
int32_t offset; // sub result column value of arithmetic expression.
|
||||||
|
int16_t resColId; // result column id
|
||||||
} SSqlExpr;
|
} SSqlExpr;
|
||||||
|
|
||||||
typedef struct SColumnIndex {
|
typedef struct SColumnIndex {
|
||||||
|
@ -251,6 +252,7 @@ typedef struct SQueryInfo {
|
||||||
int64_t clauseLimit; // limit for current sub clause
|
int64_t clauseLimit; // limit for current sub clause
|
||||||
int64_t prjOffset; // offset value in the original sql expression, only applied at client side
|
int64_t prjOffset; // offset value in the original sql expression, only applied at client side
|
||||||
int32_t udColumnId; // current user-defined constant output field column id, monotonically decreases from TSDB_UD_COLUMN_INDEX
|
int32_t udColumnId; // current user-defined constant output field column id, monotonically decreases from TSDB_UD_COLUMN_INDEX
|
||||||
|
int16_t resColumnId; // result column id
|
||||||
} SQueryInfo;
|
} SQueryInfo;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
@ -462,17 +464,16 @@ int32_t tscSQLSyntaxErrMsg(char* msg, const char* additionalInfo, const char* s
|
||||||
|
|
||||||
int32_t tscToSQLCmd(SSqlObj *pSql, struct SSqlInfo *pInfo);
|
int32_t tscToSQLCmd(SSqlObj *pSql, struct SSqlInfo *pInfo);
|
||||||
|
|
||||||
static FORCE_INLINE void tscGetResultColumnChr(SSqlRes* pRes, SFieldInfo* pFieldInfo, int32_t columnIndex) {
|
static FORCE_INLINE void tscGetResultColumnChr(SSqlRes* pRes, SFieldInfo* pFieldInfo, int32_t columnIndex, int32_t offset) {
|
||||||
SInternalField* pInfo = (SInternalField*) TARRAY_GET_ELEM(pFieldInfo->internalField, columnIndex);
|
SInternalField* pInfo = (SInternalField*) TARRAY_GET_ELEM(pFieldInfo->internalField, columnIndex);
|
||||||
assert(pInfo->pSqlExpr != NULL);
|
|
||||||
|
|
||||||
int32_t type = pInfo->pSqlExpr->resType;
|
int32_t type = pInfo->field.type;
|
||||||
int32_t bytes = pInfo->pSqlExpr->resBytes;
|
int32_t bytes = pInfo->field.bytes;
|
||||||
|
|
||||||
char* pData = pRes->data + (int32_t)(pInfo->pSqlExpr->offset * pRes->numOfRows + bytes * pRes->row);
|
char* pData = pRes->data + (int32_t)(offset * pRes->numOfRows + bytes * pRes->row);
|
||||||
|
|
||||||
// user defined constant value output columns
|
// user defined constant value output columns
|
||||||
if (TSDB_COL_IS_UD_COL(pInfo->pSqlExpr->colInfo.flag)) {
|
if (pInfo->pSqlExpr != NULL && TSDB_COL_IS_UD_COL(pInfo->pSqlExpr->colInfo.flag)) {
|
||||||
if (type == TSDB_DATA_TYPE_NCHAR || type == TSDB_DATA_TYPE_BINARY) {
|
if (type == TSDB_DATA_TYPE_NCHAR || type == TSDB_DATA_TYPE_BINARY) {
|
||||||
pData = pInfo->pSqlExpr->param[1].pz;
|
pData = pInfo->pSqlExpr->param[1].pz;
|
||||||
pRes->length[columnIndex] = pInfo->pSqlExpr->param[1].nLen;
|
pRes->length[columnIndex] = pInfo->pSqlExpr->param[1].nLen;
|
||||||
|
@ -517,6 +518,7 @@ extern SRpcCorEpSet tscMgmtEpSet;
|
||||||
extern int (*tscBuildMsg[TSDB_SQL_MAX])(SSqlObj *pSql, SSqlInfo *pInfo);
|
extern int (*tscBuildMsg[TSDB_SQL_MAX])(SSqlObj *pSql, SSqlInfo *pInfo);
|
||||||
|
|
||||||
void tscBuildVgroupTableInfo(SSqlObj* pSql, STableMetaInfo* pTableMetaInfo, SArray* tables);
|
void tscBuildVgroupTableInfo(SSqlObj* pSql, STableMetaInfo* pTableMetaInfo, SArray* tables);
|
||||||
|
int16_t getNewResColId(SQueryInfo* pQueryInfo);
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
}
|
}
|
||||||
|
|
|
@ -351,7 +351,7 @@ void tscProcessFetchRow(SSchedMsg *pMsg) {
|
||||||
SInternalField* pSup = taosArrayGet(pQueryInfo->fieldsInfo.internalField, i);
|
SInternalField* pSup = taosArrayGet(pQueryInfo->fieldsInfo.internalField, i);
|
||||||
|
|
||||||
if (pSup->pSqlExpr != NULL) {
|
if (pSup->pSqlExpr != NULL) {
|
||||||
tscGetResultColumnChr(pRes, &pQueryInfo->fieldsInfo, i);
|
tscGetResultColumnChr(pRes, &pQueryInfo->fieldsInfo, i, 0);
|
||||||
} else {
|
} else {
|
||||||
// todo add
|
// todo add
|
||||||
}
|
}
|
||||||
|
|
|
@ -2695,17 +2695,18 @@ static void apercentile_func_second_merge(SQLFunctionCtx *pCtx) {
|
||||||
}
|
}
|
||||||
|
|
||||||
SAPercentileInfo *pOutput = getAPerctInfo(pCtx);
|
SAPercentileInfo *pOutput = getAPerctInfo(pCtx);
|
||||||
SHistogramInfo * pHisto = pOutput->pHisto;
|
SHistogramInfo *pHisto = pOutput->pHisto;
|
||||||
|
|
||||||
if (pHisto->numOfElems <= 0) {
|
if (pHisto->numOfElems <= 0) {
|
||||||
memcpy(pHisto, pInput->pHisto, sizeof(SHistogramInfo) + sizeof(SHistBin) * (MAX_HISTOGRAM_BIN + 1));
|
memcpy(pHisto, pInput->pHisto, sizeof(SHistogramInfo) + sizeof(SHistBin) * (MAX_HISTOGRAM_BIN + 1));
|
||||||
pHisto->elems = (SHistBin*) ((char *)pHisto + sizeof(SHistogramInfo));
|
pHisto->elems = (SHistBin*) ((char *)pHisto + sizeof(SHistogramInfo));
|
||||||
} else {
|
} else {
|
||||||
|
//TODO(dengyihao): avoid memcpy
|
||||||
pHisto->elems = (SHistBin*) ((char *)pHisto + sizeof(SHistogramInfo));
|
pHisto->elems = (SHistBin*) ((char *)pHisto + sizeof(SHistogramInfo));
|
||||||
|
|
||||||
SHistogramInfo *pRes = tHistogramMerge(pHisto, pInput->pHisto, MAX_HISTOGRAM_BIN);
|
SHistogramInfo *pRes = tHistogramMerge(pHisto, pInput->pHisto, MAX_HISTOGRAM_BIN);
|
||||||
tHistogramDestroy(&pOutput->pHisto);
|
memcpy(pHisto, pRes, sizeof(SHistogramInfo) + sizeof(SHistBin) * MAX_HISTOGRAM_BIN);
|
||||||
pOutput->pHisto = pRes;
|
pHisto->elems = (SHistBin*) ((char *)pHisto + sizeof(SHistogramInfo));
|
||||||
|
tHistogramDestroy(&pRes);
|
||||||
}
|
}
|
||||||
|
|
||||||
SResultRowCellInfo *pResInfo = GET_RES_INFO(pCtx);
|
SResultRowCellInfo *pResInfo = GET_RES_INFO(pCtx);
|
||||||
|
|
|
@ -162,7 +162,7 @@ static int32_t tscBuildTableSchemaResultFields(SSqlObj *pSql, int32_t numOfCols,
|
||||||
|
|
||||||
SInternalField* pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
SInternalField* pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY,
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY,
|
||||||
(TSDB_COL_NAME_LEN - 1) + VARSTR_HEADER_SIZE, (TSDB_COL_NAME_LEN - 1), false);
|
(TSDB_COL_NAME_LEN - 1) + VARSTR_HEADER_SIZE, -1000, (TSDB_COL_NAME_LEN - 1), false);
|
||||||
|
|
||||||
rowLen += ((TSDB_COL_NAME_LEN - 1) + VARSTR_HEADER_SIZE);
|
rowLen += ((TSDB_COL_NAME_LEN - 1) + VARSTR_HEADER_SIZE);
|
||||||
|
|
||||||
|
@ -172,7 +172,7 @@ static int32_t tscBuildTableSchemaResultFields(SSqlObj *pSql, int32_t numOfCols,
|
||||||
|
|
||||||
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY, (int16_t)(typeColLength + VARSTR_HEADER_SIZE),
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY, (int16_t)(typeColLength + VARSTR_HEADER_SIZE),
|
||||||
typeColLength, false);
|
-1000, typeColLength, false);
|
||||||
|
|
||||||
rowLen += typeColLength + VARSTR_HEADER_SIZE;
|
rowLen += typeColLength + VARSTR_HEADER_SIZE;
|
||||||
|
|
||||||
|
@ -182,7 +182,7 @@ static int32_t tscBuildTableSchemaResultFields(SSqlObj *pSql, int32_t numOfCols,
|
||||||
|
|
||||||
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_INT, sizeof(int32_t),
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_INT, sizeof(int32_t),
|
||||||
sizeof(int32_t), false);
|
-1000, sizeof(int32_t), false);
|
||||||
|
|
||||||
rowLen += sizeof(int32_t);
|
rowLen += sizeof(int32_t);
|
||||||
|
|
||||||
|
@ -192,7 +192,7 @@ static int32_t tscBuildTableSchemaResultFields(SSqlObj *pSql, int32_t numOfCols,
|
||||||
|
|
||||||
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY, (int16_t)(noteColLength + VARSTR_HEADER_SIZE),
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY, (int16_t)(noteColLength + VARSTR_HEADER_SIZE),
|
||||||
noteColLength, false);
|
-1000, noteColLength, false);
|
||||||
|
|
||||||
rowLen += noteColLength + VARSTR_HEADER_SIZE;
|
rowLen += noteColLength + VARSTR_HEADER_SIZE;
|
||||||
return rowLen;
|
return rowLen;
|
||||||
|
@ -407,8 +407,7 @@ static int32_t tscSCreateBuildResultFields(SSqlObj *pSql, BuildType type, const
|
||||||
}
|
}
|
||||||
|
|
||||||
SInternalField* pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
SInternalField* pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY,
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY, f.bytes, -1000, f.bytes - VARSTR_HEADER_SIZE, false);
|
||||||
f.bytes, f.bytes - VARSTR_HEADER_SIZE, false);
|
|
||||||
|
|
||||||
rowLen += f.bytes;
|
rowLen += f.bytes;
|
||||||
|
|
||||||
|
@ -422,7 +421,7 @@ static int32_t tscSCreateBuildResultFields(SSqlObj *pSql, BuildType type, const
|
||||||
|
|
||||||
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
pInfo = tscFieldInfoAppend(&pQueryInfo->fieldsInfo, &f);
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY,
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index, TSDB_DATA_TYPE_BINARY,
|
||||||
(int16_t)(ddlLen + VARSTR_HEADER_SIZE), ddlLen, false);
|
(int16_t)(ddlLen + VARSTR_HEADER_SIZE), -1000, ddlLen, false);
|
||||||
|
|
||||||
rowLen += ddlLen + VARSTR_HEADER_SIZE;
|
rowLen += ddlLen + VARSTR_HEADER_SIZE;
|
||||||
|
|
||||||
|
@ -619,7 +618,11 @@ static int32_t tscRebuildDDLForNormalTable(SSqlObj *pSql, const char *tableName,
|
||||||
for (int32_t i = 0; i < numOfRows; ++i) {
|
for (int32_t i = 0; i < numOfRows; ++i) {
|
||||||
uint8_t type = pSchema[i].type;
|
uint8_t type = pSchema[i].type;
|
||||||
if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
|
if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s(%d),", pSchema[i].name,tDataTypeDesc[pSchema[i].type].aName,pSchema->bytes);
|
int32_t bytes = pSchema[i].bytes - VARSTR_HEADER_SIZE;
|
||||||
|
if (type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
|
bytes = bytes/TSDB_NCHAR_SIZE;
|
||||||
|
}
|
||||||
|
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s(%d),", pSchema[i].name, tDataTypeDesc[pSchema[i].type].aName, bytes);
|
||||||
} else {
|
} else {
|
||||||
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s,", pSchema[i].name, tDataTypeDesc[pSchema[i].type].aName);
|
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s,", pSchema[i].name, tDataTypeDesc[pSchema[i].type].aName);
|
||||||
}
|
}
|
||||||
|
@ -642,7 +645,11 @@ static int32_t tscRebuildDDLForSuperTable(SSqlObj *pSql, const char *tableName,
|
||||||
for (int32_t i = 0; i < numOfRows; ++i) {
|
for (int32_t i = 0; i < numOfRows; ++i) {
|
||||||
uint8_t type = pSchema[i].type;
|
uint8_t type = pSchema[i].type;
|
||||||
if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
|
if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result),"%s %s(%d),", pSchema[i].name,tDataTypeDesc[pSchema[i].type].aName,pSchema->bytes);
|
int32_t bytes = pSchema[i].bytes - VARSTR_HEADER_SIZE;
|
||||||
|
if (type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
|
bytes = bytes/TSDB_NCHAR_SIZE;
|
||||||
|
}
|
||||||
|
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result),"%s %s(%d),", pSchema[i].name,tDataTypeDesc[pSchema[i].type].aName, bytes);
|
||||||
} else {
|
} else {
|
||||||
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s,", pSchema[i].name, tDataTypeDesc[type].aName);
|
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s,", pSchema[i].name, tDataTypeDesc[type].aName);
|
||||||
}
|
}
|
||||||
|
@ -652,7 +659,11 @@ static int32_t tscRebuildDDLForSuperTable(SSqlObj *pSql, const char *tableName,
|
||||||
for (int32_t i = numOfRows; i < totalRows; i++) {
|
for (int32_t i = numOfRows; i < totalRows; i++) {
|
||||||
uint8_t type = pSchema[i].type;
|
uint8_t type = pSchema[i].type;
|
||||||
if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
|
if (type == TSDB_DATA_TYPE_BINARY || type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s(%d),", pSchema[i].name,tDataTypeDesc[pSchema[i].type].aName,pSchema->bytes);
|
int32_t bytes = pSchema[i].bytes - VARSTR_HEADER_SIZE;
|
||||||
|
if (type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
|
bytes = bytes/TSDB_NCHAR_SIZE;
|
||||||
|
}
|
||||||
|
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s(%d),", pSchema[i].name,tDataTypeDesc[pSchema[i].type].aName, bytes);
|
||||||
} else {
|
} else {
|
||||||
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s,", pSchema[i].name, tDataTypeDesc[type].aName);
|
snprintf(result + strlen(result), TSDB_MAX_BINARY_LEN - strlen(result), "%s %s,", pSchema[i].name, tDataTypeDesc[type].aName);
|
||||||
}
|
}
|
||||||
|
|
|
@ -13,14 +13,15 @@
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
|
#include "tscLocalMerge.h"
|
||||||
|
#include "tscSubquery.h"
|
||||||
#include "os.h"
|
#include "os.h"
|
||||||
|
#include "qAst.h"
|
||||||
#include "tlosertree.h"
|
#include "tlosertree.h"
|
||||||
|
#include "tscLog.h"
|
||||||
#include "tscUtil.h"
|
#include "tscUtil.h"
|
||||||
#include "tschemautil.h"
|
#include "tschemautil.h"
|
||||||
#include "tsclient.h"
|
#include "tsclient.h"
|
||||||
#include "tutil.h"
|
|
||||||
#include "tscLog.h"
|
|
||||||
#include "tscLocalMerge.h"
|
|
||||||
|
|
||||||
typedef struct SCompareParam {
|
typedef struct SCompareParam {
|
||||||
SLocalDataSource **pLocalData;
|
SLocalDataSource **pLocalData;
|
||||||
|
@ -29,6 +30,8 @@ typedef struct SCompareParam {
|
||||||
int32_t groupOrderType;
|
int32_t groupOrderType;
|
||||||
} SCompareParam;
|
} SCompareParam;
|
||||||
|
|
||||||
|
static void doArithmeticCalculate(SQueryInfo* pQueryInfo, tFilePage* pOutput, int32_t rowSize, int32_t finalRowSize);
|
||||||
|
|
||||||
int32_t treeComparator(const void *pLeft, const void *pRight, void *param) {
|
int32_t treeComparator(const void *pLeft, const void *pRight, void *param) {
|
||||||
int32_t pLeftIdx = *(int32_t *)pLeft;
|
int32_t pLeftIdx = *(int32_t *)pLeft;
|
||||||
int32_t pRightIdx = *(int32_t *)pRight;
|
int32_t pRightIdx = *(int32_t *)pRight;
|
||||||
|
@ -132,28 +135,41 @@ static void tscInitSqlContext(SSqlCmd *pCmd, SLocalReducer *pReducer, tOrderDesc
|
||||||
}
|
}
|
||||||
|
|
||||||
static SFillColInfo* createFillColInfo(SQueryInfo* pQueryInfo) {
|
static SFillColInfo* createFillColInfo(SQueryInfo* pQueryInfo) {
|
||||||
int32_t numOfCols = (int32_t)tscSqlExprNumOfExprs(pQueryInfo);
|
int32_t numOfCols = (int32_t)tscNumOfFields(pQueryInfo);
|
||||||
int32_t offset = 0;
|
int32_t offset = 0;
|
||||||
|
|
||||||
SFillColInfo* pFillCol = calloc(numOfCols, sizeof(SFillColInfo));
|
SFillColInfo* pFillCol = calloc(numOfCols, sizeof(SFillColInfo));
|
||||||
for(int32_t i = 0; i < numOfCols; ++i) {
|
for(int32_t i = 0; i < numOfCols; ++i) {
|
||||||
SSqlExpr* pExpr = tscSqlExprGet(pQueryInfo, i);
|
SInternalField* pIField = taosArrayGet(pQueryInfo->fieldsInfo.internalField, i);
|
||||||
|
|
||||||
pFillCol[i].col.bytes = pExpr->resBytes;
|
if (pIField->pArithExprInfo == NULL) {
|
||||||
pFillCol[i].col.type = (int8_t)pExpr->resType;
|
SSqlExpr* pExpr = pIField->pSqlExpr;
|
||||||
pFillCol[i].col.colId = pExpr->colInfo.colId;
|
|
||||||
pFillCol[i].flag = pExpr->colInfo.flag;
|
pFillCol[i].col.bytes = pExpr->resBytes;
|
||||||
pFillCol[i].col.offset = offset;
|
pFillCol[i].col.type = (int8_t)pExpr->resType;
|
||||||
pFillCol[i].functionId = pExpr->functionId;
|
pFillCol[i].col.colId = pExpr->colInfo.colId;
|
||||||
pFillCol[i].fillVal.i = pQueryInfo->fillVal[i];
|
pFillCol[i].flag = pExpr->colInfo.flag;
|
||||||
offset += pExpr->resBytes;
|
pFillCol[i].col.offset = offset;
|
||||||
|
pFillCol[i].functionId = pExpr->functionId;
|
||||||
|
pFillCol[i].fillVal.i = pQueryInfo->fillVal[i];
|
||||||
|
} else {
|
||||||
|
pFillCol[i].col.bytes = pIField->field.bytes;
|
||||||
|
pFillCol[i].col.type = (int8_t)pIField->field.type;
|
||||||
|
pFillCol[i].col.colId = -100;
|
||||||
|
pFillCol[i].flag = TSDB_COL_NORMAL;
|
||||||
|
pFillCol[i].col.offset = offset;
|
||||||
|
pFillCol[i].functionId = -1;
|
||||||
|
pFillCol[i].fillVal.i = pQueryInfo->fillVal[i];
|
||||||
|
}
|
||||||
|
|
||||||
|
offset += pFillCol[i].col.bytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
return pFillCol;
|
return pFillCol;
|
||||||
}
|
}
|
||||||
|
|
||||||
void tscCreateLocalReducer(tExtMemBuffer **pMemBuffer, int32_t numOfBuffer, tOrderDescriptor *pDesc,
|
void tscCreateLocalReducer(tExtMemBuffer **pMemBuffer, int32_t numOfBuffer, tOrderDescriptor *pDesc,
|
||||||
SColumnModel *finalmodel, SSqlObj* pSql) {
|
SColumnModel *finalmodel, SColumnModel *pFFModel, SSqlObj* pSql) {
|
||||||
SSqlCmd* pCmd = &pSql->cmd;
|
SSqlCmd* pCmd = &pSql->cmd;
|
||||||
SSqlRes* pRes = &pSql->res;
|
SSqlRes* pRes = &pSql->res;
|
||||||
|
|
||||||
|
@ -342,8 +358,6 @@ void tscCreateLocalReducer(tExtMemBuffer **pMemBuffer, int32_t numOfBuffer, tOrd
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t numOfCols = tscSqlExprNumOfExprs(pQueryInfo);
|
|
||||||
|
|
||||||
pReducer->pTempBuffer->num = 0;
|
pReducer->pTempBuffer->num = 0;
|
||||||
|
|
||||||
tscCreateResPointerInfo(pRes, pQueryInfo);
|
tscCreateResPointerInfo(pRes, pQueryInfo);
|
||||||
|
@ -372,7 +386,7 @@ void tscCreateLocalReducer(tExtMemBuffer **pMemBuffer, int32_t numOfBuffer, tOrd
|
||||||
if (pQueryInfo->fillType != TSDB_FILL_NONE) {
|
if (pQueryInfo->fillType != TSDB_FILL_NONE) {
|
||||||
SFillColInfo* pFillCol = createFillColInfo(pQueryInfo);
|
SFillColInfo* pFillCol = createFillColInfo(pQueryInfo);
|
||||||
pReducer->pFillInfo = taosInitFillInfo(pQueryInfo->order.order, revisedSTime, pQueryInfo->groupbyExpr.numOfGroupCols,
|
pReducer->pFillInfo = taosInitFillInfo(pQueryInfo->order.order, revisedSTime, pQueryInfo->groupbyExpr.numOfGroupCols,
|
||||||
4096, (int32_t)numOfCols, pQueryInfo->interval.sliding, pQueryInfo->interval.slidingUnit,
|
4096, (int32_t)pQueryInfo->fieldsInfo.numOfOutput, pQueryInfo->interval.sliding, pQueryInfo->interval.slidingUnit,
|
||||||
tinfo.precision, pQueryInfo->fillType, pFillCol, pSql);
|
tinfo.precision, pQueryInfo->fillType, pFillCol, pSql);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -491,7 +505,8 @@ void tscDestroyLocalReducer(SSqlObj *pSql) {
|
||||||
pLocalReducer->pFillInfo = taosDestroyFillInfo(pLocalReducer->pFillInfo);
|
pLocalReducer->pFillInfo = taosDestroyFillInfo(pLocalReducer->pFillInfo);
|
||||||
|
|
||||||
if (pLocalReducer->pCtx != NULL) {
|
if (pLocalReducer->pCtx != NULL) {
|
||||||
for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
|
int32_t numOfExprs = (int32_t) tscSqlExprNumOfExprs(pQueryInfo);
|
||||||
|
for (int32_t i = 0; i < numOfExprs; ++i) {
|
||||||
SQLFunctionCtx *pCtx = &pLocalReducer->pCtx[i];
|
SQLFunctionCtx *pCtx = &pLocalReducer->pCtx[i];
|
||||||
|
|
||||||
tVariantDestroy(&pCtx->tag);
|
tVariantDestroy(&pCtx->tag);
|
||||||
|
@ -555,7 +570,8 @@ static int32_t createOrderDescriptor(tOrderDescriptor **pOrderDesc, SSqlCmd *pCm
|
||||||
if (numOfGroupByCols > 0) {
|
if (numOfGroupByCols > 0) {
|
||||||
|
|
||||||
if (pQueryInfo->groupbyExpr.numOfGroupCols > 0) {
|
if (pQueryInfo->groupbyExpr.numOfGroupCols > 0) {
|
||||||
int32_t startCols = pQueryInfo->fieldsInfo.numOfOutput - pQueryInfo->groupbyExpr.numOfGroupCols;
|
int32_t numOfInternalOutput = (int32_t) tscSqlExprNumOfExprs(pQueryInfo);
|
||||||
|
int32_t startCols = numOfInternalOutput - pQueryInfo->groupbyExpr.numOfGroupCols;
|
||||||
|
|
||||||
// the last "pQueryInfo->groupbyExpr.numOfGroupCols" columns are order-by columns
|
// the last "pQueryInfo->groupbyExpr.numOfGroupCols" columns are order-by columns
|
||||||
for (int32_t i = 0; i < pQueryInfo->groupbyExpr.numOfGroupCols; ++i) {
|
for (int32_t i = 0; i < pQueryInfo->groupbyExpr.numOfGroupCols; ++i) {
|
||||||
|
@ -674,6 +690,8 @@ int32_t tscLocalReducerEnvCreate(SSqlObj *pSql, tExtMemBuffer ***pMemBuffer, tOr
|
||||||
|
|
||||||
pSchema[i].bytes = pExpr->resBytes;
|
pSchema[i].bytes = pExpr->resBytes;
|
||||||
pSchema[i].type = (int8_t)pExpr->resType;
|
pSchema[i].type = (int8_t)pExpr->resType;
|
||||||
|
tstrncpy(pSchema[i].name, pExpr->aliasName, tListLen(pSchema[i].name));
|
||||||
|
|
||||||
rlen += pExpr->resBytes;
|
rlen += pExpr->resBytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -736,8 +754,8 @@ int32_t tscLocalReducerEnvCreate(SSqlObj *pSql, tExtMemBuffer ***pMemBuffer, tOr
|
||||||
}
|
}
|
||||||
|
|
||||||
*pFinalModel = createColumnModel(pSchema, (int32_t)size, capacity);
|
*pFinalModel = createColumnModel(pSchema, (int32_t)size, capacity);
|
||||||
tfree(pSchema);
|
|
||||||
|
|
||||||
|
tfree(pSchema);
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -966,10 +984,11 @@ static void doFillResult(SSqlObj *pSql, SLocalReducer *pLocalReducer, bool doneO
|
||||||
savePrevRecordAndSetupFillInfo(pLocalReducer, pQueryInfo, pFillInfo);
|
savePrevRecordAndSetupFillInfo(pLocalReducer, pQueryInfo, pFillInfo);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
int32_t offset = 0;
|
||||||
for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
|
for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
|
||||||
TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
TAOS_FIELD *pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
||||||
int16_t offset = getColumnModelOffset(pLocalReducer->resColModel, i);
|
|
||||||
memcpy(pRes->data + offset * pRes->numOfRows, pResPages[i]->data, (size_t)(pField->bytes * pRes->numOfRows));
|
memcpy(pRes->data + offset * pRes->numOfRows, pResPages[i]->data, (size_t)(pField->bytes * pRes->numOfRows));
|
||||||
|
offset += pField->bytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
pRes->numOfRowsGroup += pRes->numOfRows;
|
pRes->numOfRowsGroup += pRes->numOfRows;
|
||||||
|
@ -1222,6 +1241,10 @@ bool genFinalResults(SSqlObj *pSql, SLocalReducer *pLocalReducer, bool noMoreCur
|
||||||
|
|
||||||
tColModelCompact(pModel, pResBuf, pModel->capacity);
|
tColModelCompact(pModel, pResBuf, pModel->capacity);
|
||||||
|
|
||||||
|
if (tscIsSecondStageQuery(pQueryInfo)) {
|
||||||
|
doArithmeticCalculate(pQueryInfo, pResBuf, pModel->rowSize, pLocalReducer->finalRowSize);
|
||||||
|
}
|
||||||
|
|
||||||
#ifdef _DEBUG_VIEW
|
#ifdef _DEBUG_VIEW
|
||||||
printf("final result before interpo:\n");
|
printf("final result before interpo:\n");
|
||||||
// tColModelDisplay(pLocalReducer->resColModel, pLocalReducer->pBufForInterpo, pResBuf->num, pResBuf->num);
|
// tColModelDisplay(pLocalReducer->resColModel, pLocalReducer->pBufForInterpo, pResBuf->num, pResBuf->num);
|
||||||
|
@ -1588,3 +1611,44 @@ void tscInitResObjForLocalQuery(SSqlObj *pObj, int32_t numOfRes, int32_t rowLen)
|
||||||
pRes->pLocalReducer->pResultBuf->num = numOfRes;
|
pRes->pLocalReducer->pResultBuf->num = numOfRes;
|
||||||
pRes->data = pRes->pLocalReducer->pResultBuf->data;
|
pRes->data = pRes->pLocalReducer->pResultBuf->data;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void doArithmeticCalculate(SQueryInfo* pQueryInfo, tFilePage* pOutput, int32_t rowSize, int32_t finalRowSize) {
|
||||||
|
char* pbuf = calloc(1, pOutput->num * rowSize);
|
||||||
|
|
||||||
|
size_t size = tscNumOfFields(pQueryInfo);
|
||||||
|
SArithmeticSupport arithSup = {0};
|
||||||
|
|
||||||
|
// todo refactor
|
||||||
|
arithSup.offset = 0;
|
||||||
|
arithSup.numOfCols = (int32_t) tscSqlExprNumOfExprs(pQueryInfo);
|
||||||
|
arithSup.exprList = pQueryInfo->exprList;
|
||||||
|
arithSup.data = calloc(arithSup.numOfCols, POINTER_BYTES);
|
||||||
|
|
||||||
|
for(int32_t k = 0; k < arithSup.numOfCols; ++k) {
|
||||||
|
SSqlExpr* pExpr = tscSqlExprGet(pQueryInfo, k);
|
||||||
|
arithSup.data[k] = (pOutput->data + pOutput->num* pExpr->offset);
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t offset = 0;
|
||||||
|
|
||||||
|
for (int i = 0; i < size; ++i) {
|
||||||
|
SInternalField* pSup = TARRAY_GET_ELEM(pQueryInfo->fieldsInfo.internalField, i);
|
||||||
|
|
||||||
|
// calculate the result from several other columns
|
||||||
|
if (pSup->pArithExprInfo != NULL) {
|
||||||
|
arithSup.pArithExpr = pSup->pArithExprInfo;
|
||||||
|
tExprTreeCalcTraverse(arithSup.pArithExpr->pExpr, (int32_t) pOutput->num, pbuf + pOutput->num*offset, &arithSup, TSDB_ORDER_ASC, getArithemicInputSrc);
|
||||||
|
} else {
|
||||||
|
SSqlExpr* pExpr = pSup->pSqlExpr;
|
||||||
|
memcpy(pbuf + pOutput->num * offset, pExpr->offset * pOutput->num + pOutput->data, pExpr->resBytes * pOutput->num);
|
||||||
|
}
|
||||||
|
|
||||||
|
offset += pSup->field.bytes;
|
||||||
|
}
|
||||||
|
|
||||||
|
assert(finalRowSize <= rowSize);
|
||||||
|
memcpy(pOutput->data, pbuf, pOutput->num * finalRowSize);
|
||||||
|
|
||||||
|
tfree(pbuf);
|
||||||
|
tfree(arithSup.data);
|
||||||
|
}
|
|
@ -1148,6 +1148,10 @@ int tsParseInsertSql(SSqlObj *pSql) {
|
||||||
|
|
||||||
index = 0;
|
index = 0;
|
||||||
sToken = tStrGetToken(str, &index, false, 0, NULL);
|
sToken = tStrGetToken(str, &index, false, 0, NULL);
|
||||||
|
if (sToken.type != TK_STRING && sToken.type != TK_ID) {
|
||||||
|
code = tscInvalidSQLErrMsg(pCmd->payload, "file path is required following keyword FILE", sToken.z);
|
||||||
|
goto _error;
|
||||||
|
}
|
||||||
str += index;
|
str += index;
|
||||||
if (sToken.n == 0) {
|
if (sToken.n == 0) {
|
||||||
code = tscInvalidSQLErrMsg(pCmd->payload, "file path is required following keyword FILE", sToken.z);
|
code = tscInvalidSQLErrMsg(pCmd->payload, "file path is required following keyword FILE", sToken.z);
|
||||||
|
|
|
@ -52,7 +52,8 @@ typedef struct SConvertFunc {
|
||||||
int32_t originFuncId;
|
int32_t originFuncId;
|
||||||
int32_t execFuncId;
|
int32_t execFuncId;
|
||||||
} SConvertFunc;
|
} SConvertFunc;
|
||||||
static SSqlExpr* doAddProjectCol(SQueryInfo* pQueryInfo, int32_t outputIndex, int32_t colIndex, int32_t tableIndex);
|
|
||||||
|
static SSqlExpr* doAddProjectCol(SQueryInfo* pQueryInfo, int32_t colIndex, int32_t tableIndex);
|
||||||
|
|
||||||
static int32_t setShowInfo(SSqlObj* pSql, SSqlInfo* pInfo);
|
static int32_t setShowInfo(SSqlObj* pSql, SSqlInfo* pInfo);
|
||||||
static char* getAccountId(SSqlObj* pSql);
|
static char* getAccountId(SSqlObj* pSql);
|
||||||
|
@ -127,6 +128,10 @@ static int32_t doCheckForStream(SSqlObj* pSql, SSqlInfo* pInfo);
|
||||||
static int32_t doCheckForQuery(SSqlObj* pSql, SQuerySQL* pQuerySql, int32_t index);
|
static int32_t doCheckForQuery(SSqlObj* pSql, SQuerySQL* pQuerySql, int32_t index);
|
||||||
static int32_t exprTreeFromSqlExpr(SSqlCmd* pCmd, tExprNode **pExpr, const tSQLExpr* pSqlExpr, SQueryInfo* pQueryInfo, SArray* pCols, int64_t *uid);
|
static int32_t exprTreeFromSqlExpr(SSqlCmd* pCmd, tExprNode **pExpr, const tSQLExpr* pSqlExpr, SQueryInfo* pQueryInfo, SArray* pCols, int64_t *uid);
|
||||||
|
|
||||||
|
int16_t getNewResColId(SQueryInfo* pQueryInfo) {
|
||||||
|
return pQueryInfo->resColumnId--;
|
||||||
|
}
|
||||||
|
|
||||||
static uint8_t convertOptr(SStrToken *pToken) {
|
static uint8_t convertOptr(SStrToken *pToken) {
|
||||||
switch (pToken->type) {
|
switch (pToken->type) {
|
||||||
case TK_LT:
|
case TK_LT:
|
||||||
|
@ -1274,6 +1279,7 @@ static void tscInsertPrimaryTSSourceColumn(SQueryInfo* pQueryInfo, SColumnIndex*
|
||||||
SColumnIndex tsCol = {.tableIndex = pIndex->tableIndex, .columnIndex = PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
SColumnIndex tsCol = {.tableIndex = pIndex->tableIndex, .columnIndex = PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
||||||
tscColumnListInsert(pQueryInfo->colList, &tsCol);
|
tscColumnListInsert(pQueryInfo->colList, &tsCol);
|
||||||
}
|
}
|
||||||
|
|
||||||
static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t exprIndex, tSQLExprItem* pItem) {
|
static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t exprIndex, tSQLExprItem* pItem) {
|
||||||
const char* msg1 = "invalid column name, illegal column type, or columns in arithmetic expression from two tables";
|
const char* msg1 = "invalid column name, illegal column type, or columns in arithmetic expression from two tables";
|
||||||
const char* msg2 = "invalid arithmetic expression in select clause";
|
const char* msg2 = "invalid arithmetic expression in select clause";
|
||||||
|
@ -1305,7 +1311,7 @@ static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t
|
||||||
SColumnIndex index = {.tableIndex = tableIndex};
|
SColumnIndex index = {.tableIndex = tableIndex};
|
||||||
|
|
||||||
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_ARITHM, &index, TSDB_DATA_TYPE_DOUBLE, sizeof(double),
|
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_ARITHM, &index, TSDB_DATA_TYPE_DOUBLE, sizeof(double),
|
||||||
sizeof(double), false);
|
-1000, sizeof(double), false);
|
||||||
|
|
||||||
char* name = (pItem->aliasName != NULL)? pItem->aliasName:pItem->pNode->token.z;
|
char* name = (pItem->aliasName != NULL)? pItem->aliasName:pItem->pNode->token.z;
|
||||||
size_t len = MIN(sizeof(pExpr->aliasName), pItem->pNode->token.n + 1);
|
size_t len = MIN(sizeof(pExpr->aliasName), pItem->pNode->token.n + 1);
|
||||||
|
@ -1321,6 +1327,7 @@ static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t
|
||||||
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg2);
|
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg2);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// check for if there is a tag in the arithmetic express
|
||||||
size_t numOfNode = taosArrayGetSize(colList);
|
size_t numOfNode = taosArrayGetSize(colList);
|
||||||
for(int32_t k = 0; k < numOfNode; ++k) {
|
for(int32_t k = 0; k < numOfNode; ++k) {
|
||||||
SColIndex* pIndex = taosArrayGet(colList, k);
|
SColIndex* pIndex = taosArrayGet(colList, k);
|
||||||
|
@ -1346,9 +1353,9 @@ static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t
|
||||||
char* c = tbufGetData(&bw, false);
|
char* c = tbufGetData(&bw, false);
|
||||||
|
|
||||||
// set the serialized binary string as the parameter of arithmetic expression
|
// set the serialized binary string as the parameter of arithmetic expression
|
||||||
addExprParams(pExpr, c, TSDB_DATA_TYPE_BINARY, (int32_t)len, index.tableIndex);
|
addExprParams(pExpr, c, TSDB_DATA_TYPE_BINARY, (int32_t)len);
|
||||||
|
|
||||||
insertResultField(pQueryInfo, exprIndex, &columnList, sizeof(double), TSDB_DATA_TYPE_DOUBLE, pExpr->aliasName, pExpr);
|
insertResultField(pQueryInfo, exprIndex, &columnList, sizeof(double), TSDB_DATA_TYPE_DOUBLE, pExpr->aliasName, pExpr);
|
||||||
|
|
||||||
// add ts column
|
// add ts column
|
||||||
tscInsertPrimaryTSSourceColumn(pQueryInfo, &index);
|
tscInsertPrimaryTSSourceColumn(pQueryInfo, &index);
|
||||||
|
|
||||||
|
@ -1380,6 +1387,10 @@ static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t
|
||||||
pArithExprInfo->interBytes = sizeof(double);
|
pArithExprInfo->interBytes = sizeof(double);
|
||||||
pArithExprInfo->type = TSDB_DATA_TYPE_DOUBLE;
|
pArithExprInfo->type = TSDB_DATA_TYPE_DOUBLE;
|
||||||
|
|
||||||
|
pArithExprInfo->base.functionId = TSDB_FUNC_ARITHM;
|
||||||
|
pArithExprInfo->base.numOfParams = 1;
|
||||||
|
pArithExprInfo->base.resColId = getNewResColId(pQueryInfo);
|
||||||
|
|
||||||
int32_t ret = exprTreeFromSqlExpr(pCmd, &pArithExprInfo->pExpr, pItem->pNode, pQueryInfo, NULL, &pArithExprInfo->uid);
|
int32_t ret = exprTreeFromSqlExpr(pCmd, &pArithExprInfo->pExpr, pItem->pNode, pQueryInfo, NULL, &pArithExprInfo->uid);
|
||||||
if (ret != TSDB_CODE_SUCCESS) {
|
if (ret != TSDB_CODE_SUCCESS) {
|
||||||
tExprTreeDestroy(&pArithExprInfo->pExpr, NULL);
|
tExprTreeDestroy(&pArithExprInfo->pExpr, NULL);
|
||||||
|
@ -1388,14 +1399,30 @@ static int32_t handleArithmeticExpr(SSqlCmd* pCmd, int32_t clauseIndex, int32_t
|
||||||
|
|
||||||
pInfo->pArithExprInfo = pArithExprInfo;
|
pInfo->pArithExprInfo = pArithExprInfo;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
SBufferWriter bw = tbufInitWriter(NULL, false);
|
||||||
|
|
||||||
|
TRY(0) {
|
||||||
|
exprTreeToBinary(&bw, pInfo->pArithExprInfo->pExpr);
|
||||||
|
} CATCH(code) {
|
||||||
|
tbufCloseWriter(&bw);
|
||||||
|
UNUSED(code);
|
||||||
|
// TODO: other error handling
|
||||||
|
} END_TRY
|
||||||
|
|
||||||
|
SSqlFuncMsg* pFuncMsg = &pInfo->pArithExprInfo->base;
|
||||||
|
pFuncMsg->arg[0].argBytes = (int16_t) tbufTell(&bw);
|
||||||
|
pFuncMsg->arg[0].argValue.pz = tbufGetData(&bw, true);
|
||||||
|
pFuncMsg->arg[0].argType = TSDB_DATA_TYPE_BINARY;
|
||||||
|
|
||||||
|
// tbufCloseWriter(&bw); // TODO there is a memory leak
|
||||||
}
|
}
|
||||||
|
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
||||||
static void addProjectQueryCol(SQueryInfo* pQueryInfo, int32_t startPos, SColumnIndex* pIndex, tSQLExprItem* pItem) {
|
static void addProjectQueryCol(SQueryInfo* pQueryInfo, int32_t startPos, SColumnIndex* pIndex, tSQLExprItem* pItem) {
|
||||||
SSqlExpr* pExpr = doAddProjectCol(pQueryInfo, startPos, pIndex->columnIndex, pIndex->tableIndex);
|
SSqlExpr* pExpr = doAddProjectCol(pQueryInfo, pIndex->columnIndex, pIndex->tableIndex);
|
||||||
|
|
||||||
STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, pIndex->tableIndex);
|
STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, pIndex->tableIndex);
|
||||||
STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
|
STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
|
||||||
|
@ -1540,7 +1567,7 @@ int32_t insertResultField(SQueryInfo* pQueryInfo, int32_t outputIndex, SColumnLi
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* doAddProjectCol(SQueryInfo* pQueryInfo, int32_t outputIndex, int32_t colIndex, int32_t tableIndex) {
|
SSqlExpr* doAddProjectCol(SQueryInfo* pQueryInfo, int32_t colIndex, int32_t tableIndex) {
|
||||||
STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, tableIndex);
|
STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, tableIndex);
|
||||||
STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
|
STableMeta* pTableMeta = pTableMetaInfo->pTableMeta;
|
||||||
int32_t numOfCols = tscGetNumOfColumns(pTableMeta);
|
int32_t numOfCols = tscGetNumOfColumns(pTableMeta);
|
||||||
|
@ -1552,20 +1579,22 @@ SSqlExpr* doAddProjectCol(SQueryInfo* pQueryInfo, int32_t outputIndex, int32_t c
|
||||||
|
|
||||||
if (functionId == TSDB_FUNC_TAGPRJ) {
|
if (functionId == TSDB_FUNC_TAGPRJ) {
|
||||||
index.columnIndex = colIndex - tscGetNumOfColumns(pTableMeta);
|
index.columnIndex = colIndex - tscGetNumOfColumns(pTableMeta);
|
||||||
|
|
||||||
tscColumnListInsert(pTableMetaInfo->tagColList, &index);
|
tscColumnListInsert(pTableMetaInfo->tagColList, &index);
|
||||||
} else {
|
} else {
|
||||||
index.columnIndex = colIndex;
|
index.columnIndex = colIndex;
|
||||||
}
|
}
|
||||||
|
|
||||||
return tscSqlExprAppend(pQueryInfo, functionId, &index, pSchema->type, pSchema->bytes,
|
int16_t colId = getNewResColId(pQueryInfo);
|
||||||
pSchema->bytes, functionId == TSDB_FUNC_TAGPRJ);
|
return tscSqlExprAppend(pQueryInfo, functionId, &index, pSchema->type, pSchema->bytes, colId, pSchema->bytes,
|
||||||
|
(functionId == TSDB_FUNC_TAGPRJ));
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* tscAddSpecialColumnForSelect(SQueryInfo* pQueryInfo, int32_t outputColIndex, int16_t functionId,
|
SSqlExpr* tscAddSpecialColumnForSelect(SQueryInfo* pQueryInfo, int32_t outputColIndex, int16_t functionId,
|
||||||
SColumnIndex* pIndex, SSchema* pColSchema, int16_t flag) {
|
SColumnIndex* pIndex, SSchema* pColSchema, int16_t flag) {
|
||||||
|
int16_t colId = getNewResColId(pQueryInfo);
|
||||||
|
|
||||||
SSqlExpr* pExpr = tscSqlExprInsert(pQueryInfo, outputColIndex, functionId, pIndex, pColSchema->type,
|
SSqlExpr* pExpr = tscSqlExprInsert(pQueryInfo, outputColIndex, functionId, pIndex, pColSchema->type,
|
||||||
pColSchema->bytes, pColSchema->bytes, TSDB_COL_IS_TAG(flag));
|
pColSchema->bytes, colId, pColSchema->bytes, TSDB_COL_IS_TAG(flag));
|
||||||
tstrncpy(pExpr->aliasName, pColSchema->name, sizeof(pExpr->aliasName));
|
tstrncpy(pExpr->aliasName, pColSchema->name, sizeof(pExpr->aliasName));
|
||||||
|
|
||||||
SColumnList ids = getColumnList(1, pIndex->tableIndex, pIndex->columnIndex);
|
SColumnList ids = getColumnList(1, pIndex->tableIndex, pIndex->columnIndex);
|
||||||
|
@ -1601,7 +1630,7 @@ static int32_t doAddProjectionExprAndResultFields(SQueryInfo* pQueryInfo, SColum
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int32_t j = 0; j < numOfTotalColumns; ++j) {
|
for (int32_t j = 0; j < numOfTotalColumns; ++j) {
|
||||||
SSqlExpr* pExpr = doAddProjectCol(pQueryInfo, startPos + j, j, pIndex->tableIndex);
|
SSqlExpr* pExpr = doAddProjectCol(pQueryInfo, j, pIndex->tableIndex);
|
||||||
tstrncpy(pExpr->aliasName, pSchema[j].name, sizeof(pExpr->aliasName));
|
tstrncpy(pExpr->aliasName, pSchema[j].name, sizeof(pExpr->aliasName));
|
||||||
|
|
||||||
pIndex->columnIndex = j;
|
pIndex->columnIndex = j;
|
||||||
|
@ -1710,7 +1739,7 @@ static int32_t setExprInfoForFunctions(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SS
|
||||||
bytes = pSchema->bytes;
|
bytes = pSchema->bytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, functionID, pColIndex, type, bytes, bytes, false);
|
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, functionID, pColIndex, type, bytes, getNewResColId(pQueryInfo), bytes, false);
|
||||||
tstrncpy(pExpr->aliasName, name, tListLen(pExpr->aliasName));
|
tstrncpy(pExpr->aliasName, name, tListLen(pExpr->aliasName));
|
||||||
|
|
||||||
if (cvtFunc.originFuncId == TSDB_FUNC_LAST_ROW && cvtFunc.originFuncId != functionID) {
|
if (cvtFunc.originFuncId == TSDB_FUNC_LAST_ROW && cvtFunc.originFuncId != functionID) {
|
||||||
|
@ -1804,7 +1833,7 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
|
|
||||||
index = (SColumnIndex){0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
index = (SColumnIndex){0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
||||||
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, size, false);
|
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, getNewResColId(pQueryInfo), size, false);
|
||||||
} else if (sqlOptr == TK_INTEGER) { // select count(1) from table1
|
} else if (sqlOptr == TK_INTEGER) { // select count(1) from table1
|
||||||
char buf[8] = {0};
|
char buf[8] = {0};
|
||||||
int64_t val = -1;
|
int64_t val = -1;
|
||||||
|
@ -1816,7 +1845,7 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
if (val == 1) {
|
if (val == 1) {
|
||||||
index = (SColumnIndex){0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
index = (SColumnIndex){0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
||||||
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, size, false);
|
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, getNewResColId(pQueryInfo), size, false);
|
||||||
} else {
|
} else {
|
||||||
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg3);
|
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg3);
|
||||||
}
|
}
|
||||||
|
@ -1836,12 +1865,12 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, size, isTag);
|
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, getNewResColId(pQueryInfo), size, isTag);
|
||||||
}
|
}
|
||||||
} else { // count(*) is equalled to count(primary_timestamp_key)
|
} else { // count(*) is equalled to count(primary_timestamp_key)
|
||||||
index = (SColumnIndex){0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
index = (SColumnIndex){0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
||||||
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
int32_t size = tDataTypeDesc[TSDB_DATA_TYPE_BIGINT].nSize;
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, size, false);
|
pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, TSDB_DATA_TYPE_BIGINT, size, getNewResColId(pQueryInfo), size, false);
|
||||||
}
|
}
|
||||||
|
|
||||||
pTableMetaInfo = tscGetMetaInfo(pQueryInfo, index.tableIndex);
|
pTableMetaInfo = tscGetMetaInfo(pQueryInfo, index.tableIndex);
|
||||||
|
@ -1928,7 +1957,7 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
colIndex += 1;
|
colIndex += 1;
|
||||||
SColumnIndex indexTS = {.tableIndex = index.tableIndex, .columnIndex = 0};
|
SColumnIndex indexTS = {.tableIndex = index.tableIndex, .columnIndex = 0};
|
||||||
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &indexTS, TSDB_DATA_TYPE_TIMESTAMP, TSDB_KEYSIZE,
|
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &indexTS, TSDB_DATA_TYPE_TIMESTAMP, TSDB_KEYSIZE,
|
||||||
TSDB_KEYSIZE, false);
|
getNewResColId(pQueryInfo), TSDB_KEYSIZE, false);
|
||||||
|
|
||||||
SColumnList ids = getColumnList(1, 0, 0);
|
SColumnList ids = getColumnList(1, 0, 0);
|
||||||
insertResultField(pQueryInfo, 0, &ids, TSDB_KEYSIZE, TSDB_DATA_TYPE_TIMESTAMP, aAggs[TSDB_FUNC_TS_DUMMY].aName, pExpr);
|
insertResultField(pQueryInfo, 0, &ids, TSDB_KEYSIZE, TSDB_DATA_TYPE_TIMESTAMP, aAggs[TSDB_FUNC_TS_DUMMY].aName, pExpr);
|
||||||
|
@ -1939,7 +1968,7 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg6);
|
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg6);
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, resultType, resultSize, resultSize, false);
|
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, functionID, &index, resultType, resultSize, getNewResColId(pQueryInfo), resultSize, false);
|
||||||
|
|
||||||
if (optr == TK_LEASTSQUARES) {
|
if (optr == TK_LEASTSQUARES) {
|
||||||
/* set the leastsquares parameters */
|
/* set the leastsquares parameters */
|
||||||
|
@ -1948,14 +1977,14 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
return TSDB_CODE_TSC_INVALID_SQL;
|
return TSDB_CODE_TSC_INVALID_SQL;
|
||||||
}
|
}
|
||||||
|
|
||||||
addExprParams(pExpr, val, TSDB_DATA_TYPE_DOUBLE, DOUBLE_BYTES, 0);
|
addExprParams(pExpr, val, TSDB_DATA_TYPE_DOUBLE, DOUBLE_BYTES);
|
||||||
|
|
||||||
memset(val, 0, tListLen(val));
|
memset(val, 0, tListLen(val));
|
||||||
if (tVariantDump(&pParamElem[2].pNode->val, val, TSDB_DATA_TYPE_DOUBLE, true) < 0) {
|
if (tVariantDump(&pParamElem[2].pNode->val, val, TSDB_DATA_TYPE_DOUBLE, true) < 0) {
|
||||||
return TSDB_CODE_TSC_INVALID_SQL;
|
return TSDB_CODE_TSC_INVALID_SQL;
|
||||||
}
|
}
|
||||||
|
|
||||||
addExprParams(pExpr, val, TSDB_DATA_TYPE_DOUBLE, sizeof(double), 0);
|
addExprParams(pExpr, val, TSDB_DATA_TYPE_DOUBLE, sizeof(double));
|
||||||
}
|
}
|
||||||
|
|
||||||
SColumnList ids = {0};
|
SColumnList ids = {0};
|
||||||
|
@ -2180,8 +2209,8 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
tscInsertPrimaryTSSourceColumn(pQueryInfo, &index);
|
tscInsertPrimaryTSSourceColumn(pQueryInfo, &index);
|
||||||
colIndex += 1; // the first column is ts
|
colIndex += 1; // the first column is ts
|
||||||
|
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, functionId, &index, resultType, resultSize, resultSize, false);
|
pExpr = tscSqlExprAppend(pQueryInfo, functionId, &index, resultType, resultSize, getNewResColId(pQueryInfo), resultSize, false);
|
||||||
addExprParams(pExpr, val, TSDB_DATA_TYPE_DOUBLE, sizeof(double), 0);
|
addExprParams(pExpr, val, TSDB_DATA_TYPE_DOUBLE, sizeof(double));
|
||||||
} else {
|
} else {
|
||||||
tVariantDump(pVariant, val, TSDB_DATA_TYPE_BIGINT, true);
|
tVariantDump(pVariant, val, TSDB_DATA_TYPE_BIGINT, true);
|
||||||
|
|
||||||
|
@ -2198,8 +2227,8 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
// todo REFACTOR
|
// todo REFACTOR
|
||||||
// set the first column ts for top/bottom query
|
// set the first column ts for top/bottom query
|
||||||
SColumnIndex index1 = {0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
SColumnIndex index1 = {0, PRIMARYKEY_TIMESTAMP_COL_INDEX};
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS, &index1, TSDB_DATA_TYPE_TIMESTAMP, TSDB_KEYSIZE,
|
pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS, &index1, TSDB_DATA_TYPE_TIMESTAMP, TSDB_KEYSIZE, getNewResColId(pQueryInfo),
|
||||||
TSDB_KEYSIZE, false);
|
TSDB_KEYSIZE, false);
|
||||||
tstrncpy(pExpr->aliasName, aAggs[TSDB_FUNC_TS].aName, sizeof(pExpr->aliasName));
|
tstrncpy(pExpr->aliasName, aAggs[TSDB_FUNC_TS].aName, sizeof(pExpr->aliasName));
|
||||||
|
|
||||||
const int32_t TS_COLUMN_INDEX = PRIMARYKEY_TIMESTAMP_COL_INDEX;
|
const int32_t TS_COLUMN_INDEX = PRIMARYKEY_TIMESTAMP_COL_INDEX;
|
||||||
|
@ -2209,8 +2238,8 @@ int32_t addExprAndResultField(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, int32_t col
|
||||||
|
|
||||||
colIndex += 1; // the first column is ts
|
colIndex += 1; // the first column is ts
|
||||||
|
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, functionId, &index, resultType, resultSize, resultSize, false);
|
pExpr = tscSqlExprAppend(pQueryInfo, functionId, &index, resultType, resultSize, getNewResColId(pQueryInfo), resultSize, false);
|
||||||
addExprParams(pExpr, val, TSDB_DATA_TYPE_BIGINT, sizeof(int64_t), 0);
|
addExprParams(pExpr, val, TSDB_DATA_TYPE_BIGINT, sizeof(int64_t));
|
||||||
}
|
}
|
||||||
|
|
||||||
memset(pExpr->aliasName, 0, tListLen(pExpr->aliasName));
|
memset(pExpr->aliasName, 0, tListLen(pExpr->aliasName));
|
||||||
|
@ -2694,7 +2723,7 @@ int32_t tscTansformSQLFuncForSTableQuery(SQueryInfo* pQueryInfo) {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
tscFieldInfoUpdateOffsetForInterResult(pQueryInfo);
|
tscFieldInfoUpdateOffset(pQueryInfo);
|
||||||
return TSDB_CODE_SUCCESS;
|
return TSDB_CODE_SUCCESS;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -2922,7 +2951,7 @@ int32_t parseGroupbyClause(SQueryInfo* pQueryInfo, SArray* pList, SSqlCmd* pCmd)
|
||||||
|
|
||||||
void setColumnOffsetValueInResultset(SQueryInfo* pQueryInfo) {
|
void setColumnOffsetValueInResultset(SQueryInfo* pQueryInfo) {
|
||||||
if (QUERY_IS_STABLE_QUERY(pQueryInfo->type)) {
|
if (QUERY_IS_STABLE_QUERY(pQueryInfo->type)) {
|
||||||
tscFieldInfoUpdateOffsetForInterResult(pQueryInfo);
|
tscFieldInfoUpdateOffset(pQueryInfo);
|
||||||
} else {
|
} else {
|
||||||
tscFieldInfoUpdateOffset(pQueryInfo);
|
tscFieldInfoUpdateOffset(pQueryInfo);
|
||||||
}
|
}
|
||||||
|
@ -4437,7 +4466,7 @@ int32_t parseFillClause(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SQuerySQL* pQuery
|
||||||
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg2);
|
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg2);
|
||||||
}
|
}
|
||||||
|
|
||||||
size_t size = tscSqlExprNumOfExprs(pQueryInfo);
|
size_t size = tscNumOfFields(pQueryInfo);
|
||||||
|
|
||||||
if (pQueryInfo->fillVal == NULL) {
|
if (pQueryInfo->fillVal == NULL) {
|
||||||
pQueryInfo->fillVal = calloc(size, sizeof(int64_t));
|
pQueryInfo->fillVal = calloc(size, sizeof(int64_t));
|
||||||
|
@ -4451,12 +4480,8 @@ int32_t parseFillClause(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SQuerySQL* pQuery
|
||||||
} else if (strncasecmp(pItem->pVar.pz, "null", 4) == 0 && pItem->pVar.nLen == 4) {
|
} else if (strncasecmp(pItem->pVar.pz, "null", 4) == 0 && pItem->pVar.nLen == 4) {
|
||||||
pQueryInfo->fillType = TSDB_FILL_NULL;
|
pQueryInfo->fillType = TSDB_FILL_NULL;
|
||||||
for (int32_t i = START_INTERPO_COL_IDX; i < size; ++i) {
|
for (int32_t i = START_INTERPO_COL_IDX; i < size; ++i) {
|
||||||
TAOS_FIELD* pFields = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
TAOS_FIELD* pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
||||||
if (pFields->type == TSDB_DATA_TYPE_BINARY || pFields->type == TSDB_DATA_TYPE_NCHAR) {
|
setNull((char*)&pQueryInfo->fillVal[i], pField->type, pField->bytes);
|
||||||
setVardataNull((char*) &pQueryInfo->fillVal[i], pFields->type);
|
|
||||||
} else {
|
|
||||||
setNull((char*)&pQueryInfo->fillVal[i], pFields->type, pFields->bytes);
|
|
||||||
};
|
|
||||||
}
|
}
|
||||||
} else if (strncasecmp(pItem->pVar.pz, "prev", 4) == 0 && pItem->pVar.nLen == 4) {
|
} else if (strncasecmp(pItem->pVar.pz, "prev", 4) == 0 && pItem->pVar.nLen == 4) {
|
||||||
pQueryInfo->fillType = TSDB_FILL_PREV;
|
pQueryInfo->fillType = TSDB_FILL_PREV;
|
||||||
|
@ -4487,15 +4512,15 @@ int32_t parseFillClause(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SQuerySQL* pQuery
|
||||||
int32_t j = 1;
|
int32_t j = 1;
|
||||||
|
|
||||||
for (int32_t i = startPos; i < numOfFillVal; ++i, ++j) {
|
for (int32_t i = startPos; i < numOfFillVal; ++i, ++j) {
|
||||||
TAOS_FIELD* pFields = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
TAOS_FIELD* pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
||||||
|
|
||||||
if (pFields->type == TSDB_DATA_TYPE_BINARY || pFields->type == TSDB_DATA_TYPE_NCHAR) {
|
if (pField->type == TSDB_DATA_TYPE_BINARY || pField->type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
setVardataNull((char*) &pQueryInfo->fillVal[i], pFields->type);
|
setVardataNull((char*) &pQueryInfo->fillVal[i], pField->type);
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
tVariant* p = taosArrayGet(pFillToken, j);
|
tVariant* p = taosArrayGet(pFillToken, j);
|
||||||
int32_t ret = tVariantDump(p, (char*)&pQueryInfo->fillVal[i], pFields->type, true);
|
int32_t ret = tVariantDump(p, (char*)&pQueryInfo->fillVal[i], pField->type, true);
|
||||||
if (ret != TSDB_CODE_SUCCESS) {
|
if (ret != TSDB_CODE_SUCCESS) {
|
||||||
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg);
|
return invalidSqlErrMsg(tscGetErrorMsgPayload(pCmd), msg);
|
||||||
}
|
}
|
||||||
|
@ -4505,12 +4530,12 @@ int32_t parseFillClause(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SQuerySQL* pQuery
|
||||||
tVariantListItem* lastItem = taosArrayGetLast(pFillToken);
|
tVariantListItem* lastItem = taosArrayGetLast(pFillToken);
|
||||||
|
|
||||||
for (int32_t i = numOfFillVal; i < size; ++i) {
|
for (int32_t i = numOfFillVal; i < size; ++i) {
|
||||||
TAOS_FIELD* pFields = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
TAOS_FIELD* pField = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
||||||
|
|
||||||
if (pFields->type == TSDB_DATA_TYPE_BINARY || pFields->type == TSDB_DATA_TYPE_NCHAR) {
|
if (pField->type == TSDB_DATA_TYPE_BINARY || pField->type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
setVardataNull((char*) &pQueryInfo->fillVal[i], pFields->type);
|
setVardataNull((char*) &pQueryInfo->fillVal[i], pField->type);
|
||||||
} else {
|
} else {
|
||||||
tVariantDump(&lastItem->pVar, (char*)&pQueryInfo->fillVal[i], pFields->type, true);
|
tVariantDump(&lastItem->pVar, (char*)&pQueryInfo->fillVal[i], pField->type, true);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -5447,7 +5472,7 @@ void addGroupInfoForSubquery(SSqlObj* pParentObj, SSqlObj* pSql, int32_t subClau
|
||||||
int16_t type = pTagSchema->type;
|
int16_t type = pTagSchema->type;
|
||||||
int16_t bytes = pTagSchema->bytes;
|
int16_t bytes = pTagSchema->bytes;
|
||||||
|
|
||||||
pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TAG, &index, type, bytes, bytes, true);
|
pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TAG, &index, type, bytes, getNewResColId(pQueryInfo), bytes, true);
|
||||||
pExpr->colInfo.flag = TSDB_COL_TAG;
|
pExpr->colInfo.flag = TSDB_COL_TAG;
|
||||||
|
|
||||||
// NOTE: tag column does not add to source column list
|
// NOTE: tag column does not add to source column list
|
||||||
|
@ -5750,7 +5775,7 @@ static int32_t doAddGroupbyColumnsOnDemand(SSqlCmd* pCmd, SQueryInfo* pQueryInfo
|
||||||
|
|
||||||
if (TSDB_COL_IS_TAG(pColIndex->flag)) {
|
if (TSDB_COL_IS_TAG(pColIndex->flag)) {
|
||||||
SColumnIndex index = {.tableIndex = pQueryInfo->groupbyExpr.tableIndex, .columnIndex = colIndex};
|
SColumnIndex index = {.tableIndex = pQueryInfo->groupbyExpr.tableIndex, .columnIndex = colIndex};
|
||||||
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TAG, &index, type, bytes, bytes, true);
|
SSqlExpr* pExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TAG, &index, type, bytes, getNewResColId(pQueryInfo), bytes, true);
|
||||||
|
|
||||||
memset(pExpr->aliasName, 0, sizeof(pExpr->aliasName));
|
memset(pExpr->aliasName, 0, sizeof(pExpr->aliasName));
|
||||||
tstrncpy(pExpr->aliasName, name, sizeof(pExpr->aliasName));
|
tstrncpy(pExpr->aliasName, name, sizeof(pExpr->aliasName));
|
||||||
|
@ -5913,7 +5938,7 @@ int32_t doLocalQueryProcess(SSqlCmd* pCmd, SQueryInfo* pQueryInfo, SQuerySQL* pQ
|
||||||
|
|
||||||
SColumnIndex ind = {0};
|
SColumnIndex ind = {0};
|
||||||
SSqlExpr* pExpr1 = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TAG_DUMMY, &ind, TSDB_DATA_TYPE_INT,
|
SSqlExpr* pExpr1 = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TAG_DUMMY, &ind, TSDB_DATA_TYPE_INT,
|
||||||
tDataTypeDesc[TSDB_DATA_TYPE_INT].nSize, tDataTypeDesc[TSDB_DATA_TYPE_INT].nSize, false);
|
tDataTypeDesc[TSDB_DATA_TYPE_INT].nSize, getNewResColId(pQueryInfo), tDataTypeDesc[TSDB_DATA_TYPE_INT].nSize, false);
|
||||||
|
|
||||||
const char* name = (pExprList->a[0].aliasName != NULL)? pExprList->a[0].aliasName:functionsInfo[index].name;
|
const char* name = (pExprList->a[0].aliasName != NULL)? pExprList->a[0].aliasName:functionsInfo[index].name;
|
||||||
tstrncpy(pExpr1->aliasName, name, tListLen(pExpr1->aliasName));
|
tstrncpy(pExpr1->aliasName, name, tListLen(pExpr1->aliasName));
|
||||||
|
@ -6585,6 +6610,7 @@ int32_t exprTreeFromSqlExpr(SSqlCmd* pCmd, tExprNode **pExpr, const tSQLExpr* pS
|
||||||
if (strcmp((*pExpr)->pSchema->name, p1->aliasName) == 0) {
|
if (strcmp((*pExpr)->pSchema->name, p1->aliasName) == 0) {
|
||||||
(*pExpr)->pSchema->type = (uint8_t)p1->resType;
|
(*pExpr)->pSchema->type = (uint8_t)p1->resType;
|
||||||
(*pExpr)->pSchema->bytes = p1->resBytes;
|
(*pExpr)->pSchema->bytes = p1->resBytes;
|
||||||
|
(*pExpr)->pSchema->colId = p1->resColId;
|
||||||
|
|
||||||
if (uid != NULL) {
|
if (uid != NULL) {
|
||||||
*uid = p1->uid;
|
*uid = p1->uid;
|
||||||
|
|
|
@ -698,7 +698,7 @@ int tscBuildQueryMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
|
||||||
pQueryMsg->queryType = htonl(pQueryInfo->type);
|
pQueryMsg->queryType = htonl(pQueryInfo->type);
|
||||||
|
|
||||||
size_t numOfOutput = tscSqlExprNumOfExprs(pQueryInfo);
|
size_t numOfOutput = tscSqlExprNumOfExprs(pQueryInfo);
|
||||||
pQueryMsg->numOfOutput = htons((int16_t)numOfOutput);
|
pQueryMsg->numOfOutput = htons((int16_t)numOfOutput); // this is the stage one output column number
|
||||||
|
|
||||||
// set column list ids
|
// set column list ids
|
||||||
size_t numOfCols = taosArrayGetSize(pQueryInfo->colList);
|
size_t numOfCols = taosArrayGetSize(pQueryInfo->colList);
|
||||||
|
@ -760,12 +760,15 @@ int tscBuildQueryMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
|
||||||
return TSDB_CODE_TSC_INVALID_SQL;
|
return TSDB_CODE_TSC_INVALID_SQL;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
assert(pExpr->resColId < 0);
|
||||||
|
|
||||||
pSqlFuncExpr->colInfo.colId = htons(pExpr->colInfo.colId);
|
pSqlFuncExpr->colInfo.colId = htons(pExpr->colInfo.colId);
|
||||||
pSqlFuncExpr->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
|
pSqlFuncExpr->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
|
||||||
pSqlFuncExpr->colInfo.flag = htons(pExpr->colInfo.flag);
|
pSqlFuncExpr->colInfo.flag = htons(pExpr->colInfo.flag);
|
||||||
|
|
||||||
pSqlFuncExpr->functionId = htons(pExpr->functionId);
|
pSqlFuncExpr->functionId = htons(pExpr->functionId);
|
||||||
pSqlFuncExpr->numOfParams = htons(pExpr->numOfParams);
|
pSqlFuncExpr->numOfParams = htons(pExpr->numOfParams);
|
||||||
|
pSqlFuncExpr->resColId = htons(pExpr->resColId);
|
||||||
pMsg += sizeof(SSqlFuncMsg);
|
pMsg += sizeof(SSqlFuncMsg);
|
||||||
|
|
||||||
for (int32_t j = 0; j < pExpr->numOfParams; ++j) {
|
for (int32_t j = 0; j < pExpr->numOfParams; ++j) {
|
||||||
|
@ -783,7 +786,73 @@ int tscBuildQueryMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
|
||||||
|
|
||||||
pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
|
pSqlFuncExpr = (SSqlFuncMsg *)pMsg;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
if(tscIsSecondStageQuery(pQueryInfo)) {
|
||||||
|
size_t output = tscNumOfFields(pQueryInfo);
|
||||||
|
pQueryMsg->secondStageOutput = htonl((int32_t) output);
|
||||||
|
|
||||||
|
SSqlFuncMsg *pSqlFuncExpr1 = (SSqlFuncMsg *)pMsg;
|
||||||
|
|
||||||
|
for (int32_t i = 0; i < output; ++i) {
|
||||||
|
SInternalField* pField = tscFieldInfoGetInternalField(&pQueryInfo->fieldsInfo, i);
|
||||||
|
SSqlExpr *pExpr = pField->pSqlExpr;
|
||||||
|
if (pExpr != NULL) {
|
||||||
|
if (!tscValidateColumnId(pTableMetaInfo, pExpr->colInfo.colId, pExpr->numOfParams)) {
|
||||||
|
tscError("%p table schema is not matched with parsed sql", pSql);
|
||||||
|
return TSDB_CODE_TSC_INVALID_SQL;
|
||||||
|
}
|
||||||
|
|
||||||
|
pSqlFuncExpr1->colInfo.colId = htons(pExpr->colInfo.colId);
|
||||||
|
pSqlFuncExpr1->colInfo.colIndex = htons(pExpr->colInfo.colIndex);
|
||||||
|
pSqlFuncExpr1->colInfo.flag = htons(pExpr->colInfo.flag);
|
||||||
|
|
||||||
|
pSqlFuncExpr1->functionId = htons(pExpr->functionId);
|
||||||
|
pSqlFuncExpr1->numOfParams = htons(pExpr->numOfParams);
|
||||||
|
pMsg += sizeof(SSqlFuncMsg);
|
||||||
|
|
||||||
|
for (int32_t j = 0; j < pExpr->numOfParams; ++j) {
|
||||||
|
// todo add log
|
||||||
|
pSqlFuncExpr1->arg[j].argType = htons((uint16_t)pExpr->param[j].nType);
|
||||||
|
pSqlFuncExpr1->arg[j].argBytes = htons(pExpr->param[j].nLen);
|
||||||
|
|
||||||
|
if (pExpr->param[j].nType == TSDB_DATA_TYPE_BINARY) {
|
||||||
|
memcpy(pMsg, pExpr->param[j].pz, pExpr->param[j].nLen);
|
||||||
|
pMsg += pExpr->param[j].nLen;
|
||||||
|
} else {
|
||||||
|
pSqlFuncExpr1->arg[j].argValue.i64 = htobe64(pExpr->param[j].i64Key);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pSqlFuncExpr1 = (SSqlFuncMsg *)pMsg;
|
||||||
|
} else {
|
||||||
|
assert(pField->pArithExprInfo != NULL);
|
||||||
|
SExprInfo* pExprInfo = pField->pArithExprInfo;
|
||||||
|
|
||||||
|
pSqlFuncExpr1->colInfo.colId = htons(pExprInfo->base.colInfo.colId);
|
||||||
|
pSqlFuncExpr1->functionId = htons(pExprInfo->base.functionId);
|
||||||
|
pSqlFuncExpr1->numOfParams = htons(pExprInfo->base.numOfParams);
|
||||||
|
pMsg += sizeof(SSqlFuncMsg);
|
||||||
|
|
||||||
|
for (int32_t j = 0; j < pExprInfo->base.numOfParams; ++j) {
|
||||||
|
// todo add log
|
||||||
|
pSqlFuncExpr1->arg[j].argType = htons((uint16_t)pExprInfo->base.arg[j].argType);
|
||||||
|
pSqlFuncExpr1->arg[j].argBytes = htons(pExprInfo->base.arg[j].argBytes);
|
||||||
|
|
||||||
|
if (pExprInfo->base.arg[j].argType == TSDB_DATA_TYPE_BINARY) {
|
||||||
|
memcpy(pMsg, pExprInfo->base.arg[j].argValue.pz, pExprInfo->base.arg[j].argBytes);
|
||||||
|
pMsg += pExprInfo->base.arg[j].argBytes;
|
||||||
|
} else {
|
||||||
|
pSqlFuncExpr1->arg[j].argValue.i64 = htobe64(pExprInfo->base.arg[j].argValue.i64);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pSqlFuncExpr1 = (SSqlFuncMsg *)pMsg;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
pQueryMsg->secondStageOutput = 0;
|
||||||
|
}
|
||||||
|
|
||||||
// serialize the table info (sid, uid, tags)
|
// serialize the table info (sid, uid, tags)
|
||||||
pMsg = doSerializeTableInfo(pQueryMsg, pSql, pMsg);
|
pMsg = doSerializeTableInfo(pQueryMsg, pSql, pMsg);
|
||||||
|
|
||||||
|
@ -810,7 +879,7 @@ int tscBuildQueryMsg(SSqlObj *pSql, SSqlInfo *pInfo) {
|
||||||
}
|
}
|
||||||
|
|
||||||
if (pQueryInfo->fillType != TSDB_FILL_NONE) {
|
if (pQueryInfo->fillType != TSDB_FILL_NONE) {
|
||||||
for (int32_t i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
|
for (int32_t i = 0; i < tscSqlExprNumOfExprs(pQueryInfo); ++i) {
|
||||||
*((int64_t *)pMsg) = htobe64(pQueryInfo->fillVal[i]);
|
*((int64_t *)pMsg) = htobe64(pQueryInfo->fillVal[i]);
|
||||||
pMsg += sizeof(pQueryInfo->fillVal[0]);
|
pMsg += sizeof(pQueryInfo->fillVal[0]);
|
||||||
}
|
}
|
||||||
|
@ -1946,7 +2015,7 @@ int tscProcessShowRsp(SSqlObj *pSql) {
|
||||||
SInternalField* pInfo = tscFieldInfoAppend(pFieldInfo, &f);
|
SInternalField* pInfo = tscFieldInfoAppend(pFieldInfo, &f);
|
||||||
|
|
||||||
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index,
|
pInfo->pSqlExpr = tscSqlExprAppend(pQueryInfo, TSDB_FUNC_TS_DUMMY, &index,
|
||||||
pTableSchema[i].type, pTableSchema[i].bytes, pTableSchema[i].bytes, false);
|
pTableSchema[i].type, pTableSchema[i].bytes, getNewResColId(pQueryInfo), pTableSchema[i].bytes, false);
|
||||||
}
|
}
|
||||||
|
|
||||||
pCmd->numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
|
pCmd->numOfCols = pQueryInfo->fieldsInfo.numOfOutput;
|
||||||
|
|
|
@ -28,7 +28,6 @@
|
||||||
#include "tutil.h"
|
#include "tutil.h"
|
||||||
#include "ttimer.h"
|
#include "ttimer.h"
|
||||||
#include "tscProfile.h"
|
#include "tscProfile.h"
|
||||||
#include "ttimer.h"
|
|
||||||
|
|
||||||
static bool validImpl(const char* str, size_t maxsize) {
|
static bool validImpl(const char* str, size_t maxsize) {
|
||||||
if (str == NULL) {
|
if (str == NULL) {
|
||||||
|
@ -482,7 +481,7 @@ int taos_fetch_block_impl(TAOS_RES *res, TAOS_ROW *rows) {
|
||||||
|
|
||||||
assert(0);
|
assert(0);
|
||||||
for (int i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
|
for (int i = 0; i < pQueryInfo->fieldsInfo.numOfOutput; ++i) {
|
||||||
tscGetResultColumnChr(pRes, &pQueryInfo->fieldsInfo, i);
|
tscGetResultColumnChr(pRes, &pQueryInfo->fieldsInfo, i, 0);
|
||||||
}
|
}
|
||||||
|
|
||||||
*rows = pRes->tsrow;
|
*rows = pRes->tsrow;
|
||||||
|
|
|
@ -1642,9 +1642,9 @@ int32_t tscHandleMasterSTableQuery(SSqlObj *pSql) {
|
||||||
}
|
}
|
||||||
|
|
||||||
tExtMemBuffer ** pMemoryBuf = NULL;
|
tExtMemBuffer ** pMemoryBuf = NULL;
|
||||||
tOrderDescriptor *pDesc = NULL;
|
tOrderDescriptor *pDesc = NULL;
|
||||||
SColumnModel * pModel = NULL;
|
SColumnModel *pModel = NULL;
|
||||||
|
|
||||||
pRes->qhandle = 0x1; // hack the qhandle check
|
pRes->qhandle = 0x1; // hack the qhandle check
|
||||||
|
|
||||||
const uint32_t nBufferSize = (1u << 16); // 64KB
|
const uint32_t nBufferSize = (1u << 16); // 64KB
|
||||||
|
@ -1707,7 +1707,7 @@ int32_t tscHandleMasterSTableQuery(SSqlObj *pSql) {
|
||||||
trs->subqueryIndex = i;
|
trs->subqueryIndex = i;
|
||||||
trs->pParentSql = pSql;
|
trs->pParentSql = pSql;
|
||||||
trs->pFinalColModel = pModel;
|
trs->pFinalColModel = pModel;
|
||||||
|
|
||||||
SSqlObj *pNew = tscCreateSTableSubquery(pSql, trs, NULL);
|
SSqlObj *pNew = tscCreateSTableSubquery(pSql, trs, NULL);
|
||||||
if (pNew == NULL) {
|
if (pNew == NULL) {
|
||||||
tscError("%p failed to malloc buffer for subObj, orderOfSub:%d, reason:%s", pSql, i, strerror(errno));
|
tscError("%p failed to malloc buffer for subObj, orderOfSub:%d, reason:%s", pSql, i, strerror(errno));
|
||||||
|
@ -1762,10 +1762,6 @@ static void tscFreeRetrieveSup(SSqlObj *pSql) {
|
||||||
}
|
}
|
||||||
|
|
||||||
tscDebug("%p start to free subquery supp obj:%p", pSql, trsupport);
|
tscDebug("%p start to free subquery supp obj:%p", pSql, trsupport);
|
||||||
// int32_t index = trsupport->subqueryIndex;
|
|
||||||
// SSqlObj *pParentSql = trsupport->pParentSql;
|
|
||||||
|
|
||||||
// assert(pSql == pParentSql->pSubs[index]);
|
|
||||||
tfree(trsupport->localBuffer);
|
tfree(trsupport->localBuffer);
|
||||||
tfree(trsupport);
|
tfree(trsupport);
|
||||||
}
|
}
|
||||||
|
@ -1956,7 +1952,7 @@ static void tscAllDataRetrievedFromDnode(SRetrieveSupport *trsupport, SSqlObj* p
|
||||||
SQueryInfo *pPQueryInfo = tscGetQueryInfoDetail(&pParentSql->cmd, 0);
|
SQueryInfo *pPQueryInfo = tscGetQueryInfoDetail(&pParentSql->cmd, 0);
|
||||||
tscClearInterpInfo(pPQueryInfo);
|
tscClearInterpInfo(pPQueryInfo);
|
||||||
|
|
||||||
tscCreateLocalReducer(trsupport->pExtMemBuffer, pState->numOfSub, pDesc, trsupport->pFinalColModel, pParentSql);
|
tscCreateLocalReducer(trsupport->pExtMemBuffer, pState->numOfSub, pDesc, trsupport->pFinalColModel, trsupport->pFFColModel, pParentSql);
|
||||||
tscDebug("%p build loser tree completed", pParentSql);
|
tscDebug("%p build loser tree completed", pParentSql);
|
||||||
|
|
||||||
pParentSql->res.precision = pSql->res.precision;
|
pParentSql->res.precision = pSql->res.precision;
|
||||||
|
@ -2418,7 +2414,7 @@ static void transferNcharData(SSqlObj *pSql, int32_t columnIndex, TAOS_FIELD *pF
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
static char *getArithemicInputSrc(void *param, const char *name, int32_t colId) {
|
char *getArithemicInputSrc(void *param, const char *name, int32_t colId) {
|
||||||
SArithmeticSupport *pSupport = (SArithmeticSupport *) param;
|
SArithmeticSupport *pSupport = (SArithmeticSupport *) param;
|
||||||
|
|
||||||
int32_t index = -1;
|
int32_t index = -1;
|
||||||
|
@ -2449,48 +2445,22 @@ TAOS_ROW doSetResultRowData(SSqlObj *pSql, bool finalResult) {
|
||||||
SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
|
SQueryInfo *pQueryInfo = tscGetQueryInfoDetail(pCmd, pCmd->clauseIndex);
|
||||||
|
|
||||||
size_t size = tscNumOfFields(pQueryInfo);
|
size_t size = tscNumOfFields(pQueryInfo);
|
||||||
|
int32_t offset = 0;
|
||||||
|
|
||||||
for (int i = 0; i < size; ++i) {
|
for (int i = 0; i < size; ++i) {
|
||||||
SInternalField* pSup = TARRAY_GET_ELEM(pQueryInfo->fieldsInfo.internalField, i);
|
tscGetResultColumnChr(pRes, &pQueryInfo->fieldsInfo, i, offset);
|
||||||
if (pSup->pSqlExpr != NULL) {
|
TAOS_FIELD *pField = TARRAY_GET_ELEM(pQueryInfo->fieldsInfo.internalField, i);
|
||||||
tscGetResultColumnChr(pRes, &pQueryInfo->fieldsInfo, i);
|
|
||||||
}
|
offset += pField->bytes;
|
||||||
|
|
||||||
// primary key column cannot be null in interval query, no need to check
|
// primary key column cannot be null in interval query, no need to check
|
||||||
if (i == 0 && pQueryInfo->interval.interval > 0) {
|
if (i == 0 && pQueryInfo->interval.interval > 0) {
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
TAOS_FIELD *pField = TARRAY_GET_ELEM(pQueryInfo->fieldsInfo.internalField, i);
|
|
||||||
if (pRes->tsrow[i] != NULL && pField->type == TSDB_DATA_TYPE_NCHAR) {
|
if (pRes->tsrow[i] != NULL && pField->type == TSDB_DATA_TYPE_NCHAR) {
|
||||||
transferNcharData(pSql, i, pField);
|
transferNcharData(pSql, i, pField);
|
||||||
}
|
}
|
||||||
|
|
||||||
// calculate the result from several other columns
|
|
||||||
if (pSup->pArithExprInfo != NULL) {
|
|
||||||
if (pRes->pArithSup == NULL) {
|
|
||||||
pRes->pArithSup = (SArithmeticSupport*)calloc(1, sizeof(SArithmeticSupport));
|
|
||||||
}
|
|
||||||
|
|
||||||
pRes->pArithSup->offset = 0;
|
|
||||||
pRes->pArithSup->pArithExpr = pSup->pArithExprInfo;
|
|
||||||
pRes->pArithSup->numOfCols = (int32_t)tscSqlExprNumOfExprs(pQueryInfo);
|
|
||||||
pRes->pArithSup->exprList = pQueryInfo->exprList;
|
|
||||||
pRes->pArithSup->data = calloc(pRes->pArithSup->numOfCols, POINTER_BYTES);
|
|
||||||
|
|
||||||
if (pRes->buffer[i] == NULL) {
|
|
||||||
TAOS_FIELD* field = tscFieldInfoGetField(&pQueryInfo->fieldsInfo, i);
|
|
||||||
pRes->buffer[i] = malloc(field->bytes);
|
|
||||||
}
|
|
||||||
|
|
||||||
for(int32_t k = 0; k < pRes->pArithSup->numOfCols; ++k) {
|
|
||||||
SSqlExpr* pExpr = tscSqlExprGet(pQueryInfo, k);
|
|
||||||
pRes->pArithSup->data[k] = (pRes->data + pRes->numOfRows* pExpr->offset) + pRes->row*pExpr->resBytes;
|
|
||||||
}
|
|
||||||
|
|
||||||
tExprTreeCalcTraverse(pRes->pArithSup->pArithExpr->pExpr, 1, pRes->buffer[i], pRes->pArithSup,
|
|
||||||
TSDB_ORDER_ASC, getArithemicInputSrc);
|
|
||||||
pRes->tsrow[i] = (unsigned char*)pRes->buffer[i];
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
pRes->row++; // index increase one-step
|
pRes->row++; // index increase one-step
|
||||||
|
|
|
@ -219,6 +219,24 @@ bool tscIsPointInterpQuery(SQueryInfo* pQueryInfo) {
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
bool tscIsSecondStageQuery(SQueryInfo* pQueryInfo) {
|
||||||
|
size_t numOfOutput = tscNumOfFields(pQueryInfo);
|
||||||
|
size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
|
||||||
|
|
||||||
|
if (numOfOutput == numOfExprs) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
for(int32_t i = 0; i < numOfOutput; ++i) {
|
||||||
|
SExprInfo* pExprInfo = tscFieldInfoGetInternalField(&pQueryInfo->fieldsInfo, i)->pArithExprInfo;
|
||||||
|
if (pExprInfo != NULL) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
bool tscIsTWAQuery(SQueryInfo* pQueryInfo) {
|
bool tscIsTWAQuery(SQueryInfo* pQueryInfo) {
|
||||||
size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
|
size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
|
||||||
for (int32_t i = 0; i < numOfExprs; ++i) {
|
for (int32_t i = 0; i < numOfExprs; ++i) {
|
||||||
|
@ -855,28 +873,11 @@ void tscFieldInfoUpdateOffset(SQueryInfo* pQueryInfo) {
|
||||||
|
|
||||||
SSqlExpr* pExpr = taosArrayGetP(pQueryInfo->exprList, 0);
|
SSqlExpr* pExpr = taosArrayGetP(pQueryInfo->exprList, 0);
|
||||||
pExpr->offset = 0;
|
pExpr->offset = 0;
|
||||||
|
|
||||||
for (int32_t i = 1; i < numOfExprs; ++i) {
|
|
||||||
SSqlExpr* prev = taosArrayGetP(pQueryInfo->exprList, i - 1);
|
|
||||||
SSqlExpr* p = taosArrayGetP(pQueryInfo->exprList, i);
|
|
||||||
|
|
||||||
p->offset = prev->offset + prev->resBytes;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
void tscFieldInfoUpdateOffsetForInterResult(SQueryInfo* pQueryInfo) {
|
|
||||||
if (tscSqlExprNumOfExprs(pQueryInfo) == 0) {
|
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
SSqlExpr* pExpr = taosArrayGetP(pQueryInfo->exprList, 0);
|
|
||||||
pExpr->offset = 0;
|
|
||||||
|
|
||||||
size_t numOfExprs = tscSqlExprNumOfExprs(pQueryInfo);
|
|
||||||
for (int32_t i = 1; i < numOfExprs; ++i) {
|
for (int32_t i = 1; i < numOfExprs; ++i) {
|
||||||
SSqlExpr* prev = taosArrayGetP(pQueryInfo->exprList, i - 1);
|
SSqlExpr* prev = taosArrayGetP(pQueryInfo->exprList, i - 1);
|
||||||
SSqlExpr* p = taosArrayGetP(pQueryInfo->exprList, i);
|
SSqlExpr* p = taosArrayGetP(pQueryInfo->exprList, i);
|
||||||
|
|
||||||
p->offset = prev->offset + prev->resBytes;
|
p->offset = prev->offset + prev->resBytes;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -944,6 +945,14 @@ void tscFieldInfoClear(SFieldInfo* pFieldInfo) {
|
||||||
|
|
||||||
if (pInfo->pArithExprInfo != NULL) {
|
if (pInfo->pArithExprInfo != NULL) {
|
||||||
tExprTreeDestroy(&pInfo->pArithExprInfo->pExpr, NULL);
|
tExprTreeDestroy(&pInfo->pArithExprInfo->pExpr, NULL);
|
||||||
|
|
||||||
|
SSqlFuncMsg* pFuncMsg = &pInfo->pArithExprInfo->base;
|
||||||
|
for(int32_t j = 0; j < pFuncMsg->numOfParams; ++j) {
|
||||||
|
if (pFuncMsg->arg[j].argType == TSDB_DATA_TYPE_BINARY) {
|
||||||
|
tfree(pFuncMsg->arg[j].argValue.pz);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
tfree(pInfo->pArithExprInfo);
|
tfree(pInfo->pArithExprInfo);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -955,7 +964,7 @@ void tscFieldInfoClear(SFieldInfo* pFieldInfo) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static SSqlExpr* doBuildSqlExpr(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
static SSqlExpr* doBuildSqlExpr(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
||||||
int16_t size, int16_t interSize, int32_t colType) {
|
int16_t size, int16_t resColId, int16_t interSize, int32_t colType) {
|
||||||
STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, pColIndex->tableIndex);
|
STableMetaInfo* pTableMetaInfo = tscGetMetaInfo(pQueryInfo, pColIndex->tableIndex);
|
||||||
|
|
||||||
SSqlExpr* pExpr = calloc(1, sizeof(SSqlExpr));
|
SSqlExpr* pExpr = calloc(1, sizeof(SSqlExpr));
|
||||||
|
@ -988,8 +997,9 @@ static SSqlExpr* doBuildSqlExpr(SQueryInfo* pQueryInfo, int16_t functionId, SCol
|
||||||
|
|
||||||
pExpr->resType = type;
|
pExpr->resType = type;
|
||||||
pExpr->resBytes = size;
|
pExpr->resBytes = size;
|
||||||
|
pExpr->resColId = resColId;
|
||||||
pExpr->interBytes = interSize;
|
pExpr->interBytes = interSize;
|
||||||
|
|
||||||
if (pTableMetaInfo->pTableMeta) {
|
if (pTableMetaInfo->pTableMeta) {
|
||||||
pExpr->uid = pTableMetaInfo->pTableMeta->id.uid;
|
pExpr->uid = pTableMetaInfo->pTableMeta->id.uid;
|
||||||
}
|
}
|
||||||
|
@ -998,20 +1008,20 @@ static SSqlExpr* doBuildSqlExpr(SQueryInfo* pQueryInfo, int16_t functionId, SCol
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* tscSqlExprInsert(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
SSqlExpr* tscSqlExprInsert(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
||||||
int16_t size, int16_t interSize, bool isTagCol) {
|
int16_t size, int16_t resColId, int16_t interSize, bool isTagCol) {
|
||||||
int32_t num = (int32_t)taosArrayGetSize(pQueryInfo->exprList);
|
int32_t num = (int32_t)taosArrayGetSize(pQueryInfo->exprList);
|
||||||
if (index == num) {
|
if (index == num) {
|
||||||
return tscSqlExprAppend(pQueryInfo, functionId, pColIndex, type, size, interSize, isTagCol);
|
return tscSqlExprAppend(pQueryInfo, functionId, pColIndex, type, size, resColId, interSize, isTagCol);
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* pExpr = doBuildSqlExpr(pQueryInfo, functionId, pColIndex, type, size, interSize, isTagCol);
|
SSqlExpr* pExpr = doBuildSqlExpr(pQueryInfo, functionId, pColIndex, type, size, resColId, interSize, isTagCol);
|
||||||
taosArrayInsert(pQueryInfo->exprList, index, &pExpr);
|
taosArrayInsert(pQueryInfo->exprList, index, &pExpr);
|
||||||
return pExpr;
|
return pExpr;
|
||||||
}
|
}
|
||||||
|
|
||||||
SSqlExpr* tscSqlExprAppend(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
SSqlExpr* tscSqlExprAppend(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
|
||||||
int16_t size, int16_t interSize, bool isTagCol) {
|
int16_t size, int16_t resColId, int16_t interSize, bool isTagCol) {
|
||||||
SSqlExpr* pExpr = doBuildSqlExpr(pQueryInfo, functionId, pColIndex, type, size, interSize, isTagCol);
|
SSqlExpr* pExpr = doBuildSqlExpr(pQueryInfo, functionId, pColIndex, type, size, resColId, interSize, isTagCol);
|
||||||
taosArrayPush(pQueryInfo->exprList, &pExpr);
|
taosArrayPush(pQueryInfo->exprList, &pExpr);
|
||||||
return pExpr;
|
return pExpr;
|
||||||
}
|
}
|
||||||
|
@ -1039,16 +1049,14 @@ size_t tscSqlExprNumOfExprs(SQueryInfo* pQueryInfo) {
|
||||||
return taosArrayGetSize(pQueryInfo->exprList);
|
return taosArrayGetSize(pQueryInfo->exprList);
|
||||||
}
|
}
|
||||||
|
|
||||||
void addExprParams(SSqlExpr* pExpr, char* argument, int32_t type, int32_t bytes, int16_t tableIndex) {
|
void addExprParams(SSqlExpr* pExpr, char* argument, int32_t type, int32_t bytes) {
|
||||||
if (pExpr == NULL || argument == NULL || bytes == 0) {
|
assert (pExpr != NULL || argument != NULL || bytes != 0);
|
||||||
return;
|
|
||||||
}
|
|
||||||
|
|
||||||
// set parameter value
|
// set parameter value
|
||||||
// transfer to tVariant from byte data/no ascii data
|
// transfer to tVariant from byte data/no ascii data
|
||||||
tVariantCreateFromBinary(&pExpr->param[pExpr->numOfParams], argument, bytes, type);
|
tVariantCreateFromBinary(&pExpr->param[pExpr->numOfParams], argument, bytes, type);
|
||||||
|
|
||||||
pExpr->numOfParams += 1;
|
pExpr->numOfParams += 1;
|
||||||
|
|
||||||
assert(pExpr->numOfParams <= 3);
|
assert(pExpr->numOfParams <= 3);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -1601,6 +1609,7 @@ void tscInitQueryInfo(SQueryInfo* pQueryInfo) {
|
||||||
pQueryInfo->exprList = taosArrayInit(4, POINTER_BYTES);
|
pQueryInfo->exprList = taosArrayInit(4, POINTER_BYTES);
|
||||||
pQueryInfo->colList = taosArrayInit(4, POINTER_BYTES);
|
pQueryInfo->colList = taosArrayInit(4, POINTER_BYTES);
|
||||||
pQueryInfo->udColumnId = TSDB_UD_COLUMN_INDEX;
|
pQueryInfo->udColumnId = TSDB_UD_COLUMN_INDEX;
|
||||||
|
pQueryInfo->resColumnId= -1000;
|
||||||
}
|
}
|
||||||
|
|
||||||
int32_t tscAddSubqueryInfo(SSqlCmd* pCmd) {
|
int32_t tscAddSubqueryInfo(SSqlCmd* pCmd) {
|
||||||
|
|
|
@ -197,7 +197,8 @@ public class TSDBConnection implements Connection {
|
||||||
|
|
||||||
public SQLWarning getWarnings() throws SQLException {
|
public SQLWarning getWarnings() throws SQLException {
|
||||||
//todo: implement getWarnings according to the warning messages returned from TDengine
|
//todo: implement getWarnings according to the warning messages returned from TDengine
|
||||||
throw new SQLException(TSDBConstants.UNSUPPORT_METHOD_EXCEPTIONZ_MSG);
|
return null;
|
||||||
|
// throw new SQLException(TSDBConstants.UNSUPPORT_METHOD_EXCEPTIONZ_MSG);
|
||||||
}
|
}
|
||||||
|
|
||||||
public void clearWarnings() throws SQLException {
|
public void clearWarnings() throws SQLException {
|
||||||
|
|
|
@ -392,6 +392,7 @@ typedef struct SColIndex {
|
||||||
typedef struct SSqlFuncMsg {
|
typedef struct SSqlFuncMsg {
|
||||||
int16_t functionId;
|
int16_t functionId;
|
||||||
int16_t numOfParams;
|
int16_t numOfParams;
|
||||||
|
int16_t resColId; // result column id, id of the current output column
|
||||||
|
|
||||||
SColIndex colInfo;
|
SColIndex colInfo;
|
||||||
struct ArgElem {
|
struct ArgElem {
|
||||||
|
@ -461,11 +462,6 @@ typedef struct STimeWindow {
|
||||||
TSKEY ekey;
|
TSKEY ekey;
|
||||||
} STimeWindow;
|
} STimeWindow;
|
||||||
|
|
||||||
/*
|
|
||||||
* the outputCols is equalled to or larger than numOfCols
|
|
||||||
* e.g., select min(colName), max(colName), avg(colName) from table
|
|
||||||
* the outputCols will be 3 while the numOfCols is 1.
|
|
||||||
*/
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
SMsgHead head;
|
SMsgHead head;
|
||||||
STimeWindow window;
|
STimeWindow window;
|
||||||
|
@ -483,13 +479,14 @@ typedef struct {
|
||||||
uint32_t queryType; // denote another query process
|
uint32_t queryType; // denote another query process
|
||||||
int16_t numOfOutput; // final output columns numbers
|
int16_t numOfOutput; // final output columns numbers
|
||||||
int16_t tagNameRelType; // relation of tag criteria and tbname criteria
|
int16_t tagNameRelType; // relation of tag criteria and tbname criteria
|
||||||
int16_t fillType; // interpolate type
|
int16_t fillType; // interpolate type
|
||||||
uint64_t fillVal; // default value array list
|
uint64_t fillVal; // default value array list
|
||||||
int32_t tsOffset; // offset value in current msg body, NOTE: ts list is compressed
|
int32_t secondStageOutput;
|
||||||
int32_t tsLen; // total length of ts comp block
|
int32_t tsOffset; // offset value in current msg body, NOTE: ts list is compressed
|
||||||
int32_t tsNumOfBlocks; // ts comp block numbers
|
int32_t tsLen; // total length of ts comp block
|
||||||
int32_t tsOrder; // ts comp block order
|
int32_t tsNumOfBlocks; // ts comp block numbers
|
||||||
int32_t numOfTags; // number of tags columns involved
|
int32_t tsOrder; // ts comp block order
|
||||||
|
int32_t numOfTags; // number of tags columns involved
|
||||||
SColumnInfo colList[];
|
SColumnInfo colList[];
|
||||||
} SQueryTableMsg;
|
} SQueryTableMsg;
|
||||||
|
|
||||||
|
|
|
@ -40,19 +40,22 @@ typedef struct {
|
||||||
enum _show_db_index {
|
enum _show_db_index {
|
||||||
TSDB_SHOW_DB_NAME_INDEX,
|
TSDB_SHOW_DB_NAME_INDEX,
|
||||||
TSDB_SHOW_DB_CREATED_TIME_INDEX,
|
TSDB_SHOW_DB_CREATED_TIME_INDEX,
|
||||||
TSDB_SHOW_DB_VGROUPS_INDEX,
|
|
||||||
TSDB_SHOW_DB_NTABLES_INDEX,
|
TSDB_SHOW_DB_NTABLES_INDEX,
|
||||||
|
TSDB_SHOW_DB_VGROUPS_INDEX,
|
||||||
TSDB_SHOW_DB_REPLICA_INDEX,
|
TSDB_SHOW_DB_REPLICA_INDEX,
|
||||||
|
TSDB_SHOW_DB_QUORUM_INDEX,
|
||||||
TSDB_SHOW_DB_DAYS_INDEX,
|
TSDB_SHOW_DB_DAYS_INDEX,
|
||||||
TSDB_SHOW_DB_KEEP_INDEX,
|
TSDB_SHOW_DB_KEEP_INDEX,
|
||||||
TSDB_SHOW_DB_TABLES_INDEX,
|
|
||||||
TSDB_SHOW_DB_ROWS_INDEX,
|
|
||||||
TSDB_SHOW_DB_CACHE_INDEX,
|
TSDB_SHOW_DB_CACHE_INDEX,
|
||||||
TSDB_SHOW_DB_ABLOCKS_INDEX,
|
TSDB_SHOW_DB_BLOCKS_INDEX,
|
||||||
TSDB_SHOW_DB_TBLOCKS_INDEX,
|
TSDB_SHOW_DB_MINROWS_INDEX,
|
||||||
TSDB_SHOW_DB_CTIME_INDEX,
|
TSDB_SHOW_DB_MAXROWS_INDEX,
|
||||||
TSDB_SHOW_DB_CLOG_INDEX,
|
TSDB_SHOW_DB_WALLEVEL_INDEX,
|
||||||
|
TSDB_SHOW_DB_FSYNC_INDEX,
|
||||||
TSDB_SHOW_DB_COMP_INDEX,
|
TSDB_SHOW_DB_COMP_INDEX,
|
||||||
|
TSDB_SHOW_DB_PRECISION_INDEX,
|
||||||
|
TSDB_SHOW_DB_UPDATE_INDEX,
|
||||||
|
TSDB_SHOW_DB_STATUS_INDEX,
|
||||||
TSDB_MAX_SHOW_DB
|
TSDB_MAX_SHOW_DB
|
||||||
};
|
};
|
||||||
|
|
||||||
|
@ -90,17 +93,23 @@ extern char version[];
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
char name[TSDB_DB_NAME_LEN + 1];
|
char name[TSDB_DB_NAME_LEN + 1];
|
||||||
int32_t replica;
|
int32_t tables;
|
||||||
int32_t days;
|
int32_t vgroups;
|
||||||
int32_t keep;
|
int16_t replications;
|
||||||
int32_t tables;
|
int16_t quorum;
|
||||||
int32_t rows;
|
int16_t daysPerFile;
|
||||||
int32_t cache;
|
int16_t daysToKeep;
|
||||||
int32_t ablocks;
|
int16_t daysToKeep1;
|
||||||
int32_t tblocks;
|
int16_t daysToKeep2;
|
||||||
int32_t ctime;
|
int32_t cacheBlockSize; //MB
|
||||||
int32_t clog;
|
int32_t totalBlocks;
|
||||||
int32_t comp;
|
int32_t minRowsPerFileBlock;
|
||||||
|
int32_t maxRowsPerFileBlock;
|
||||||
|
int8_t walLevel;
|
||||||
|
int32_t fsyncPeriod;
|
||||||
|
int8_t compression;
|
||||||
|
int8_t precision; // time resolution
|
||||||
|
int8_t update;
|
||||||
} SDbInfo;
|
} SDbInfo;
|
||||||
|
|
||||||
typedef struct {
|
typedef struct {
|
||||||
|
@ -173,6 +182,7 @@ static struct argp_option options[] = {
|
||||||
{"start-time", 'S', "START_TIME", 0, "Start time to dump.", 3},
|
{"start-time", 'S', "START_TIME", 0, "Start time to dump.", 3},
|
||||||
{"end-time", 'E', "END_TIME", 0, "End time to dump.", 3},
|
{"end-time", 'E', "END_TIME", 0, "End time to dump.", 3},
|
||||||
{"data-batch", 'N', "DATA_BATCH", 0, "Number of data point per insert statement. Default is 1.", 3},
|
{"data-batch", 'N', "DATA_BATCH", 0, "Number of data point per insert statement. Default is 1.", 3},
|
||||||
|
{"max-sql-len", 'L', "SQL_LEN", 0, "Max length of one sql. Default is 65480.", 3},
|
||||||
{"table-batch", 't', "TABLE_BATCH", 0, "Number of table dumpout into one output file. Default is 1.", 3},
|
{"table-batch", 't', "TABLE_BATCH", 0, "Number of table dumpout into one output file. Default is 1.", 3},
|
||||||
{"thread_num", 'T', "THREAD_NUM", 0, "Number of thread for dump in file. Default is 5.", 3},
|
{"thread_num", 'T', "THREAD_NUM", 0, "Number of thread for dump in file. Default is 5.", 3},
|
||||||
{"allow-sys", 'a', 0, 0, "Allow to dump sys database", 3},
|
{"allow-sys", 'a', 0, 0, "Allow to dump sys database", 3},
|
||||||
|
@ -200,6 +210,7 @@ struct arguments {
|
||||||
int64_t start_time;
|
int64_t start_time;
|
||||||
int64_t end_time;
|
int64_t end_time;
|
||||||
int32_t data_batch;
|
int32_t data_batch;
|
||||||
|
int32_t max_sql_len;
|
||||||
int32_t table_batch; // num of table which will be dump into one output file.
|
int32_t table_batch; // num of table which will be dump into one output file.
|
||||||
bool allow_sys;
|
bool allow_sys;
|
||||||
// other options
|
// other options
|
||||||
|
@ -298,6 +309,17 @@ static error_t parse_opt(int key, char *arg, struct argp_state *state) {
|
||||||
case 'N':
|
case 'N':
|
||||||
arguments->data_batch = atoi(arg);
|
arguments->data_batch = atoi(arg);
|
||||||
break;
|
break;
|
||||||
|
case 'L':
|
||||||
|
{
|
||||||
|
int32_t len = atoi(arg);
|
||||||
|
if (len > TSDB_MAX_ALLOWED_SQL_LEN) {
|
||||||
|
len = TSDB_MAX_ALLOWED_SQL_LEN;
|
||||||
|
} else if (len < TSDB_MAX_SQL_LEN) {
|
||||||
|
len = TSDB_MAX_SQL_LEN;
|
||||||
|
}
|
||||||
|
arguments->max_sql_len = len;
|
||||||
|
break;
|
||||||
|
}
|
||||||
case 't':
|
case 't':
|
||||||
arguments->table_batch = atoi(arg);
|
arguments->table_batch = atoi(arg);
|
||||||
break;
|
break;
|
||||||
|
@ -360,6 +382,7 @@ struct arguments tsArguments = {
|
||||||
0,
|
0,
|
||||||
INT64_MAX,
|
INT64_MAX,
|
||||||
1,
|
1,
|
||||||
|
TSDB_MAX_SQL_LEN,
|
||||||
1,
|
1,
|
||||||
false,
|
false,
|
||||||
// other options
|
// other options
|
||||||
|
@ -415,7 +438,9 @@ int main(int argc, char *argv[]) {
|
||||||
printf("start_time: %" PRId64 "\n", tsArguments.start_time);
|
printf("start_time: %" PRId64 "\n", tsArguments.start_time);
|
||||||
printf("end_time: %" PRId64 "\n", tsArguments.end_time);
|
printf("end_time: %" PRId64 "\n", tsArguments.end_time);
|
||||||
printf("data_batch: %d\n", tsArguments.data_batch);
|
printf("data_batch: %d\n", tsArguments.data_batch);
|
||||||
|
printf("max_sql_len: %d\n", tsArguments.max_sql_len);
|
||||||
printf("table_batch: %d\n", tsArguments.table_batch);
|
printf("table_batch: %d\n", tsArguments.table_batch);
|
||||||
|
printf("thread_num: %d\n", tsArguments.thread_num);
|
||||||
printf("allow_sys: %d\n", tsArguments.allow_sys);
|
printf("allow_sys: %d\n", tsArguments.allow_sys);
|
||||||
printf("abort: %d\n", tsArguments.abort);
|
printf("abort: %d\n", tsArguments.abort);
|
||||||
printf("isDumpIn: %d\n", tsArguments.isDumpIn);
|
printf("isDumpIn: %d\n", tsArguments.isDumpIn);
|
||||||
|
@ -682,8 +707,8 @@ int taosDumpOut(struct arguments *arguments) {
|
||||||
TAOS_FIELD *fields = taos_fetch_fields(result);
|
TAOS_FIELD *fields = taos_fetch_fields(result);
|
||||||
|
|
||||||
while ((row = taos_fetch_row(result)) != NULL) {
|
while ((row = taos_fetch_row(result)) != NULL) {
|
||||||
// sys database name : 'monitor', but subsequent version changed to 'log'
|
// sys database name : 'log', but subsequent version changed to 'log'
|
||||||
if (strncasecmp(row[TSDB_SHOW_DB_NAME_INDEX], "monitor", fields[TSDB_SHOW_DB_NAME_INDEX].bytes) == 0 &&
|
if (strncasecmp(row[TSDB_SHOW_DB_NAME_INDEX], "log", fields[TSDB_SHOW_DB_NAME_INDEX].bytes) == 0 &&
|
||||||
(!arguments->allow_sys))
|
(!arguments->allow_sys))
|
||||||
continue;
|
continue;
|
||||||
|
|
||||||
|
@ -711,20 +736,27 @@ int taosDumpOut(struct arguments *arguments) {
|
||||||
}
|
}
|
||||||
|
|
||||||
strncpy(dbInfos[count]->name, (char *)row[TSDB_SHOW_DB_NAME_INDEX], fields[TSDB_SHOW_DB_NAME_INDEX].bytes);
|
strncpy(dbInfos[count]->name, (char *)row[TSDB_SHOW_DB_NAME_INDEX], fields[TSDB_SHOW_DB_NAME_INDEX].bytes);
|
||||||
#if 0
|
#if 0
|
||||||
dbInfos[count]->replica = (int)(*((int16_t *)row[TSDB_SHOW_DB_REPLICA_INDEX]));
|
if (arguments->with_property) {
|
||||||
dbInfos[count]->days = (int)(*((int16_t *)row[TSDB_SHOW_DB_DAYS_INDEX]));
|
dbInfos[count]->tables = *((int32_t *)row[TSDB_SHOW_DB_NTABLES_INDEX]);
|
||||||
dbInfos[count]->keep = *((int *)row[TSDB_SHOW_DB_KEEP_INDEX]);
|
dbInfos[count]->vgroups = *((int32_t *)row[TSDB_SHOW_DB_VGROUPS_INDEX]);
|
||||||
dbInfos[count]->tables = *((int *)row[TSDB_SHOW_DB_TABLES_INDEX]);
|
dbInfos[count]->replications = *((int16_t *)row[TSDB_SHOW_DB_REPLICA_INDEX]);
|
||||||
dbInfos[count]->rows = *((int *)row[TSDB_SHOW_DB_ROWS_INDEX]);
|
dbInfos[count]->quorum = *((int16_t *)row[TSDB_SHOW_DB_QUORUM_INDEX]);
|
||||||
dbInfos[count]->cache = *((int *)row[TSDB_SHOW_DB_CACHE_INDEX]);
|
dbInfos[count]->daysPerFile = *((int16_t *)row[TSDB_SHOW_DB_DAYS_INDEX]);
|
||||||
dbInfos[count]->ablocks = *((int *)row[TSDB_SHOW_DB_ABLOCKS_INDEX]);
|
dbInfos[count]->daysToKeep = *((int16_t *)row[TSDB_SHOW_DB_KEEP_INDEX]);
|
||||||
dbInfos[count]->tblocks = (int)(*((int16_t *)row[TSDB_SHOW_DB_TBLOCKS_INDEX]));
|
dbInfos[count]->daysToKeep1;
|
||||||
dbInfos[count]->ctime = *((int *)row[TSDB_SHOW_DB_CTIME_INDEX]);
|
dbInfos[count]->daysToKeep2;
|
||||||
dbInfos[count]->clog = (int)(*((int8_t *)row[TSDB_SHOW_DB_CLOG_INDEX]));
|
dbInfos[count]->cacheBlockSize = *((int32_t *)row[TSDB_SHOW_DB_CACHE_INDEX]);
|
||||||
dbInfos[count]->comp = (int)(*((int8_t *)row[TSDB_SHOW_DB_COMP_INDEX]));
|
dbInfos[count]->totalBlocks = *((int32_t *)row[TSDB_SHOW_DB_BLOCKS_INDEX]);
|
||||||
|
dbInfos[count]->minRowsPerFileBlock = *((int32_t *)row[TSDB_SHOW_DB_MINROWS_INDEX]);
|
||||||
|
dbInfos[count]->maxRowsPerFileBlock = *((int32_t *)row[TSDB_SHOW_DB_MAXROWS_INDEX]);
|
||||||
|
dbInfos[count]->walLevel = *((int8_t *)row[TSDB_SHOW_DB_WALLEVEL_INDEX]);
|
||||||
|
dbInfos[count]->fsyncPeriod = *((int32_t *)row[TSDB_SHOW_DB_FSYNC_INDEX]);
|
||||||
|
dbInfos[count]->compression = (int8_t)(*((int8_t *)row[TSDB_SHOW_DB_COMP_INDEX]));
|
||||||
|
dbInfos[count]->precision = *((int8_t *)row[TSDB_SHOW_DB_PRECISION_INDEX]);
|
||||||
|
dbInfos[count]->update = *((int8_t *)row[TSDB_SHOW_DB_UPDATE_INDEX]);
|
||||||
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
count++;
|
count++;
|
||||||
|
|
||||||
if (arguments->databases) {
|
if (arguments->databases) {
|
||||||
|
@ -1037,10 +1069,13 @@ void taosDumpCreateDbClause(SDbInfo *dbInfo, bool isDumpProperty, FILE *fp) {
|
||||||
|
|
||||||
pstr += sprintf(pstr, "CREATE DATABASE IF NOT EXISTS %s", dbInfo->name);
|
pstr += sprintf(pstr, "CREATE DATABASE IF NOT EXISTS %s", dbInfo->name);
|
||||||
if (isDumpProperty) {
|
if (isDumpProperty) {
|
||||||
|
#if 0
|
||||||
pstr += sprintf(pstr,
|
pstr += sprintf(pstr,
|
||||||
" REPLICA %d DAYS %d KEEP %d TABLES %d ROWS %d CACHE %d ABLOCKS %d TBLOCKS %d CTIME %d CLOG %d COMP %d",
|
"TABLES %d vgroups %d REPLICA %d quorum %d DAYS %d KEEP %d CACHE %d BLOCKS %d MINROWS %d MAXROWS %d WALLEVEL %d FYNC %d COMP %d PRECISION %s UPDATE %d",
|
||||||
dbInfo->replica, dbInfo->days, dbInfo->keep, dbInfo->tables, dbInfo->rows, dbInfo->cache,
|
dbInfo->tables, dbInfo->vgroups, dbInfo->replications, dbInfo->quorum, dbInfo->daysPerFile, dbInfo->daysToKeep, dbInfo->cacheBlockSize,
|
||||||
dbInfo->ablocks, dbInfo->tblocks, dbInfo->ctime, dbInfo->clog, dbInfo->comp);
|
dbInfo->totalBlocks, dbInfo->minRowsPerFileBlock, dbInfo->maxRowsPerFileBlock, dbInfo->walLevel, dbInfo->fsyncPeriod, dbInfo->compression,
|
||||||
|
dbInfo->precision, dbInfo->update);
|
||||||
|
#endif
|
||||||
}
|
}
|
||||||
|
|
||||||
pstr += sprintf(pstr, ";");
|
pstr += sprintf(pstr, ";");
|
||||||
|
@ -1459,7 +1494,8 @@ int taosDumpTableData(FILE *fp, char *tbname, struct arguments *arguments, TAOS*
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
char* tmpBuffer = (char *)calloc(1, COMMAND_SIZE);
|
int32_t sql_buf_len = arguments->max_sql_len;
|
||||||
|
char* tmpBuffer = (char *)calloc(1, sql_buf_len + 128);
|
||||||
if (tmpBuffer == NULL) {
|
if (tmpBuffer == NULL) {
|
||||||
fprintf(stderr, "failed to allocate memory\n");
|
fprintf(stderr, "failed to allocate memory\n");
|
||||||
free(tmpCommand);
|
free(tmpCommand);
|
||||||
|
@ -1502,85 +1538,83 @@ int taosDumpTableData(FILE *fp, char *tbname, struct arguments *arguments, TAOS*
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
char sqlStr[8] = "\0";
|
|
||||||
if (arguments->mysqlFlag) {
|
|
||||||
sprintf(sqlStr, "INSERT");
|
|
||||||
} else {
|
|
||||||
sprintf(sqlStr, "IMPORT");
|
|
||||||
}
|
|
||||||
|
|
||||||
int rowFlag = 0;
|
int rowFlag = 0;
|
||||||
|
int32_t curr_sqlstr_len = 0;
|
||||||
|
int32_t total_sqlstr_len = 0;
|
||||||
count = 0;
|
count = 0;
|
||||||
while ((row = taos_fetch_row(tmpResult)) != NULL) {
|
while ((row = taos_fetch_row(tmpResult)) != NULL) {
|
||||||
pstr = tmpBuffer;
|
pstr = tmpBuffer;
|
||||||
|
curr_sqlstr_len = 0;
|
||||||
|
|
||||||
int32_t* length = taos_fetch_lengths(tmpResult); // act len
|
int32_t* length = taos_fetch_lengths(tmpResult); // act len
|
||||||
|
|
||||||
if (count == 0) {
|
if (count == 0) {
|
||||||
pstr += sprintf(pstr, "%s INTO %s VALUES (", sqlStr, tbname);
|
total_sqlstr_len = 0;
|
||||||
} else {
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "INSERT INTO %s VALUES (", tbname);
|
||||||
|
} else {
|
||||||
if (arguments->mysqlFlag) {
|
if (arguments->mysqlFlag) {
|
||||||
if (0 == rowFlag) {
|
if (0 == rowFlag) {
|
||||||
pstr += sprintf(pstr, "(");
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "(");
|
||||||
rowFlag++;
|
rowFlag++;
|
||||||
} else {
|
} else {
|
||||||
pstr += sprintf(pstr, ", (");
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, ", (");
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
pstr += sprintf(pstr, "(");
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "(");
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
for (int col = 0; col < numFields; col++) {
|
for (int col = 0; col < numFields; col++) {
|
||||||
if (col != 0) pstr += sprintf(pstr, ", ");
|
if (col != 0) curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, ", ");
|
||||||
|
|
||||||
if (row[col] == NULL) {
|
if (row[col] == NULL) {
|
||||||
pstr += sprintf(pstr, "NULL");
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "NULL");
|
||||||
continue;
|
continue;
|
||||||
}
|
}
|
||||||
|
|
||||||
switch (fields[col].type) {
|
switch (fields[col].type) {
|
||||||
case TSDB_DATA_TYPE_BOOL:
|
case TSDB_DATA_TYPE_BOOL:
|
||||||
pstr += sprintf(pstr, "%d", ((((int32_t)(*((char *)row[col]))) == 1) ? 1 : 0));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%d", ((((int32_t)(*((char *)row[col]))) == 1) ? 1 : 0));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_TINYINT:
|
case TSDB_DATA_TYPE_TINYINT:
|
||||||
pstr += sprintf(pstr, "%d", *((int8_t *)row[col]));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%d", *((int8_t *)row[col]));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_SMALLINT:
|
case TSDB_DATA_TYPE_SMALLINT:
|
||||||
pstr += sprintf(pstr, "%d", *((int16_t *)row[col]));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%d", *((int16_t *)row[col]));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_INT:
|
case TSDB_DATA_TYPE_INT:
|
||||||
pstr += sprintf(pstr, "%d", *((int32_t *)row[col]));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%d", *((int32_t *)row[col]));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_BIGINT:
|
case TSDB_DATA_TYPE_BIGINT:
|
||||||
pstr += sprintf(pstr, "%" PRId64 "", *((int64_t *)row[col]));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%" PRId64 "", *((int64_t *)row[col]));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_FLOAT:
|
case TSDB_DATA_TYPE_FLOAT:
|
||||||
pstr += sprintf(pstr, "%f", GET_FLOAT_VAL(row[col]));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%f", GET_FLOAT_VAL(row[col]));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_DOUBLE:
|
case TSDB_DATA_TYPE_DOUBLE:
|
||||||
pstr += sprintf(pstr, "%f", GET_DOUBLE_VAL(row[col]));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%f", GET_DOUBLE_VAL(row[col]));
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_BINARY:
|
case TSDB_DATA_TYPE_BINARY:
|
||||||
*(pstr++) = '\'';
|
//*(pstr++) = '\'';
|
||||||
converStringToReadable((char *)row[col], length[col], tbuf, COMMAND_SIZE);
|
converStringToReadable((char *)row[col], length[col], tbuf, COMMAND_SIZE);
|
||||||
pstr = stpcpy(pstr, tbuf);
|
//pstr = stpcpy(pstr, tbuf);
|
||||||
*(pstr++) = '\'';
|
//*(pstr++) = '\'';
|
||||||
|
pstr += sprintf(pstr + curr_sqlstr_len, "\'%s\'", tbuf);
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_NCHAR:
|
case TSDB_DATA_TYPE_NCHAR:
|
||||||
convertNCharToReadable((char *)row[col], length[col], tbuf, COMMAND_SIZE);
|
convertNCharToReadable((char *)row[col], length[col], tbuf, COMMAND_SIZE);
|
||||||
pstr += sprintf(pstr, "\'%s\'", tbuf);
|
pstr += sprintf(pstr + curr_sqlstr_len, "\'%s\'", tbuf);
|
||||||
break;
|
break;
|
||||||
case TSDB_DATA_TYPE_TIMESTAMP:
|
case TSDB_DATA_TYPE_TIMESTAMP:
|
||||||
if (!arguments->mysqlFlag) {
|
if (!arguments->mysqlFlag) {
|
||||||
pstr += sprintf(pstr, "%" PRId64 "", *(int64_t *)row[col]);
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "%" PRId64 "", *(int64_t *)row[col]);
|
||||||
} else {
|
} else {
|
||||||
char buf[64] = "\0";
|
char buf[64] = "\0";
|
||||||
int64_t ts = *((int64_t *)row[col]);
|
int64_t ts = *((int64_t *)row[col]);
|
||||||
time_t tt = (time_t)(ts / 1000);
|
time_t tt = (time_t)(ts / 1000);
|
||||||
struct tm *ptm = localtime(&tt);
|
struct tm *ptm = localtime(&tt);
|
||||||
strftime(buf, 64, "%y-%m-%d %H:%M:%S", ptm);
|
strftime(buf, 64, "%y-%m-%d %H:%M:%S", ptm);
|
||||||
pstr += sprintf(pstr, "\'%s.%03d\'", buf, (int)(ts % 1000));
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, "\'%s.%03d\'", buf, (int)(ts % 1000));
|
||||||
}
|
}
|
||||||
break;
|
break;
|
||||||
default:
|
default:
|
||||||
|
@ -1588,13 +1622,15 @@ int taosDumpTableData(FILE *fp, char *tbname, struct arguments *arguments, TAOS*
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
pstr += sprintf(pstr, ") ");
|
curr_sqlstr_len += sprintf(pstr + curr_sqlstr_len, ") ");
|
||||||
|
|
||||||
totalRows++;
|
totalRows++;
|
||||||
count++;
|
count++;
|
||||||
fprintf(fp, "%s", tmpBuffer);
|
fprintf(fp, "%s", tmpBuffer);
|
||||||
|
|
||||||
if (count >= arguments->data_batch) {
|
total_sqlstr_len += curr_sqlstr_len;
|
||||||
|
|
||||||
|
if ((count >= arguments->data_batch) || (sql_buf_len - total_sqlstr_len < TSDB_MAX_BYTES_PER_ROW)) {
|
||||||
fprintf(fp, ";\n");
|
fprintf(fp, ";\n");
|
||||||
count = 0;
|
count = 0;
|
||||||
} //else {
|
} //else {
|
||||||
|
|
|
@ -37,7 +37,7 @@ extern "C" {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifndef TAOS_OS_DEF_EPOLL
|
#ifndef TAOS_OS_DEF_EPOLL
|
||||||
#define TAOS_EPOLL_WAIT_TIME -1
|
#define TAOS_EPOLL_WAIT_TIME 500
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#ifdef TAOS_RANDOM_NETWORK_FAIL
|
#ifdef TAOS_RANDOM_NETWORK_FAIL
|
||||||
|
|
|
@ -152,7 +152,10 @@ typedef struct SQuery {
|
||||||
SLimitVal limit;
|
SLimitVal limit;
|
||||||
int32_t rowSize;
|
int32_t rowSize;
|
||||||
SSqlGroupbyExpr* pGroupbyExpr;
|
SSqlGroupbyExpr* pGroupbyExpr;
|
||||||
SExprInfo* pSelectExpr;
|
SExprInfo* pExpr1;
|
||||||
|
SExprInfo* pExpr2;
|
||||||
|
int32_t numOfExpr2;
|
||||||
|
|
||||||
SColumnInfo* colList;
|
SColumnInfo* colList;
|
||||||
SColumnInfo* tagColList;
|
SColumnInfo* tagColList;
|
||||||
int32_t numOfFilterCols;
|
int32_t numOfFilterCols;
|
||||||
|
|
|
@ -43,7 +43,8 @@ typedef struct SHistogramInfo {
|
||||||
int32_t numOfElems;
|
int32_t numOfElems;
|
||||||
int32_t numOfEntries;
|
int32_t numOfEntries;
|
||||||
int32_t maxEntries;
|
int32_t maxEntries;
|
||||||
|
double min;
|
||||||
|
double max;
|
||||||
#if defined(USE_ARRAYLIST)
|
#if defined(USE_ARRAYLIST)
|
||||||
SHistBin* elems;
|
SHistBin* elems;
|
||||||
#else
|
#else
|
||||||
|
@ -52,9 +53,6 @@ typedef struct SHistogramInfo {
|
||||||
int32_t maxIndex;
|
int32_t maxIndex;
|
||||||
bool ordered;
|
bool ordered;
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
double min;
|
|
||||||
double max;
|
|
||||||
} SHistogramInfo;
|
} SHistogramInfo;
|
||||||
|
|
||||||
SHistogramInfo* tHistogramCreate(int32_t numOfBins);
|
SHistogramInfo* tHistogramCreate(int32_t numOfBins);
|
||||||
|
|
|
@ -48,7 +48,7 @@ static FORCE_INLINE SResultRow *getResultRow(SWindowResInfo *pWindowResInfo, int
|
||||||
}
|
}
|
||||||
|
|
||||||
#define curTimeWindowIndex(_winres) ((_winres)->curIndex)
|
#define curTimeWindowIndex(_winres) ((_winres)->curIndex)
|
||||||
#define GET_ROW_PARAM_FOR_MULTIOUTPUT(_q, tbq, sq) (((tbq) && (!sq))? (_q)->pSelectExpr[1].base.arg->argValue.i64:1)
|
#define GET_ROW_PARAM_FOR_MULTIOUTPUT(_q, tbq, sq) (((tbq) && (!sq))? (_q)->pExpr1[1].base.arg->argValue.i64:1)
|
||||||
|
|
||||||
bool isWindowResClosed(SWindowResInfo *pWindowResInfo, int32_t slot);
|
bool isWindowResClosed(SWindowResInfo *pWindowResInfo, int32_t slot);
|
||||||
|
|
||||||
|
@ -62,7 +62,7 @@ static FORCE_INLINE char *getPosInResultPage(SQueryRuntimeEnv *pRuntimeEnv, int3
|
||||||
|
|
||||||
int32_t realRowId = (int32_t)(pResult->rowId * GET_ROW_PARAM_FOR_MULTIOUTPUT(pQuery, pRuntimeEnv->topBotQuery, pRuntimeEnv->stableQuery));
|
int32_t realRowId = (int32_t)(pResult->rowId * GET_ROW_PARAM_FOR_MULTIOUTPUT(pQuery, pRuntimeEnv->topBotQuery, pRuntimeEnv->stableQuery));
|
||||||
return ((char *)page->data) + pRuntimeEnv->offset[columnIndex] * pRuntimeEnv->numOfRowsPerPage +
|
return ((char *)page->data) + pRuntimeEnv->offset[columnIndex] * pRuntimeEnv->numOfRowsPerPage +
|
||||||
pQuery->pSelectExpr[columnIndex].bytes * realRowId;
|
pQuery->pExpr1[columnIndex].bytes * realRowId;
|
||||||
}
|
}
|
||||||
|
|
||||||
bool isNull_filter(SColumnFilterElem *pFilter, char* minval, char* maxval);
|
bool isNull_filter(SColumnFilterElem *pFilter, char* minval, char* maxval);
|
||||||
|
|
|
@ -128,7 +128,7 @@ typedef struct SArithmeticSupport {
|
||||||
SExprInfo *pArithExpr;
|
SExprInfo *pArithExpr;
|
||||||
int32_t numOfCols;
|
int32_t numOfCols;
|
||||||
SColumnInfo *colList;
|
SColumnInfo *colList;
|
||||||
SArray* exprList; // client side used
|
void *exprList; // client side used
|
||||||
int32_t offset;
|
int32_t offset;
|
||||||
char** data;
|
char** data;
|
||||||
} SArithmeticSupport;
|
} SArithmeticSupport;
|
||||||
|
|
File diff suppressed because it is too large
Load Diff
|
@ -225,7 +225,7 @@ tSQLExpr *tSQLExprCreate(tSQLExpr *pLeft, tSQLExpr *pRight, int32_t optrType) {
|
||||||
tSQLExprDestroy(pLeft);
|
tSQLExprDestroy(pLeft);
|
||||||
tSQLExprDestroy(pRight);
|
tSQLExprDestroy(pRight);
|
||||||
|
|
||||||
} else if (pLeft->nSQLOptr == TK_FLOAT || pRight->nSQLOptr == TK_FLOAT) {
|
} else if ((pLeft->nSQLOptr == TK_FLOAT && pRight->nSQLOptr == TK_INTEGER) || (pLeft->nSQLOptr == TK_INTEGER && pRight->nSQLOptr == TK_FLOAT)) {
|
||||||
pExpr->val.nType = TSDB_DATA_TYPE_DOUBLE;
|
pExpr->val.nType = TSDB_DATA_TYPE_DOUBLE;
|
||||||
pExpr->nSQLOptr = TK_FLOAT;
|
pExpr->nSQLOptr = TK_FLOAT;
|
||||||
|
|
||||||
|
|
|
@ -24,7 +24,7 @@ int32_t getOutputInterResultBufSize(SQuery* pQuery) {
|
||||||
int32_t size = 0;
|
int32_t size = 0;
|
||||||
|
|
||||||
for (int32_t i = 0; i < pQuery->numOfOutput; ++i) {
|
for (int32_t i = 0; i < pQuery->numOfOutput; ++i) {
|
||||||
size += pQuery->pSelectExpr[i].interBytes;
|
size += pQuery->pExpr1[i].interBytes;
|
||||||
}
|
}
|
||||||
|
|
||||||
assert(size >= 0);
|
assert(size >= 0);
|
||||||
|
@ -237,7 +237,7 @@ void clearResultRow(SQueryRuntimeEnv *pRuntimeEnv, SResultRow *pWindowRes) {
|
||||||
SResultRowCellInfo *pResultInfo = &pWindowRes->pCellInfo[i];
|
SResultRowCellInfo *pResultInfo = &pWindowRes->pCellInfo[i];
|
||||||
|
|
||||||
char * s = getPosInResultPage(pRuntimeEnv, i, pWindowRes, page);
|
char * s = getPosInResultPage(pRuntimeEnv, i, pWindowRes, page);
|
||||||
size_t size = pRuntimeEnv->pQuery->pSelectExpr[i].bytes;
|
size_t size = pRuntimeEnv->pQuery->pExpr1[i].bytes;
|
||||||
memset(s, 0, size);
|
memset(s, 0, size);
|
||||||
|
|
||||||
RESET_RESULT_INFO(pResultInfo);
|
RESET_RESULT_INFO(pResultInfo);
|
||||||
|
@ -280,7 +280,7 @@ void copyResultRow(SQueryRuntimeEnv *pRuntimeEnv, SResultRow *dst, const SResult
|
||||||
|
|
||||||
tFilePage *srcpage = getResBufPage(pRuntimeEnv->pResultBuf, src->pageId);
|
tFilePage *srcpage = getResBufPage(pRuntimeEnv->pResultBuf, src->pageId);
|
||||||
char * srcBuf = getPosInResultPage(pRuntimeEnv, i, (SResultRow *)src, srcpage);
|
char * srcBuf = getPosInResultPage(pRuntimeEnv, i, (SResultRow *)src, srcpage);
|
||||||
size_t s = pRuntimeEnv->pQuery->pSelectExpr[i].bytes;
|
size_t s = pRuntimeEnv->pQuery->pExpr1[i].bytes;
|
||||||
|
|
||||||
memcpy(dstBuf, srcBuf, s);
|
memcpy(dstBuf, srcBuf, s);
|
||||||
}
|
}
|
||||||
|
|
|
@ -171,40 +171,17 @@ void *taosInitTcpServer(uint32_t ip, uint16_t port, char *label, int numOfThread
|
||||||
}
|
}
|
||||||
|
|
||||||
static void taosStopTcpThread(SThreadObj* pThreadObj) {
|
static void taosStopTcpThread(SThreadObj* pThreadObj) {
|
||||||
pThreadObj->stop = true;
|
// save thread into local variable and signal thread to stop
|
||||||
eventfd_t fd = -1;
|
|
||||||
|
|
||||||
// save thread into local variable since pThreadObj is freed when thread exits
|
|
||||||
pthread_t thread = pThreadObj->thread;
|
pthread_t thread = pThreadObj->thread;
|
||||||
|
if (!taosCheckPthreadValid(thread)) {
|
||||||
if (taosComparePthread(pThreadObj->thread, pthread_self())) {
|
return;
|
||||||
|
}
|
||||||
|
pThreadObj->stop = true;
|
||||||
|
if (taosComparePthread(thread, pthread_self())) {
|
||||||
pthread_detach(pthread_self());
|
pthread_detach(pthread_self());
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
pthread_join(thread, NULL);
|
||||||
if (taosCheckPthreadValid(pThreadObj->thread)) {
|
|
||||||
// signal the thread to stop, try graceful method first,
|
|
||||||
// and use pthread_cancel when failed
|
|
||||||
struct epoll_event event = { .events = EPOLLIN };
|
|
||||||
fd = eventfd(1, 0);
|
|
||||||
if (fd == -1) {
|
|
||||||
// failed to create eventfd, call pthread_cancel instead, which may result in data corruption:
|
|
||||||
tError("%s, failed to create eventfd(%s)", pThreadObj->label, strerror(errno));
|
|
||||||
pThreadObj->stop = true;
|
|
||||||
pthread_cancel(pThreadObj->thread);
|
|
||||||
} else if (epoll_ctl(pThreadObj->pollFd, EPOLL_CTL_ADD, fd, &event) < 0) {
|
|
||||||
// failed to call epoll_ctl, call pthread_cancel instead, which may result in data corruption:
|
|
||||||
tError("%s, failed to call epoll_ctl(%s)", pThreadObj->label, strerror(errno));
|
|
||||||
pthread_cancel(pThreadObj->thread);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// at this step, pThreadObj has already been released
|
|
||||||
if (taosCheckPthreadValid(thread)) {
|
|
||||||
pthread_join(thread, NULL);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (fd != -1) taosCloseSocket(fd);
|
|
||||||
}
|
}
|
||||||
|
|
||||||
void taosStopTcpServer(void *handle) {
|
void taosStopTcpServer(void *handle) {
|
||||||
|
|
|
@ -301,31 +301,14 @@ static SThreadObj *taosGetTcpThread(SPoolObj *pPool) {
|
||||||
}
|
}
|
||||||
|
|
||||||
static void taosStopPoolThread(SThreadObj *pThread) {
|
static void taosStopPoolThread(SThreadObj *pThread) {
|
||||||
|
pthread_t thread = pThread->thread;
|
||||||
|
if (!taosCheckPthreadValid(thread)) {
|
||||||
|
return;
|
||||||
|
}
|
||||||
pThread->stop = true;
|
pThread->stop = true;
|
||||||
|
if (taosComparePthread(thread, pthread_self())) {
|
||||||
if (pThread->thread == pthread_self()) {
|
|
||||||
pthread_detach(pthread_self());
|
pthread_detach(pthread_self());
|
||||||
return;
|
return;
|
||||||
}
|
}
|
||||||
|
|
||||||
// save thread ID into a local variable, since pThread is freed when the thread exits
|
|
||||||
pthread_t thread = pThread->thread;
|
|
||||||
|
|
||||||
// signal the thread to stop, try graceful method first,
|
|
||||||
// and use pthread_cancel when failed
|
|
||||||
struct epoll_event event = {.events = EPOLLIN};
|
|
||||||
eventfd_t fd = eventfd(1, 0);
|
|
||||||
if (fd == -1) {
|
|
||||||
// failed to create eventfd, call pthread_cancel instead, which may result in data corruption
|
|
||||||
sError("failed to create eventfd since %s", strerror(errno));
|
|
||||||
pthread_cancel(pThread->thread);
|
|
||||||
pThread->stop = true;
|
|
||||||
} else if (epoll_ctl(pThread->pollFd, EPOLL_CTL_ADD, fd, &event) < 0) {
|
|
||||||
// failed to call epoll_ctl, call pthread_cancel instead, which may result in data corruption
|
|
||||||
sError("failed to call epoll_ctl since %s", strerror(errno));
|
|
||||||
pthread_cancel(pThread->thread);
|
|
||||||
}
|
|
||||||
|
|
||||||
pthread_join(thread, NULL);
|
pthread_join(thread, NULL);
|
||||||
taosClose(fd);
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -282,7 +282,7 @@ static int tsdbCommitToFile(STsdbRepo *pRepo, int fid, SCommitIter *iters, SRWHe
|
||||||
|
|
||||||
_err:
|
_err:
|
||||||
tfree(dataDir);
|
tfree(dataDir);
|
||||||
tsdbCloseHelperFile(pHelper, 1, NULL);
|
tsdbCloseHelperFile(pHelper, 1, pGroup);
|
||||||
return -1;
|
return -1;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -145,6 +145,7 @@ void tSkipListPutBatch(SSkipList *pSkipList, void **ppData, int ndata) {
|
||||||
// forward to put the rest of data
|
// forward to put the rest of data
|
||||||
for (int idata = 1; idata < ndata; idata++) {
|
for (int idata = 1; idata < ndata; idata++) {
|
||||||
pDataKey = pSkipList->keyFn(ppData[idata]);
|
pDataKey = pSkipList->keyFn(ppData[idata]);
|
||||||
|
hasDup = false;
|
||||||
|
|
||||||
// Compare max key
|
// Compare max key
|
||||||
pKey = SL_GET_MAX_KEY(pSkipList);
|
pKey = SL_GET_MAX_KEY(pSkipList);
|
||||||
|
@ -153,8 +154,6 @@ void tSkipListPutBatch(SSkipList *pSkipList, void **ppData, int ndata) {
|
||||||
for (int i = 0; i < pSkipList->maxLevel; i++) {
|
for (int i = 0; i < pSkipList->maxLevel; i++) {
|
||||||
forward[i] = SL_NODE_GET_BACKWARD_POINTER(pSkipList->pTail, i);
|
forward[i] = SL_NODE_GET_BACKWARD_POINTER(pSkipList->pTail, i);
|
||||||
}
|
}
|
||||||
|
|
||||||
hasDup = false;
|
|
||||||
} else {
|
} else {
|
||||||
SSkipListNode *px = pSkipList->pHead;
|
SSkipListNode *px = pSkipList->pHead;
|
||||||
for (int i = pSkipList->maxLevel - 1; i >= 0; --i) {
|
for (int i = pSkipList->maxLevel - 1; i >= 0; --i) {
|
||||||
|
@ -173,7 +172,7 @@ void tSkipListPutBatch(SSkipList *pSkipList, void **ppData, int ndata) {
|
||||||
|
|
||||||
compare = pSkipList->comparFn(pKey, pDataKey);
|
compare = pSkipList->comparFn(pKey, pDataKey);
|
||||||
if (compare >= 0) {
|
if (compare >= 0) {
|
||||||
if (compare == 0) hasDup = true;
|
if (compare == 0 && !hasDup) hasDup = true;
|
||||||
break;
|
break;
|
||||||
} else {
|
} else {
|
||||||
px = p;
|
px = p;
|
||||||
|
|
|
@ -132,7 +132,7 @@ https://www.taosdata.com/cn/all-downloads/
|
||||||
配置完成后,在命令行内使用taos shell连接server端
|
配置完成后,在命令行内使用taos shell连接server端
|
||||||
|
|
||||||
```shell
|
```shell
|
||||||
C:\TDengine>taos
|
C:\TDengine>taos -h td01
|
||||||
Welcome to the TDengine shell from Linux, Client Version:2.0.1.1
|
Welcome to the TDengine shell from Linux, Client Version:2.0.1.1
|
||||||
Copyright (c) 2017 by TAOS Data, Inc. All rights reserved.
|
Copyright (c) 2017 by TAOS Data, Inc. All rights reserved.
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1,33 @@
|
||||||
|
README.md
|
||||||
|
target/
|
||||||
|
!.mvn/wrapper/maven-wrapper.jar
|
||||||
|
!**/src/main/**/target/
|
||||||
|
!**/src/test/**/target/
|
||||||
|
|
||||||
|
### STS ###
|
||||||
|
.apt_generated
|
||||||
|
.classpath
|
||||||
|
.factorypath
|
||||||
|
.project
|
||||||
|
.settings
|
||||||
|
.springBeans
|
||||||
|
.sts4-cache
|
||||||
|
|
||||||
|
### IntelliJ IDEA ###
|
||||||
|
.idea
|
||||||
|
*.iws
|
||||||
|
*.iml
|
||||||
|
*.ipr
|
||||||
|
|
||||||
|
### NetBeans ###
|
||||||
|
/nbproject/private/
|
||||||
|
/nbbuild/
|
||||||
|
/dist/
|
||||||
|
/nbdist/
|
||||||
|
/.nb-gradle/
|
||||||
|
build/
|
||||||
|
!**/src/main/**/build/
|
||||||
|
!**/src/test/**/build/
|
||||||
|
|
||||||
|
### VS Code ###
|
||||||
|
.vscode/
|
118
tests/examples/JDBC/mybatisplus-demo/.mvn/wrapper/MavenWrapperDownloader.java
vendored
Normal file
118
tests/examples/JDBC/mybatisplus-demo/.mvn/wrapper/MavenWrapperDownloader.java
vendored
Normal file
|
@ -0,0 +1,118 @@
|
||||||
|
/*
|
||||||
|
* Copyright 2007-present the original author or authors.
|
||||||
|
*
|
||||||
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
* you may not use this file except in compliance with the License.
|
||||||
|
* You may obtain a copy of the License at
|
||||||
|
*
|
||||||
|
* https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
*
|
||||||
|
* Unless required by applicable law or agreed to in writing, software
|
||||||
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
* See the License for the specific language governing permissions and
|
||||||
|
* limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
import java.net.*;
|
||||||
|
import java.io.*;
|
||||||
|
import java.nio.channels.*;
|
||||||
|
import java.util.Properties;
|
||||||
|
|
||||||
|
public class MavenWrapperDownloader {
|
||||||
|
|
||||||
|
private static final String WRAPPER_VERSION = "0.5.6";
|
||||||
|
/**
|
||||||
|
* Default URL to download the maven-wrapper.jar from, if no 'downloadUrl' is provided.
|
||||||
|
*/
|
||||||
|
private static final String DEFAULT_DOWNLOAD_URL = "https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/"
|
||||||
|
+ WRAPPER_VERSION + "/maven-wrapper-" + WRAPPER_VERSION + ".jar";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Path to the maven-wrapper.properties file, which might contain a downloadUrl property to
|
||||||
|
* use instead of the default one.
|
||||||
|
*/
|
||||||
|
private static final String MAVEN_WRAPPER_PROPERTIES_PATH =
|
||||||
|
".mvn/wrapper/maven-wrapper.properties";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Path where the maven-wrapper.jar will be saved to.
|
||||||
|
*/
|
||||||
|
private static final String MAVEN_WRAPPER_JAR_PATH =
|
||||||
|
".mvn/wrapper/maven-wrapper.jar";
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Name of the property which should be used to override the default download url for the wrapper.
|
||||||
|
*/
|
||||||
|
private static final String PROPERTY_NAME_WRAPPER_URL = "wrapperUrl";
|
||||||
|
|
||||||
|
public static void main(String args[]) {
|
||||||
|
System.out.println("- Downloader started");
|
||||||
|
File baseDirectory = new File(args[0]);
|
||||||
|
System.out.println("- Using base directory: " + baseDirectory.getAbsolutePath());
|
||||||
|
|
||||||
|
// If the maven-wrapper.properties exists, read it and check if it contains a custom
|
||||||
|
// wrapperUrl parameter.
|
||||||
|
File mavenWrapperPropertyFile = new File(baseDirectory, MAVEN_WRAPPER_PROPERTIES_PATH);
|
||||||
|
String url = DEFAULT_DOWNLOAD_URL;
|
||||||
|
if (mavenWrapperPropertyFile.exists()) {
|
||||||
|
FileInputStream mavenWrapperPropertyFileInputStream = null;
|
||||||
|
try {
|
||||||
|
mavenWrapperPropertyFileInputStream = new FileInputStream(mavenWrapperPropertyFile);
|
||||||
|
Properties mavenWrapperProperties = new Properties();
|
||||||
|
mavenWrapperProperties.load(mavenWrapperPropertyFileInputStream);
|
||||||
|
url = mavenWrapperProperties.getProperty(PROPERTY_NAME_WRAPPER_URL, url);
|
||||||
|
} catch (IOException e) {
|
||||||
|
System.out.println("- ERROR loading '" + MAVEN_WRAPPER_PROPERTIES_PATH + "'");
|
||||||
|
} finally {
|
||||||
|
try {
|
||||||
|
if (mavenWrapperPropertyFileInputStream != null) {
|
||||||
|
mavenWrapperPropertyFileInputStream.close();
|
||||||
|
}
|
||||||
|
} catch (IOException e) {
|
||||||
|
// Ignore ...
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
System.out.println("- Downloading from: " + url);
|
||||||
|
|
||||||
|
File outputFile = new File(baseDirectory.getAbsolutePath(), MAVEN_WRAPPER_JAR_PATH);
|
||||||
|
if (!outputFile.getParentFile().exists()) {
|
||||||
|
if (!outputFile.getParentFile().mkdirs()) {
|
||||||
|
System.out.println(
|
||||||
|
"- ERROR creating output directory '" + outputFile.getParentFile().getAbsolutePath() + "'");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
System.out.println("- Downloading to: " + outputFile.getAbsolutePath());
|
||||||
|
try {
|
||||||
|
downloadFileFromURL(url, outputFile);
|
||||||
|
System.out.println("Done");
|
||||||
|
System.exit(0);
|
||||||
|
} catch (Throwable e) {
|
||||||
|
System.out.println("- Error downloading");
|
||||||
|
e.printStackTrace();
|
||||||
|
System.exit(1);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void downloadFileFromURL(String urlString, File destination) throws Exception {
|
||||||
|
if (System.getenv("MVNW_USERNAME") != null && System.getenv("MVNW_PASSWORD") != null) {
|
||||||
|
String username = System.getenv("MVNW_USERNAME");
|
||||||
|
char[] password = System.getenv("MVNW_PASSWORD").toCharArray();
|
||||||
|
Authenticator.setDefault(new Authenticator() {
|
||||||
|
@Override
|
||||||
|
protected PasswordAuthentication getPasswordAuthentication() {
|
||||||
|
return new PasswordAuthentication(username, password);
|
||||||
|
}
|
||||||
|
});
|
||||||
|
}
|
||||||
|
URL website = new URL(urlString);
|
||||||
|
ReadableByteChannel rbc;
|
||||||
|
rbc = Channels.newChannel(website.openStream());
|
||||||
|
FileOutputStream fos = new FileOutputStream(destination);
|
||||||
|
fos.getChannel().transferFrom(rbc, 0, Long.MAX_VALUE);
|
||||||
|
fos.close();
|
||||||
|
rbc.close();
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
Binary file not shown.
|
@ -0,0 +1,2 @@
|
||||||
|
distributionUrl=https://repo.maven.apache.org/maven2/org/apache/maven/apache-maven/3.6.3/apache-maven-3.6.3-bin.zip
|
||||||
|
wrapperUrl=https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar
|
|
@ -0,0 +1,322 @@
|
||||||
|
#!/bin/sh
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
# Licensed to the Apache Software Foundation (ASF) under one
|
||||||
|
# or more contributor license agreements. See the NOTICE file
|
||||||
|
# distributed with this work for additional information
|
||||||
|
# regarding copyright ownership. The ASF licenses this file
|
||||||
|
# to you under the Apache License, Version 2.0 (the
|
||||||
|
# "License"); you may not use this file except in compliance
|
||||||
|
# with the License. You may obtain a copy of the License at
|
||||||
|
#
|
||||||
|
# https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
#
|
||||||
|
# Unless required by applicable law or agreed to in writing,
|
||||||
|
# software distributed under the License is distributed on an
|
||||||
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||||
|
# KIND, either express or implied. See the License for the
|
||||||
|
# specific language governing permissions and limitations
|
||||||
|
# under the License.
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
# Maven Start Up Batch script
|
||||||
|
#
|
||||||
|
# Required ENV vars:
|
||||||
|
# ------------------
|
||||||
|
# JAVA_HOME - location of a JDK home dir
|
||||||
|
#
|
||||||
|
# Optional ENV vars
|
||||||
|
# -----------------
|
||||||
|
# M2_HOME - location of maven2's installed home dir
|
||||||
|
# MAVEN_OPTS - parameters passed to the Java VM when running Maven
|
||||||
|
# e.g. to debug Maven itself, use
|
||||||
|
# set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000
|
||||||
|
# MAVEN_SKIP_RC - flag to disable loading of mavenrc files
|
||||||
|
# ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
if [ -z "$MAVEN_SKIP_RC" ]; then
|
||||||
|
|
||||||
|
if [ -f /etc/mavenrc ]; then
|
||||||
|
. /etc/mavenrc
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -f "$HOME/.mavenrc" ]; then
|
||||||
|
. "$HOME/.mavenrc"
|
||||||
|
fi
|
||||||
|
|
||||||
|
fi
|
||||||
|
|
||||||
|
# OS specific support. $var _must_ be set to either true or false.
|
||||||
|
cygwin=false
|
||||||
|
darwin=false
|
||||||
|
mingw=false
|
||||||
|
case "$(uname)" in
|
||||||
|
CYGWIN*) cygwin=true ;;
|
||||||
|
MINGW*) mingw=true ;;
|
||||||
|
Darwin*)
|
||||||
|
darwin=true
|
||||||
|
# Use /usr/libexec/java_home if available, otherwise fall back to /Library/Java/Home
|
||||||
|
# See https://developer.apple.com/library/mac/qa/qa1170/_index.html
|
||||||
|
if [ -z "$JAVA_HOME" ]; then
|
||||||
|
if [ -x "/usr/libexec/java_home" ]; then
|
||||||
|
export JAVA_HOME="$(/usr/libexec/java_home)"
|
||||||
|
else
|
||||||
|
export JAVA_HOME="/Library/Java/Home"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
|
||||||
|
if [ -z "$JAVA_HOME" ]; then
|
||||||
|
if [ -r /etc/gentoo-release ]; then
|
||||||
|
JAVA_HOME=$(java-config --jre-home)
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$M2_HOME" ]; then
|
||||||
|
## resolve links - $0 may be a link to maven's home
|
||||||
|
PRG="$0"
|
||||||
|
|
||||||
|
# need this for relative symlinks
|
||||||
|
while [ -h "$PRG" ]; do
|
||||||
|
ls=$(ls -ld "$PRG")
|
||||||
|
link=$(expr "$ls" : '.*-> \(.*\)$')
|
||||||
|
if expr "$link" : '/.*' >/dev/null; then
|
||||||
|
PRG="$link"
|
||||||
|
else
|
||||||
|
PRG="$(dirname "$PRG")/$link"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
saveddir=$(pwd)
|
||||||
|
|
||||||
|
M2_HOME=$(dirname "$PRG")/..
|
||||||
|
|
||||||
|
# make it fully qualified
|
||||||
|
M2_HOME=$(cd "$M2_HOME" && pwd)
|
||||||
|
|
||||||
|
cd "$saveddir"
|
||||||
|
# echo Using m2 at $M2_HOME
|
||||||
|
fi
|
||||||
|
|
||||||
|
# For Cygwin, ensure paths are in UNIX format before anything is touched
|
||||||
|
if $cygwin; then
|
||||||
|
[ -n "$M2_HOME" ] &&
|
||||||
|
M2_HOME=$(cygpath --unix "$M2_HOME")
|
||||||
|
[ -n "$JAVA_HOME" ] &&
|
||||||
|
JAVA_HOME=$(cygpath --unix "$JAVA_HOME")
|
||||||
|
[ -n "$CLASSPATH" ] &&
|
||||||
|
CLASSPATH=$(cygpath --path --unix "$CLASSPATH")
|
||||||
|
fi
|
||||||
|
|
||||||
|
# For Mingw, ensure paths are in UNIX format before anything is touched
|
||||||
|
if $mingw; then
|
||||||
|
[ -n "$M2_HOME" ] &&
|
||||||
|
M2_HOME="$( (
|
||||||
|
cd "$M2_HOME"
|
||||||
|
pwd
|
||||||
|
))"
|
||||||
|
[ -n "$JAVA_HOME" ] &&
|
||||||
|
JAVA_HOME="$( (
|
||||||
|
cd "$JAVA_HOME"
|
||||||
|
pwd
|
||||||
|
))"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$JAVA_HOME" ]; then
|
||||||
|
javaExecutable="$(which javac)"
|
||||||
|
if [ -n "$javaExecutable" ] && ! [ "$(expr \"$javaExecutable\" : '\([^ ]*\)')" = "no" ]; then
|
||||||
|
# readlink(1) is not available as standard on Solaris 10.
|
||||||
|
readLink=$(which readlink)
|
||||||
|
if [ ! $(expr "$readLink" : '\([^ ]*\)') = "no" ]; then
|
||||||
|
if $darwin; then
|
||||||
|
javaHome="$(dirname \"$javaExecutable\")"
|
||||||
|
javaExecutable="$(cd \"$javaHome\" && pwd -P)/javac"
|
||||||
|
else
|
||||||
|
javaExecutable="$(readlink -f \"$javaExecutable\")"
|
||||||
|
fi
|
||||||
|
javaHome="$(dirname \"$javaExecutable\")"
|
||||||
|
javaHome=$(expr "$javaHome" : '\(.*\)/bin')
|
||||||
|
JAVA_HOME="$javaHome"
|
||||||
|
export JAVA_HOME
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$JAVACMD" ]; then
|
||||||
|
if [ -n "$JAVA_HOME" ]; then
|
||||||
|
if [ -x "$JAVA_HOME/jre/sh/java" ]; then
|
||||||
|
# IBM's JDK on AIX uses strange locations for the executables
|
||||||
|
JAVACMD="$JAVA_HOME/jre/sh/java"
|
||||||
|
else
|
||||||
|
JAVACMD="$JAVA_HOME/bin/java"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
JAVACMD="$(which java)"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ ! -x "$JAVACMD" ]; then
|
||||||
|
echo "Error: JAVA_HOME is not defined correctly." >&2
|
||||||
|
echo " We cannot execute $JAVACMD" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ -z "$JAVA_HOME" ]; then
|
||||||
|
echo "Warning: JAVA_HOME environment variable is not set."
|
||||||
|
fi
|
||||||
|
|
||||||
|
CLASSWORLDS_LAUNCHER=org.codehaus.plexus.classworlds.launcher.Launcher
|
||||||
|
|
||||||
|
# traverses directory structure from process work directory to filesystem root
|
||||||
|
# first directory with .mvn subdirectory is considered project base directory
|
||||||
|
find_maven_basedir() {
|
||||||
|
|
||||||
|
if [ -z "$1" ]; then
|
||||||
|
echo "Path not specified to find_maven_basedir"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
basedir="$1"
|
||||||
|
wdir="$1"
|
||||||
|
while [ "$wdir" != '/' ]; do
|
||||||
|
if [ -d "$wdir"/.mvn ]; then
|
||||||
|
basedir=$wdir
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
# workaround for JBEAP-8937 (on Solaris 10/Sparc)
|
||||||
|
if [ -d "${wdir}" ]; then
|
||||||
|
wdir=$(
|
||||||
|
cd "$wdir/.."
|
||||||
|
pwd
|
||||||
|
)
|
||||||
|
fi
|
||||||
|
# end of workaround
|
||||||
|
done
|
||||||
|
echo "${basedir}"
|
||||||
|
}
|
||||||
|
|
||||||
|
# concatenates all lines of a file
|
||||||
|
concat_lines() {
|
||||||
|
if [ -f "$1" ]; then
|
||||||
|
echo "$(tr -s '\n' ' ' <"$1")"
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
BASE_DIR=$(find_maven_basedir "$(pwd)")
|
||||||
|
if [ -z "$BASE_DIR" ]; then
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
##########################################################################################
|
||||||
|
# Extension to allow automatically downloading the maven-wrapper.jar from Maven-central
|
||||||
|
# This allows using the maven wrapper in projects that prohibit checking in binary data.
|
||||||
|
##########################################################################################
|
||||||
|
if [ -r "$BASE_DIR/.mvn/wrapper/maven-wrapper.jar" ]; then
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo "Found .mvn/wrapper/maven-wrapper.jar"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo "Couldn't find .mvn/wrapper/maven-wrapper.jar, downloading it ..."
|
||||||
|
fi
|
||||||
|
if [ -n "$MVNW_REPOURL" ]; then
|
||||||
|
jarUrl="$MVNW_REPOURL/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
|
||||||
|
else
|
||||||
|
jarUrl="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
|
||||||
|
fi
|
||||||
|
while IFS="=" read key value; do
|
||||||
|
case "$key" in wrapperUrl)
|
||||||
|
jarUrl="$value"
|
||||||
|
break
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done <"$BASE_DIR/.mvn/wrapper/maven-wrapper.properties"
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo "Downloading from: $jarUrl"
|
||||||
|
fi
|
||||||
|
wrapperJarPath="$BASE_DIR/.mvn/wrapper/maven-wrapper.jar"
|
||||||
|
if $cygwin; then
|
||||||
|
wrapperJarPath=$(cygpath --path --windows "$wrapperJarPath")
|
||||||
|
fi
|
||||||
|
|
||||||
|
if command -v wget >/dev/null; then
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo "Found wget ... using wget"
|
||||||
|
fi
|
||||||
|
if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then
|
||||||
|
wget "$jarUrl" -O "$wrapperJarPath"
|
||||||
|
else
|
||||||
|
wget --http-user=$MVNW_USERNAME --http-password=$MVNW_PASSWORD "$jarUrl" -O "$wrapperJarPath"
|
||||||
|
fi
|
||||||
|
elif command -v curl >/dev/null; then
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo "Found curl ... using curl"
|
||||||
|
fi
|
||||||
|
if [ -z "$MVNW_USERNAME" ] || [ -z "$MVNW_PASSWORD" ]; then
|
||||||
|
curl -o "$wrapperJarPath" "$jarUrl" -f
|
||||||
|
else
|
||||||
|
curl --user $MVNW_USERNAME:$MVNW_PASSWORD -o "$wrapperJarPath" "$jarUrl" -f
|
||||||
|
fi
|
||||||
|
|
||||||
|
else
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo "Falling back to using Java to download"
|
||||||
|
fi
|
||||||
|
javaClass="$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.java"
|
||||||
|
# For Cygwin, switch paths to Windows format before running javac
|
||||||
|
if $cygwin; then
|
||||||
|
javaClass=$(cygpath --path --windows "$javaClass")
|
||||||
|
fi
|
||||||
|
if [ -e "$javaClass" ]; then
|
||||||
|
if [ ! -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo " - Compiling MavenWrapperDownloader.java ..."
|
||||||
|
fi
|
||||||
|
# Compiling the Java class
|
||||||
|
("$JAVA_HOME/bin/javac" "$javaClass")
|
||||||
|
fi
|
||||||
|
if [ -e "$BASE_DIR/.mvn/wrapper/MavenWrapperDownloader.class" ]; then
|
||||||
|
# Running the downloader
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo " - Running MavenWrapperDownloader.java ..."
|
||||||
|
fi
|
||||||
|
("$JAVA_HOME/bin/java" -cp .mvn/wrapper MavenWrapperDownloader "$MAVEN_PROJECTBASEDIR")
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
##########################################################################################
|
||||||
|
# End of extension
|
||||||
|
##########################################################################################
|
||||||
|
|
||||||
|
export MAVEN_PROJECTBASEDIR=${MAVEN_BASEDIR:-"$BASE_DIR"}
|
||||||
|
if [ "$MVNW_VERBOSE" = true ]; then
|
||||||
|
echo $MAVEN_PROJECTBASEDIR
|
||||||
|
fi
|
||||||
|
MAVEN_OPTS="$(concat_lines "$MAVEN_PROJECTBASEDIR/.mvn/jvm.config") $MAVEN_OPTS"
|
||||||
|
|
||||||
|
# For Cygwin, switch paths to Windows format before running java
|
||||||
|
if $cygwin; then
|
||||||
|
[ -n "$M2_HOME" ] &&
|
||||||
|
M2_HOME=$(cygpath --path --windows "$M2_HOME")
|
||||||
|
[ -n "$JAVA_HOME" ] &&
|
||||||
|
JAVA_HOME=$(cygpath --path --windows "$JAVA_HOME")
|
||||||
|
[ -n "$CLASSPATH" ] &&
|
||||||
|
CLASSPATH=$(cygpath --path --windows "$CLASSPATH")
|
||||||
|
[ -n "$MAVEN_PROJECTBASEDIR" ] &&
|
||||||
|
MAVEN_PROJECTBASEDIR=$(cygpath --path --windows "$MAVEN_PROJECTBASEDIR")
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Provide a "standardized" way to retrieve the CLI args that will
|
||||||
|
# work with both Windows and non-Windows executions.
|
||||||
|
MAVEN_CMD_LINE_ARGS="$MAVEN_CONFIG $@"
|
||||||
|
export MAVEN_CMD_LINE_ARGS
|
||||||
|
|
||||||
|
WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain
|
||||||
|
|
||||||
|
exec "$JAVACMD" \
|
||||||
|
$MAVEN_OPTS \
|
||||||
|
-classpath "$MAVEN_PROJECTBASEDIR/.mvn/wrapper/maven-wrapper.jar" \
|
||||||
|
"-Dmaven.home=${M2_HOME}" "-Dmaven.multiModuleProjectDirectory=${MAVEN_PROJECTBASEDIR}" \
|
||||||
|
${WRAPPER_LAUNCHER} $MAVEN_CONFIG "$@"
|
|
@ -0,0 +1,182 @@
|
||||||
|
@REM ----------------------------------------------------------------------------
|
||||||
|
@REM Licensed to the Apache Software Foundation (ASF) under one
|
||||||
|
@REM or more contributor license agreements. See the NOTICE file
|
||||||
|
@REM distributed with this work for additional information
|
||||||
|
@REM regarding copyright ownership. The ASF licenses this file
|
||||||
|
@REM to you under the Apache License, Version 2.0 (the
|
||||||
|
@REM "License"); you may not use this file except in compliance
|
||||||
|
@REM with the License. You may obtain a copy of the License at
|
||||||
|
@REM
|
||||||
|
@REM https://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
@REM
|
||||||
|
@REM Unless required by applicable law or agreed to in writing,
|
||||||
|
@REM software distributed under the License is distributed on an
|
||||||
|
@REM "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||||
|
@REM KIND, either express or implied. See the License for the
|
||||||
|
@REM specific language governing permissions and limitations
|
||||||
|
@REM under the License.
|
||||||
|
@REM ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
@REM ----------------------------------------------------------------------------
|
||||||
|
@REM Maven Start Up Batch script
|
||||||
|
@REM
|
||||||
|
@REM Required ENV vars:
|
||||||
|
@REM JAVA_HOME - location of a JDK home dir
|
||||||
|
@REM
|
||||||
|
@REM Optional ENV vars
|
||||||
|
@REM M2_HOME - location of maven2's installed home dir
|
||||||
|
@REM MAVEN_BATCH_ECHO - set to 'on' to enable the echoing of the batch commands
|
||||||
|
@REM MAVEN_BATCH_PAUSE - set to 'on' to wait for a keystroke before ending
|
||||||
|
@REM MAVEN_OPTS - parameters passed to the Java VM when running Maven
|
||||||
|
@REM e.g. to debug Maven itself, use
|
||||||
|
@REM set MAVEN_OPTS=-Xdebug -Xrunjdwp:transport=dt_socket,server=y,suspend=y,address=8000
|
||||||
|
@REM MAVEN_SKIP_RC - flag to disable loading of mavenrc files
|
||||||
|
@REM ----------------------------------------------------------------------------
|
||||||
|
|
||||||
|
@REM Begin all REM lines with '@' in case MAVEN_BATCH_ECHO is 'on'
|
||||||
|
@echo off
|
||||||
|
@REM set title of command window
|
||||||
|
title %0
|
||||||
|
@REM enable echoing by setting MAVEN_BATCH_ECHO to 'on'
|
||||||
|
@if "%MAVEN_BATCH_ECHO%" == "on" echo %MAVEN_BATCH_ECHO%
|
||||||
|
|
||||||
|
@REM set %HOME% to equivalent of $HOME
|
||||||
|
if "%HOME%" == "" (set "HOME=%HOMEDRIVE%%HOMEPATH%")
|
||||||
|
|
||||||
|
@REM Execute a user defined script before this one
|
||||||
|
if not "%MAVEN_SKIP_RC%" == "" goto skipRcPre
|
||||||
|
@REM check for pre script, once with legacy .bat ending and once with .cmd ending
|
||||||
|
if exist "%HOME%\mavenrc_pre.bat" call "%HOME%\mavenrc_pre.bat"
|
||||||
|
if exist "%HOME%\mavenrc_pre.cmd" call "%HOME%\mavenrc_pre.cmd"
|
||||||
|
:skipRcPre
|
||||||
|
|
||||||
|
@setlocal
|
||||||
|
|
||||||
|
set ERROR_CODE=0
|
||||||
|
|
||||||
|
@REM To isolate internal variables from possible post scripts, we use another setlocal
|
||||||
|
@setlocal
|
||||||
|
|
||||||
|
@REM ==== START VALIDATION ====
|
||||||
|
if not "%JAVA_HOME%" == "" goto OkJHome
|
||||||
|
|
||||||
|
echo.
|
||||||
|
echo Error: JAVA_HOME not found in your environment. >&2
|
||||||
|
echo Please set the JAVA_HOME variable in your environment to match the >&2
|
||||||
|
echo location of your Java installation. >&2
|
||||||
|
echo.
|
||||||
|
goto error
|
||||||
|
|
||||||
|
:OkJHome
|
||||||
|
if exist "%JAVA_HOME%\bin\java.exe" goto init
|
||||||
|
|
||||||
|
echo.
|
||||||
|
echo Error: JAVA_HOME is set to an invalid directory. >&2
|
||||||
|
echo JAVA_HOME = "%JAVA_HOME%" >&2
|
||||||
|
echo Please set the JAVA_HOME variable in your environment to match the >&2
|
||||||
|
echo location of your Java installation. >&2
|
||||||
|
echo.
|
||||||
|
goto error
|
||||||
|
|
||||||
|
@REM ==== END VALIDATION ====
|
||||||
|
|
||||||
|
:init
|
||||||
|
|
||||||
|
@REM Find the project base dir, i.e. the directory that contains the folder ".mvn".
|
||||||
|
@REM Fallback to current working directory if not found.
|
||||||
|
|
||||||
|
set MAVEN_PROJECTBASEDIR=%MAVEN_BASEDIR%
|
||||||
|
IF NOT "%MAVEN_PROJECTBASEDIR%"=="" goto endDetectBaseDir
|
||||||
|
|
||||||
|
set EXEC_DIR=%CD%
|
||||||
|
set WDIR=%EXEC_DIR%
|
||||||
|
:findBaseDir
|
||||||
|
IF EXIST "%WDIR%"\.mvn goto baseDirFound
|
||||||
|
cd ..
|
||||||
|
IF "%WDIR%"=="%CD%" goto baseDirNotFound
|
||||||
|
set WDIR=%CD%
|
||||||
|
goto findBaseDir
|
||||||
|
|
||||||
|
:baseDirFound
|
||||||
|
set MAVEN_PROJECTBASEDIR=%WDIR%
|
||||||
|
cd "%EXEC_DIR%"
|
||||||
|
goto endDetectBaseDir
|
||||||
|
|
||||||
|
:baseDirNotFound
|
||||||
|
set MAVEN_PROJECTBASEDIR=%EXEC_DIR%
|
||||||
|
cd "%EXEC_DIR%"
|
||||||
|
|
||||||
|
:endDetectBaseDir
|
||||||
|
|
||||||
|
IF NOT EXIST "%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config" goto endReadAdditionalConfig
|
||||||
|
|
||||||
|
@setlocal EnableExtensions EnableDelayedExpansion
|
||||||
|
for /F "usebackq delims=" %%a in ("%MAVEN_PROJECTBASEDIR%\.mvn\jvm.config") do set JVM_CONFIG_MAVEN_PROPS=!JVM_CONFIG_MAVEN_PROPS! %%a
|
||||||
|
@endlocal & set JVM_CONFIG_MAVEN_PROPS=%JVM_CONFIG_MAVEN_PROPS%
|
||||||
|
|
||||||
|
:endReadAdditionalConfig
|
||||||
|
|
||||||
|
SET MAVEN_JAVA_EXE="%JAVA_HOME%\bin\java.exe"
|
||||||
|
set WRAPPER_JAR="%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.jar"
|
||||||
|
set WRAPPER_LAUNCHER=org.apache.maven.wrapper.MavenWrapperMain
|
||||||
|
|
||||||
|
set DOWNLOAD_URL="https://repo.maven.apache.org/maven2/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
|
||||||
|
|
||||||
|
FOR /F "tokens=1,2 delims==" %%A IN ("%MAVEN_PROJECTBASEDIR%\.mvn\wrapper\maven-wrapper.properties") DO (
|
||||||
|
IF "%%A"=="wrapperUrl" SET DOWNLOAD_URL=%%B
|
||||||
|
)
|
||||||
|
|
||||||
|
@REM Extension to allow automatically downloading the maven-wrapper.jar from Maven-central
|
||||||
|
@REM This allows using the maven wrapper in projects that prohibit checking in binary data.
|
||||||
|
if exist %WRAPPER_JAR% (
|
||||||
|
if "%MVNW_VERBOSE%" == "true" (
|
||||||
|
echo Found %WRAPPER_JAR%
|
||||||
|
)
|
||||||
|
) else (
|
||||||
|
if not "%MVNW_REPOURL%" == "" (
|
||||||
|
SET DOWNLOAD_URL="%MVNW_REPOURL%/io/takari/maven-wrapper/0.5.6/maven-wrapper-0.5.6.jar"
|
||||||
|
)
|
||||||
|
if "%MVNW_VERBOSE%" == "true" (
|
||||||
|
echo Couldn't find %WRAPPER_JAR%, downloading it ...
|
||||||
|
echo Downloading from: %DOWNLOAD_URL%
|
||||||
|
)
|
||||||
|
|
||||||
|
powershell -Command "&{"^
|
||||||
|
"$webclient = new-object System.Net.WebClient;"^
|
||||||
|
"if (-not ([string]::IsNullOrEmpty('%MVNW_USERNAME%') -and [string]::IsNullOrEmpty('%MVNW_PASSWORD%'))) {"^
|
||||||
|
"$webclient.Credentials = new-object System.Net.NetworkCredential('%MVNW_USERNAME%', '%MVNW_PASSWORD%');"^
|
||||||
|
"}"^
|
||||||
|
"[Net.ServicePointManager]::SecurityProtocol = [Net.SecurityProtocolType]::Tls12; $webclient.DownloadFile('%DOWNLOAD_URL%', '%WRAPPER_JAR%')"^
|
||||||
|
"}"
|
||||||
|
if "%MVNW_VERBOSE%" == "true" (
|
||||||
|
echo Finished downloading %WRAPPER_JAR%
|
||||||
|
)
|
||||||
|
)
|
||||||
|
@REM End of extension
|
||||||
|
|
||||||
|
@REM Provide a "standardized" way to retrieve the CLI args that will
|
||||||
|
@REM work with both Windows and non-Windows executions.
|
||||||
|
set MAVEN_CMD_LINE_ARGS=%*
|
||||||
|
|
||||||
|
%MAVEN_JAVA_EXE% %JVM_CONFIG_MAVEN_PROPS% %MAVEN_OPTS% %MAVEN_DEBUG_OPTS% -classpath %WRAPPER_JAR% "-Dmaven.multiModuleProjectDirectory=%MAVEN_PROJECTBASEDIR%" %WRAPPER_LAUNCHER% %MAVEN_CONFIG% %*
|
||||||
|
if ERRORLEVEL 1 goto error
|
||||||
|
goto end
|
||||||
|
|
||||||
|
:error
|
||||||
|
set ERROR_CODE=1
|
||||||
|
|
||||||
|
:end
|
||||||
|
@endlocal & set ERROR_CODE=%ERROR_CODE%
|
||||||
|
|
||||||
|
if not "%MAVEN_SKIP_RC%" == "" goto skipRcPost
|
||||||
|
@REM check for post script, once with legacy .bat ending and once with .cmd ending
|
||||||
|
if exist "%HOME%\mavenrc_post.bat" call "%HOME%\mavenrc_post.bat"
|
||||||
|
if exist "%HOME%\mavenrc_post.cmd" call "%HOME%\mavenrc_post.cmd"
|
||||||
|
:skipRcPost
|
||||||
|
|
||||||
|
@REM pause the script if MAVEN_BATCH_PAUSE is set to 'on'
|
||||||
|
if "%MAVEN_BATCH_PAUSE%" == "on" pause
|
||||||
|
|
||||||
|
if "%MAVEN_TERMINATE_CMD%" == "on" exit %ERROR_CODE%
|
||||||
|
|
||||||
|
exit /B %ERROR_CODE%
|
|
@ -0,0 +1,101 @@
|
||||||
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
|
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||||
|
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||||
|
<modelVersion>4.0.0</modelVersion>
|
||||||
|
<parent>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-starter-parent</artifactId>
|
||||||
|
<version>2.4.0</version>
|
||||||
|
<relativePath/> <!-- lookup parent from repository -->
|
||||||
|
</parent>
|
||||||
|
<groupId>com.taosdata.example</groupId>
|
||||||
|
<artifactId>mybatisplus-demo</artifactId>
|
||||||
|
<version>0.0.1-SNAPSHOT</version>
|
||||||
|
<name>mybatisplus-demo</name>
|
||||||
|
<description>Demo project for tdengine</description>
|
||||||
|
|
||||||
|
<properties>
|
||||||
|
<java.version>1.8</java.version>
|
||||||
|
</properties>
|
||||||
|
|
||||||
|
<dependencies>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-starter</artifactId>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.projectlombok</groupId>
|
||||||
|
<artifactId>lombok</artifactId>
|
||||||
|
<optional>true</optional>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>com.baomidou</groupId>
|
||||||
|
<artifactId>mybatis-plus-boot-starter</artifactId>
|
||||||
|
<version>3.1.2</version>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>com.h2database</groupId>
|
||||||
|
<artifactId>h2</artifactId>
|
||||||
|
<scope>runtime</scope>
|
||||||
|
</dependency>
|
||||||
|
|
||||||
|
<dependency>
|
||||||
|
<groupId>com.taosdata.jdbc</groupId>
|
||||||
|
<artifactId>taos-jdbcdriver</artifactId>
|
||||||
|
<version>2.0.11</version>
|
||||||
|
</dependency>
|
||||||
|
|
||||||
|
<!-- https://mvnrepository.com/artifact/mysql/mysql-connector-java -->
|
||||||
|
<dependency>
|
||||||
|
<groupId>mysql</groupId>
|
||||||
|
<artifactId>mysql-connector-java</artifactId>
|
||||||
|
<version>5.1.47</version>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-starter-web</artifactId>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-devtools</artifactId>
|
||||||
|
<scope>runtime</scope>
|
||||||
|
<optional>true</optional>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-starter-test</artifactId>
|
||||||
|
<scope>test</scope>
|
||||||
|
</dependency>
|
||||||
|
<dependency>
|
||||||
|
<groupId>junit</groupId>
|
||||||
|
<artifactId>junit</artifactId>
|
||||||
|
<version>4.12</version>
|
||||||
|
<scope>test</scope>
|
||||||
|
</dependency>
|
||||||
|
</dependencies>
|
||||||
|
|
||||||
|
<build>
|
||||||
|
<plugins>
|
||||||
|
<plugin>
|
||||||
|
<groupId>org.springframework.boot</groupId>
|
||||||
|
<artifactId>spring-boot-maven-plugin</artifactId>
|
||||||
|
</plugin>
|
||||||
|
|
||||||
|
<plugin>
|
||||||
|
<groupId>org.apache.maven.plugins</groupId>
|
||||||
|
<artifactId>maven-surefire-plugin</artifactId>
|
||||||
|
<version>2.17</version>
|
||||||
|
<configuration>
|
||||||
|
<includes>
|
||||||
|
<include>**/*Test.java</include>
|
||||||
|
</includes>
|
||||||
|
<excludes>
|
||||||
|
<exclude>**/Abstract*.java</exclude>
|
||||||
|
</excludes>
|
||||||
|
</configuration>
|
||||||
|
</plugin>
|
||||||
|
</plugins>
|
||||||
|
</build>
|
||||||
|
|
||||||
|
|
||||||
|
</project>
|
|
@ -0,0 +1,15 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo;
|
||||||
|
|
||||||
|
import org.mybatis.spring.annotation.MapperScan;
|
||||||
|
import org.springframework.boot.SpringApplication;
|
||||||
|
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||||
|
|
||||||
|
@SpringBootApplication
|
||||||
|
@MapperScan("com.taosdata.example.mybatisplusdemo.mapper")
|
||||||
|
public class MybatisplusDemoApplication {
|
||||||
|
|
||||||
|
public static void main(String[] args) {
|
||||||
|
SpringApplication.run(MybatisplusDemoApplication.class, args);
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,34 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.config;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.PaginationInterceptor;
|
||||||
|
import org.springframework.context.annotation.Bean;
|
||||||
|
import org.springframework.context.annotation.Configuration;
|
||||||
|
|
||||||
|
@Configuration
|
||||||
|
public class MybatisPlusConfig {
|
||||||
|
|
||||||
|
|
||||||
|
/** mybatis 3.4.1 pagination config start ***/
|
||||||
|
// @Bean
|
||||||
|
// public MybatisPlusInterceptor mybatisPlusInterceptor() {
|
||||||
|
// MybatisPlusInterceptor interceptor = new MybatisPlusInterceptor();
|
||||||
|
// interceptor.addInnerInterceptor(new PaginationInnerInterceptor());
|
||||||
|
// return interceptor;
|
||||||
|
// }
|
||||||
|
|
||||||
|
// @Bean
|
||||||
|
// public ConfigurationCustomizer configurationCustomizer() {
|
||||||
|
// return configuration -> configuration.setUseDeprecatedExecutor(false);
|
||||||
|
// }
|
||||||
|
|
||||||
|
@Bean
|
||||||
|
public PaginationInterceptor paginationInterceptor() {
|
||||||
|
// return new PaginationInterceptor();
|
||||||
|
PaginationInterceptor paginationInterceptor = new PaginationInterceptor();
|
||||||
|
//TODO: mybatis-plus do not support TDengine, use postgresql Dialect
|
||||||
|
paginationInterceptor.setDialectType("postgresql");
|
||||||
|
|
||||||
|
return paginationInterceptor;
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,15 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.domain;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
public class Temperature {
|
||||||
|
|
||||||
|
private Timestamp ts;
|
||||||
|
private float temperature;
|
||||||
|
private String location;
|
||||||
|
private int tbIndex;
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,15 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.domain;
|
||||||
|
|
||||||
|
import lombok.Data;
|
||||||
|
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
|
||||||
|
@Data
|
||||||
|
public class Weather {
|
||||||
|
|
||||||
|
private Timestamp ts;
|
||||||
|
private float temperature;
|
||||||
|
private int humidity;
|
||||||
|
private String location;
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,23 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
|
import com.taosdata.example.mybatisplusdemo.domain.Temperature;
|
||||||
|
import org.apache.ibatis.annotations.Insert;
|
||||||
|
import org.apache.ibatis.annotations.Param;
|
||||||
|
import org.apache.ibatis.annotations.Update;
|
||||||
|
|
||||||
|
public interface TemperatureMapper extends BaseMapper<Temperature> {
|
||||||
|
|
||||||
|
@Update("CREATE TABLE if not exists temperature(ts timestamp, temperature float) tags(location nchar(64), tbIndex int)")
|
||||||
|
int createSuperTable();
|
||||||
|
|
||||||
|
@Update("create table #{tbName} using temperature tags( #{location}, #{tbindex})")
|
||||||
|
int createTable(@Param("tbName") String tbName, @Param("location") String location, @Param("tbindex") int tbindex);
|
||||||
|
|
||||||
|
@Update("drop table if exists temperature")
|
||||||
|
void dropSuperTable();
|
||||||
|
|
||||||
|
@Insert("insert into t${tbIndex}(ts, temperature) values(#{ts}, #{temperature})")
|
||||||
|
int insertOne(Temperature one);
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,8 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
|
||||||
|
import com.taosdata.example.mybatisplusdemo.domain.Weather;
|
||||||
|
|
||||||
|
public interface WeatherMapper extends BaseMapper<Weather> {
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,34 @@
|
||||||
|
spring:
|
||||||
|
datasource:
|
||||||
|
# driver-class-name: org.h2.Driver
|
||||||
|
# schema: classpath:db/schema-mysql.sql
|
||||||
|
# data: classpath:db/data-mysql.sql
|
||||||
|
# url: jdbc:h2:mem:test
|
||||||
|
# username: root
|
||||||
|
# password: test
|
||||||
|
|
||||||
|
# driver-class-name: com.mysql.jdbc.Driver
|
||||||
|
# url: jdbc:mysql://master:3306/test?useSSL=false
|
||||||
|
# username: root
|
||||||
|
# password: 123456
|
||||||
|
|
||||||
|
driver-class-name: com.taosdata.jdbc.TSDBDriver
|
||||||
|
url: jdbc:TAOS://localhost:6030/mp_test
|
||||||
|
user: root
|
||||||
|
password: taosdata
|
||||||
|
charset: UTF-8
|
||||||
|
locale: en_US.UTF-8
|
||||||
|
timezone: UTC-8
|
||||||
|
|
||||||
|
mybatis-plus:
|
||||||
|
configuration:
|
||||||
|
map-underscore-to-camel-case: false
|
||||||
|
|
||||||
|
logging:
|
||||||
|
level:
|
||||||
|
com:
|
||||||
|
taosdata:
|
||||||
|
example:
|
||||||
|
mybatisplusdemo:
|
||||||
|
mapper: debug
|
||||||
|
|
|
@ -0,0 +1,140 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||||
|
import com.taosdata.example.mybatisplusdemo.domain.Temperature;
|
||||||
|
import org.junit.After;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Before;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.boot.test.context.SpringBootTest;
|
||||||
|
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
||||||
|
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
|
@RunWith(SpringJUnit4ClassRunner.class)
|
||||||
|
@SpringBootTest
|
||||||
|
public class TemperatureMapperTest {
|
||||||
|
|
||||||
|
private static Random random = new Random(System.currentTimeMillis());
|
||||||
|
private static String[] locations = {"北京", "上海", "深圳", "广州", "杭州"};
|
||||||
|
|
||||||
|
@Before
|
||||||
|
public void before() {
|
||||||
|
mapper.dropSuperTable();
|
||||||
|
// create table temperature
|
||||||
|
mapper.createSuperTable();
|
||||||
|
// create table t_X using temperature
|
||||||
|
for (int i = 0; i < 10; i++) {
|
||||||
|
mapper.createTable("t" + i, locations[random.nextInt(locations.length)], i);
|
||||||
|
}
|
||||||
|
// insert into table
|
||||||
|
int affectRows = 0;
|
||||||
|
// insert 10 tables
|
||||||
|
for (int i = 0; i < 10; i++) {
|
||||||
|
// each table insert 5 rows
|
||||||
|
for (int j = 0; j < 5; j++) {
|
||||||
|
Temperature one = new Temperature();
|
||||||
|
one.setTs(new Timestamp(1605024000000l));
|
||||||
|
one.setTemperature(random.nextFloat() * 50);
|
||||||
|
one.setLocation("望京");
|
||||||
|
one.setTbIndex(i);
|
||||||
|
affectRows += mapper.insertOne(one);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Assert.assertEquals(50, affectRows);
|
||||||
|
}
|
||||||
|
|
||||||
|
@After
|
||||||
|
public void after() {
|
||||||
|
mapper.dropSuperTable();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private TemperatureMapper mapper;
|
||||||
|
|
||||||
|
/***
|
||||||
|
* test SelectList
|
||||||
|
* **/
|
||||||
|
@Test
|
||||||
|
public void testSelectList() {
|
||||||
|
List<Temperature> temperatureList = mapper.selectList(null);
|
||||||
|
temperatureList.forEach(System.out::println);
|
||||||
|
}
|
||||||
|
|
||||||
|
/***
|
||||||
|
* test InsertOne which is a custom metheod
|
||||||
|
* ***/
|
||||||
|
@Test
|
||||||
|
public void testInsert() {
|
||||||
|
Temperature one = new Temperature();
|
||||||
|
one.setTs(new Timestamp(1605024000000l));
|
||||||
|
one.setTemperature(random.nextFloat() * 50);
|
||||||
|
one.setLocation("望京");
|
||||||
|
int affectRows = mapper.insertOne(one);
|
||||||
|
Assert.assertEquals(1, affectRows);
|
||||||
|
}
|
||||||
|
|
||||||
|
/***
|
||||||
|
* test SelectOne
|
||||||
|
* **/
|
||||||
|
@Test
|
||||||
|
public void testSelectOne() {
|
||||||
|
QueryWrapper<Temperature> wrapper = new QueryWrapper<>();
|
||||||
|
wrapper.eq("location", "beijing");
|
||||||
|
Temperature one = mapper.selectOne(wrapper);
|
||||||
|
System.out.println(one);
|
||||||
|
Assert.assertNotNull(one);
|
||||||
|
}
|
||||||
|
|
||||||
|
/***
|
||||||
|
* test select By map
|
||||||
|
* ***/
|
||||||
|
@Test
|
||||||
|
public void testSelectByMap() {
|
||||||
|
Map<String, Object> map = new HashMap<>();
|
||||||
|
map.put("location", "beijing");
|
||||||
|
List<Temperature> temperatures = mapper.selectByMap(map);
|
||||||
|
Assert.assertEquals(1, temperatures.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
/***
|
||||||
|
* test selectObjs
|
||||||
|
* **/
|
||||||
|
@Test
|
||||||
|
public void testSelectObjs() {
|
||||||
|
List<Object> ts = mapper.selectObjs(null);
|
||||||
|
System.out.println(ts);
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* test selectC ount
|
||||||
|
* **/
|
||||||
|
@Test
|
||||||
|
public void testSelectCount() {
|
||||||
|
int count = mapper.selectCount(null);
|
||||||
|
Assert.assertEquals(5, count);
|
||||||
|
}
|
||||||
|
|
||||||
|
/****
|
||||||
|
* 分页
|
||||||
|
*/
|
||||||
|
@Test
|
||||||
|
public void testSelectPage() {
|
||||||
|
IPage page = new Page(1, 2);
|
||||||
|
IPage<Temperature> temperatureIPage = mapper.selectPage(page, null);
|
||||||
|
System.out.println("total : " + temperatureIPage.getTotal());
|
||||||
|
System.out.println("pages : " + temperatureIPage.getPages());
|
||||||
|
for (Temperature temperature : temperatureIPage.getRecords()) {
|
||||||
|
System.out.println(temperature);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -0,0 +1,88 @@
|
||||||
|
package com.taosdata.example.mybatisplusdemo.mapper;
|
||||||
|
|
||||||
|
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||||||
|
import com.baomidou.mybatisplus.core.metadata.IPage;
|
||||||
|
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||||||
|
import com.taosdata.example.mybatisplusdemo.domain.Weather;
|
||||||
|
import org.junit.Assert;
|
||||||
|
import org.junit.Test;
|
||||||
|
import org.junit.runner.RunWith;
|
||||||
|
import org.springframework.beans.factory.annotation.Autowired;
|
||||||
|
import org.springframework.boot.test.context.SpringBootTest;
|
||||||
|
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
|
||||||
|
|
||||||
|
import java.sql.Timestamp;
|
||||||
|
import java.util.HashMap;
|
||||||
|
import java.util.List;
|
||||||
|
import java.util.Map;
|
||||||
|
import java.util.Random;
|
||||||
|
|
||||||
|
@RunWith(SpringJUnit4ClassRunner.class)
|
||||||
|
@SpringBootTest
|
||||||
|
public class WeatherMapperTest {
|
||||||
|
|
||||||
|
private static Random random = new Random(System.currentTimeMillis());
|
||||||
|
|
||||||
|
@Autowired
|
||||||
|
private WeatherMapper mapper;
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectList() {
|
||||||
|
List<Weather> weathers = mapper.selectList(null);
|
||||||
|
weathers.forEach(System.out::println);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testInsert() {
|
||||||
|
Weather one = new Weather();
|
||||||
|
one.setTs(new Timestamp(1605024000000l));
|
||||||
|
one.setTemperature(random.nextFloat() * 50);
|
||||||
|
one.setHumidity(random.nextInt(100));
|
||||||
|
one.setLocation("望京");
|
||||||
|
int affectRows = mapper.insert(one);
|
||||||
|
Assert.assertEquals(1, affectRows);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectOne() {
|
||||||
|
QueryWrapper<Weather> wrapper = new QueryWrapper<>();
|
||||||
|
wrapper.eq("location", "beijing");
|
||||||
|
Weather one = mapper.selectOne(wrapper);
|
||||||
|
System.out.println(one);
|
||||||
|
Assert.assertEquals(12.22f, one.getTemperature(), 0.00f);
|
||||||
|
Assert.assertEquals("beijing", one.getLocation());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectByMap() {
|
||||||
|
Map<String, Object> map = new HashMap<>();
|
||||||
|
map.put("location", "beijing");
|
||||||
|
List<Weather> weathers = mapper.selectByMap(map);
|
||||||
|
Assert.assertEquals(1, weathers.size());
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectObjs() {
|
||||||
|
List<Object> ts = mapper.selectObjs(null);
|
||||||
|
System.out.println(ts);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectCount() {
|
||||||
|
int count = mapper.selectCount(null);
|
||||||
|
// Assert.assertEquals(5, count);
|
||||||
|
System.out.println(count);
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testSelectPage() {
|
||||||
|
IPage page = new Page(1, 2);
|
||||||
|
IPage<Weather> weatherIPage = mapper.selectPage(page, null);
|
||||||
|
System.out.println("total : " + weatherIPage.getTotal());
|
||||||
|
System.out.println("pages : " + weatherIPage.getPages());
|
||||||
|
for (Weather weather : weatherIPage.getRecords()) {
|
||||||
|
System.out.println(weather);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
|
@ -50,10 +50,10 @@ static void queryDB(TAOS *taos, char *command) {
|
||||||
taos_free_result(pSql);
|
taos_free_result(pSql);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
void Test(char *qstr, const char *input, int i);
|
||||||
|
|
||||||
int main(int argc, char *argv[]) {
|
int main(int argc, char *argv[]) {
|
||||||
TAOS * taos;
|
|
||||||
char qstr[1024];
|
char qstr[1024];
|
||||||
TAOS_RES *result;
|
|
||||||
|
|
||||||
// connect to server
|
// connect to server
|
||||||
if (argc < 2) {
|
if (argc < 2) {
|
||||||
|
@ -63,41 +63,26 @@ int main(int argc, char *argv[]) {
|
||||||
|
|
||||||
// init TAOS
|
// init TAOS
|
||||||
taos_init();
|
taos_init();
|
||||||
|
for (int i = 0; i < 4000000; i++) {
|
||||||
taos = taos_connect(argv[1], "root", "taosdata", NULL, 0);
|
Test(qstr, argv[1], i);
|
||||||
|
}
|
||||||
|
taos_cleanup();
|
||||||
|
}
|
||||||
|
void Test(char *qstr, const char *input, int index) {
|
||||||
|
TAOS *taos = taos_connect(input, "root", "taosdata", NULL, 0);
|
||||||
|
printf("==================test at %d\n================================", index);
|
||||||
|
queryDB(taos, "drop database if exists demo");
|
||||||
|
queryDB(taos, "create database demo");
|
||||||
|
TAOS_RES *result;
|
||||||
if (taos == NULL) {
|
if (taos == NULL) {
|
||||||
printf("failed to connect to server, reason:%s\n", "null taos"/*taos_errstr(taos)*/);
|
printf("failed to connect to server, reason:%s\n", "null taos"/*taos_errstr(taos)*/);
|
||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
printf("success to connect to server\n");
|
|
||||||
|
|
||||||
|
|
||||||
//taos_query(taos, "drop database demo");
|
|
||||||
queryDB(taos, "drop database if exists demo");
|
|
||||||
|
|
||||||
//result = taos_query(taos, "create database demo");
|
|
||||||
//if (result == NULL) {
|
|
||||||
// printf("failed to create database, reason:%s\n", "null result"/*taos_errstr(taos)*/);
|
|
||||||
// exit(1);
|
|
||||||
//}
|
|
||||||
queryDB(taos, "create database demo");
|
|
||||||
printf("success to create database\n");
|
|
||||||
|
|
||||||
//taos_query(taos, "use demo");
|
|
||||||
queryDB(taos, "use demo");
|
queryDB(taos, "use demo");
|
||||||
|
|
||||||
// create table
|
|
||||||
//if (taos_query(taos, "create table m1 (ts timestamp, ti tinyint, si smallint, i int, bi bigint, f float, d double, b binary(10))") == 0) {
|
|
||||||
// printf("failed to create table, reason:%s\n", taos_errstr(result));
|
|
||||||
// exit(1);
|
|
||||||
//}
|
|
||||||
queryDB(taos, "create table m1 (ts timestamp, ti tinyint, si smallint, i int, bi bigint, f float, d double, b binary(10))");
|
queryDB(taos, "create table m1 (ts timestamp, ti tinyint, si smallint, i int, bi bigint, f float, d double, b binary(10))");
|
||||||
printf("success to create table\n");
|
printf("success to create table\n");
|
||||||
|
|
||||||
// sleep for one second to make sure table is created on data node
|
|
||||||
// taosMsleep(1000);
|
|
||||||
|
|
||||||
// insert 10 records
|
|
||||||
int i = 0;
|
int i = 0;
|
||||||
for (i = 0; i < 10; ++i) {
|
for (i = 0; i < 10; ++i) {
|
||||||
sprintf(qstr, "insert into m1 values (%" PRId64 ", %d, %d, %d, %d, %f, %lf, '%s')", 1546300800000 + i * 1000, i, i, i, i*10000000, i*1.0, i*2.0, "hello");
|
sprintf(qstr, "insert into m1 values (%" PRId64 ", %d, %d, %d, %d, %f, %lf, '%s')", 1546300800000 + i * 1000, i, i, i, i*10000000, i*1.0, i*2.0, "hello");
|
||||||
|
@ -117,7 +102,6 @@ int main(int argc, char *argv[]) {
|
||||||
}
|
}
|
||||||
taos_free_result(result);
|
taos_free_result(result);
|
||||||
|
|
||||||
//sleep(1);
|
|
||||||
}
|
}
|
||||||
printf("success to insert rows, total %d rows\n", i);
|
printf("success to insert rows, total %d rows\n", i);
|
||||||
|
|
||||||
|
@ -147,5 +131,6 @@ int main(int argc, char *argv[]) {
|
||||||
|
|
||||||
taos_free_result(result);
|
taos_free_result(result);
|
||||||
printf("====demo end====\n\n");
|
printf("====demo end====\n\n");
|
||||||
return getchar();
|
taos_close(taos);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -0,0 +1 @@
|
||||||
|
Subproject commit f2ffd30521b8e8afbc9d25c75f8eeeb6a48bd030
|
|
@ -1,3 +0,0 @@
|
||||||
/target
|
|
||||||
**/*.rs.bk
|
|
||||||
Cargo.lock
|
|
|
@ -1,7 +0,0 @@
|
||||||
[package]
|
|
||||||
name = "tdengine"
|
|
||||||
version = "0.1.0"
|
|
||||||
authors = ["Chunhua Jiang <jiangch@3reality.com>"]
|
|
||||||
edition = "2018"
|
|
||||||
|
|
||||||
[dependencies]
|
|
|
@ -1,20 +0,0 @@
|
||||||
# TDengine driver connector for Rust
|
|
||||||
|
|
||||||
It's a rust implementation for [TDengine](https://github.com/taosdata/TDengine), an open-sourced big data platform designed and optimized for the Internet of Things (IoT), Connected Cars, Industrial IoT, and IT Infrastructure and Application Monitoring.
|
|
||||||
|
|
||||||
## Dependencies
|
|
||||||
- Rust:
|
|
||||||
```
|
|
||||||
curl https://sh.rustup.rs -sSf | sh
|
|
||||||
```
|
|
||||||
|
|
||||||
## Run with Sample
|
|
||||||
|
|
||||||
Build and run basic sample:
|
|
||||||
```
|
|
||||||
cargo run --example demo
|
|
||||||
```
|
|
||||||
Build and run subscribe sample:
|
|
||||||
```
|
|
||||||
cargo run --example subscribe
|
|
||||||
```
|
|
|
@ -1,10 +0,0 @@
|
||||||
// build.rs
|
|
||||||
|
|
||||||
use std::env;
|
|
||||||
|
|
||||||
fn main() {
|
|
||||||
let project_dir = env::var("CARGO_MANIFEST_DIR").unwrap();
|
|
||||||
|
|
||||||
println!("cargo:rustc-link-search={}", project_dir); // the "-L" flag
|
|
||||||
println!("cargo:rustc-link-lib=taos"); // the "-l" flag
|
|
||||||
}
|
|
|
@ -1,19 +0,0 @@
|
||||||
use std::process;
|
|
||||||
use tdengine::Tdengine;
|
|
||||||
|
|
||||||
fn main() {
|
|
||||||
let tde = Tdengine::new("127.0.0.1", "root", "taosdata", "demo", 0)
|
|
||||||
.unwrap_or_else(|err| {
|
|
||||||
eprintln!("Can't create Tdengine: {}", err);
|
|
||||||
process::exit(1)
|
|
||||||
});
|
|
||||||
|
|
||||||
tde.query("drop database demo");
|
|
||||||
tde.query("create database demo");
|
|
||||||
tde.query("use demo");
|
|
||||||
tde.query("create table m1 (ts timestamp, speed int)");
|
|
||||||
|
|
||||||
for i in 0..10 {
|
|
||||||
tde.query(format!("insert into m1 values (now+{}s, {})", i, i).as_str());
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,19 +0,0 @@
|
||||||
use std::process;
|
|
||||||
use tdengine::Subscriber;
|
|
||||||
|
|
||||||
fn main() {
|
|
||||||
let subscriber = Subscriber::new("127.0.0.1", "root", "taosdata", "demo", "m1", 0, 1000)
|
|
||||||
.unwrap_or_else(|err| {
|
|
||||||
eprintln!("Can't create Subscriber: {}", err);
|
|
||||||
process::exit(1)
|
|
||||||
});
|
|
||||||
|
|
||||||
loop {
|
|
||||||
let row = subscriber.consume().unwrap_or_else(|err| {
|
|
||||||
eprintln!("consume exit: {}", err);
|
|
||||||
process::exit(1)
|
|
||||||
});
|
|
||||||
|
|
||||||
subscriber.print_row(&row);
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,332 +0,0 @@
|
||||||
/* automatically generated by rust-bindgen */
|
|
||||||
#![allow(unused)]
|
|
||||||
#![allow(non_camel_case_types)]
|
|
||||||
|
|
||||||
pub const _STDINT_H: u32 = 1;
|
|
||||||
pub const _FEATURES_H: u32 = 1;
|
|
||||||
pub const _DEFAULT_SOURCE: u32 = 1;
|
|
||||||
pub const __USE_ISOC11: u32 = 1;
|
|
||||||
pub const __USE_ISOC99: u32 = 1;
|
|
||||||
pub const __USE_ISOC95: u32 = 1;
|
|
||||||
pub const __USE_POSIX_IMPLICITLY: u32 = 1;
|
|
||||||
pub const _POSIX_SOURCE: u32 = 1;
|
|
||||||
pub const _POSIX_C_SOURCE: u32 = 200809;
|
|
||||||
pub const __USE_POSIX: u32 = 1;
|
|
||||||
pub const __USE_POSIX2: u32 = 1;
|
|
||||||
pub const __USE_POSIX199309: u32 = 1;
|
|
||||||
pub const __USE_POSIX199506: u32 = 1;
|
|
||||||
pub const __USE_XOPEN2K: u32 = 1;
|
|
||||||
pub const __USE_XOPEN2K8: u32 = 1;
|
|
||||||
pub const _ATFILE_SOURCE: u32 = 1;
|
|
||||||
pub const __USE_MISC: u32 = 1;
|
|
||||||
pub const __USE_ATFILE: u32 = 1;
|
|
||||||
pub const __USE_FORTIFY_LEVEL: u32 = 0;
|
|
||||||
pub const _STDC_PREDEF_H: u32 = 1;
|
|
||||||
pub const __STDC_IEC_559__: u32 = 1;
|
|
||||||
pub const __STDC_IEC_559_COMPLEX__: u32 = 1;
|
|
||||||
pub const __STDC_ISO_10646__: u32 = 201505;
|
|
||||||
pub const __STDC_NO_THREADS__: u32 = 1;
|
|
||||||
pub const __GNU_LIBRARY__: u32 = 6;
|
|
||||||
pub const __GLIBC__: u32 = 2;
|
|
||||||
pub const __GLIBC_MINOR__: u32 = 23;
|
|
||||||
pub const _SYS_CDEFS_H: u32 = 1;
|
|
||||||
pub const __WORDSIZE: u32 = 64;
|
|
||||||
pub const __WORDSIZE_TIME64_COMPAT32: u32 = 1;
|
|
||||||
pub const __SYSCALL_WORDSIZE: u32 = 64;
|
|
||||||
pub const _BITS_WCHAR_H: u32 = 1;
|
|
||||||
pub const INT8_MIN: i32 = -128;
|
|
||||||
pub const INT16_MIN: i32 = -32768;
|
|
||||||
pub const INT32_MIN: i32 = -2147483648;
|
|
||||||
pub const INT8_MAX: u32 = 127;
|
|
||||||
pub const INT16_MAX: u32 = 32767;
|
|
||||||
pub const INT32_MAX: u32 = 2147483647;
|
|
||||||
pub const UINT8_MAX: u32 = 255;
|
|
||||||
pub const UINT16_MAX: u32 = 65535;
|
|
||||||
pub const UINT32_MAX: u32 = 4294967295;
|
|
||||||
pub const INT_LEAST8_MIN: i32 = -128;
|
|
||||||
pub const INT_LEAST16_MIN: i32 = -32768;
|
|
||||||
pub const INT_LEAST32_MIN: i32 = -2147483648;
|
|
||||||
pub const INT_LEAST8_MAX: u32 = 127;
|
|
||||||
pub const INT_LEAST16_MAX: u32 = 32767;
|
|
||||||
pub const INT_LEAST32_MAX: u32 = 2147483647;
|
|
||||||
pub const UINT_LEAST8_MAX: u32 = 255;
|
|
||||||
pub const UINT_LEAST16_MAX: u32 = 65535;
|
|
||||||
pub const UINT_LEAST32_MAX: u32 = 4294967295;
|
|
||||||
pub const INT_FAST8_MIN: i32 = -128;
|
|
||||||
pub const INT_FAST16_MIN: i64 = -9223372036854775808;
|
|
||||||
pub const INT_FAST32_MIN: i64 = -9223372036854775808;
|
|
||||||
pub const INT_FAST8_MAX: u32 = 127;
|
|
||||||
pub const INT_FAST16_MAX: u64 = 9223372036854775807;
|
|
||||||
pub const INT_FAST32_MAX: u64 = 9223372036854775807;
|
|
||||||
pub const UINT_FAST8_MAX: u32 = 255;
|
|
||||||
pub const UINT_FAST16_MAX: i32 = -1;
|
|
||||||
pub const UINT_FAST32_MAX: i32 = -1;
|
|
||||||
pub const INTPTR_MIN: i64 = -9223372036854775808;
|
|
||||||
pub const INTPTR_MAX: u64 = 9223372036854775807;
|
|
||||||
pub const UINTPTR_MAX: i32 = -1;
|
|
||||||
pub const PTRDIFF_MIN: i64 = -9223372036854775808;
|
|
||||||
pub const PTRDIFF_MAX: u64 = 9223372036854775807;
|
|
||||||
pub const SIG_ATOMIC_MIN: i32 = -2147483648;
|
|
||||||
pub const SIG_ATOMIC_MAX: u32 = 2147483647;
|
|
||||||
pub const SIZE_MAX: i32 = -1;
|
|
||||||
pub const WINT_MIN: u32 = 0;
|
|
||||||
pub const WINT_MAX: u32 = 4294967295;
|
|
||||||
pub const TSDB_DATA_TYPE_NULL: u32 = 0;
|
|
||||||
pub const TSDB_DATA_TYPE_BOOL: u32 = 1;
|
|
||||||
pub const TSDB_DATA_TYPE_TINYINT: u32 = 2;
|
|
||||||
pub const TSDB_DATA_TYPE_SMALLINT: u32 = 3;
|
|
||||||
pub const TSDB_DATA_TYPE_INT: u32 = 4;
|
|
||||||
pub const TSDB_DATA_TYPE_BIGINT: u32 = 5;
|
|
||||||
pub const TSDB_DATA_TYPE_FLOAT: u32 = 6;
|
|
||||||
pub const TSDB_DATA_TYPE_DOUBLE: u32 = 7;
|
|
||||||
pub const TSDB_DATA_TYPE_BINARY: u32 = 8;
|
|
||||||
pub const TSDB_DATA_TYPE_TIMESTAMP: u32 = 9;
|
|
||||||
pub const TSDB_DATA_TYPE_NCHAR: u32 = 10;
|
|
||||||
pub type int_least8_t = ::std::os::raw::c_schar;
|
|
||||||
pub type int_least16_t = ::std::os::raw::c_short;
|
|
||||||
pub type int_least32_t = ::std::os::raw::c_int;
|
|
||||||
pub type int_least64_t = ::std::os::raw::c_long;
|
|
||||||
pub type uint_least8_t = ::std::os::raw::c_uchar;
|
|
||||||
pub type uint_least16_t = ::std::os::raw::c_ushort;
|
|
||||||
pub type uint_least32_t = ::std::os::raw::c_uint;
|
|
||||||
pub type uint_least64_t = ::std::os::raw::c_ulong;
|
|
||||||
pub type int_fast8_t = ::std::os::raw::c_schar;
|
|
||||||
pub type int_fast16_t = ::std::os::raw::c_long;
|
|
||||||
pub type int_fast32_t = ::std::os::raw::c_long;
|
|
||||||
pub type int_fast64_t = ::std::os::raw::c_long;
|
|
||||||
pub type uint_fast8_t = ::std::os::raw::c_uchar;
|
|
||||||
pub type uint_fast16_t = ::std::os::raw::c_ulong;
|
|
||||||
pub type uint_fast32_t = ::std::os::raw::c_ulong;
|
|
||||||
pub type uint_fast64_t = ::std::os::raw::c_ulong;
|
|
||||||
pub type intmax_t = ::std::os::raw::c_long;
|
|
||||||
pub type uintmax_t = ::std::os::raw::c_ulong;
|
|
||||||
pub const TSDB_OPTION_TSDB_OPTION_LOCALE: TSDB_OPTION = 0;
|
|
||||||
pub const TSDB_OPTION_TSDB_OPTION_CHARSET: TSDB_OPTION = 1;
|
|
||||||
pub const TSDB_OPTION_TSDB_OPTION_TIMEZONE: TSDB_OPTION = 2;
|
|
||||||
pub const TSDB_OPTION_TSDB_OPTION_CONFIGDIR: TSDB_OPTION = 3;
|
|
||||||
pub const TSDB_OPTION_TSDB_OPTION_SHELL_ACTIVITY_TIMER: TSDB_OPTION = 4;
|
|
||||||
pub const TSDB_OPTION_TSDB_MAX_OPTIONS: TSDB_OPTION = 5;
|
|
||||||
pub type TSDB_OPTION = u32;
|
|
||||||
#[repr(C)]
|
|
||||||
#[derive(Copy, Clone)]
|
|
||||||
pub struct taosField {
|
|
||||||
pub name: [::std::os::raw::c_char; 64usize],
|
|
||||||
pub bytes: ::std::os::raw::c_short,
|
|
||||||
pub type_: ::std::os::raw::c_char,
|
|
||||||
}
|
|
||||||
#[test]
|
|
||||||
fn bindgen_test_layout_taosField() {
|
|
||||||
assert_eq!(
|
|
||||||
::std::mem::size_of::<taosField>(),
|
|
||||||
68usize,
|
|
||||||
concat!("Size of: ", stringify!(taosField))
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
::std::mem::align_of::<taosField>(),
|
|
||||||
2usize,
|
|
||||||
concat!("Alignment of ", stringify!(taosField))
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
unsafe { &(*(::std::ptr::null::<taosField>())).name as *const _ as usize },
|
|
||||||
0usize,
|
|
||||||
concat!(
|
|
||||||
"Offset of field: ",
|
|
||||||
stringify!(taosField),
|
|
||||||
"::",
|
|
||||||
stringify!(name)
|
|
||||||
)
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
unsafe { &(*(::std::ptr::null::<taosField>())).bytes as *const _ as usize },
|
|
||||||
64usize,
|
|
||||||
concat!(
|
|
||||||
"Offset of field: ",
|
|
||||||
stringify!(taosField),
|
|
||||||
"::",
|
|
||||||
stringify!(bytes)
|
|
||||||
)
|
|
||||||
);
|
|
||||||
assert_eq!(
|
|
||||||
unsafe { &(*(::std::ptr::null::<taosField>())).type_ as *const _ as usize },
|
|
||||||
66usize,
|
|
||||||
concat!(
|
|
||||||
"Offset of field: ",
|
|
||||||
stringify!(taosField),
|
|
||||||
"::",
|
|
||||||
stringify!(type_)
|
|
||||||
)
|
|
||||||
);
|
|
||||||
}
|
|
||||||
pub type TAOS_FIELD = taosField;
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_init();
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_options(
|
|
||||||
option: TSDB_OPTION,
|
|
||||||
arg: *const ::std::os::raw::c_void,
|
|
||||||
...
|
|
||||||
) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_connect(
|
|
||||||
ip: *mut ::std::os::raw::c_char,
|
|
||||||
user: *mut ::std::os::raw::c_char,
|
|
||||||
pass: *mut ::std::os::raw::c_char,
|
|
||||||
db: *mut ::std::os::raw::c_char,
|
|
||||||
port: ::std::os::raw::c_int,
|
|
||||||
) -> *mut ::std::os::raw::c_void;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_close(taos: *mut ::std::os::raw::c_void);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_query(
|
|
||||||
taos: *mut ::std::os::raw::c_void,
|
|
||||||
sqlstr: *mut ::std::os::raw::c_char,
|
|
||||||
) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_use_result(taos: *mut ::std::os::raw::c_void) -> *mut ::std::os::raw::c_void;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_fetch_row(res: *mut ::std::os::raw::c_void) -> *mut *mut ::std::os::raw::c_void;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_result_precision(res: *mut ::std::os::raw::c_void) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_free_result(res: *mut ::std::os::raw::c_void);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_field_count(taos: *mut ::std::os::raw::c_void) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_num_fields(res: *mut ::std::os::raw::c_void) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_affected_rows(taos: *mut ::std::os::raw::c_void) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_fetch_fields(res: *mut ::std::os::raw::c_void) -> *mut TAOS_FIELD;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_select_db(
|
|
||||||
taos: *mut ::std::os::raw::c_void,
|
|
||||||
db: *mut ::std::os::raw::c_char,
|
|
||||||
) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_print_row(
|
|
||||||
str: *mut ::std::os::raw::c_char,
|
|
||||||
row: *mut *mut ::std::os::raw::c_void,
|
|
||||||
fields: *mut TAOS_FIELD,
|
|
||||||
num_fields: ::std::os::raw::c_int,
|
|
||||||
) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_stop_query(res: *mut ::std::os::raw::c_void);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_fetch_block(
|
|
||||||
res: *mut ::std::os::raw::c_void,
|
|
||||||
rows: *mut *mut *mut ::std::os::raw::c_void,
|
|
||||||
) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_validate_sql(
|
|
||||||
taos: *mut ::std::os::raw::c_void,
|
|
||||||
sql: *mut ::std::os::raw::c_char,
|
|
||||||
) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_get_server_info(taos: *mut ::std::os::raw::c_void) -> *mut ::std::os::raw::c_char;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_get_client_info() -> *mut ::std::os::raw::c_char;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_errstr(taos: *mut ::std::os::raw::c_void) -> *mut ::std::os::raw::c_char;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_errno(taos: *mut ::std::os::raw::c_void) -> ::std::os::raw::c_int;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_query_a(
|
|
||||||
taos: *mut ::std::os::raw::c_void,
|
|
||||||
sqlstr: *mut ::std::os::raw::c_char,
|
|
||||||
fp: ::std::option::Option<
|
|
||||||
unsafe extern "C" fn(
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
arg1: *mut ::std::os::raw::c_void,
|
|
||||||
code: ::std::os::raw::c_int,
|
|
||||||
),
|
|
||||||
>,
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_fetch_rows_a(
|
|
||||||
res: *mut ::std::os::raw::c_void,
|
|
||||||
fp: ::std::option::Option<
|
|
||||||
unsafe extern "C" fn(
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
arg1: *mut ::std::os::raw::c_void,
|
|
||||||
numOfRows: ::std::os::raw::c_int,
|
|
||||||
),
|
|
||||||
>,
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_fetch_row_a(
|
|
||||||
res: *mut ::std::os::raw::c_void,
|
|
||||||
fp: ::std::option::Option<
|
|
||||||
unsafe extern "C" fn(
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
arg1: *mut ::std::os::raw::c_void,
|
|
||||||
row: *mut *mut ::std::os::raw::c_void,
|
|
||||||
),
|
|
||||||
>,
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_subscribe(
|
|
||||||
host: *mut ::std::os::raw::c_char,
|
|
||||||
user: *mut ::std::os::raw::c_char,
|
|
||||||
pass: *mut ::std::os::raw::c_char,
|
|
||||||
db: *mut ::std::os::raw::c_char,
|
|
||||||
table: *mut ::std::os::raw::c_char,
|
|
||||||
time: i64,
|
|
||||||
mseconds: ::std::os::raw::c_int,
|
|
||||||
) -> *mut ::std::os::raw::c_void;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_consume(tsub: *mut ::std::os::raw::c_void) -> *mut *mut ::std::os::raw::c_void;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_unsubscribe(tsub: *mut ::std::os::raw::c_void);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_open_stream(
|
|
||||||
taos: *mut ::std::os::raw::c_void,
|
|
||||||
sqlstr: *mut ::std::os::raw::c_char,
|
|
||||||
fp: ::std::option::Option<
|
|
||||||
unsafe extern "C" fn(
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
arg1: *mut ::std::os::raw::c_void,
|
|
||||||
row: *mut *mut ::std::os::raw::c_void,
|
|
||||||
),
|
|
||||||
>,
|
|
||||||
stime: i64,
|
|
||||||
param: *mut ::std::os::raw::c_void,
|
|
||||||
callback: ::std::option::Option<unsafe extern "C" fn(arg1: *mut ::std::os::raw::c_void)>,
|
|
||||||
) -> *mut ::std::os::raw::c_void;
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub fn taos_close_stream(tstr: *mut ::std::os::raw::c_void);
|
|
||||||
}
|
|
||||||
extern "C" {
|
|
||||||
pub static mut configDir: [::std::os::raw::c_char; 0usize];
|
|
||||||
}
|
|
|
@ -1,10 +0,0 @@
|
||||||
#![allow(unused)]
|
|
||||||
#![allow(non_camel_case_types)]
|
|
||||||
|
|
||||||
pub mod subscriber;
|
|
||||||
pub use subscriber::*;
|
|
||||||
|
|
||||||
pub mod tdengine;
|
|
||||||
pub use tdengine::*;
|
|
||||||
|
|
||||||
pub mod utils;
|
|
|
@ -1,77 +0,0 @@
|
||||||
#![allow(non_camel_case_types)]
|
|
||||||
#![allow(non_snake_case)]
|
|
||||||
|
|
||||||
#[path = "utils.rs"]
|
|
||||||
mod utils;
|
|
||||||
use utils::*;
|
|
||||||
use utils::bindings::*;
|
|
||||||
|
|
||||||
use std::os::raw::{c_void, c_char, c_int, c_long};
|
|
||||||
|
|
||||||
pub struct Subscriber {
|
|
||||||
tsub: *mut c_void,
|
|
||||||
fields: *mut taosField,
|
|
||||||
fcount: c_int,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Subscriber {
|
|
||||||
pub fn new(host: &str,
|
|
||||||
username: &str,
|
|
||||||
passwd: &str,
|
|
||||||
db: &str,
|
|
||||||
table:&str,
|
|
||||||
time: i64,
|
|
||||||
mseconds: i32
|
|
||||||
) -> Result<Subscriber, &'static str> {
|
|
||||||
unsafe {
|
|
||||||
let mut tsub = taos_subscribe(str_into_raw(host),
|
|
||||||
str_into_raw(username),
|
|
||||||
str_into_raw(passwd),
|
|
||||||
str_into_raw(db),
|
|
||||||
str_into_raw(table),
|
|
||||||
time as c_long,
|
|
||||||
mseconds as c_int);
|
|
||||||
if tsub.is_null() {
|
|
||||||
return Err("subscribe error")
|
|
||||||
}
|
|
||||||
println!("subscribed to {} user:{}, db:{}, tb:{}, time:{}, mseconds:{}",
|
|
||||||
host, username, db, table, time, mseconds);
|
|
||||||
|
|
||||||
let mut fields = taos_fetch_fields(tsub);
|
|
||||||
if fields.is_null() {
|
|
||||||
taos_unsubscribe(tsub);
|
|
||||||
return Err("fetch fields error")
|
|
||||||
}
|
|
||||||
|
|
||||||
let fcount = taos_field_count(tsub);
|
|
||||||
if fcount == 0 {
|
|
||||||
taos_unsubscribe(tsub);
|
|
||||||
return Err("fields count is 0")
|
|
||||||
}
|
|
||||||
|
|
||||||
Ok(Subscriber{tsub, fields, fcount})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn consume(self: &Subscriber) -> Result<Row, &'static str> {
|
|
||||||
unsafe {
|
|
||||||
let taosRow = taos_consume(self.tsub);
|
|
||||||
if taosRow.is_null() {
|
|
||||||
return Err("consume error")
|
|
||||||
}
|
|
||||||
let taosRow= std::slice::from_raw_parts(taosRow, self.fcount as usize);
|
|
||||||
let row = raw_into_row(self.fields, self.fcount, &taosRow);
|
|
||||||
Ok(row)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn print_row(self: &Subscriber, row: &Row) {
|
|
||||||
println!("{}", format_row(row));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Drop for Subscriber {
|
|
||||||
fn drop(&mut self) {
|
|
||||||
unsafe {taos_unsubscribe(self.tsub);}
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,65 +0,0 @@
|
||||||
#[path = "bindings.rs"]
|
|
||||||
mod bindings;
|
|
||||||
use bindings::*;
|
|
||||||
|
|
||||||
#[path = "utils.rs"]
|
|
||||||
mod utils;
|
|
||||||
use utils::*;
|
|
||||||
|
|
||||||
use std::os::raw::c_void;
|
|
||||||
use std::os::raw::c_char;
|
|
||||||
use std::os::raw::c_int;
|
|
||||||
use std::os::raw::c_long;
|
|
||||||
|
|
||||||
pub struct Tdengine {
|
|
||||||
conn: *mut c_void,
|
|
||||||
}
|
|
||||||
|
|
||||||
/// - **TODO**: doc
|
|
||||||
impl Tdengine {
|
|
||||||
|
|
||||||
//! - **TODO**: implement default param.
|
|
||||||
//!
|
|
||||||
//! > refer to https://stackoverflow.com/questions/24047686/default-function-arguments-in-rust
|
|
||||||
pub fn new(ip: &str, username: &str, passwd: &str, db: &str, port: i32) -> Result<Tdengine, &'static str> {
|
|
||||||
unsafe {
|
|
||||||
taos_init();
|
|
||||||
let mut conn = taos_connect(str_into_raw(ip),
|
|
||||||
str_into_raw(username),
|
|
||||||
str_into_raw(passwd),
|
|
||||||
str_into_raw(db),
|
|
||||||
port as c_int);
|
|
||||||
if conn.is_null() {
|
|
||||||
Err("connect error")
|
|
||||||
} else {
|
|
||||||
println!("connected to {}:{} user:{}, db:{}", ip, port, username, db);
|
|
||||||
Ok(Tdengine {conn})
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// - **TODO**: check error code
|
|
||||||
pub fn query(self: &Tdengine, s: &str) {
|
|
||||||
unsafe {
|
|
||||||
if taos_query(self.conn, str_into_raw(s)) == 0 {
|
|
||||||
println!("query '{}' ok", s);
|
|
||||||
} else {
|
|
||||||
println!("query '{}' error: {}", s, raw_into_str(taos_errstr(self.conn)));
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Drop for Tdengine {
|
|
||||||
fn drop(&mut self) {
|
|
||||||
unsafe {taos_close(self.conn);}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[cfg(test)]
|
|
||||||
mod tests {
|
|
||||||
#[test]
|
|
||||||
fn it_works() {
|
|
||||||
assert_eq!(2 + 2, 4);
|
|
||||||
}
|
|
||||||
}
|
|
|
@ -1,127 +0,0 @@
|
||||||
#[path = "bindings.rs"]
|
|
||||||
pub mod bindings;
|
|
||||||
use bindings::*;
|
|
||||||
|
|
||||||
use std::fmt;
|
|
||||||
use std::fmt::Display;
|
|
||||||
use std::os::raw::{c_void, c_char, c_int};
|
|
||||||
use std::ffi::{CString, CStr};
|
|
||||||
|
|
||||||
// #[derive(Debug)]
|
|
||||||
pub enum Field {
|
|
||||||
tinyInt(i8),
|
|
||||||
smallInt(i16),
|
|
||||||
normalInt(i32),
|
|
||||||
bigInt(i64),
|
|
||||||
float(f32),
|
|
||||||
double(f64),
|
|
||||||
binary(String),
|
|
||||||
timeStamp(i64),
|
|
||||||
boolType(bool),
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
impl fmt::Display for Field {
|
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
||||||
match &*self {
|
|
||||||
Field::tinyInt(v) => write!(f, "{}", v),
|
|
||||||
Field::smallInt(v) => write!(f, "{}", v),
|
|
||||||
Field::normalInt(v) => write!(f, "{}", v),
|
|
||||||
Field::bigInt(v) => write!(f, "{}", v),
|
|
||||||
Field::float(v) => write!(f, "{}", v),
|
|
||||||
Field::double(v) => write!(f, "{}", v),
|
|
||||||
Field::binary(v) => write!(f, "{}", v),
|
|
||||||
Field::tinyInt(v) => write!(f, "{}", v),
|
|
||||||
Field::timeStamp(v) => write!(f, "{}", v),
|
|
||||||
Field::boolType(v) => write!(f, "{}", v),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// pub type Fields = Vec<Field>;
|
|
||||||
pub type Row = Vec<Field>;
|
|
||||||
|
|
||||||
pub fn format_row(row: &Row) -> String {
|
|
||||||
let mut s = String::new();
|
|
||||||
for field in row {
|
|
||||||
s.push_str(format!("{} ", field).as_str());
|
|
||||||
// println!("{}", field);
|
|
||||||
}
|
|
||||||
s
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn str_into_raw(s: &str) -> *mut c_char {
|
|
||||||
if s.is_empty() {
|
|
||||||
0 as *mut c_char
|
|
||||||
} else {
|
|
||||||
CString::new(s).unwrap().into_raw()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn raw_into_str<'a>(raw: *mut c_char) -> &'static str {
|
|
||||||
unsafe {CStr::from_ptr(raw).to_str().unwrap()}
|
|
||||||
}
|
|
||||||
|
|
||||||
|
|
||||||
pub fn raw_into_field(raw: *mut TAOS_FIELD, fcount: c_int) -> Vec<taosField> {
|
|
||||||
let mut fields: Vec<taosField> = Vec::new();
|
|
||||||
|
|
||||||
for i in 0..fcount as isize {
|
|
||||||
fields.push(
|
|
||||||
taosField {
|
|
||||||
name: unsafe {(*raw.offset(i as isize))}.name,
|
|
||||||
bytes: unsafe {(*raw.offset(i as isize))}.bytes,
|
|
||||||
type_: unsafe {(*raw.offset(i as isize))}.type_,
|
|
||||||
}
|
|
||||||
);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// TODO: error[E0382]: use of moved value: `fields`
|
|
||||||
// for field in &fields {
|
|
||||||
// println!("type: {}, bytes: {}", field.type_, field.bytes);
|
|
||||||
// }
|
|
||||||
|
|
||||||
fields
|
|
||||||
}
|
|
||||||
|
|
||||||
pub fn raw_into_row(fields: *mut TAOS_FIELD, fcount: c_int, raw_row: &[*mut c_void]) -> Row {
|
|
||||||
let mut row: Row= Vec::new();
|
|
||||||
let fields = raw_into_field(fields, fcount);
|
|
||||||
|
|
||||||
for (i, field) in fields.iter().enumerate() {
|
|
||||||
// println!("index: {}, type: {}, bytes: {}", i, field.type_, field.bytes);
|
|
||||||
unsafe {
|
|
||||||
match field.type_ as u32 {
|
|
||||||
TSDB_DATA_TYPE_TINYINT => {
|
|
||||||
row.push(Field::tinyInt(*(raw_row[i] as *mut i8)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_SMALLINT => {
|
|
||||||
row.push(Field::smallInt(*(raw_row[i] as *mut i16)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_INT => {
|
|
||||||
row.push(Field::normalInt(*(raw_row[i] as *mut i32)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_BIGINT => {
|
|
||||||
row.push(Field::bigInt(*(raw_row[i] as *mut i64)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_FLOAT => {
|
|
||||||
row.push(Field::float(*(raw_row[i] as *mut f32)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_DOUBLE => {
|
|
||||||
row.push(Field::double(*(raw_row[i] as *mut f64)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_BINARY | TSDB_DATA_TYPE_NCHAR => {
|
|
||||||
// row.push(Field::binary(*(raw_row[i] as *mut f64)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_TIMESTAMP => {
|
|
||||||
row.push(Field::timeStamp(*(raw_row[i] as *mut i64)));
|
|
||||||
}
|
|
||||||
TSDB_DATA_TYPE_BOOL => {
|
|
||||||
// row.push(Field::boolType(*(raw_row[i] as *mut i8) as bool));
|
|
||||||
}
|
|
||||||
_ => println!(""),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
row
|
|
||||||
}
|
|
|
@ -16,112 +16,202 @@ import sys
|
||||||
import json
|
import json
|
||||||
import time
|
import time
|
||||||
import random
|
import random
|
||||||
# query sql
|
import requests
|
||||||
query_sql = [
|
from requests.auth import HTTPBasicAuth
|
||||||
# first supertable
|
func_list=['avg','count','twa','sum','stddev','leastsquares','min',
|
||||||
"select count(*) from test.meters ;",
|
'max','first','last','top','bottom','percentile','apercentile',
|
||||||
"select count(*) from test.meters where t3 > 2;",
|
'last_row','diff','spread']
|
||||||
"select count(*) from test.meters where ts <> '2020-05-13 10:00:00.002';",
|
condition_list=[
|
||||||
"select count(*) from test.meters where t7 like 'taos_1%';",
|
"where _c0 > now -10d ",
|
||||||
"select count(*) from test.meters where t7 like '_____2';",
|
'interval(10s)',
|
||||||
"select count(*) from test.meters where t8 like '%思%';",
|
'limit 10',
|
||||||
"select count(*) from test.meters interval(1n) order by ts desc;",
|
'group by',
|
||||||
#"select max(c0) from test.meters group by tbname",
|
'order by',
|
||||||
"select first(ts) from test.meters where t5 >5000 and t5<5100;",
|
'fill(null)'
|
||||||
"select last(ts) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select last_row(*) from test.meters;",
|
|
||||||
"select twa(c1) from test.t1 where ts > 1500000001000 and ts < 1500000101000" ,
|
|
||||||
"select avg(c1) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select bottom(c1, 2) from test.t1;",
|
|
||||||
"select diff(c1) from test.t1;",
|
|
||||||
"select leastsquares(c1, 1, 1) from test.t1 ;",
|
|
||||||
"select max(c1) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select min(c1) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select c1 + c2 + c1 / c5 + c4 + c2 from test.t1;",
|
|
||||||
"select percentile(c1, 50) from test.t1;",
|
|
||||||
"select spread(c1) from test.t1 ;",
|
|
||||||
"select stddev(c1) from test.t1;",
|
|
||||||
"select sum(c1) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select top(c1, 2) from test.meters where t5 >5000 and t5<5100;"
|
|
||||||
"select twa(c4) from test.t1 where ts > 1500000001000 and ts < 1500000101000" ,
|
|
||||||
"select avg(c4) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select bottom(c4, 2) from test.t1 where t5 >5000 and t5<5100;",
|
|
||||||
"select diff(c4) from test.t1 where t5 >5000 and t5<5100;",
|
|
||||||
"select leastsquares(c4, 1, 1) from test.t1 ;",
|
|
||||||
"select max(c4) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select min(c4) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select c5 + c2 + c4 / c5 + c4 + c2 from test.t1 ;",
|
|
||||||
"select percentile(c5, 50) from test.t1;",
|
|
||||||
"select spread(c5) from test.t1 ;",
|
|
||||||
"select stddev(c5) from test.t1 where t5 >5000 and t5<5100;",
|
|
||||||
"select sum(c5) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
"select top(c5, 2) from test.meters where t5 >5000 and t5<5100;",
|
|
||||||
#all vnode
|
|
||||||
"select count(*) from test.meters where t5 >5000 and t5<5100",
|
|
||||||
"select max(c0),avg(c1) from test.meters where t5 >5000 and t5<5100",
|
|
||||||
"select sum(c5),avg(c1) from test.meters where t5 >5000 and t5<5100",
|
|
||||||
"select max(c0),min(c5) from test.meters where t5 >5000 and t5<5100",
|
|
||||||
"select min(c0),avg(c5) from test.meters where t5 >5000 and t5<5100",
|
|
||||||
# second supertable
|
|
||||||
"select count(*) from test.meters1 where t3 > 2;",
|
|
||||||
"select count(*) from test.meters1 where ts <> '2020-05-13 10:00:00.002';",
|
|
||||||
"select count(*) from test.meters where t7 like 'taos_1%';",
|
|
||||||
"select count(*) from test.meters where t7 like '_____2';",
|
|
||||||
"select count(*) from test.meters where t8 like '%思%';",
|
|
||||||
"select count(*) from test.meters1 interval(1n) order by ts desc;",
|
|
||||||
#"select max(c0) from test.meters1 group by tbname",
|
|
||||||
"select first(ts) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select last(ts) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select last_row(*) from test.meters1 ;",
|
|
||||||
"select twa(c1) from test.m1 where ts > 1500000001000 and ts < 1500000101000" ,
|
|
||||||
"select avg(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select bottom(c1, 2) from test.m1 where t5 >5000 and t5<5100;",
|
|
||||||
"select diff(c1) from test.m1 ;",
|
|
||||||
"select leastsquares(c1, 1, 1) from test.m1 ;",
|
|
||||||
"select max(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select min(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select c1 + c2 + c1 / c0 + c2 from test.m1 ;",
|
|
||||||
"select percentile(c1, 50) from test.m1;",
|
|
||||||
"select spread(c1) from test.m1 ;",
|
|
||||||
"select stddev(c1) from test.m1;",
|
|
||||||
"select sum(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select top(c1, 2) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select twa(c5) from test.m1 where ts > 1500000001000 and ts < 1500000101000" ,
|
|
||||||
"select avg(c5) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select bottom(c5, 2) from test.m1;",
|
|
||||||
"select diff(c5) from test.m1;",
|
|
||||||
"select leastsquares(c5, 1, 1) from test.m1 ;",
|
|
||||||
"select max(c5) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select min(c5) from test.meters1 where t5 >5000 and t5<5100;",
|
|
||||||
"select c5 + c2 + c4 / c5 + c0 from test.m1;",
|
|
||||||
"select percentile(c4, 50) from test.m1;",
|
|
||||||
"select spread(c4) from test.m1 ;",
|
|
||||||
"select stddev(c4) from test.m1;",
|
|
||||||
"select sum(c4) from test.meters1 where t5 >5100 and t5<5300;",
|
|
||||||
"select top(c4, 2) from test.meters1 where t5 >5100 and t5<5300;",
|
|
||||||
"select count(*) from test.meters1 where t5 >5100 and t5<5300",
|
|
||||||
#all vnode
|
|
||||||
"select count(*) from test.meters1 where t5 >5100 and t5<5300",
|
|
||||||
"select max(c0),avg(c1) from test.meters1 where t5 >5000 and t5<5100",
|
|
||||||
"select sum(c5),avg(c1) from test.meters1 where t5 >5000 and t5<5100",
|
|
||||||
"select max(c0),min(c5) from test.meters1 where t5 >5000 and t5<5100",
|
|
||||||
"select min(c0),avg(c5) from test.meters1 where t5 >5000 and t5<5100",
|
|
||||||
#join
|
|
||||||
# "select * from meters,meters1 where meters.ts = meters1.ts and meters.t5 = meters1.t5",
|
|
||||||
# "select * from meters,meters1 where meters.ts = meters1.ts and meters.t7 = meters1.t7",
|
|
||||||
# "select * from meters,meters1 where meters.ts = meters1.ts and meters.t8 = meters1.t8",
|
|
||||||
# "select meters.ts,meters1.c2 from meters,meters1 where meters.ts = meters1.ts and meters.t8 = meters1.t8"
|
|
||||||
]
|
]
|
||||||
|
where_list = ['_c0>now-10d',' <50'," like \'%a%\'"]
|
||||||
class ConcurrentInquiry:
|
class ConcurrentInquiry:
|
||||||
def initConnection(self):
|
def __init__(self,n_Therads=25,r_Therads=25):
|
||||||
self.numOfTherads = 50
|
self.n_numOfTherads = n_Therads
|
||||||
|
self.r_numOfTherads = r_Therads
|
||||||
self.ts=1500000001000
|
self.ts=1500000001000
|
||||||
|
self.dbname='test'
|
||||||
|
self.stb_list=[]
|
||||||
|
self.subtb_list=[]
|
||||||
|
self.stb_stru_list=[]
|
||||||
|
self.subtb_stru_list=[]
|
||||||
|
self.stb_tag_list=[]
|
||||||
|
self.subtb_tag_list=[]
|
||||||
|
|
||||||
def SetThreadsNum(self,num):
|
def SetThreadsNum(self,num):
|
||||||
self.numOfTherads=num
|
self.numOfTherads=num
|
||||||
def query_thread(self,threadID):
|
|
||||||
host = "10.211.55.14"
|
def ret_fcol(self,cl,sql): #返回结果的第一列
|
||||||
|
cl.execute(sql)
|
||||||
|
fcol_list=[]
|
||||||
|
for data in cl:
|
||||||
|
fcol_list.append(data[0])
|
||||||
|
return fcol_list
|
||||||
|
|
||||||
|
def r_stb_list(self,cl): #返回超级表列表
|
||||||
|
sql='show '+self.dbname+'.stables'
|
||||||
|
self.stb_list=self.ret_fcol(cl,sql)
|
||||||
|
|
||||||
|
def r_subtb_list(self,cl,stablename): #每个超级表返回2个子表
|
||||||
|
sql='select tbname from '+self.dbname+'.'+stablename+' limit 2;'
|
||||||
|
self.subtb_list+=self.ret_fcol(cl,sql)
|
||||||
|
|
||||||
|
def cal_struct(self,cl,tbname): #查看表结构
|
||||||
|
tb=[]
|
||||||
|
tag=[]
|
||||||
|
sql='describe '+self.dbname+'.'+tbname+';'
|
||||||
|
cl.execute(sql)
|
||||||
|
for data in cl:
|
||||||
|
if data[3]:
|
||||||
|
tag.append(data[0])
|
||||||
|
else:
|
||||||
|
tb.append(data[0])
|
||||||
|
return tb,tag
|
||||||
|
|
||||||
|
def r_stb_stru(self,cl): #获取所有超级表的表结构
|
||||||
|
for i in self.stb_list:
|
||||||
|
tb,tag=self.cal_struct(cl,i)
|
||||||
|
self.stb_stru_list.append(tb)
|
||||||
|
self.stb_tag_list.append(tag)
|
||||||
|
|
||||||
|
def r_subtb_stru(self,cl): #返回所有子表的表结构
|
||||||
|
for i in self.subtb_list:
|
||||||
|
tb,tag=self.cal_struct(cl,i)
|
||||||
|
self.subtb_stru_list.append(tb)
|
||||||
|
self.subtb_tag_list.append(tag)
|
||||||
|
|
||||||
|
def get_full(self): #获取所有的表、表结构
|
||||||
|
host = "127.0.0.1"
|
||||||
|
user = "root"
|
||||||
|
password = "taosdata"
|
||||||
|
conn = taos.connect(
|
||||||
|
host,
|
||||||
|
user,
|
||||||
|
password,
|
||||||
|
)
|
||||||
|
cl = conn.cursor()
|
||||||
|
self.r_stb_list(cl)
|
||||||
|
for i in self.stb_list:
|
||||||
|
self.r_subtb_list(cl,i)
|
||||||
|
self.r_stb_stru(cl)
|
||||||
|
self.r_subtb_stru(cl)
|
||||||
|
cl.close()
|
||||||
|
conn.close()
|
||||||
|
|
||||||
|
#query condition
|
||||||
|
def con_where(self,tlist):
|
||||||
|
l=[]
|
||||||
|
for i in range(random.randint(0,len(tlist))):
|
||||||
|
c = random.choice(where_list)
|
||||||
|
if c == '_c0>now-10d':
|
||||||
|
l.append(c)
|
||||||
|
else:
|
||||||
|
l.append(random.choice(tlist)+c)
|
||||||
|
return 'where '+random.choice([' and ',' or ']).join(l)
|
||||||
|
|
||||||
|
def con_interval(self,tlist):
|
||||||
|
return random.choice(['interval(10s)','interval(10d)','interval(1n)'])
|
||||||
|
|
||||||
|
def con_limit(self,tlist):
|
||||||
|
return random.choice(['limit 10','limit 10 offset 10','slimit 10','slimit 10 offset 10','limit 10 slimit 10','limit 10 offset 5 slimit 5 soffset 10'])
|
||||||
|
|
||||||
|
def con_fill(self,tlist):
|
||||||
|
return random.choice(['fill(null)','fill(prev)','fill(none)','fill(LINEAR)'])
|
||||||
|
|
||||||
|
def con_group(self,tlist):
|
||||||
|
return 'group by '+random.choice(tlist)
|
||||||
|
|
||||||
|
def con_order(self,tlist):
|
||||||
|
return 'order by '+random.choice(tlist)
|
||||||
|
|
||||||
|
def gen_query_sql(self): #生成查询语句
|
||||||
|
tbi=random.randint(0,len(self.subtb_list)+len(self.stb_list)) #随机决定查询哪张表
|
||||||
|
tbname=''
|
||||||
|
col_list=[]
|
||||||
|
tag_list=[]
|
||||||
|
is_stb=0
|
||||||
|
if tbi>len(self.stb_list) :
|
||||||
|
tbi=tbi-len(self.stb_list)
|
||||||
|
tbname=self.subtb_list[tbi-1]
|
||||||
|
col_list=self.subtb_stru_list[tbi-1]
|
||||||
|
tag_list=self.subtb_tag_list[tbi-1]
|
||||||
|
else:
|
||||||
|
tbname=self.stb_list[tbi-1]
|
||||||
|
col_list=self.stb_stru_list[tbi-1]
|
||||||
|
tag_list=self.stb_tag_list[tbi-1]
|
||||||
|
is_stb=1
|
||||||
|
tlist=col_list+tag_list
|
||||||
|
con_rand=random.randint(0,len(condition_list))
|
||||||
|
func_rand=random.randint(0,len(func_list))
|
||||||
|
col_rand=random.randint(0,len(col_list))
|
||||||
|
tag_rand=random.randint(0,len(tag_list))
|
||||||
|
t_rand=random.randint(0,len(tlist))
|
||||||
|
sql='select ' #select
|
||||||
|
random.shuffle(col_list)
|
||||||
|
random.shuffle(func_list)
|
||||||
|
sel_col_list=[]
|
||||||
|
col_rand=random.randint(0,len(col_list))
|
||||||
|
for i,j in zip(col_list[0:col_rand],func_list): #决定每个被查询col的函数
|
||||||
|
if j == 'leastsquares':
|
||||||
|
sel_col_list.append(j+'('+i+',1,1)')
|
||||||
|
elif j == 'top' or j == 'bottom' or j == 'percentile' or j == 'apercentile':
|
||||||
|
sel_col_list.append(j+'('+i+',1)')
|
||||||
|
else:
|
||||||
|
sel_col_list.append(j+'('+i+')')
|
||||||
|
sql=sql+','.join(sel_col_list)+' from '+random.choice(self.stb_list+self.subtb_list)+' ' #select col & func
|
||||||
|
con_func=[self.con_where,self.con_interval,self.con_limit,self.con_group,self.con_order,self.con_fill]
|
||||||
|
sel_con=random.sample(con_func,random.randint(0,len(con_func)))
|
||||||
|
sel_con_list=[]
|
||||||
|
for i in sel_con:
|
||||||
|
sel_con_list.append(i(tlist)) #获取对应的条件函数
|
||||||
|
sql+=' '.join(sel_con_list) # condition
|
||||||
|
print(sql)
|
||||||
|
return sql
|
||||||
|
|
||||||
|
def rest_query(self,sql): #rest 接口
|
||||||
|
host = "127.0.0.1"
|
||||||
|
user = "root"
|
||||||
|
password = "taosdata"
|
||||||
|
port =6041
|
||||||
|
url = "http://{}:{}/rest/sql".format(host, port )
|
||||||
|
try:
|
||||||
|
r = requests.post(url,
|
||||||
|
data = 'use test',
|
||||||
|
auth = HTTPBasicAuth('root', 'taosdata'))
|
||||||
|
r = requests.post(url,
|
||||||
|
data = sql,
|
||||||
|
auth = HTTPBasicAuth('root', 'taosdata'))
|
||||||
|
except:
|
||||||
|
print("REST API Failure (TODO: more info here)")
|
||||||
|
raise
|
||||||
|
rj = r.json()
|
||||||
|
if ('status' not in rj):
|
||||||
|
raise RuntimeError("No status in REST response")
|
||||||
|
|
||||||
|
if rj['status'] == 'error': # clearly reported error
|
||||||
|
if ('code' not in rj): # error without code
|
||||||
|
raise RuntimeError("REST error return without code")
|
||||||
|
errno = rj['code'] # May need to massage this in the future
|
||||||
|
# print("Raising programming error with REST return: {}".format(rj))
|
||||||
|
raise taos.error.ProgrammingError(
|
||||||
|
rj['desc'], errno) # todo: check existance of 'desc'
|
||||||
|
|
||||||
|
if rj['status'] != 'succ': # better be this
|
||||||
|
raise RuntimeError(
|
||||||
|
"Unexpected REST return status: {}".format(
|
||||||
|
rj['status']))
|
||||||
|
|
||||||
|
nRows = rj['rows'] if ('rows' in rj) else 0
|
||||||
|
return nRows
|
||||||
|
|
||||||
|
def query_thread_n(self,threadID): #使用原生python接口查询
|
||||||
|
host = "127.0.0.1"
|
||||||
user = "root"
|
user = "root"
|
||||||
password = "taosdata"
|
password = "taosdata"
|
||||||
conn = taos.connect(
|
conn = taos.connect(
|
||||||
|
@ -135,35 +225,59 @@ class ConcurrentInquiry:
|
||||||
print("Thread %d: starting" % threadID)
|
print("Thread %d: starting" % threadID)
|
||||||
|
|
||||||
while True:
|
while True:
|
||||||
ran_query_sql=query_sql
|
|
||||||
random.shuffle(ran_query_sql)
|
|
||||||
for i in ran_query_sql:
|
|
||||||
print("Thread %d : %s"% (threadID,i))
|
|
||||||
try:
|
try:
|
||||||
|
sql=self.gen_query_sql()
|
||||||
|
print("sql is ",sql)
|
||||||
start = time.time()
|
start = time.time()
|
||||||
cl.execute(i)
|
cl.execute(sql)
|
||||||
cl.fetchall
|
cl.fetchall()
|
||||||
end = time.time()
|
end = time.time()
|
||||||
print("time cost :",end-start)
|
print("time cost :",end-start)
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
print(
|
print(
|
||||||
"Failure thread%d, sql: %s,exception: %s" %
|
"Failure thread%d, sql: %s,exception: %s" %
|
||||||
(threadID, str(i),str(e)))
|
(threadID, str(sql),str(e)))
|
||||||
exit(-1)
|
#exit(-1)
|
||||||
|
|
||||||
|
|
||||||
print("Thread %d: finishing" % threadID)
|
print("Thread %d: finishing" % threadID)
|
||||||
|
|
||||||
|
def query_thread_r(self,threadID): #使用rest接口查询
|
||||||
|
print("Thread %d: starting" % threadID)
|
||||||
|
while True:
|
||||||
|
try:
|
||||||
|
sql=self.gen_query_sql()
|
||||||
|
print("sql is ",sql)
|
||||||
|
start = time.time()
|
||||||
|
self.rest_query(sql)
|
||||||
|
end = time.time()
|
||||||
|
print("time cost :",end-start)
|
||||||
|
except Exception as e:
|
||||||
|
print(
|
||||||
|
"Failure thread%d, sql: %s,exception: %s" %
|
||||||
|
(threadID, str(sql),str(e)))
|
||||||
|
#exit(-1)
|
||||||
|
|
||||||
|
|
||||||
|
print("Thread %d: finishing" % threadID)
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
|
print(self.n_numOfTherads,self.r_numOfTherads)
|
||||||
threads = []
|
threads = []
|
||||||
for i in range(self.numOfTherads):
|
for i in range(self.n_numOfTherads):
|
||||||
thread = threading.Thread(target=self.query_thread, args=(i,))
|
thread = threading.Thread(target=self.query_thread_n, args=(i,))
|
||||||
threads.append(thread)
|
threads.append(thread)
|
||||||
thread.start()
|
thread.start()
|
||||||
|
for i in range(self.r_numOfTherads):
|
||||||
q = ConcurrentInquiry()
|
# for i in range(1):
|
||||||
q.initConnection()
|
thread = threading.Thread(target=self.query_thread_r, args=(i,))
|
||||||
|
threads.append(thread)
|
||||||
|
thread.start()
|
||||||
|
if len(sys.argv)>1:
|
||||||
|
q = ConcurrentInquiry(n_Therads=sys.argv[1],r_Therads=sys.argv[2])
|
||||||
|
else:
|
||||||
|
q = ConcurrentInquiry()
|
||||||
|
q.get_full()
|
||||||
|
#q.gen_query_sql()
|
||||||
q.run()
|
q.run()
|
||||||
|
|
|
@ -0,0 +1,169 @@
|
||||||
|
###################################################################
|
||||||
|
# Copyright (c) 2016 by TAOS Technologies, Inc.
|
||||||
|
# All rights reserved.
|
||||||
|
#
|
||||||
|
# This file is proprietary and confidential to TAOS Technologies.
|
||||||
|
# No part of this file may be reproduced, stored, transmitted,
|
||||||
|
# disclosed or used in any form or by any means other than as
|
||||||
|
# expressly provided by the written permission from Jianhui Tao
|
||||||
|
#
|
||||||
|
###################################################################
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
import threading
|
||||||
|
import taos
|
||||||
|
import sys
|
||||||
|
import json
|
||||||
|
import time
|
||||||
|
import random
|
||||||
|
# query sql
|
||||||
|
query_sql = [
|
||||||
|
# first supertable
|
||||||
|
"select count(*) from test.meters ;",
|
||||||
|
"select count(*) from test.meters where t3 > 2;",
|
||||||
|
"select count(*) from test.meters where ts <> '2020-05-13 10:00:00.002';",
|
||||||
|
"select count(*) from test.meters where t7 like 'taos_1%';",
|
||||||
|
"select count(*) from test.meters where t7 like '_____2';",
|
||||||
|
"select count(*) from test.meters where t8 like '%思%';",
|
||||||
|
"select count(*) from test.meters interval(1n) order by ts desc;",
|
||||||
|
#"select max(c0) from test.meters group by tbname",
|
||||||
|
"select first(ts) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select last(ts) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select last_row(*) from test.meters;",
|
||||||
|
"select twa(c1) from test.t1 where ts > 1500000001000 and ts < 1500000101000" ,
|
||||||
|
"select avg(c1) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select bottom(c1, 2) from test.t1;",
|
||||||
|
"select diff(c1) from test.t1;",
|
||||||
|
"select leastsquares(c1, 1, 1) from test.t1 ;",
|
||||||
|
"select max(c1) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select min(c1) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select c1 + c2 + c1 / c5 + c4 + c2 from test.t1;",
|
||||||
|
"select percentile(c1, 50) from test.t1;",
|
||||||
|
"select spread(c1) from test.t1 ;",
|
||||||
|
"select stddev(c1) from test.t1;",
|
||||||
|
"select sum(c1) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select top(c1, 2) from test.meters where t5 >5000 and t5<5100;"
|
||||||
|
"select twa(c4) from test.t1 where ts > 1500000001000 and ts < 1500000101000" ,
|
||||||
|
"select avg(c4) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select bottom(c4, 2) from test.t1 where t5 >5000 and t5<5100;",
|
||||||
|
"select diff(c4) from test.t1 where t5 >5000 and t5<5100;",
|
||||||
|
"select leastsquares(c4, 1, 1) from test.t1 ;",
|
||||||
|
"select max(c4) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select min(c4) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select c5 + c2 + c4 / c5 + c4 + c2 from test.t1 ;",
|
||||||
|
"select percentile(c5, 50) from test.t1;",
|
||||||
|
"select spread(c5) from test.t1 ;",
|
||||||
|
"select stddev(c5) from test.t1 where t5 >5000 and t5<5100;",
|
||||||
|
"select sum(c5) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
"select top(c5, 2) from test.meters where t5 >5000 and t5<5100;",
|
||||||
|
#all vnode
|
||||||
|
"select count(*) from test.meters where t5 >5000 and t5<5100",
|
||||||
|
"select max(c0),avg(c1) from test.meters where t5 >5000 and t5<5100",
|
||||||
|
"select sum(c5),avg(c1) from test.meters where t5 >5000 and t5<5100",
|
||||||
|
"select max(c0),min(c5) from test.meters where t5 >5000 and t5<5100",
|
||||||
|
"select min(c0),avg(c5) from test.meters where t5 >5000 and t5<5100",
|
||||||
|
# second supertable
|
||||||
|
"select count(*) from test.meters1 where t3 > 2;",
|
||||||
|
"select count(*) from test.meters1 where ts <> '2020-05-13 10:00:00.002';",
|
||||||
|
"select count(*) from test.meters where t7 like 'taos_1%';",
|
||||||
|
"select count(*) from test.meters where t7 like '_____2';",
|
||||||
|
"select count(*) from test.meters where t8 like '%思%';",
|
||||||
|
"select count(*) from test.meters1 interval(1n) order by ts desc;",
|
||||||
|
#"select max(c0) from test.meters1 group by tbname",
|
||||||
|
"select first(ts) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select last(ts) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select last_row(*) from test.meters1 ;",
|
||||||
|
"select twa(c1) from test.m1 where ts > 1500000001000 and ts < 1500000101000" ,
|
||||||
|
"select avg(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select bottom(c1, 2) from test.m1 where t5 >5000 and t5<5100;",
|
||||||
|
"select diff(c1) from test.m1 ;",
|
||||||
|
"select leastsquares(c1, 1, 1) from test.m1 ;",
|
||||||
|
"select max(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select min(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select c1 + c2 + c1 / c0 + c2 from test.m1 ;",
|
||||||
|
"select percentile(c1, 50) from test.m1;",
|
||||||
|
"select spread(c1) from test.m1 ;",
|
||||||
|
"select stddev(c1) from test.m1;",
|
||||||
|
"select sum(c1) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select top(c1, 2) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select twa(c5) from test.m1 where ts > 1500000001000 and ts < 1500000101000" ,
|
||||||
|
"select avg(c5) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select bottom(c5, 2) from test.m1;",
|
||||||
|
"select diff(c5) from test.m1;",
|
||||||
|
"select leastsquares(c5, 1, 1) from test.m1 ;",
|
||||||
|
"select max(c5) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select min(c5) from test.meters1 where t5 >5000 and t5<5100;",
|
||||||
|
"select c5 + c2 + c4 / c5 + c0 from test.m1;",
|
||||||
|
"select percentile(c4, 50) from test.m1;",
|
||||||
|
"select spread(c4) from test.m1 ;",
|
||||||
|
"select stddev(c4) from test.m1;",
|
||||||
|
"select sum(c4) from test.meters1 where t5 >5100 and t5<5300;",
|
||||||
|
"select top(c4, 2) from test.meters1 where t5 >5100 and t5<5300;",
|
||||||
|
"select count(*) from test.meters1 where t5 >5100 and t5<5300",
|
||||||
|
#all vnode
|
||||||
|
"select count(*) from test.meters1 where t5 >5100 and t5<5300",
|
||||||
|
"select max(c0),avg(c1) from test.meters1 where t5 >5000 and t5<5100",
|
||||||
|
"select sum(c5),avg(c1) from test.meters1 where t5 >5000 and t5<5100",
|
||||||
|
"select max(c0),min(c5) from test.meters1 where t5 >5000 and t5<5100",
|
||||||
|
"select min(c0),avg(c5) from test.meters1 where t5 >5000 and t5<5100",
|
||||||
|
#join
|
||||||
|
# "select * from meters,meters1 where meters.ts = meters1.ts and meters.t5 = meters1.t5",
|
||||||
|
# "select * from meters,meters1 where meters.ts = meters1.ts and meters.t7 = meters1.t7",
|
||||||
|
# "select * from meters,meters1 where meters.ts = meters1.ts and meters.t8 = meters1.t8",
|
||||||
|
# "select meters.ts,meters1.c2 from meters,meters1 where meters.ts = meters1.ts and meters.t8 = meters1.t8"
|
||||||
|
]
|
||||||
|
|
||||||
|
class ConcurrentInquiry:
|
||||||
|
def initConnection(self):
|
||||||
|
self.numOfTherads = 50
|
||||||
|
self.ts=1500000001000
|
||||||
|
|
||||||
|
def SetThreadsNum(self,num):
|
||||||
|
self.numOfTherads=num
|
||||||
|
def query_thread(self,threadID):
|
||||||
|
host = "10.211.55.14"
|
||||||
|
user = "root"
|
||||||
|
password = "taosdata"
|
||||||
|
conn = taos.connect(
|
||||||
|
host,
|
||||||
|
user,
|
||||||
|
password,
|
||||||
|
)
|
||||||
|
cl = conn.cursor()
|
||||||
|
cl.execute("use test;")
|
||||||
|
|
||||||
|
print("Thread %d: starting" % threadID)
|
||||||
|
|
||||||
|
while True:
|
||||||
|
ran_query_sql=query_sql
|
||||||
|
random.shuffle(ran_query_sql)
|
||||||
|
for i in ran_query_sql:
|
||||||
|
print("Thread %d : %s"% (threadID,i))
|
||||||
|
try:
|
||||||
|
start = time.time()
|
||||||
|
cl.execute(i)
|
||||||
|
cl.fetchall()
|
||||||
|
end = time.time()
|
||||||
|
print("time cost :",end-start)
|
||||||
|
except Exception as e:
|
||||||
|
print(
|
||||||
|
"Failure thread%d, sql: %s,exception: %s" %
|
||||||
|
(threadID, str(i),str(e)))
|
||||||
|
exit(-1)
|
||||||
|
|
||||||
|
|
||||||
|
print("Thread %d: finishing" % threadID)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
|
||||||
|
threads = []
|
||||||
|
for i in range(self.numOfTherads):
|
||||||
|
thread = threading.Thread(target=self.query_thread, args=(i,))
|
||||||
|
threads.append(thread)
|
||||||
|
thread.start()
|
||||||
|
|
||||||
|
q = ConcurrentInquiry()
|
||||||
|
q.initConnection()
|
||||||
|
q.run()
|
|
@ -0,0 +1,71 @@
|
||||||
|
###################################################################
|
||||||
|
# Copyright (c) 2016 by TAOS Technologies, Inc.
|
||||||
|
# All rights reserved.
|
||||||
|
#
|
||||||
|
# This file is proprietary and confidential to TAOS Technologies.
|
||||||
|
# No part of this file may be reproduced, stored, transmitted,
|
||||||
|
# disclosed or used in any form or by any means other than as
|
||||||
|
# expressly provided by the written permission from Jianhui Tao
|
||||||
|
#
|
||||||
|
###################################################################
|
||||||
|
|
||||||
|
# -*- coding: utf-8 -*-
|
||||||
|
|
||||||
|
import sys
|
||||||
|
import taos
|
||||||
|
from util.log import *
|
||||||
|
from util.cases import *
|
||||||
|
from util.sql import *
|
||||||
|
import numpy as np
|
||||||
|
|
||||||
|
|
||||||
|
class TDTestCase:
|
||||||
|
def init(self, conn, logSql):
|
||||||
|
tdLog.debug("start to execute %s" % __file__)
|
||||||
|
tdSql.init(conn.cursor())
|
||||||
|
|
||||||
|
self.rowNum = 10
|
||||||
|
self.ts = 1537146000000
|
||||||
|
|
||||||
|
def run(self):
|
||||||
|
tdSql.prepare()
|
||||||
|
|
||||||
|
tdSql.execute('''create table test(ts timestamp, col1 int, col2 int) tags(loc nchar(20))''')
|
||||||
|
tdSql.execute("create table test1 using test tags('beijing')")
|
||||||
|
tdSql.execute("create table test2 using test tags('shanghai')")
|
||||||
|
for i in range(self.rowNum):
|
||||||
|
tdSql.execute("insert into test1 values(%d, %d, %d)" % (self.ts + i, i + 1, i + 1))
|
||||||
|
tdSql.execute("insert into test2 values(%d, %d, %d)" % (self.ts + i, i + 1, i + 1))
|
||||||
|
|
||||||
|
# arithmetic verifacation
|
||||||
|
tdSql.query("select 0.1 + 0.1 from test")
|
||||||
|
tdSql.checkRows(self.rowNum * 2)
|
||||||
|
for i in range(self.rowNum * 2):
|
||||||
|
tdSql.checkData(0, 0, 0.20000000)
|
||||||
|
|
||||||
|
tdSql.query("select 4 * avg(col1) from test")
|
||||||
|
tdSql.checkRows(1)
|
||||||
|
tdSql.checkData(0, 0, 22)
|
||||||
|
|
||||||
|
tdSql.query("select 4 * sum(col1) from test")
|
||||||
|
tdSql.checkRows(1)
|
||||||
|
tdSql.checkData(0, 0, 440)
|
||||||
|
|
||||||
|
tdSql.query("select 4 * avg(col1) * sum(col2) from test")
|
||||||
|
tdSql.checkRows(1)
|
||||||
|
tdSql.checkData(0, 0, 2420)
|
||||||
|
|
||||||
|
tdSql.query("select 4 * avg(col1) * sum(col2) from test group by loc")
|
||||||
|
tdSql.checkRows(2)
|
||||||
|
tdSql.checkData(0, 0, 1210)
|
||||||
|
tdSql.checkData(1, 0, 1210)
|
||||||
|
|
||||||
|
tdSql.error("select avg(col1 * 2)from test group by loc")
|
||||||
|
|
||||||
|
|
||||||
|
def stop(self):
|
||||||
|
tdSql.close()
|
||||||
|
tdLog.success("%s successfully executed" % __file__)
|
||||||
|
|
||||||
|
tdCases.addWindows(__file__, TDTestCase())
|
||||||
|
tdCases.addLinux(__file__, TDTestCase())
|
|
@ -15,25 +15,28 @@ import requests
|
||||||
import threading
|
import threading
|
||||||
import random
|
import random
|
||||||
import time
|
import time
|
||||||
|
import argparse
|
||||||
|
|
||||||
class RestfulInsert:
|
class RestfulInsert:
|
||||||
def init(self):
|
def __init__(self, host, dbname, threads, tables, records, batchSize, tbNamePerfix, outOfOrder):
|
||||||
self.header = {'Authorization': 'Basic cm9vdDp0YW9zZGF0YQ=='}
|
self.header = {'Authorization': 'Basic cm9vdDp0YW9zZGF0YQ=='}
|
||||||
self.url = "http://127.0.0.1:6041/rest/sql"
|
self.url = "http://%s:6041/rest/sql" % host
|
||||||
self.ts = 1500000000000
|
self.ts = 1500000000000
|
||||||
self.numOfThreads = 20
|
self.dbname = dbname
|
||||||
self.numOfTables = 10000
|
self.numOfThreads = threads
|
||||||
self.recordsPerTable = 10000
|
self.numOfTables = tables
|
||||||
self.batchSize = 1000
|
self.recordsPerTable = records
|
||||||
self.tableNamePerfix = 't'
|
self.batchSize = batchSize
|
||||||
|
self.tableNamePerfix = tbNamePerfix
|
||||||
|
self.outOfOrder = outOfOrder
|
||||||
|
|
||||||
def createTable(self, threadID):
|
def createTable(self, threadID):
|
||||||
tablesPerThread = int (self.numOfTables / self.numOfThreads)
|
tablesPerThread = int (self.numOfTables / self.numOfThreads)
|
||||||
print("create table %d to %d" % (tablesPerThread * threadID, tablesPerThread * (threadID + 1) - 1))
|
print("create table %d to %d" % (tablesPerThread * threadID, tablesPerThread * (threadID + 1) - 1))
|
||||||
for i in range(tablesPerThread):
|
for i in range(tablesPerThread):
|
||||||
tableID = threadID * tablesPerThread
|
tableID = threadID * tablesPerThread
|
||||||
name = 'beijing' if tableID % 2 == 0 else 'shanghai'
|
name = 'beijing' if tableID % 2 == 0 else 'shanghai'
|
||||||
data = "create table test.%s%d using test.meters tags(%d, '%s')" % (self.tableNamePerfix, tableID + i, tableID + i, name)
|
data = "create table %s.%s%d using %s.meters tags(%d, '%s')" % (self.dbname, self.tableNamePerfix, tableID + i, self.dbname, tableID + i, name)
|
||||||
requests.post(self.url, data, headers = self.header)
|
requests.post(self.url, data, headers = self.header)
|
||||||
|
|
||||||
def insertData(self, threadID):
|
def insertData(self, threadID):
|
||||||
|
@ -43,17 +46,42 @@ class RestfulInsert:
|
||||||
tableID = i + threadID * tablesPerThread
|
tableID = i + threadID * tablesPerThread
|
||||||
start = self.ts
|
start = self.ts
|
||||||
for j in range(int(self.recordsPerTable / self.batchSize)):
|
for j in range(int(self.recordsPerTable / self.batchSize)):
|
||||||
data = "insert into test.%s%d values" % (self.tableNamePerfix, tableID)
|
data = "insert into %s.%s%d values" % (self.dbname, self.tableNamePerfix, tableID)
|
||||||
|
values = []
|
||||||
for k in range(self.batchSize):
|
for k in range(self.batchSize):
|
||||||
data += "(%d, %d, %d, %d)" % (start + j * self.batchSize + k, random.randint(1, 100), random.randint(1, 100), random.randint(1, 100))
|
data += "(%d, %d, %d, %d)" % (start + j * self.batchSize + k, random.randint(1, 100), random.randint(1, 100), random.randint(1, 100))
|
||||||
|
requests.post(self.url, data, headers = self.header)
|
||||||
|
|
||||||
|
def insertUnlimitedData(self, threadID):
|
||||||
|
print("thread %d started" % threadID)
|
||||||
|
tablesPerThread = int (self.numOfTables / self.numOfThreads)
|
||||||
|
while True:
|
||||||
|
i = 0
|
||||||
|
start = self.ts
|
||||||
|
|
||||||
|
for i in range(tablesPerThread):
|
||||||
|
tableID = i + threadID * tablesPerThread
|
||||||
|
|
||||||
|
data = "insert into %s.%s%d values" % (self.dbname, self.tableNamePerfix, tableID)
|
||||||
|
values = []
|
||||||
|
for k in range(self.batchSize):
|
||||||
|
values.append("(%d, %d, %d, %d)" % (start + j * self.batchSize + k, random.randint(1, 100), random.randint(1, 100), random.randint(1, 100)))
|
||||||
|
|
||||||
|
if(self.outOfOrder == False):
|
||||||
|
for k in range(len(values)):
|
||||||
|
data += values[k]
|
||||||
|
else:
|
||||||
|
random.shuffle(values)
|
||||||
|
for k in range(len(values)):
|
||||||
|
data += values[k]
|
||||||
requests.post(self.url, data, headers = self.header)
|
requests.post(self.url, data, headers = self.header)
|
||||||
|
|
||||||
def run(self):
|
def run(self):
|
||||||
data = "drop database if exists test"
|
data = "drop database if exists %s" % self.dbname
|
||||||
requests.post(self.url, data, headers = self.header)
|
requests.post(self.url, data, headers = self.header)
|
||||||
data = "create database test"
|
data = "create database %s" % self.dbname
|
||||||
requests.post(self.url, data, headers = self.header)
|
requests.post(self.url, data, headers = self.header)
|
||||||
data = "create table test.meters(ts timestamp, f1 int, f2 int, f3 int) tags(id int, loc nchar(20))"
|
data = "create table %s.meters(ts timestamp, f1 int, f2 int, f3 int) tags(id int, loc nchar(20))" % self.dbname
|
||||||
requests.post(self.url, data, headers = self.header)
|
requests.post(self.url, data, headers = self.header)
|
||||||
|
|
||||||
threads = []
|
threads = []
|
||||||
|
@ -70,7 +98,10 @@ class RestfulInsert:
|
||||||
threads = []
|
threads = []
|
||||||
startTime = time.time()
|
startTime = time.time()
|
||||||
for i in range(self.numOfThreads):
|
for i in range(self.numOfThreads):
|
||||||
thread = threading.Thread(target=self.insertData, args=(i,))
|
if(self.recordsPerTable != -1):
|
||||||
|
thread = threading.Thread(target=self.insertData, args=(i,))
|
||||||
|
else:
|
||||||
|
thread = threading.Thread(target=self.insertUnlimitedData, args=(i,))
|
||||||
thread.start()
|
thread.start()
|
||||||
threads.append(thread)
|
threads.append(thread)
|
||||||
|
|
||||||
|
@ -78,6 +109,62 @@ class RestfulInsert:
|
||||||
threads[i].join()
|
threads[i].join()
|
||||||
print("inserting %d records takes %d seconds" % (self.numOfTables * self.recordsPerTable, (time.time() - startTime)))
|
print("inserting %d records takes %d seconds" % (self.numOfTables * self.recordsPerTable, (time.time() - startTime)))
|
||||||
|
|
||||||
ri = RestfulInsert()
|
parser = argparse.ArgumentParser()
|
||||||
ri.init()
|
parser.add_argument(
|
||||||
|
'-H',
|
||||||
|
'--host-name',
|
||||||
|
action='store',
|
||||||
|
default='127.0.0.1',
|
||||||
|
type=str,
|
||||||
|
help='host name to be connected (default: 127.0.0.1)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-d',
|
||||||
|
'--db-name',
|
||||||
|
action='store',
|
||||||
|
default='test',
|
||||||
|
type=str,
|
||||||
|
help='Database name to be created (default: test)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-t',
|
||||||
|
'--number-of-threads',
|
||||||
|
action='store',
|
||||||
|
default=10,
|
||||||
|
type=int,
|
||||||
|
help='Number of threads to create tables and insert datas (default: 10)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-T',
|
||||||
|
'--number-of-tables',
|
||||||
|
action='store',
|
||||||
|
default=1000,
|
||||||
|
type=int,
|
||||||
|
help='Number of tables to be created (default: 1000)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-r',
|
||||||
|
'--number-of-records',
|
||||||
|
action='store',
|
||||||
|
default=1000,
|
||||||
|
type=int,
|
||||||
|
help='Number of record to be created for each table (default: 1000, -1 for unlimited records)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-s',
|
||||||
|
'--batch-size',
|
||||||
|
action='store',
|
||||||
|
default='1000',
|
||||||
|
type=int,
|
||||||
|
help='Number of tables to be created (default: 1000)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-p',
|
||||||
|
'--table-name-prefix',
|
||||||
|
action='store',
|
||||||
|
default='t',
|
||||||
|
type=str,
|
||||||
|
help='Number of tables to be created (default: 1000)')
|
||||||
|
parser.add_argument(
|
||||||
|
'-o',
|
||||||
|
'--out-of-order',
|
||||||
|
action='store_true',
|
||||||
|
help='The order of test data (default: False)')
|
||||||
|
|
||||||
|
args = parser.parse_args()
|
||||||
|
ri = RestfulInsert(args.host_name, args.db_name, args.number_of_threads, args.number_of_tables, args.number_of_records, args.batch_size, args.table_name_prefix, args.out_of_order)
|
||||||
ri.run()
|
ri.run()
|
|
@ -851,11 +851,205 @@ endi
|
||||||
print =====================>td-1442
|
print =====================>td-1442
|
||||||
sql_error select count(*) from m_fl_tb0 interval(1s) fill(prev);
|
sql_error select count(*) from m_fl_tb0 interval(1s) fill(prev);
|
||||||
|
|
||||||
print =============== clear
|
print =====================> aggregation + arithmetic + fill
|
||||||
sql drop database $db
|
#sql select avg(cpu_taosd) - first(cpu_taosd) from dn1 where ts<'2020-11-13 11:00:00' and ts>'2020-11-13 10:50:00' interval(10s) fill(value, 99)
|
||||||
sql show databases
|
#sql select count(*), first(k), avg(k), avg(k)-first(k) from tm0 where ts>'2020-1-1 1:1:1' and ts<'2020-1-1 1:02:59' interval(10s) fill(value, 99);
|
||||||
if $rows != 0 then
|
#sql select count(*), first(k), avg(k), avg(k)-first(k) from tm0 where ts>'2020-1-1 1:1:1' and ts<'2020-1-1 1:02:59' interval(10s) fill(NULL);
|
||||||
|
|
||||||
|
print =====================> td-2060
|
||||||
|
sql create table m1 (ts timestamp, k int ) tags(a int);
|
||||||
|
sql create table if not exists tm0 using m1 tags(1);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:1:1', 1);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:1:2', 2);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:1:3', 3);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:2:4', 4);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:2:5', 5);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:2:6', 6);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:3:7', 7);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:3:8', 8);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:3:9', 9);
|
||||||
|
sql insert into tm0 values('2020-1-1 1:4:10', 10);
|
||||||
|
|
||||||
|
sql select max(k)-min(k),last(k)-first(k),0-spread(k) from tm0 where ts>='2020-1-1 1:1:1' and ts<='2020-1-1 1:2:15' interval(10s) fill(value, 99,91,90,89,88,87,86,85);
|
||||||
|
if $rows != 8 then
|
||||||
return -1
|
return -1
|
||||||
endi
|
endi
|
||||||
|
|
||||||
|
if $data00 != @20-01-01 01:01:00.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data01 != 2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data02 != 2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data03 != -2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data10 != @20-01-01 01:01:10.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data11 != 99.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data12 != 91.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data13 != 90.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data60 != @20-01-01 01:02:00.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data61 != 2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data62 != 2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data63 != -2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data70 != @20-01-01 01:02:10.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data71 != 99.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data72 != 91.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data73 != 90.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
sql select first(k)-avg(k),0-spread(k) from tm0 where ts>='2020-1-1 1:1:1' and ts<='2020-1-1 1:2:15' interval(10s) fill(NULL);
|
||||||
|
if $rows != 8 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data00 != @20-01-01 01:01:00.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data01 != -1.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data02 != -2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data10 != @20-01-01 01:01:10.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data11 != NULL then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data12 != NULL then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
sql select max(k)-min(k),last(k)-first(k),0-spread(k) from tm0 where ts>='2020-1-1 1:1:1' and ts<='2020-1-1 4:2:15' interval(500a) fill(value, 99,91,90,89,88,87,86,85) order by ts asc;
|
||||||
|
if $rows != 21749 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
sql select max(k)-min(k),last(k)-first(k),0-spread(k),count(1) from m1 where ts>='2020-1-1 1:1:1' and ts<='2020-1-1 1:2:15' interval(10s) fill(value, 99,91,90,89,88,87,86,85) order by ts asc;
|
||||||
|
if $rows != 8 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data00 != @20-01-01 01:01:00.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data00 != @20-01-01 01:01:00.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
if $data1
|
||||||
|
if $data01 != 2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data02 != 2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data03 != -2.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data04 != 3 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data10 != @20-01-01 01:01:10.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data11 != 99.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data12 != 91.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data13 != 90.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data14 != 89 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
print ==================> td-2115
|
||||||
|
sql select count(*), min(c3)-max(c3) from m_fl_mt0 group by tgcol
|
||||||
|
if $rows != 10 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data00 != 5 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data01 != -4.000000000 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data02 != 0 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data12 != 1 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
|
||||||
|
print =============== clear
|
||||||
|
#sql drop database $db
|
||||||
|
#sql show databases
|
||||||
|
#if $rows != 0 then
|
||||||
|
# return -1
|
||||||
|
#endi
|
||||||
|
|
||||||
system sh/exec.sh -n dnode1 -s stop -x SIGINT
|
system sh/exec.sh -n dnode1 -s stop -x SIGINT
|
|
@ -212,4 +212,157 @@ sql select count(join_mt0.c1), first(join_mt0.c1)/count(*), first(join_mt1.c9) f
|
||||||
sql select count(join_mt0.c1), first(join_mt0.c1)-last(join_mt0.c1), first(join_mt1.c9) from join_mt0, join_mt1 where join_mt0.t1=join_mt1.t1 and join_mt0.ts=join_mt1.ts;", NULL);
|
sql select count(join_mt0.c1), first(join_mt0.c1)-last(join_mt0.c1), first(join_mt1.c9) from join_mt0, join_mt1 where join_mt0.t1=join_mt1.t1 and join_mt0.ts=join_mt1.ts;", NULL);
|
||||||
sql select last(join_mt0.c1) from join_mt0, join_mt1 where join_mt0.t1=join_mt1.t1 and join_mt0.ts=join_mt1.ts;", NULL);
|
sql select last(join_mt0.c1) from join_mt0, join_mt1 where join_mt0.t1=join_mt1.t1 and join_mt0.ts=join_mt1.ts;", NULL);
|
||||||
|
|
||||||
|
sql create database disorder_db;
|
||||||
|
sql use disorder_db;
|
||||||
|
sql create table m1(ts timestamp, k int) tags(a int);
|
||||||
|
sql create table tm0 using m1 tags(0);
|
||||||
|
sql create table tm1 using m1 tags(1);
|
||||||
|
sql create table tm2 using m1 tags(2);
|
||||||
|
sql create table tm3 using m1 tags(3);
|
||||||
|
sql create table tm4 using m1 tags(4);
|
||||||
|
sql create table tm5 using m1 tags(5);
|
||||||
|
sql create table tm6 using m1 tags(6);
|
||||||
|
sql create table tm7 using m1 tags(7);
|
||||||
|
|
||||||
|
sql show vgroups
|
||||||
|
if $rows != 2 then
|
||||||
|
print maxTablesPerVnode set to 4 is not active.
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
sql insert into tm0 values('2020-1-1 1:1:1', 0);
|
||||||
|
sql insert into tm1 values('2020-1-1 1:1:1', 1);
|
||||||
|
sql insert into tm2 values('2020-1-1 1:1:1', 2);
|
||||||
|
sql insert into tm3 values('2020-1-1 1:1:1', 3);
|
||||||
|
sql insert into tm4 values('2020-1-1 1:1:1', 4);
|
||||||
|
sql insert into tm5 values('2020-1-1 1:1:1', 5);
|
||||||
|
sql insert into tm6 values('2020-1-1 1:1:1', 6);
|
||||||
|
sql insert into tm7 values('2020-1-1 1:1:1', 7);
|
||||||
|
|
||||||
|
sql create table m2(ts timestamp, k int) tags(b int);
|
||||||
|
sql create table t0 using m2 tags(0);
|
||||||
|
sql create table t1 using m2 tags(4);
|
||||||
|
sql create table t2 using m2 tags(92);
|
||||||
|
sql create table t3 using m2 tags(93);
|
||||||
|
sql create table t4 using m2 tags(1);
|
||||||
|
sql create table t5 using m2 tags(5);
|
||||||
|
sql create table t6 using m2 tags(96);
|
||||||
|
sql create table t7 using m2 tags(97);
|
||||||
|
|
||||||
|
sql show vgroups
|
||||||
|
if $rows != 4 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
sql insert into t0 values('2020-1-1 1:1:1', 10);
|
||||||
|
sql insert into t1 values('2020-1-1 1:1:1', 11);
|
||||||
|
sql insert into t2 values('2020-1-1 1:1:1', 12);
|
||||||
|
sql insert into t3 values('2020-1-1 1:1:1', 13);
|
||||||
|
sql insert into t4 values('2020-1-1 1:1:1', 14);
|
||||||
|
sql insert into t5 values('2020-1-1 1:1:1', 15);
|
||||||
|
sql insert into t6 values('2020-1-1 1:1:1', 16);
|
||||||
|
sql insert into t7 values('2020-1-1 1:1:1', 17);
|
||||||
|
|
||||||
|
sql select m1.ts,m1.tbname,m1.a, m2.ts,m2.tbname,m2.b from m1,m2 where m1.a=m2.b and m1.ts=m2.ts;
|
||||||
|
if $rows != 4 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data00 != @20-01-01 01:01:01.000@ then
|
||||||
|
print expect 20-01-01 01:01:01.000, actual:$data00
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data01 != @tm0@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data02 != 0 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data03 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data04 != @t0@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data05 != 0 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data10 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data11 != @tm1@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data12 != 1 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data13 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data14 != @t4@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data15 != 1 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data20 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data21 != @tm4@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data22 != 4 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data23 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data24 != @t1@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data25 != 4 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data30 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data31 != @tm5@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data32 != 5 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data33 != @20-01-01 01:01:01.000@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data34 != @t5@ then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
|
if $data35 != 5 then
|
||||||
|
return -1
|
||||||
|
endi
|
||||||
|
|
||||||
system sh/exec.sh -n dnode1 -s stop -x SIGINT
|
system sh/exec.sh -n dnode1 -s stop -x SIGINT
|
Loading…
Reference in New Issue