319 lines
11 KiB
C
319 lines
11 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_VNODE_DEF_H_
|
|
#define _TD_VNODE_DEF_H_
|
|
|
|
#include "executor.h"
|
|
#include "filter.h"
|
|
#include "qworker.h"
|
|
#include "sync.h"
|
|
#include "tRealloc.h"
|
|
#include "tchecksum.h"
|
|
#include "tcoding.h"
|
|
#include "tcompare.h"
|
|
#include "tcompression.h"
|
|
#include "tdatablock.h"
|
|
#include "tdb.h"
|
|
#include "tencode.h"
|
|
#include "tfs.h"
|
|
#include "tglobal.h"
|
|
#include "tjson.h"
|
|
#include "tlist.h"
|
|
#include "tlockfree.h"
|
|
#include "tlosertree.h"
|
|
#include "tlrucache.h"
|
|
#include "tmsgcb.h"
|
|
#include "tref.h"
|
|
#include "tskiplist.h"
|
|
#include "tstream.h"
|
|
#include "ttime.h"
|
|
#include "ttimer.h"
|
|
#include "wal.h"
|
|
|
|
#include "vnode.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
typedef struct SVnodeInfo SVnodeInfo;
|
|
typedef struct SMeta SMeta;
|
|
typedef struct SSma SSma;
|
|
typedef struct STsdb STsdb;
|
|
typedef struct STQ STQ;
|
|
typedef struct SVState SVState;
|
|
typedef struct SVBufPool SVBufPool;
|
|
typedef struct SQWorker SQHandle;
|
|
typedef struct STsdbKeepCfg STsdbKeepCfg;
|
|
typedef struct SMetaSnapReader SMetaSnapReader;
|
|
typedef struct SMetaSnapWriter SMetaSnapWriter;
|
|
typedef struct STsdbSnapReader STsdbSnapReader;
|
|
typedef struct STsdbSnapWriter STsdbSnapWriter;
|
|
typedef struct SSnapDataHdr SSnapDataHdr;
|
|
|
|
#define VNODE_META_DIR "meta"
|
|
#define VNODE_TSDB_DIR "tsdb"
|
|
#define VNODE_TQ_DIR "tq"
|
|
#define VNODE_WAL_DIR "wal"
|
|
#define VNODE_TSMA_DIR "tsma"
|
|
#define VNODE_RSMA_DIR "rsma"
|
|
#define VNODE_RSMA0_DIR "tsdb"
|
|
#define VNODE_RSMA1_DIR "rsma1"
|
|
#define VNODE_RSMA2_DIR "rsma2"
|
|
|
|
// vnd.h
|
|
void* vnodeBufPoolMalloc(SVBufPool* pPool, int size);
|
|
void vnodeBufPoolFree(SVBufPool* pPool, void* p);
|
|
|
|
// meta
|
|
typedef struct SMCtbCursor SMCtbCursor;
|
|
typedef struct SMStbCursor SMStbCursor;
|
|
typedef struct STbUidStore STbUidStore;
|
|
|
|
int metaOpen(SVnode* pVnode, SMeta** ppMeta);
|
|
int metaClose(SMeta* pMeta);
|
|
int metaBegin(SMeta* pMeta);
|
|
int metaCommit(SMeta* pMeta);
|
|
int metaCreateSTable(SMeta* pMeta, int64_t version, SVCreateStbReq* pReq);
|
|
int metaAlterSTable(SMeta* pMeta, int64_t version, SVCreateStbReq* pReq);
|
|
int metaDropSTable(SMeta* pMeta, int64_t verison, SVDropStbReq* pReq);
|
|
int metaCreateTable(SMeta* pMeta, int64_t version, SVCreateTbReq* pReq);
|
|
int metaDropTable(SMeta* pMeta, int64_t version, SVDropTbReq* pReq, SArray* tbUids);
|
|
int metaTtlDropTable(SMeta* pMeta, int64_t ttl, SArray* tbUids);
|
|
int metaAlterTable(SMeta* pMeta, int64_t version, SVAlterTbReq* pReq, STableMetaRsp* pMetaRsp);
|
|
SSchemaWrapper* metaGetTableSchema(SMeta* pMeta, tb_uid_t uid, int32_t sver, bool isinline);
|
|
STSchema* metaGetTbTSchema(SMeta* pMeta, tb_uid_t uid, int32_t sver);
|
|
int32_t metaGetTbTSchemaEx(SMeta* pMeta, tb_uid_t suid, tb_uid_t uid, int32_t sver, STSchema** ppTSchema);
|
|
int metaGetTableEntryByName(SMetaReader* pReader, const char* name);
|
|
tb_uid_t metaGetTableEntryUidByName(SMeta* pMeta, const char* name);
|
|
int metaGetTbNum(SMeta* pMeta);
|
|
SMCtbCursor* metaOpenCtbCursor(SMeta* pMeta, tb_uid_t uid);
|
|
void metaCloseCtbCursor(SMCtbCursor* pCtbCur);
|
|
tb_uid_t metaCtbCursorNext(SMCtbCursor* pCtbCur);
|
|
SMStbCursor* metaOpenStbCursor(SMeta* pMeta, tb_uid_t uid);
|
|
void metaCloseStbCursor(SMStbCursor* pStbCur);
|
|
tb_uid_t metaStbCursorNext(SMStbCursor* pStbCur);
|
|
STSma* metaGetSmaInfoByIndex(SMeta* pMeta, int64_t indexUid);
|
|
STSmaWrapper* metaGetSmaInfoByTable(SMeta* pMeta, tb_uid_t uid, bool deepCopy);
|
|
SArray* metaGetSmaIdsByTable(SMeta* pMeta, tb_uid_t uid);
|
|
SArray* metaGetSmaTbUids(SMeta* pMeta);
|
|
void* metaGetIdx(SMeta* pMeta);
|
|
void* metaGetIvtIdx(SMeta* pMeta);
|
|
int metaTtlSmaller(SMeta* pMeta, uint64_t time, SArray* uidList);
|
|
|
|
int32_t metaCreateTSma(SMeta* pMeta, int64_t version, SSmaCfg* pCfg);
|
|
int32_t metaDropTSma(SMeta* pMeta, int64_t indexUid);
|
|
|
|
// tsdb
|
|
int tsdbOpen(SVnode* pVnode, STsdb** ppTsdb, const char* dir, STsdbKeepCfg* pKeepCfg);
|
|
int tsdbClose(STsdb** pTsdb);
|
|
int32_t tsdbBegin(STsdb* pTsdb);
|
|
int32_t tsdbCommit(STsdb* pTsdb);
|
|
int tsdbScanAndConvertSubmitMsg(STsdb* pTsdb, SSubmitReq* pMsg);
|
|
int tsdbInsertData(STsdb* pTsdb, int64_t version, SSubmitReq* pMsg, SSubmitRsp* pRsp);
|
|
int32_t tsdbInsertTableData(STsdb* pTsdb, int64_t version, SSubmitMsgIter* pMsgIter, SSubmitBlk* pBlock,
|
|
SSubmitBlkRsp* pRsp);
|
|
int32_t tsdbDeleteTableData(STsdb* pTsdb, int64_t version, tb_uid_t suid, tb_uid_t uid, TSKEY sKey, TSKEY eKey);
|
|
STsdbReader tsdbQueryCacheLastT(STsdb* tsdb, SQueryTableDataCond* pCond, STableListInfo* tableList, uint64_t qId,
|
|
void* pMemRef);
|
|
|
|
// tq
|
|
int tqInit();
|
|
void tqCleanUp();
|
|
STQ* tqOpen(const char* path, SVnode* pVnode, SWal* pWal);
|
|
void tqClose(STQ*);
|
|
int tqPushMsg(STQ*, void* msg, int32_t msgLen, tmsg_t msgType, int64_t ver);
|
|
int tqCommit(STQ*);
|
|
int32_t tqUpdateTbUidList(STQ* pTq, const SArray* tbUidList, bool isAdd);
|
|
int32_t tqProcessVgChangeReq(STQ* pTq, char* msg, int32_t msgLen);
|
|
int32_t tqProcessVgDeleteReq(STQ* pTq, char* msg, int32_t msgLen);
|
|
int32_t tqProcessOffsetCommitReq(STQ* pTq, char* msg, int32_t msgLen);
|
|
int32_t tqProcessPollReq(STQ* pTq, SRpcMsg* pMsg, int32_t workerId);
|
|
int32_t tqProcessTaskDeployReq(STQ* pTq, char* msg, int32_t msgLen);
|
|
int32_t tqProcessTaskDropReq(STQ* pTq, char* msg, int32_t msgLen);
|
|
int32_t tqProcessStreamTrigger(STQ* pTq, SSubmitReq* data);
|
|
int32_t tqProcessTaskRunReq(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tqProcessTaskDispatchReq(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tqProcessTaskRecoverReq(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tqProcessTaskDispatchRsp(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tqProcessTaskRecoverRsp(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tqProcessTaskRetrieveReq(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tqProcessTaskRetrieveRsp(STQ* pTq, SRpcMsg* pMsg);
|
|
int32_t tsdbGetStbIdList(SMeta* pMeta, int64_t suid, SArray* list);
|
|
|
|
SSubmitReq* tdBlockToSubmit(const SArray* pBlocks, const STSchema* pSchema, bool createTb, int64_t suid,
|
|
const char* stbFullName, int32_t vgId);
|
|
|
|
// sma
|
|
int32_t smaInit();
|
|
void smaCleanUp();
|
|
int32_t smaOpen(SVnode* pVnode);
|
|
int32_t smaClose(SSma* pSma);
|
|
int32_t smaBegin(SSma* pSma);
|
|
int32_t smaPreCommit(SSma* pSma);
|
|
int32_t smaCommit(SSma* pSma);
|
|
int32_t smaPostCommit(SSma* pSma);
|
|
|
|
int32_t tdProcessTSmaCreate(SSma* pSma, int64_t version, const char* msg);
|
|
int32_t tdProcessTSmaInsert(SSma* pSma, int64_t indexUid, const char* msg);
|
|
int64_t tdRSmaGetMaxSubmitVer(SSma* pSma, int8_t level);
|
|
|
|
int32_t tdProcessRSmaCreate(SVnode* pVnode, SVCreateStbReq* pReq);
|
|
int32_t tdProcessRSmaSubmit(SSma* pSma, void* pMsg, int32_t inputType);
|
|
int32_t tdFetchTbUidList(SSma* pSma, STbUidStore** ppStore, tb_uid_t suid, tb_uid_t uid);
|
|
int32_t tdUpdateTbUidList(SSma* pSma, STbUidStore* pUidStore);
|
|
void tdUidStoreDestory(STbUidStore* pStore);
|
|
void* tdUidStoreFree(STbUidStore* pStore);
|
|
|
|
// SMetaSnapReader ========================================
|
|
int32_t metaSnapReaderOpen(SMeta* pMeta, int64_t sver, int64_t ever, SMetaSnapReader** ppReader);
|
|
int32_t metaSnapReaderClose(SMetaSnapReader** ppReader);
|
|
int32_t metaSnapRead(SMetaSnapReader* pReader, uint8_t** ppData);
|
|
// SMetaSnapWriter ========================================
|
|
int32_t metaSnapWriterOpen(SMeta* pMeta, int64_t sver, int64_t ever, SMetaSnapWriter** ppWriter);
|
|
int32_t metaSnapWrite(SMetaSnapWriter* pWriter, uint8_t* pData, uint32_t nData);
|
|
int32_t metaSnapWriterClose(SMetaSnapWriter** ppWriter, int8_t rollback);
|
|
// STsdbSnapReader ========================================
|
|
int32_t tsdbSnapReaderOpen(STsdb* pTsdb, int64_t sver, int64_t ever, STsdbSnapReader** ppReader);
|
|
int32_t tsdbSnapReaderClose(STsdbSnapReader** ppReader);
|
|
int32_t tsdbSnapRead(STsdbSnapReader* pReader, uint8_t** ppData);
|
|
// STsdbSnapWriter ========================================
|
|
int32_t tsdbSnapWriterOpen(STsdb* pTsdb, int64_t sver, int64_t ever, STsdbSnapWriter** ppWriter);
|
|
int32_t tsdbSnapWrite(STsdbSnapWriter* pWriter, uint8_t* pData, uint32_t nData);
|
|
int32_t tsdbSnapWriterClose(STsdbSnapWriter** ppWriter, int8_t rollback);
|
|
|
|
typedef struct {
|
|
int8_t streamType; // sma or other
|
|
int8_t dstType;
|
|
int16_t padding;
|
|
int32_t smaId;
|
|
int64_t tbUid;
|
|
int64_t lastReceivedVer;
|
|
int64_t lastCommittedVer;
|
|
} SStreamSinkInfo;
|
|
|
|
typedef struct {
|
|
SVnode* pVnode;
|
|
SHashObj* pHash; // streamId -> SStreamSinkInfo
|
|
} SSink;
|
|
|
|
// SVState
|
|
struct SVState {
|
|
int64_t committed;
|
|
int64_t applied;
|
|
int64_t applyTerm;
|
|
int64_t commitID;
|
|
int64_t commitTerm;
|
|
};
|
|
|
|
struct SVnodeInfo {
|
|
SVnodeCfg config;
|
|
SVState state;
|
|
};
|
|
|
|
typedef enum {
|
|
TSDB_TYPE_TSDB = 0, // TSDB
|
|
TSDB_TYPE_TSMA = 1, // TSMA
|
|
TSDB_TYPE_RSMA_L0 = 2, // RSMA Level 0
|
|
TSDB_TYPE_RSMA_L1 = 3, // RSMA Level 1
|
|
TSDB_TYPE_RSMA_L2 = 4, // RSMA Level 2
|
|
} ETsdbType;
|
|
|
|
struct STsdbKeepCfg {
|
|
int8_t precision; // precision always be used with below keep cfgs
|
|
int32_t days;
|
|
int32_t keep0;
|
|
int32_t keep1;
|
|
int32_t keep2;
|
|
};
|
|
|
|
struct SVnode {
|
|
char* path;
|
|
SVnodeCfg config;
|
|
SVState state;
|
|
STfs* pTfs;
|
|
SMsgCb msgCb;
|
|
SVBufPool* pPool;
|
|
SVBufPool* inUse;
|
|
SVBufPool* onCommit;
|
|
SVBufPool* onRecycle;
|
|
SMeta* pMeta;
|
|
SSma* pSma;
|
|
STsdb* pTsdb;
|
|
SWal* pWal;
|
|
STQ* pTq;
|
|
SSink* pSink;
|
|
tsem_t canCommit;
|
|
int64_t sync;
|
|
int32_t blockCount;
|
|
tsem_t syncSem;
|
|
SQHandle* pQuery;
|
|
};
|
|
|
|
#define TD_VID(PVNODE) ((PVNODE)->config.vgId)
|
|
|
|
#define VND_TSDB(vnd) ((vnd)->pTsdb)
|
|
#define VND_RSMA0(vnd) ((vnd)->pTsdb)
|
|
#define VND_RSMA1(vnd) ((vnd)->pSma->pRSmaTsdb1)
|
|
#define VND_RSMA2(vnd) ((vnd)->pSma->pRSmaTsdb2)
|
|
#define VND_RETENTIONS(vnd) (&(vnd)->config.tsdbCfg.retentions)
|
|
#define VND_IS_RSMA(v) ((v)->config.isRsma == 1)
|
|
#define VND_IS_TSMA(v) ((v)->config.isTsma == 1)
|
|
|
|
struct STbUidStore {
|
|
tb_uid_t suid;
|
|
SArray* tbUids;
|
|
SHashObj* uidHash;
|
|
};
|
|
|
|
struct SSma {
|
|
bool locked;
|
|
TdThreadMutex mutex;
|
|
SVnode* pVnode;
|
|
STsdb* pRSmaTsdb1;
|
|
STsdb* pRSmaTsdb2;
|
|
void* pTSmaEnv;
|
|
void* pRSmaEnv;
|
|
};
|
|
|
|
#define SMA_CFG(s) (&(s)->pVnode->config)
|
|
#define SMA_TSDB_CFG(s) (&(s)->pVnode->config.tsdbCfg)
|
|
#define SMA_RETENTION(s) ((SRetention*)&(s)->pVnode->config.tsdbCfg.retentions)
|
|
#define SMA_LOCKED(s) ((s)->locked)
|
|
#define SMA_META(s) ((s)->pVnode->pMeta)
|
|
#define SMA_VID(s) TD_VID((s)->pVnode)
|
|
#define SMA_TFS(s) ((s)->pVnode->pTfs)
|
|
#define SMA_TSMA_ENV(s) ((s)->pTSmaEnv)
|
|
#define SMA_RSMA_ENV(s) ((s)->pRSmaEnv)
|
|
#define SMA_RSMA_TSDB0(s) ((s)->pVnode->pTsdb)
|
|
#define SMA_RSMA_TSDB1(s) ((s)->pRSmaTsdb1)
|
|
#define SMA_RSMA_TSDB2(s) ((s)->pRSmaTsdb2)
|
|
|
|
// sma
|
|
void smaHandleRes(void* pVnode, int64_t smaId, const SArray* data);
|
|
|
|
struct SSnapDataHdr {
|
|
int8_t type;
|
|
int64_t size;
|
|
uint8_t data[];
|
|
};
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_VNODE_DEF_H_*/
|