122 lines
5.1 KiB
C
122 lines
5.1 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 "query.h"
|
|
#include "querynodes.h"
|
|
|
|
struct SCatalogReq;
|
|
struct SMetaData;
|
|
|
|
typedef struct SStmtCallback {
|
|
TAOS_STMT* pStmt;
|
|
int32_t (*getTbNameFn)(TAOS_STMT*, char**);
|
|
int32_t (*setInfoFn)(TAOS_STMT*, STableMeta*, void*, char*, bool, SHashObj*, SHashObj*, const char*);
|
|
int32_t (*getExecInfoFn)(TAOS_STMT*, SHashObj**, SHashObj**);
|
|
} SStmtCallback;
|
|
|
|
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 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;
|
|
bool isSuperUser;
|
|
bool enableSysInfo;
|
|
bool async;
|
|
int8_t schemalessType;
|
|
const char* svrVer;
|
|
bool nodeOffline;
|
|
SArray* pTableMetaPos; // sql table pos => catalog data pos
|
|
SArray* pTableVgroupPos; // sql table pos => catalog data pos
|
|
int64_t allocatorId;
|
|
} SParseContext;
|
|
|
|
int32_t qParseSql(SParseContext* pCxt, SQuery** pQuery);
|
|
bool qIsInsertValuesSql(const char* pStr, size_t length);
|
|
|
|
// for async mode
|
|
int32_t qParseSqlSyntax(SParseContext* pCxt, SQuery** pQuery, struct SCatalogReq* pCatalogReq);
|
|
int32_t qAnalyseSqlSemantic(SParseContext* pCxt, const struct SCatalogReq* pCatalogReq,
|
|
const struct SMetaData* pMetaData, SQuery* pQuery);
|
|
int32_t qContinueParseSql(SParseContext* pCxt, struct SCatalogReq* pCatalogReq, const struct SMetaData* pMetaData,
|
|
SQuery* pQuery);
|
|
|
|
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);
|
|
void qCleanupKeywordsTable();
|
|
|
|
int32_t qBuildStmtOutput(SQuery* pQuery, SHashObj* pVgHash, SHashObj* pBlockHash);
|
|
int32_t qResetStmtDataBlock(void* block, bool keepBuf);
|
|
int32_t qCloneStmtDataBlock(void** pDst, void* pSrc);
|
|
void qFreeStmtDataBlock(void* pDataBlock);
|
|
int32_t qRebuildStmtDataBlock(void** pDst, void* pSrc, uint64_t uid, int32_t vgId);
|
|
void qDestroyStmtDataBlock(void* pBlock);
|
|
STableMeta* qGetTableMetaInDataBlock(void* pDataBlock);
|
|
|
|
int32_t qStmtBindParams(SQuery* pQuery, TAOS_MULTI_BIND* pParams, int32_t colIdx);
|
|
int32_t qStmtParseQuerySql(SParseContext* pCxt, SQuery* pQuery);
|
|
int32_t qBindStmtColsValue(void* pBlock, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen);
|
|
int32_t qBindStmtSingleColValue(void* pBlock, TAOS_MULTI_BIND* bind, char* msgBuf, int32_t msgBufLen, int32_t colIdx,
|
|
int32_t rowNum);
|
|
int32_t qBuildStmtColFields(void* pDataBlock, int32_t* fieldNum, TAOS_FIELD_E** 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 destroyBoundColumnInfo(void* pBoundInfo);
|
|
int32_t qCreateSName(SName* pName, const char* pTableName, int32_t acctId, char* dbName, char* msgBuf,
|
|
int32_t msgBufLen);
|
|
|
|
void* smlInitHandle(SQuery* pQuery);
|
|
void smlDestroyHandle(void* pHandle);
|
|
int32_t smlBindData(void* handle, SArray* tags, SArray* colsSchema, SArray* cols, bool format, STableMeta* pTableMeta,
|
|
char* tableName, const char* sTableName, int32_t sTableNameLen, char* msgBuf, int16_t msgBufLen);
|
|
int32_t smlBuildOutput(void* handle, SHashObj* pVgHash);
|
|
|
|
int32_t rewriteToVnodeModifyOpStmt(SQuery* pQuery, SArray* pBufArray);
|
|
SArray* serializeVgroupsCreateTableBatch(SHashObj* pVgroupHashmap);
|
|
SArray* serializeVgroupsDropTableBatch(SHashObj* pVgroupHashmap);
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_PARSER_H_*/
|