255 lines
7.2 KiB
C
255 lines
7.2 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 TDENGINE_CLIENTSTMT2_H
|
|
#define TDENGINE_CLIENTSTMT2_H
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
#include "catalog.h"
|
|
/*
|
|
typedef enum {
|
|
STMT_TYPE_INSERT = 1,
|
|
STMT_TYPE_MULTI_INSERT,
|
|
STMT_TYPE_QUERY,
|
|
} STMT_TYPE;
|
|
|
|
typedef enum {
|
|
STMT_INIT = 1,
|
|
STMT_PREPARE,
|
|
STMT_SETTBNAME,
|
|
STMT_SETTAGS,
|
|
STMT_FETCH_FIELDS,
|
|
STMT_BIND,
|
|
STMT_BIND_COL,
|
|
STMT_ADD_BATCH,
|
|
STMT_EXECUTE,
|
|
STMT_MAX,
|
|
} STMT_STATUS;
|
|
|
|
#define STMT_TABLE_COLS_NUM 1000
|
|
|
|
typedef struct SStmtTableCache {
|
|
STableDataCxt *pDataCtx;
|
|
void *boundTags;
|
|
} SStmtTableCache;
|
|
|
|
typedef struct SStmtQueryResInfo {
|
|
TAOS_FIELD *fields;
|
|
TAOS_FIELD *userFields;
|
|
uint32_t numOfCols;
|
|
int32_t precision;
|
|
} SStmtQueryResInfo;
|
|
|
|
typedef struct SStmtBindInfo {
|
|
bool needParse;
|
|
bool inExecCache;
|
|
uint64_t tbUid;
|
|
uint64_t tbSuid;
|
|
int32_t tbVgId;
|
|
int32_t sBindRowNum;
|
|
int32_t sBindLastIdx;
|
|
int8_t tbType;
|
|
bool tagsCached;
|
|
void *boundTags;
|
|
char tbName[TSDB_TABLE_FNAME_LEN];
|
|
char tbFName[TSDB_TABLE_FNAME_LEN];
|
|
char stbFName[TSDB_TABLE_FNAME_LEN];
|
|
SName sname;
|
|
|
|
char statbName[TSDB_TABLE_FNAME_LEN];
|
|
} SStmtBindInfo;
|
|
|
|
typedef struct SStmtAsyncParam {
|
|
STableColsData *pTbData;
|
|
void* pStmt;
|
|
} SStmtAsyncParam;
|
|
|
|
typedef struct SStmtExecInfo {
|
|
int32_t affectedRows;
|
|
SRequestObj *pRequest;
|
|
SHashObj *pBlockHash;
|
|
STableDataCxt *pCurrBlock;
|
|
SSubmitTbData *pCurrTbData;
|
|
} SStmtExecInfo;
|
|
*/
|
|
typedef struct {
|
|
bool stbInterlaceMode;
|
|
STMT_TYPE type;
|
|
STMT_STATUS status;
|
|
uint64_t suid;
|
|
uint64_t runTimes;
|
|
SHashObj *pTableCache; // SHash<SStmtTableCache>
|
|
SQuery *pQuery;
|
|
char *sqlStr;
|
|
int32_t sqlLen;
|
|
SArray *nodeList;
|
|
SStmtQueryResInfo queryRes;
|
|
bool autoCreateTbl;
|
|
SHashObj *pVgHash;
|
|
SBindInfo2 *pBindInfo;
|
|
bool bindRowFormat;
|
|
bool fixValueTags;
|
|
SVCreateTbReq *fixValueTbReq;
|
|
|
|
SStbInterlaceInfo siInfo;
|
|
} SStmtSQLInfo2;
|
|
/*
|
|
typedef struct SStmtStatInfo {
|
|
int64_t ctgGetTbMetaNum;
|
|
int64_t getCacheTbInfo;
|
|
int64_t parseSqlNum;
|
|
int64_t bindDataNum;
|
|
int64_t setTbNameUs;
|
|
int64_t bindDataUs1;
|
|
int64_t bindDataUs2;
|
|
int64_t bindDataUs3;
|
|
int64_t bindDataUs4;
|
|
int64_t addBatchUs;
|
|
int64_t execWaitUs;
|
|
int64_t execUseUs;
|
|
} SStmtStatInfo;
|
|
|
|
typedef struct SStmtQNode {
|
|
bool restoreTbCols;
|
|
STableColsData tblData;
|
|
struct SStmtQNode* next;
|
|
} SStmtQNode;
|
|
|
|
typedef struct SStmtQueue {
|
|
bool stopQueue;
|
|
SStmtQNode* head;
|
|
SStmtQNode* tail;
|
|
uint64_t qRemainNum;
|
|
} SStmtQueue;
|
|
*/
|
|
typedef struct {
|
|
TAOS_STMT2 *stmt;
|
|
TAOS_STMT2_BINDV *bindv;
|
|
int32_t col_idx;
|
|
__taos_async_fn_t fp;
|
|
void *param;
|
|
} ThreadArgs;
|
|
|
|
typedef struct AsyncBindParam {
|
|
TdThreadMutex mutex;
|
|
TdThreadCond waitCond;
|
|
uint8_t asyncBindNum;
|
|
} AsyncBindParam;
|
|
|
|
typedef struct {
|
|
STscObj *taos;
|
|
SCatalog *pCatalog;
|
|
int32_t affectedRows;
|
|
uint32_t seqId;
|
|
uint32_t seqIds[STMT_MAX];
|
|
bool bindThreadInUse;
|
|
TdThread bindThread;
|
|
TAOS_STMT2_OPTION options;
|
|
bool stbInterlaceMode;
|
|
SStmtQueue queue;
|
|
|
|
SStmtSQLInfo2 sql;
|
|
SStmtExecInfo exec;
|
|
SStmtBindInfo bInfo;
|
|
|
|
char *db;
|
|
int64_t reqid;
|
|
int32_t errCode;
|
|
tsem_t asyncExecSem;
|
|
bool execSemWaited;
|
|
AsyncBindParam asyncBindParam;
|
|
SStmtStatInfo stat;
|
|
} STscStmt2;
|
|
/*
|
|
extern char *gStmtStatusStr[];
|
|
|
|
#define STMT_LOG_SEQ(n) \
|
|
do { \
|
|
(pStmt)->seqId++; \
|
|
(pStmt)->seqIds[n]++; \
|
|
STMT_DLOG("the %dth:%d %s", (pStmt)->seqIds[n], (pStmt)->seqId, gStmtStatusStr[n]); \
|
|
} while (0)
|
|
|
|
#define STMT_STATUS_NE(S) (pStmt->sql.status != STMT_##S)
|
|
#define STMT_STATUS_EQ(S) (pStmt->sql.status == STMT_##S)
|
|
|
|
#define STMT_ERR_RET(c) \
|
|
do { \
|
|
int32_t _code = c; \
|
|
if (_code != TSDB_CODE_SUCCESS) { \
|
|
terrno = _code; \
|
|
pStmt->errCode = _code; \
|
|
return _code; \
|
|
} \
|
|
} while (0)
|
|
#define STMT_RET(c) \
|
|
do { \
|
|
int32_t _code = c; \
|
|
if (_code != TSDB_CODE_SUCCESS) { \
|
|
terrno = _code; \
|
|
pStmt->errCode = _code; \
|
|
} \
|
|
return _code; \
|
|
} while (0)
|
|
#define STMT_ERR_JRET(c) \
|
|
do { \
|
|
code = c; \
|
|
if (code != TSDB_CODE_SUCCESS) { \
|
|
terrno = code; \
|
|
pStmt->errCode = code; \
|
|
goto _return; \
|
|
} \
|
|
} while (0)
|
|
#define STMT_ERRI_JRET(c) \
|
|
do { \
|
|
code = c; \
|
|
if (code != TSDB_CODE_SUCCESS) { \
|
|
terrno = code; \
|
|
goto _return; \
|
|
} \
|
|
} while (0)
|
|
|
|
|
|
#define STMT_FLOG(param, ...) qFatal("stmt:%p, " param, pStmt, __VA_ARGS__)
|
|
#define STMT_ELOG(param, ...) qError("stmt:%p, " param, pStmt, __VA_ARGS__)
|
|
#define STMT_DLOG(param, ...) qDebug("stmt:%p, " param, pStmt, __VA_ARGS__)
|
|
|
|
#define STMT_ELOG_E(param) qError("stmt:%p, " param, pStmt)
|
|
#define STMT_DLOG_E(param) qDebug("stmt:%p, " param, pStmt)
|
|
*/
|
|
TAOS_STMT2 *stmtInit2(STscObj *taos, TAOS_STMT2_OPTION *pOptions);
|
|
int stmtClose2(TAOS_STMT2 *stmt);
|
|
int stmtExec2(TAOS_STMT2 *stmt, int *affected_rows);
|
|
int stmtPrepare2(TAOS_STMT2 *stmt, const char *sql, unsigned long length);
|
|
int stmtSetTbName2(TAOS_STMT2 *stmt, const char *tbName);
|
|
int stmtSetTbTags2(TAOS_STMT2 *stmt, TAOS_STMT2_BIND *tags, SVCreateTbReq **pCreateTbReq);
|
|
int stmtCheckTags2(TAOS_STMT2 *stmt, SVCreateTbReq **pCreateTbReq);
|
|
int stmtBindBatch2(TAOS_STMT2 *stmt, TAOS_STMT2_BIND *bind, int32_t colIdx, SVCreateTbReq *pCreateTbReq);
|
|
int stmtGetStbColFields2(TAOS_STMT2 *stmt, int *nums, TAOS_FIELD_ALL **fields);
|
|
int stmtGetParamNum2(TAOS_STMT2 *stmt, int *nums);
|
|
int stmtIsInsert2(TAOS_STMT2 *stmt, int *insert);
|
|
TAOS_RES *stmtUseResult2(TAOS_STMT2 *stmt);
|
|
const char *stmtErrstr2(TAOS_STMT2 *stmt);
|
|
int stmt2AsyncBind(TAOS_STMT2 *stmt, TAOS_STMT2_BINDV *bindv, int32_t col_idx, __taos_async_fn_t fp, void *param);
|
|
int stmtAsyncBindThreadFunc(void *args);
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif // TDENGINE_CLIENTSTMT2_H
|