tscUtil.h 11.0 KB
Newer Older
H
hzcheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25
/*
 * 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_TSCUTIL_H
#define TDENGINE_TSCUTIL_H

#ifdef __cplusplus
extern "C" {
#endif

/*
 * @date   2018/09/30
 */
26
#include "exception.h"
H
Haojun Liao 已提交
27 28
#include "os.h"
#include "qExtbuffer.h"
29
#include "taosdef.h"
H
Haojun Liao 已提交
30
#include "tbuffer.h"
31
#include "tscLocalMerge.h"
S
slguan 已提交
32
#include "tsclient.h"
H
hzcheng 已提交
33

weixin_48148422's avatar
weixin_48148422 已提交
34
#define UTIL_TABLE_IS_SUPER_TABLE(metaInfo)  \
H
hjxilinx 已提交
35
  (((metaInfo)->pTableMeta != NULL) && ((metaInfo)->pTableMeta->tableType == TSDB_SUPER_TABLE))
36
#define UTIL_TABLE_IS_CHILD_TABLE(metaInfo) \
H
hjxilinx 已提交
37
  (((metaInfo)->pTableMeta != NULL) && ((metaInfo)->pTableMeta->tableType == TSDB_CHILD_TABLE))
H
hjxilinx 已提交
38
  
weixin_48148422's avatar
weixin_48148422 已提交
39 40
#define UTIL_TABLE_IS_NORMAL_TABLE(metaInfo)\
  (!(UTIL_TABLE_IS_SUPER_TABLE(metaInfo) || UTIL_TABLE_IS_CHILD_TABLE(metaInfo)))
H
hzcheng 已提交
41

S
slguan 已提交
42 43
#define TSDB_COL_IS_TAG(f) (((f)&TSDB_COL_TAG) != 0)

H
hzcheng 已提交
44 45
typedef struct SParsedColElem {
  int16_t colIndex;
B
Bomin Zhang 已提交
46
  uint16_t offset;
H
hzcheng 已提交
47 48 49
} SParsedColElem;

typedef struct SParsedDataColInfo {
50 51
  int16_t        numOfCols;
  int16_t        numOfAssignedCols;
H
hzcheng 已提交
52 53 54 55
  SParsedColElem elems[TSDB_MAX_COLUMNS];
  bool           hasVal[TSDB_MAX_COLUMNS];
} SParsedDataColInfo;

56 57 58 59 60 61 62 63
typedef struct STidTags {
  int64_t  uid;
  int32_t  tid;
  int32_t  vgId;
  char     tag[];
} STidTags;

typedef struct SJoinSupporter {
S
slguan 已提交
64 65 66
  SSubqueryState* pState;
  SSqlObj*        pObj;           // parent SqlObj
  int32_t         subqueryIndex;  // index of sub query
H
Haojun Liao 已提交
67 68
  int64_t         intervalTime;   // interval time
  int64_t         slidingTime;    // sliding time
S
slguan 已提交
69 70
  SLimitVal       limit;          // limit info
  uint64_t        uid;            // query meter uid
H
hjxilinx 已提交
71 72
  SArray*         colList;        // previous query information, no need to use this attribute, and the corresponding attribution
  SArray*         exprList;
S
slguan 已提交
73 74 75
  SFieldInfo      fieldsInfo;
  STagCond        tagCond;
  SSqlGroupbyExpr groupbyExpr;
H
hjxilinx 已提交
76 77
  struct STSBuf*  pTSBuf;          // the TSBuf struct that holds the compressed timestamp array
  FILE*           f;               // temporary file in order to create TSBuf
78 79 80 81 82 83 84 85 86 87 88
  char            path[PATH_MAX];  // temporary file path, todo dynamic allocate memory
  int32_t         tagSize;         // the length of each in the first filter stage
  char*           pIdTagList;      // result of first stage tags
  int32_t         totalLen;
  int32_t         num;
} SJoinSupporter;

typedef struct SVgroupTableInfo {
  SCMVgroupInfo vgInfo;
  SArray*       itemList;   //SArray<STableIdInfo>
} SVgroupTableInfo;
S
slguan 已提交
89

H
Haojun Liao 已提交
90 91 92 93 94 95 96 97 98 99
static FORCE_INLINE SQueryInfo* tscGetQueryInfoDetail(SSqlCmd* pCmd, int32_t subClauseIndex) {
  assert(pCmd != NULL && subClauseIndex >= 0 && subClauseIndex < TSDB_MAX_UNION_CLAUSE);

  if (pCmd->pQueryInfo == NULL || subClauseIndex >= pCmd->numOfClause) {
    return NULL;
  }

  return pCmd->pQueryInfo[subClauseIndex];
}

