mirror of
https://github.com/taosdata/TDengine
synced 2026-05-24 10:09:01 +00:00
ac23a4fc55
* enh: add client processing * enh: add mnode vtables processing * enh: add mnode vtable processing * enh: add normal child vtable support * fix: compile issues * fix: compile issues * fix: create stream issues * fix: multi stream scan issue * fix: remove debug info * fix: agg task and task level issues * fix: correct task output type * fix: split vtablescan from agg * fix: memory leak issues * fix: add limitations * Update 09-error-code.md * Update 09-error-code.md * fix: remove usless case * feat(stream): extract original table data in source scan task Implemented functionality in the source task to extract data corresponding to the virtual table from the original table using WAL. The extracted data is then sent to the downstream merge task for further processing. * feat(stream): multi-way merge using loser tree in virtual merge task Implemented multi-way merge in the merge task using a loser tree to combine data from multiple original table into a single virtual table. The merged virtual table data is then pushed downstream for further processing. Introduced memory limit handling during the merge process with configurable behavior when the memory limit is reached. * fix(test): remove useless cases --------- Co-authored-by: dapan1121 <wpan@taosdata.com> Co-authored-by: Pan Wei <72057773+dapan1121@users.noreply.github.com>
231 lines
10 KiB
C
231 lines
10 KiB
C
/*
|
|
* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
|
|
*
|
|
* This program is free software: you can use, redistribute, and/or modify
|
|
* it under the terms of the GNU Affero General Public License, version 3
|
|
* or later ("AGPL"), as published by the Free Software Foundation.
|
|
*
|
|
* This program is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE.
|
|
*
|
|
* You should have received a copy of the GNU Affero General Public License
|
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef _TD_PARSER_H_
|
|
#define _TD_PARSER_H_
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
#include "catalog.h"
|
|
#include "query.h"
|
|
#include "querynodes.h"
|
|
|
|
#define PAR_ERR_RET(c) \
|
|
do { \
|
|
int32_t _code = c; \
|
|
if (_code != TSDB_CODE_SUCCESS) { \
|
|
terrno = _code; \
|
|
return _code; \
|
|
} \
|
|
} while (0)
|
|
#define PAR_RET(c) \
|
|
do { \
|
|
int32_t _code = c; \
|
|
if (_code != TSDB_CODE_SUCCESS) { \
|
|
terrno = _code; \
|
|
} \
|
|
return _code; \
|
|
} while (0)
|
|
#define PAR_ERR_JRET(c) \
|
|
do { \
|
|
code = c; \
|
|
if (code != TSDB_CODE_SUCCESS) { \
|
|
terrno = code; \
|
|
goto _return; \
|
|
} \
|
|
} while (0)
|
|
|
|
typedef struct SStmtCallback {
|
|
TAOS_STMT* pStmt;
|
|
int32_t (*getTbNameFn)(TAOS_STMT*, char**);
|
|
int32_t (*setInfoFn)(TAOS_STMT*, STableMeta*, void*, SName*, bool, SHashObj*, SHashObj*, const char*, bool);
|
|
int32_t (*getExecInfoFn)(TAOS_STMT*, SHashObj**, SHashObj**);
|
|
} SStmtCallback;
|
|
|
|
typedef enum {
|
|
PARSE_SQL_RES_QUERY = 1,
|
|
PARSE_SQL_RES_SCHEMA,
|
|
} SParseResType;
|
|
|
|
typedef struct SParseSchemaRes {
|
|
int8_t precision;
|
|
int32_t numOfCols;
|
|
SSchema* pSchema;
|
|
} SParseSchemaRes;
|
|
|
|
typedef struct SParseQueryRes {
|
|
SNode* pQuery;
|
|
SCatalogReq* pCatalogReq;
|
|
SMetaData meta;
|
|
} SParseQueryRes;
|
|
|
|
typedef struct SParseSqlRes {
|
|
SParseResType resType;
|
|
union {
|
|
SParseSchemaRes schemaRes;
|
|
SParseQueryRes queryRes;
|
|
};
|
|
} SParseSqlRes;
|
|
|
|
typedef int32_t (*parseSqlFn)(void*, const char*, const char*, bool, const char*, SParseSqlRes*);
|
|
|
|
typedef struct SParseCsvCxt {
|
|
TdFilePtr fp; // last parsed file
|
|
int32_t tableNo; // last parsed table
|
|
SName tableName; // last parsed table
|
|
const char* pLastSqlPos; // the location of the last parsed sql
|
|
} SParseCsvCxt;
|
|
|
|
typedef void (*setQueryFn)(int64_t);
|
|
|
|
typedef struct SParseContext {
|
|
uint64_t requestId;
|
|
int64_t requestRid;
|
|
int32_t acctId;
|
|
const char* db;
|
|
bool topicQuery;
|
|
void* pTransporter;
|
|
SEpSet mgmtEpSet;
|
|
const char* pSql; // sql string
|
|
size_t sqlLen; // length of the sql string
|
|
char* pMsg; // extended error message if exists to help identifying the problem in sql statement.
|
|
int32_t msgLen; // max length of the msg
|
|
struct SCatalog* pCatalog;
|
|
SStmtCallback* pStmtCb;
|
|
const char* pUser;
|
|
const char* pEffectiveUser;
|
|
bool parseOnly;
|
|
bool isSuperUser;
|
|
bool enableSysInfo;
|
|
bool async;
|
|
bool hasInvisibleCol;
|
|
bool isView;
|
|
bool isAudit;
|
|
bool nodeOffline;
|
|
bool isStmtBind;
|
|
const char* svrVer;
|
|
SArray* pTableMetaPos; // sql table pos => catalog data pos
|
|
SArray* pTableVgroupPos; // sql table pos => catalog data pos
|
|
int64_t allocatorId;
|
|
parseSqlFn parseSqlFp;
|
|
void* parseSqlParam;
|
|
int8_t biMode;
|
|
SArray* pSubMetaList;
|
|
setQueryFn setQueryFp;
|
|
timezone_t timezone;
|
|
void *charsetCxt;
|
|
bool streamRunHistory;
|
|
} SParseContext;
|
|
|
|
int32_t qParseSql(SParseContext* pCxt, SQuery** pQuery);
|
|
bool qIsInsertValuesSql(const char* pStr, size_t length);
|
|
bool qParseDbName(const char* pStr, size_t length, char** pDbName);
|
|
|
|
// for async mode
|
|
int32_t qParseSqlSyntax(SParseContext* pCxt, SQuery** pQuery, struct SCatalogReq* pCatalogReq);
|
|
int32_t qAnalyseSqlSemantic(SParseContext* pCxt, const struct SCatalogReq* pCatalogReq,
|
|
struct SMetaData* pMetaData, SQuery* pQuery);
|
|
int32_t qContinueParseSql(SParseContext* pCxt, struct SCatalogReq* pCatalogReq, const struct SMetaData* pMetaData,
|
|
SQuery* pQuery);
|
|
int32_t qContinueParsePostQuery(SParseContext* pCxt, SQuery* pQuery, SSDataBlock* pBlock);
|
|
|
|
void qDestroyParseContext(SParseContext* pCxt);
|
|
|
|
void qDestroyQuery(SQuery* pQueryNode);
|
|
|
|
int32_t qExtractResultSchema(const SNode* pRoot, int32_t* numOfCols, SSchema** pSchema);
|
|
int32_t qSetSTableIdForRsma(SNode* pStmt, int64_t uid);
|
|
int32_t qInitKeywordsTable();
|
|
void qCleanupKeywordsTable();
|
|
|
|
int32_t qAppendStmtTableOutput(SQuery* pQuery, SHashObj* pAllVgHash, STableColsData* pTbData, STableDataCxt* pTbCtx,
|
|
SStbInterlaceInfo* pBuildInfo, SVCreateTbReq* ctbReq);
|
|
int32_t qBuildStmtFinOutput(SQuery* pQuery, SHashObj* pAllVgHash, SArray* pVgDataBlocks);
|
|
// int32_t qBuildStmtOutputFromTbList(SQuery* pQuery, SHashObj* pVgHash, SArray* pBlockList, STableDataCxt* pTbCtx,
|
|
// int32_t tbNum);
|
|
int32_t qBuildStmtOutput(SQuery* pQuery, SHashObj* pVgHash, SHashObj* pBlockHash);
|
|
int32_t qResetStmtColumns(SArray* pCols, bool deepClear);
|
|
int32_t qResetStmtDataBlock(STableDataCxt* block, bool keepBuf);
|
|
int32_t qCloneStmtDataBlock(STableDataCxt** pDst, STableDataCxt* pSrc, bool reset);
|
|
int32_t qRebuildStmtDataBlock(STableDataCxt** pDst, STableDataCxt* pSrc, uint64_t uid, uint64_t suid, int32_t vgId,
|
|
bool rebuildCreateTb);
|
|
void qDestroyStmtDataBlock(STableDataCxt* pBlock);
|
|
STableMeta* qGetTableMetaInDataBlock(STableDataCxt* pDataBlock);
|
|
int32_t qCloneCurrentTbData(STableDataCxt* pDataBlock, SSubmitTbData** pData);
|
|
|
|
int32_t qStmtBindParams(SQuery* pQuery, TAOS_MULTI_BIND* pParams, int32_t colIdx, void *charsetCxt);
|
|
int32_t qStmtParseQuerySql(SParseContext* pCxt, SQuery* pQuery);
|
|
int32_t qBindStmtStbColsValue(void* pBlock, SArray* pCols, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen,
|
|
STSchema** pTSchema, SBindInfo* pBindInfos, void* charsetCxt);
|
|
int32_t qBindStmtColsValue(void* pBlock, SArray* pCols, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen, void* charsetCxt);
|
|
int32_t qBindStmtSingleColValue(void* pBlock, SArray* pCols, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen,
|
|
int32_t colIdx, int32_t rowNum, void* charsetCxt);
|
|
int32_t qBuildStmtColFields(void* pDataBlock, int32_t* fieldNum, TAOS_FIELD_E** fields);
|
|
int32_t qBuildStmtStbColFields(void* pBlock, void* boundTags, bool hasCtbName, int32_t* fieldNum,
|
|
TAOS_FIELD_ALL** fields);
|
|
int32_t qBuildStmtTagFields(void* pBlock, void* boundTags, int32_t* fieldNum, TAOS_FIELD_E** fields);
|
|
int32_t qBindStmtTagsValue(void* pBlock, void* boundTags, int64_t suid, const char* sTableName, char* tName,
|
|
TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen, void* charsetCxt);
|
|
|
|
int32_t qStmtBindParams2(SQuery* pQuery, TAOS_STMT2_BIND* pParams, int32_t colIdx, void* charsetCxt);
|
|
int32_t qBindStmtStbColsValue2(void* pBlock, SArray* pCols, TAOS_STMT2_BIND* bind, char* msgBuf, int32_t msgBufLen,
|
|
STSchema** pTSchema, SBindInfo2* pBindInfos, void *charsetCxt);
|
|
int32_t qBindStmtColsValue2(void* pBlock, SArray* pCols, TAOS_STMT2_BIND* bind, char* msgBuf, int32_t msgBufLen,
|
|
void* charsetCxt);
|
|
int32_t qBindStmtSingleColValue2(void* pBlock, SArray* pCols, TAOS_STMT2_BIND* bind, char* msgBuf, int32_t msgBufLen,
|
|
int32_t colIdx, int32_t rowNum, void* charsetCxt);
|
|
int32_t qBindStmt2RowValue(void* pBlock, SArray* pCols, TAOS_STMT2_BIND* bind, char* msgBuf, int32_t msgBufLen,
|
|
STSchema** pTSchema, SBindInfo2* pBindInfos, void* charsetCxt);
|
|
int32_t qBindStmtTagsValue2(void* pBlock, void* boundTags, int64_t suid, const char* sTableName, char* tName,
|
|
TAOS_STMT2_BIND* bind, char* msgBuf, int32_t msgBufLen, void* charsetCxt,
|
|
SVCreateTbReq* pCreateTbReq);
|
|
|
|
void destroyBoundColumnInfo(void* pBoundInfo);
|
|
int32_t qCreateSName(SName* pName, const char* pTableName, int32_t acctId, char* dbName, char* msgBuf,
|
|
int32_t msgBufLen);
|
|
|
|
void qDestroyBoundColInfo(void* pInfo);
|
|
|
|
int32_t smlInitHandle(SQuery** query);
|
|
int32_t smlBuildRow(STableDataCxt* pTableCxt);
|
|
int32_t smlBuildCol(STableDataCxt* pTableCxt, SSchema* schema, void* kv, int32_t index, void* charsetCxt);
|
|
int32_t smlInitTableDataCtx(SQuery* query, STableMeta* pTableMeta, STableDataCxt** cxt);
|
|
|
|
void clearColValArraySml(SArray* pCols);
|
|
int32_t smlBindData(SQuery* handle, bool dataFormat, SArray* tags, SArray* colsSchema, SArray* cols,
|
|
STableMeta* pTableMeta, char* tableName, const char* sTableName, int32_t sTableNameLen, int32_t ttl,
|
|
char* msgBuf, int32_t msgBufLen, void* charsetCxt);
|
|
int32_t smlBuildOutput(SQuery* handle, SHashObj* pVgHash);
|
|
int32_t smlBuildOutputRaw(SQuery* handle, SHashObj* pVgHash);
|
|
int rawBlockBindRawData(SHashObj* pVgroupHash, SArray* pVgroupList, STableMeta* pTableMeta, void* data);
|
|
int rawBlockBindData(SQuery* query, STableMeta* pTableMeta, void* data, SVCreateTbReq* pCreateTb, void* fields,
|
|
int numFields, bool needChangeLength, char* errstr, int32_t errstrLen, bool raw);
|
|
int32_t checkSchema(SSchema* pColSchema, SSchemaExt* pColExtSchema, int8_t* fields, char* errstr, int32_t errstrLen);
|
|
|
|
int32_t rewriteToVnodeModifyOpStmt(SQuery* pQuery, SArray* pBufArray);
|
|
int32_t serializeVgroupsCreateTableBatch(SHashObj* pVgroupHashmap, SArray** pOut);
|
|
int32_t serializeVgroupsDropTableBatch(SHashObj* pVgroupHashmap, SArray** pOut);
|
|
void destoryCatalogReq(SCatalogReq* pCatalogReq);
|
|
bool isPrimaryKeyImpl(SNode* pExpr);
|
|
int32_t insAppendStmtTableDataCxt(SHashObj* pAllVgHash, STableColsData* pTbData, STableDataCxt* pTbCtx,
|
|
SStbInterlaceInfo* pBuildInfo, SVCreateTbReq* ctbReq);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_PARSER_H_*/
|