tscUtil.h 9.7 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
 */
S
slguan 已提交
26
#include <limits.h>
H
hzcheng 已提交
27 28
#include <stdio.h>
#include "textbuffer.h"
H
hjxilinx 已提交
29
#include "tscSecondaryMerge.h"
S
slguan 已提交
30 31
#include "tsclient.h"
#include "tsdb.h"
H
hzcheng 已提交
32

H
hjxilinx 已提交
33 34
#define UTIL_METER_IS_METRIC(metaInfo) \
  (((metaInfo)->pMeterMeta != NULL) && ((metaInfo)->pMeterMeta->meterType == TSDB_METER_METRIC))
H
hjxilinx 已提交
35 36 37
#define UTIL_METER_IS_NOMRAL_METER(metaInfo) (!(UTIL_METER_IS_METRIC(metaInfo)))
#define UTIL_METER_IS_CREATE_FROM_METRIC(metaInfo) \
  (((metaInfo)->pMeterMeta != NULL) && ((metaInfo)->pMeterMeta->meterType == TSDB_METER_MTABLE))
H
hzcheng 已提交
38

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

H
hzcheng 已提交
41 42 43 44 45 46
typedef struct SParsedColElem {
  int16_t colIndex;
  int16_t offset;
} SParsedColElem;

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

S
slguan 已提交
53 54 55 56 57 58 59 60 61 62 63 64 65
typedef struct SJoinSubquerySupporter {
  SSubqueryState* pState;
  SSqlObj*        pObj;           // parent SqlObj
  bool            hasMore;        // has data from vnode to fetch
  int32_t         subqueryIndex;  // index of sub query
  int64_t         interval;       // interval time
  SLimitVal       limit;          // limit info
  uint64_t        uid;            // query meter uid
  SColumnBaseInfo colList;        // previous query information
  SSqlExprInfo    exprsInfo;
  SFieldInfo      fieldsInfo;
  STagCond        tagCond;
  SSqlGroupbyExpr groupbyExpr;
H
hjxilinx 已提交
66 67 68
  struct STSBuf*  pTSBuf;          // the TSBuf struct that holds the compressed timestamp array
  FILE*           f;               // temporary file in order to create TSBuf
  char            path[PATH_MAX];  // temporary file path
S
slguan 已提交
69 70
} SJoinSubquerySupporter;

H
hjxilinx 已提交
71
void              tscDestroyDataBlock(STableDataBlocks* pDataBlock);
S
slguan 已提交
72
STableDataBlocks* tscCreateDataBlock(int32_t size);
H
hjxilinx 已提交
73 74 75 76 77 78 79 80 81
void              tscAppendDataBlock(SDataBlockList* pList, STableDataBlocks* pBlocks);
SParamInfo*       tscAddParamToDataBlock(STableDataBlocks* pDataBlock, char type, uint8_t timePrec, short bytes,
                                         uint32_t offset);

SDataBlockList*   tscCreateBlockArrayList();
void*             tscDestroyBlockArrayList(SDataBlockList* pList);
int32_t           tscCopyDataBlockToPayload(SSqlObj* pSql, STableDataBlocks* pDataBlock);
void              tscFreeUnusedDataBlocks(SDataBlockList* pList);
int32_t           tscMergeTableDataBlocks(SSqlObj* pSql, SDataBlockList* pDataList);
S
slguan 已提交
82 83 84
STableDataBlocks* tscGetDataBlockFromList(void* pHashList, SDataBlockList* pDataBlockList, int64_t id, int32_t size,
                                          int32_t startOffset, int32_t rowSize, char* tableId);
STableDataBlocks* tscCreateDataBlockEx(size_t size, int32_t rowSize, int32_t startOffset, char* name);
H
hzcheng 已提交
85

H
hjxilinx 已提交
86
SVnodeSidList*    tscGetVnodeSidList(SMetricMeta* pMetricmeta, int32_t vnodeIdx);
H
hzcheng 已提交
87 88 89 90 91
SMeterSidExtInfo* tscGetMeterSidInfo(SVnodeSidList* pSidList, int32_t idx);

/**
 *
 * for the projection query on metric or point interpolation query on metric,
S
slguan 已提交
92
 * we iterate all the meters, instead of invoke query on all qualified meters simultaneously.
H
hzcheng 已提交
93 94 95 96 97
 *
 * @param pSql  sql object
 * @return
 */
bool tscIsPointInterpQuery(SSqlCmd* pCmd);
S
slguan 已提交
98 99 100 101 102 103 104 105 106
bool tscIsTWAQuery(SSqlCmd* pCmd);
bool tscProjectionQueryOnMetric(SSqlCmd* pCmd);
bool tscIsTwoStageMergeMetricQuery(SSqlCmd* pCmd);
bool tscQueryOnMetric(SSqlCmd* pCmd);
bool tscQueryMetricTags(SSqlCmd* pCmd);
bool tscIsSelectivityWithTagQuery(SSqlCmd* pCmd);