H
hjxilinx 已提交
100
int32_t tscCreateDataBlock(size_t initialSize, int32_t rowSize, int32_t startOffset, const char* name,
H
hjxilinx 已提交
101
                           STableMeta* pTableMeta, STableDataBlocks** dataBlocks);
H
hjxilinx 已提交
102 103
void tscDestroyDataBlock(STableDataBlocks* pDataBlock);
void tscSortRemoveDataBlockDupRows(STableDataBlocks* dataBuf);
H
hjxilinx 已提交
104 105 106 107

SParamInfo* tscAddParamToDataBlock(STableDataBlocks* pDataBlock, char type, uint8_t timePrec, short bytes,
                                   uint32_t offset);

108
void*   tscDestroyBlockArrayList(SArray* pDataBlockList);
H
hjxilinx 已提交
109
int32_t tscCopyDataBlockToPayload(SSqlObj* pSql, STableDataBlocks* pDataBlock);
110 111 112
void    tscFreeUnusedDataBlocks(SArray* pDataBlockList);
int32_t tscMergeTableDataBlocks(SSqlObj* pSql, SArray* pDataList);
int32_t tscGetDataBlockFromList(void* pHashList, SArray* pDataBlockList, int64_t id, int32_t size,
H
hjxilinx 已提交
113
                                int32_t startOffset, int32_t rowSize, const char* tableId, STableMeta* pTableMeta,
H
hjxilinx 已提交
114
                                STableDataBlocks** dataBlocks);
H
hjxilinx 已提交
115

H
hzcheng 已提交
116 117
/**
 * for the projection query on metric or point interpolation query on metric,
S
slguan 已提交
118
 * we iterate all the meters, instead of invoke query on all qualified meters simultaneously.
H
hzcheng 已提交
119 120 121 122
 *
 * @param pSql  sql object
 * @return
 */
123 124
bool tscIsPointInterpQuery(SQueryInfo* pQueryInfo);
bool tscIsTWAQuery(SQueryInfo* pQueryInfo);
125

126
bool tscNonOrderedProjectionQueryOnSTable(SQueryInfo *pQueryInfo, int32_t tableIndex);
127 128 129
bool tscOrderedProjectionQueryOnSTable(SQueryInfo* pQueryInfo, int32_t tableIndex);
bool tscIsProjectionQueryOnSTable(SQueryInfo* pQueryInfo, int32_t tableIndex);

H
Haojun Liao 已提交
130
bool tscIsProjectionQuery(SQueryInfo* pQueryInfo);
H
hjxilinx 已提交
131

H
hjxilinx 已提交
132
bool tscIsTwoStageSTableQuery(SQueryInfo* pQueryInfo, int32_t tableIndex);
H
hjxilinx 已提交
133
bool tscQueryTags(SQueryInfo* pQueryInfo);
S
slguan 已提交
134

135
void tscAddSpecialColumnForSelect(SQueryInfo* pQueryInfo, int32_t outputColIndex, int16_t functionId, SColumnIndex* pIndex,
H
Haojun Liao 已提交
136
                                  SSchema* pColSchema, int16_t colType);
H
hzcheng 已提交
137

H
Haojun Liao 已提交
138
int32_t tscSetTableFullName(STableMetaInfo* pTableMetaInfo, SSQLToken* pzTableName, SSqlObj* pSql);
139
void    tscClearInterpInfo(SQueryInfo* pQueryInfo);
H
hzcheng 已提交
140

H
hjxilinx 已提交
141
bool tscIsInsertData(char* sqlstr);
H
hzcheng 已提交
142 143

/* use for keep current db info temporarily, for handle table with db prefix */
H
hjxilinx 已提交
144
// todo remove it
H
Haojun Liao 已提交
145
void tscGetDBInfoFromTableFullName(char* tableId, char* db);
H
hzcheng 已提交
146

S
slguan 已提交
147
int tscAllocPayload(SSqlCmd* pCmd, int size);
H
hzcheng 已提交
148

H
hjxilinx 已提交
149 150 151 152 153 154 155
TAOS_FIELD tscCreateField(int8_t type, const char* name, int16_t bytes);

SFieldSupInfo* tscFieldInfoAppend(SFieldInfo* pFieldInfo, TAOS_FIELD* pField);
SFieldSupInfo* tscFieldInfoInsert(SFieldInfo* pFieldInfo, int32_t index, TAOS_FIELD* field);

