vnodeUtil.h 3.2 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 26 27 28 29 30 31
/*
 * 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_VNODE_UTIL_H
#define TDENGINE_VNODE_UTIL_H

#ifdef __cplusplus
extern "C" {
#endif

/* get the qinfo struct address from the query struct address */
#define GET_COLUMN_BYTES(query, colidx) \
  ((query)->colList[(query)->pSelectExpr[colidx].pBase.colInfo.colIdxInBuf].data.bytes)
#define GET_COLUMN_TYPE(query, colidx) \
  ((query)->colList[(query)->pSelectExpr[colidx].pBase.colInfo.colIdxInBuf].data.type)

#define QUERY_IS_ASC_QUERY(q) (GET_FORWARD_DIRECTION_FACTOR((q)->order.order) == QUERY_ASC_FORWARD_STEP)
#define EXTRA_BYTES 2            // for possible compression deflation

S
slguan 已提交
32 33
#define GET_COL_DATA_POS(query, index, step) ((query)->pos + (index)*(step))

H
hzcheng 已提交
34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67
int vnodeGetEid(int days);

int vnodeCheckFileIntegrity(FILE *fp);

void vnodeCreateFileHeader(FILE *fp);

void vnodeCreateFileHeaderFd(int fd);

void vnodeGetHeadFileHeaderInfo(int fd, SVnodeHeadInfo *pHeadInfo);

void vnodeUpdateHeadFileHeader(int fd, SVnodeHeadInfo *pHeadInfo);

/**
 * check if two schema is identical or not
 * This function does not check if a schema is valid or not
 *
 * @param pSSchemaFirst
 * @param numOfCols1
 * @param pSSchemaSecond
 * @param numOfCols2
 * @return
 */
bool vnodeMeterSchemaIdentical(SColumn *pSchema1, int32_t numOfCols1, SColumn *pSchema2, int32_t numOfCols2);

/**
 * free SFields in SQuery
 * vnodeFreeFields must be called before free(pQuery->pBlock);
 * @param pQuery
 */
void vnodeFreeFields(SQuery *pQuery);

void vnodeUpdateFilterColumnIndex(SQuery* pQuery);
void vnodeUpdateQueryColumnIndex(SQuery* pQuery, SMeterObj* pMeterObj);

68
int32_t vnodeCreateFilterInfo(void* pQInfo, SQuery *pQuery);
H
hzcheng 已提交
69 70 71 72 73 74 75 76 77 78 79

bool vnodeFilterData(SQuery* pQuery, int32_t* numOfActualRead, int32_t index);
bool vnodeDoFilterData(SQuery* pQuery, int32_t elemPos);

bool vnodeIsProjectionQuery(SSqlFunctionExpr *pExpr, int32_t numOfOutput);

int32_t vnodeIncQueryRefCount(SQueryMeterMsg *pQueryMsg, SMeterSidExtInfo **pSids, SMeterObj **pMeterObjList,
                              int32_t *numOfInc);

void vnodeDecQueryRefCount(SQueryMeterMsg *pQueryMsg, SMeterObj **pMeterObjList, int32_t numOfInc);

S
slguan 已提交
80
int32_t vnodeSetMeterState(SMeterObj* pMeterObj, int32_t state);
S
slguan 已提交
81 82 83
void vnodeClearMeterState(SMeterObj* pMeterObj, int32_t state);
bool vnodeIsMeterState(SMeterObj* pMeterObj, int32_t state);
void vnodeSetMeterDeleting(SMeterObj* pMeterObj);
H
hjxilinx 已提交
84 85
int32_t vnodeSetMeterInsertImportStateEx(SMeterObj* pObj, int32_t st);

S
slguan 已提交
86
bool vnodeIsSafeToDeleteMeter(SVnodeObj* pVnode, int32_t sid);
S
slguan 已提交
87 88
void vnodeFreeColumnInfo(SColumnInfo* pColumnInfo);
bool isGroupbyNormalCol(SSqlGroupbyExpr* pExpr);
S
slguan 已提交
89

H
hzcheng 已提交
90 91 92 93 94
#ifdef __cplusplus
}
#endif

#endif  // TDENGINE_VNODE_UTIL_H