void tscAddSpecialColumnForSelect(SSqlCmd* pCmd, int32_t outputColIndex, int16_t functionId, SColumnIndex* pIndex,
                                  SSchema* pColSchema, int16_t isTag);
H
hzcheng 已提交
107

S
slguan 已提交
108
void addRequiredTagColumn(SSqlCmd* pCmd, int32_t tagColIndex, int32_t tableIndex);
H
hjxilinx 已提交
109

S
slguan 已提交
110
int32_t setMeterID(SSqlObj* pSql, SSQLToken* pzTableName, int32_t tableIndex);
H
hjxilinx 已提交
111
void    tscClearInterpInfo(SSqlCmd* pCmd);
H
hzcheng 已提交
112 113 114 115 116 117

bool tscIsInsertOrImportData(char* sqlstr);

/* use for keep current db info temporarily, for handle table with db prefix */
void tscGetDBInfoFromMeterId(char* meterId, char* db);

S
slguan 已提交
118
int tscAllocPayload(SSqlCmd* pCmd, int size);
H
hzcheng 已提交
119 120 121

void tscFieldInfoSetValFromSchema(SFieldInfo* pFieldInfo, int32_t index, SSchema* pSchema);
void tscFieldInfoSetValFromField(SFieldInfo* pFieldInfo, int32_t index, TAOS_FIELD* pField);
H
hjxilinx 已提交
122
void tscFieldInfoSetValue(SFieldInfo* pFieldInfo, int32_t index, int8_t type, const char* name, int16_t bytes);
S
slguan 已提交
123
void tscFieldInfoUpdateVisible(SFieldInfo* pFieldInfo, int32_t index, bool visible);
H
hzcheng 已提交
124 125

void tscFieldInfoCalOffset(SSqlCmd* pCmd);
S
slguan 已提交
126 127 128
void tscFieldInfoUpdateOffset(SSqlCmd* pCmd);
void tscFieldInfoCopy(SFieldInfo* src, SFieldInfo* dst, const int32_t* indexList, int32_t size);
void tscFieldInfoCopyAll(SFieldInfo* src, SFieldInfo* dst);
H
hzcheng 已提交
129 130

TAOS_FIELD* tscFieldInfoGetField(SSqlCmd* pCmd, int32_t index);
H
hjxilinx 已提交
131 132 133
int16_t     tscFieldInfoGetOffset(SSqlCmd* pCmd, int32_t index);
int32_t     tscGetResRowLength(SSqlCmd* pCmd);
void        tscClearFieldInfo(SFieldInfo* pFieldInfo);
H
hzcheng 已提交
134

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

S
slguan 已提交
137
SSqlExpr* tscSqlExprInsert(SSqlCmd* pCmd, int32_t index, int16_t functionId, SColumnIndex* pColIndex, int16_t type,
H
hjxilinx 已提交
138 139 140
                           int16_t size, int16_t interSize);
SSqlExpr* tscSqlExprInsertEmpty(SSqlCmd* pCmd, int32_t index, int16_t functionId);

H
hzcheng 已提交
141 142 143 144
SSqlExpr* tscSqlExprUpdate(SSqlCmd* pCmd, int32_t index, int16_t functionId, int16_t srcColumnIndex, int16_t type,
                           int16_t size);

SSqlExpr* tscSqlExprGet(SSqlCmd* pCmd, int32_t index);
H
hjxilinx 已提交
145
void      tscSqlExprCopy(SSqlExprInfo* dst, const SSqlExprInfo* src, uint64_t uid);
S
slguan 已提交
146 147

SColumnBase* tscColumnBaseInfoInsert(SSqlCmd* pCmd, SColumnIndex* colIndex);
H
hjxilinx 已提交
148 149
void         tscColumnFilterInfoCopy(SColumnFilterInfo* dst, const SColumnFilterInfo* src);
void         tscColumnBaseCopy(SColumnBase* dst, const SColumnBase* src);
H
hzcheng 已提交
150

H
hjxilinx 已提交
151
void         tscColumnBaseInfoCopy(SColumnBaseInfo* dst, const SColumnBaseInfo* src, int16_t tableIndex);
S
slguan 已提交
152
SColumnBase* tscColumnBaseInfoGet(SColumnBaseInfo* pColumnBaseInfo, int32_t index);
H
hjxilinx 已提交
153
void         tscColumnBaseInfoUpdateTableIndex(SColumnBaseInfo* pColList, int16_t tableIndex);
H
hzcheng 已提交
154