SFieldSupInfo* tscFieldInfoGetSupp(SFieldInfo* pFieldInfo, int32_t index);
TAOS_FIELD* tscFieldInfoGetField(SFieldInfo* pFieldInfo, int32_t index);
H
hzcheng 已提交
156

H
hjxilinx 已提交
157 158
void tscFieldInfoUpdateOffset(SQueryInfo* pQueryInfo);
void tscFieldInfoCopy(SFieldInfo* dst, const SFieldInfo* src);
159
void tscFieldInfoUpdateOffsetForInterResult(SQueryInfo* pQueryInfo);
H
hzcheng 已提交
160

H
hjxilinx 已提交
161 162
int16_t tscFieldInfoGetOffset(SQueryInfo* pQueryInfo, int32_t index);
void    tscFieldInfoClear(SFieldInfo* pFieldInfo);
H
Haojun Liao 已提交
163 164 165

static FORCE_INLINE int32_t tscNumOfFields(SQueryInfo* pQueryInfo) { return pQueryInfo->fieldsInfo.numOfOutput; }

H
hjxilinx 已提交
166
int32_t tscFieldInfoCompare(const SFieldInfo* pFieldInfo1, const SFieldInfo* pFieldInfo2);
H
hzcheng 已提交
167

S
slguan 已提交
168
void addExprParams(SSqlExpr* pExpr, char* argument, int32_t type, int32_t bytes, int16_t tableIndex);
H
hzcheng 已提交
169

H
hjxilinx 已提交
170
int32_t   tscGetResRowLength(SArray* pExprList);
H
hjxilinx 已提交
171 172

SSqlExpr* tscSqlExprInsert(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
173
    int16_t size, int16_t interSize, bool isTagCol);
H
hjxilinx 已提交
174 175

SSqlExpr* tscSqlExprAppend(SQueryInfo* pQueryInfo, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
176
                           int16_t size, int16_t interSize, bool isTagCol);
H
hjxilinx 已提交
177

178
SSqlExpr* tscSqlExprUpdate(SQueryInfo* pQueryInfo, int32_t index, int16_t functionId, int16_t srcColumnIndex, int16_t type,
H
hzcheng 已提交
179
                           int16_t size);
180
size_t   tscSqlExprNumOfExprs(SQueryInfo* pQueryInfo);
H
hzcheng 已提交
181

182
SSqlExpr* tscSqlExprGet(SQueryInfo* pQueryInfo, int32_t index);
H
hjxilinx 已提交
183
void      tscSqlExprCopy(SArray* dst, const SArray* src, uint64_t uid, bool deepcopy);
H
hjxilinx 已提交
184
void      tscSqlExprInfoDestroy(SArray* pExprInfo);
S
slguan 已提交
185

186 187
SColumn* tscColumnClone(const SColumn* src);
SColumn* tscColumnListInsert(SArray* pColList, SColumnIndex* colIndex);
188
SArray* tscColumnListClone(const SArray* src, int16_t tableIndex);
189
void tscColumnListDestroy(SArray* pColList);
H
hzcheng 已提交
190 191 192 193 194

int32_t tscValidateName(SSQLToken* pToken);

void tscIncStreamExecutionCount(void* pStream);

H
hjxilinx 已提交
195
bool tscValidateColumnId(STableMetaInfo* pTableMetaInfo, int32_t colId);
H
hzcheng 已提交
196

S
slguan 已提交
197
// get starter position of metric query condition (query on tags) in SSqlCmd.payload
198
SCond* tsGetSTableQueryCond(STagCond* pCond, uint64_t uid);
199
void   tsSetSTableQueryCond(STagCond* pTagCond, uint64_t uid, SBufferWriter* bw);
S
slguan 已提交
200 201

void tscTagCondCopy(STagCond* dest, const STagCond* src);
H
hzcheng 已提交
202 203
void tscTagCondRelease(STagCond* pCond);

204
void tscGetSrcColumnInfo(SSrcColumnInfo* pColInfo, SQueryInfo* pQueryInfo);
H
hzcheng 已提交
205 206 207

void tscSetFreeHeatBeat(STscObj* pObj);
bool tscShouldFreeHeatBeat(SSqlObj* pHb);
H
hjxilinx 已提交
208
bool tscShouldBeFreed(SSqlObj* pSql);
S
slguan 已提交
209

210
STableMetaInfo* tscGetTableMetaInfoFromCmd(SSqlCmd *pCmd, int32_t subClauseIndex, int32_t tableIndex);
H
hjxilinx 已提交
211
STableMetaInfo* tscGetMetaInfo(SQueryInfo *pQueryInfo, int32_t tableIndex);
212 213