S
slguan 已提交
155 156
void tscColumnBaseInfoReserve(SColumnBaseInfo* pColumnBaseInfo, int32_t size);
void tscColumnBaseInfoDestroy(SColumnBaseInfo* pColumnBaseInfo);
H
hzcheng 已提交
157 158 159 160 161 162 163

int32_t tscValidateName(SSQLToken* pToken);

void tscIncStreamExecutionCount(void* pStream);

bool tscValidateColumnId(SSqlCmd* pCmd, int32_t colId);

S
slguan 已提交
164
// get starter position of metric query condition (query on tags) in SSqlCmd.payload
S
slguan 已提交
165
SCond* tsGetMetricQueryCondPos(STagCond* pCond, uint64_t tableIndex);
H
hjxilinx 已提交
166
void   tsSetMetricQueryCond(STagCond* pTagCond, uint64_t uid, const char* str);
S
slguan 已提交
167 168

void tscTagCondCopy(STagCond* dest, const STagCond* src);
H
hzcheng 已提交
169 170 171 172 173 174 175 176 177
void tscTagCondRelease(STagCond* pCond);
void tscTagCondSetQueryCondType(STagCond* pCond, int16_t type);

void tscGetSrcColumnInfo(SSrcColumnInfo* pColInfo, SSqlCmd* pCmd);

void tscSetFreeHeatBeat(STscObj* pObj);
bool tscShouldFreeHeatBeat(SSqlObj* pHb);
void tscCleanSqlCmd(SSqlCmd* pCmd);
bool tscShouldFreeAsyncSqlObj(SSqlObj* pSql);
S
slguan 已提交
178

H
hjxilinx 已提交
179
void            tscRemoveAllMeterMetaInfo(SSqlCmd* pCmd, bool removeFromCache);
S
slguan 已提交
180 181
SMeterMetaInfo* tscGetMeterMetaInfo(SSqlCmd* pCmd, int32_t index);
SMeterMetaInfo* tscGetMeterMetaInfoByUid(SSqlCmd* pCmd, uint64_t uid, int32_t* index);
H
hjxilinx 已提交
182
void            tscClearMeterMetaInfo(SMeterMetaInfo* pMeterMetaInfo, bool removeFromCache);
S
slguan 已提交
183 184 185 186 187 188

SMeterMetaInfo* tscAddMeterMetaInfo(SSqlCmd* pCmd, const char* name, SMeterMeta* pMeterMeta, SMetricMeta* pMetricMeta,
                                    int16_t numOfTags, int16_t* tags);
SMeterMetaInfo* tscAddEmptyMeterMetaInfo(SSqlCmd* pCmd);

void tscGetMetricMetaCacheKey(SSqlCmd* pCmd, char* keyStr, uint64_t uid);
H
hjxilinx 已提交
189 190 191
int  tscGetMetricMeta(SSqlObj* pSql);
int  tscGetMeterMeta(SSqlObj* pSql, char* meterId, int32_t tableIndex);
int  tscGetMeterMetaEx(SSqlObj* pSql, char* meterId, bool createIfNotExists);
S
slguan 已提交
192 193 194 195

void tscResetForNextRetrieve(SSqlRes* pRes);

void tscAddTimestampColumn(SSqlCmd* pCmd, int16_t functionId, int16_t tableIndex);
H
hzcheng 已提交
196 197
void tscDoQuery(SSqlObj* pSql);

S
slguan 已提交
198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
/**
 * 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
hjxilinx 已提交
215 216
SSqlObj* createSubqueryObj(SSqlObj* pSql, int16_t tableIndex, void (*fp)(), void* param, SSqlObj* pPrevSql);
void     addGroupInfoForSubquery(SSqlObj* pParentObj, SSqlObj* pSql, int32_t tableIndex);
S
slguan 已提交
217 218 219

void doAddGroupColumnForSubquery(SSqlCmd* pCmd, int32_t tagIndex);

H
hjxilinx 已提交
220
int16_t tscGetJoinTagColIndexByUid(STagCond* pTagCond, uint64_t uid);
S
slguan 已提交
221

L
lihui 已提交
222
TAOS* taos_connect_a(char* ip, char* user, char* pass, char* db, uint16_t port, void (*fp)(void*, TAOS_RES*, int),
S
slguan 已提交
223 224
                     void* param, void** taos);

S
slguan 已提交
225
void sortRemoveDuplicates(STableDataBlocks* dataBuf);
H
hzcheng 已提交
226 227 228 229 230
#ifdef __cplusplus
}
#endif

#endif  // TDENGINE_TSCUTIL_H