SQueryInfo *tscGetQueryInfoDetail(SSqlCmd* pCmd, int32_t subClauseIndex);
214
int32_t tscGetQueryInfoDetailSafely(SSqlCmd *pCmd, int32_t subClauseIndex, SQueryInfo** pQueryInfo);
215

216
void tscClearTableMetaInfo(STableMetaInfo* pTableMetaInfo, bool removeFromCache);
S
slguan 已提交
217

218
STableMetaInfo* tscAddTableMetaInfo(SQueryInfo* pQueryInfo, const char* name, STableMeta* pTableMeta,
219
    SVgroupsInfo* vgroupList, SArray* pTagCols);
220

H
hjxilinx 已提交
221
STableMetaInfo* tscAddEmptyMetaInfo(SQueryInfo *pQueryInfo);
222
int32_t tscAddSubqueryInfo(SSqlCmd *pCmd);
H
hjxilinx 已提交
223 224 225

void tscInitQueryInfo(SQueryInfo* pQueryInfo);

226
void tscClearSubqueryInfo(SSqlCmd* pCmd);
S
slguan 已提交
227

H
hjxilinx 已提交
228
int  tscGetSTableVgroupInfo(SSqlObj* pSql, int32_t clauseIndex);
H
hjxilinx 已提交
229
int  tscGetTableMeta(SSqlObj* pSql, STableMetaInfo* pTableMetaInfo);
H
hjxilinx 已提交
230
int  tscGetMeterMetaEx(SSqlObj* pSql, STableMetaInfo* pTableMetaInfo, bool createIfNotExists);
S
slguan 已提交
231 232 233

void tscResetForNextRetrieve(SSqlRes* pRes);

234
void tscAddTimestampColumn(SQueryInfo* pQueryInfo, int16_t functionId, int16_t tableIndex);
H
hzcheng 已提交
235 236
void tscDoQuery(SSqlObj* pSql);

S
slguan 已提交
237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 253
/**
 * The create object function must be successful expect for the out of memory issue.
 *
 * Therefore, the metermeta/metricmeta object is directly passed to the newly created subquery object from the
 * previous sql object, instead of retrieving the metermeta/metricmeta from cache.
 *
 * Because the metermeta/metricmeta may have been released by other threads, resulting in the retrieving failed as
 * well as the create function.
 *
 * @param pSql
 * @param vnodeIndex
 * @param tableIndex
 * @param fp
 * @param param
 * @param pPrevSql
 * @return
 */
H
Haojun Liao 已提交
254 255
SSqlObj* createSimpleSubObj(SSqlObj* pSql, void (*fp)(), void* param, int32_t cmd);

256
SSqlObj* createSubqueryObj(SSqlObj* pSql, int16_t tableIndex, void (*fp)(), void* param, int32_t cmd, SSqlObj* pPrevSql);
257
void     addGroupInfoForSubquery(SSqlObj* pParentObj, SSqlObj* pSql, int32_t subClauseIndex, int32_t tableIndex);
S
slguan 已提交
258

259
void doAddGroupColumnForSubquery(SQueryInfo* pQueryInfo, int32_t tagIndex);
S
slguan 已提交
260

H
Haojun Liao 已提交
261
int16_t tscGetJoinTagColIdByUid(STagCond* pTagCond, uint64_t uid);
S
slguan 已提交
262

H
hjxilinx 已提交
263
void tscPrintSelectClause(SSqlObj* pSql, int32_t subClauseIndex);
H
hjxilinx 已提交
264

265
bool hasMoreVnodesToTry(SSqlObj *pSql);
H
Haojun Liao 已提交
266 267
bool hasMoreClauseToTry(SSqlObj* pSql);

268 269
void tscTryQueryNextVnode(SSqlObj *pSql, __async_cb_func_t fp);
void tscAsyncQuerySingleRowForNextVnode(void *param, TAOS_RES *tres, int numOfRows);
H
Haojun Liao 已提交
270
void tscTryQueryNextClause(SSqlObj* pSql, __async_cb_func_t fp);
271
int  tscSetMgmtEpSetFromCfg(const char *first, const char *second);
272

weixin_48148422's avatar
weixin_48148422 已提交
273 274 275 276
void* malloc_throw(size_t size);
void* calloc_throw(size_t nmemb, size_t size);
char* strdup_throw(const char* str);

H
hzcheng 已提交
277 278 279 280 281
#ifdef __cplusplus
}
#endif

#endif  // TDENGINE_TSCUTIL_H