vnodeInt.h 11.5 KB
Newer Older
H
save  
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15
/*
 * 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/>.
 */

H
refact  
Hongze Cheng 已提交
16 17
#ifndef _TD_VNODE_DEF_H_
#define _TD_VNODE_DEF_H_
H
save  
Hongze Cheng 已提交
18

H
Hongze Cheng 已提交
19
#include "executor.h"
H
Hongze Cheng 已提交
20 21 22
#include "filter.h"
#include "qworker.h"
#include "sync.h"
H
Hongze Cheng 已提交
23
#include "tRealloc.h"
H
Hongze Cheng 已提交
24
#include "tchecksum.h"
H
more  
Hongze Cheng 已提交
25
#include "tcoding.h"
H
Hongze Cheng 已提交
26
#include "tcompare.h"
H
Hongze Cheng 已提交
27
#include "tcompression.h"
L
Liu Jicong 已提交
28
#include "tdatablock.h"
H
Hongze Cheng 已提交
29
#include "tdb.h"
H
Hongze Cheng 已提交
30
#include "tencode.h"
H
refact  
Hongze Cheng 已提交
31
#include "tfs.h"
H
Hongze Cheng 已提交
32
#include "tglobal.h"
H
Hongze Cheng 已提交
33
#include "tjson.h"
H
Hongze Cheng 已提交
34
#include "tlist.h"
H
refact  
Hongze Cheng 已提交
35
#include "tlockfree.h"
H
Hongze Cheng 已提交
36
#include "tlosertree.h"
H
Hongze Cheng 已提交
37
#include "tlrucache.h"
38
#include "tmsgcb.h"
H
Hongze Cheng 已提交
39
#include "tref.h"
H
Hongze Cheng 已提交
40
#include "tskiplist.h"
H
Hongze Cheng 已提交
41
#include "tstream.h"
H
Hongze Cheng 已提交
42
#include "ttime.h"
H
Hongze Cheng 已提交
43 44
#include "ttimer.h"
#include "wal.h"
H
save  
Hongze Cheng 已提交
45

H
Hongze Cheng 已提交
46 47
#include "vnode.h"

H
save  
Hongze Cheng 已提交
48 49 50 51
#ifdef __cplusplus
extern "C" {
#endif

H
Hongze Cheng 已提交
52 53 54 55 56 57 58 59 60 61
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;
H
Hongze Cheng 已提交
62
typedef struct SMetaSnapWriter SMetaSnapWriter;
H
Hongze Cheng 已提交
63
typedef struct STsdbSnapReader STsdbSnapReader;
H
Hongze Cheng 已提交
64
typedef struct STsdbSnapWriter STsdbSnapWriter;
H
Hongze Cheng 已提交
65
typedef struct SSnapDataHdr    SSnapDataHdr;
H
refact  
Hongze Cheng 已提交
66

C
Cary Xu 已提交
67 68 69 70 71
#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"
C
Cary Xu 已提交
72
#define VNODE_RSMA_DIR  "rsma"
C
Cary Xu 已提交
73
#define VNODE_RSMA0_DIR "tsdb"
C
Cary Xu 已提交
74 75
#define VNODE_RSMA1_DIR "rsma1"
#define VNODE_RSMA2_DIR "rsma2"
H
Hongze Cheng 已提交
76

H
Hongze Cheng 已提交
77
// vnd.h
H
Hongze Cheng 已提交
78 79
void* vnodeBufPoolMalloc(SVBufPool* pPool, int size);
void  vnodeBufPoolFree(SVBufPool* pPool, void* p);
H
Hongze Cheng 已提交
80 81 82

// meta
typedef struct SMCtbCursor SMCtbCursor;
C
Cary Xu 已提交
83
typedef struct SMStbCursor SMStbCursor;
H
Hongze Cheng 已提交
84 85 86 87 88 89 90
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);
H
Hongze Cheng 已提交
91
int             metaAlterSTable(SMeta* pMeta, int64_t version, SVCreateStbReq* pReq);
H
Hongze Cheng 已提交
92
int             metaDropSTable(SMeta* pMeta, int64_t verison, SVDropStbReq* pReq);
H
Hongze Cheng 已提交
93
int             metaCreateTable(SMeta* pMeta, int64_t version, SVCreateTbReq* pReq);
94
int             metaDropTable(SMeta* pMeta, int64_t version, SVDropTbReq* pReq, SArray* tbUids);
H
Hongze Cheng 已提交
95
int             metaTtlDropTable(SMeta* pMeta, int64_t ttl, SArray* tbUids);
H
more  
Hongze Cheng 已提交
96
int             metaAlterTable(SMeta* pMeta, int64_t version, SVAlterTbReq* pReq, STableMetaRsp* pMetaRsp);
H
Hongze Cheng 已提交
97 98
SSchemaWrapper* metaGetTableSchema(SMeta* pMeta, tb_uid_t uid, int32_t sver, bool isinline);
STSchema*       metaGetTbTSchema(SMeta* pMeta, tb_uid_t uid, int32_t sver);
H
Hongze Cheng 已提交
99
int32_t         metaGetTbTSchemaEx(SMeta* pMeta, tb_uid_t suid, tb_uid_t uid, int32_t sver, STSchema** ppTSchema);
H
Hongze Cheng 已提交
100
int             metaGetTableEntryByName(SMetaReader* pReader, const char* name);
H
Hongze Cheng 已提交
101
tb_uid_t        metaGetTableEntryUidByName(SMeta* pMeta, const char* name);
H
Hongze Cheng 已提交
102 103
int             metaGetTbNum(SMeta* pMeta);
SMCtbCursor*    metaOpenCtbCursor(SMeta* pMeta, tb_uid_t uid);
C
Cary Xu 已提交
104
void            metaCloseCtbCursor(SMCtbCursor* pCtbCur);
H
Hongze Cheng 已提交
105
tb_uid_t        metaCtbCursorNext(SMCtbCursor* pCtbCur);
C
Cary Xu 已提交
106 107 108
SMStbCursor*    metaOpenStbCursor(SMeta* pMeta, tb_uid_t uid);
void            metaCloseStbCursor(SMStbCursor* pStbCur);
tb_uid_t        metaStbCursorNext(SMStbCursor* pStbCur);
C
Cary Xu 已提交
109 110 111 112
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);
dengyihao's avatar
dengyihao 已提交
113
void*           metaGetIdx(SMeta* pMeta);
dengyihao's avatar
dengyihao 已提交
114
void*           metaGetIvtIdx(SMeta* pMeta);
H
Hongze Cheng 已提交
115
int             metaTtlSmaller(SMeta* pMeta, uint64_t time, SArray* uidList);
C
Cary Xu 已提交
116

117 118
int32_t metaCreateTSma(SMeta* pMeta, int64_t version, SSmaCfg* pCfg);
int32_t metaDropTSma(SMeta* pMeta, int64_t indexUid);
H
Hongze Cheng 已提交
119 120

// tsdb
H
Hongze Cheng 已提交
121 122 123 124
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);
H
Hongze Cheng 已提交
125
int32_t     tsdbDoRetention(STsdb* pTsdb, int64_t now);
H
Hongze Cheng 已提交
126 127 128 129 130 131 132
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);
H
Hongze Cheng 已提交
133 134

// tq
L
Liu Jicong 已提交
135 136
int     tqInit();
void    tqCleanUp();
H
Hongze Cheng 已提交
137 138 139 140
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*);
141
int32_t tqUpdateTbUidList(STQ* pTq, const SArray* tbUidList, bool isAdd);
H
Hongze Cheng 已提交
142
int32_t tqProcessVgChangeReq(STQ* pTq, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
143
int32_t tqProcessVgDeleteReq(STQ* pTq, char* msg, int32_t msgLen);
144
int32_t tqProcessOffsetCommitReq(STQ* pTq, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
145
int32_t tqProcessPollReq(STQ* pTq, SRpcMsg* pMsg, int32_t workerId);
L
Liu Jicong 已提交
146
int32_t tqProcessTaskDeployReq(STQ* pTq, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
147
int32_t tqProcessTaskDropReq(STQ* pTq, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
148
int32_t tqProcessStreamTrigger(STQ* pTq, SSubmitReq* data);
L
Liu Jicong 已提交
149 150 151 152 153
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);
L
Liu Jicong 已提交
154 155
int32_t tqProcessTaskRetrieveReq(STQ* pTq, SRpcMsg* pMsg);
int32_t tqProcessTaskRetrieveRsp(STQ* pTq, SRpcMsg* pMsg);
H
Hongze Cheng 已提交
156
int32_t tsdbGetStbIdList(SMeta* pMeta, int64_t suid, SArray* list);
H
Hongze Cheng 已提交
157

C
Cary Xu 已提交
158 159 160
SSubmitReq* tdBlockToSubmit(const SArray* pBlocks, const STSchema* pSchema, bool createTb, int64_t suid,
                            const char* stbFullName, int32_t vgId);

C
Cary Xu 已提交
161
// sma
C
Cary Xu 已提交
162 163
int32_t smaInit();
void    smaCleanUp();
164 165
int32_t smaOpen(SVnode* pVnode);
int32_t smaClose(SSma* pSma);
C
Cary Xu 已提交
166
int32_t smaBegin(SSma* pSma);
C
Cary Xu 已提交
167 168 169
int32_t smaPreCommit(SSma* pSma);
int32_t smaCommit(SSma* pSma);
int32_t smaPostCommit(SSma* pSma);
C
Cary Xu 已提交
170

C
Cary Xu 已提交
171
int32_t tdProcessTSmaCreate(SSma* pSma, int64_t version, const char* msg);
172
int32_t tdProcessTSmaInsert(SSma* pSma, int64_t indexUid, const char* msg);
173
int64_t tdRSmaGetMaxSubmitVer(SSma* pSma, int8_t level);
174

175
int32_t tdProcessRSmaCreate(SSma* pSma, SVCreateStbReq* pReq);
176
int32_t tdProcessRSmaSubmit(SSma* pSma, void* pMsg, int32_t inputType);
177
int32_t tdProcessRSmaDrop(SSma* pSma, SVDropStbReq* pReq);
178 179 180 181 182
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);

H
Hongze Cheng 已提交
183 184 185
// SMetaSnapReader ========================================
int32_t metaSnapReaderOpen(SMeta* pMeta, int64_t sver, int64_t ever, SMetaSnapReader** ppReader);
int32_t metaSnapReaderClose(SMetaSnapReader** ppReader);
H
Hongze Cheng 已提交
186
int32_t metaSnapRead(SMetaSnapReader* pReader, uint8_t** ppData);
H
Hongze Cheng 已提交
187
// SMetaSnapWriter ========================================
H
Hongze Cheng 已提交
188 189 190
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);
H
Hongze Cheng 已提交
191 192 193
// STsdbSnapReader ========================================
int32_t tsdbSnapReaderOpen(STsdb* pTsdb, int64_t sver, int64_t ever, STsdbSnapReader** ppReader);
int32_t tsdbSnapReaderClose(STsdbSnapReader** ppReader);
H
Hongze Cheng 已提交
194
int32_t tsdbSnapRead(STsdbSnapReader* pReader, uint8_t** ppData);
H
Hongze Cheng 已提交
195
// STsdbSnapWriter ========================================
H
Hongze Cheng 已提交
196 197 198 199
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);

L
Liu Jicong 已提交
200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
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;

H
refact  
Hongze Cheng 已提交
215 216 217 218
// SVState
struct SVState {
  int64_t committed;
  int64_t applied;
H
Hongze Cheng 已提交
219
  int64_t applyTerm;
H
Hongze Cheng 已提交
220
  int64_t commitID;
H
Hongze Cheng 已提交
221
  int64_t commitTerm;
H
refact  
Hongze Cheng 已提交
222 223
};

H
Hongze Cheng 已提交
224 225 226 227 228
struct SVnodeInfo {
  SVnodeCfg config;
  SVState   state;
};

C
Cary Xu 已提交
229 230 231 232 233 234 235 236
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;

L
Liu Jicong 已提交
237
struct STsdbKeepCfg {
C
Cary Xu 已提交
238 239 240 241 242
  int8_t  precision;  // precision always be used with below keep cfgs
  int32_t days;
  int32_t keep0;
  int32_t keep1;
  int32_t keep2;
243
};
C
Cary Xu 已提交
244

H
save  
Hongze Cheng 已提交
245
struct SVnode {
H
refact  
Hongze Cheng 已提交
246 247 248
  char*      path;
  SVnodeCfg  config;
  SVState    state;
H
Hongze Cheng 已提交
249 250
  STfs*      pTfs;
  SMsgCb     msgCb;
H
Hongze Cheng 已提交
251 252 253 254
  SVBufPool* pPool;
  SVBufPool* inUse;
  SVBufPool* onCommit;
  SVBufPool* onRecycle;
H
refact  
Hongze Cheng 已提交
255
  SMeta*     pMeta;
256
  SSma*      pSma;
H
refact  
Hongze Cheng 已提交
257 258
  STsdb*     pTsdb;
  SWal*      pWal;
L
Liu Jicong 已提交
259 260
  STQ*       pTq;
  SSink*     pSink;
H
refact  
Hongze Cheng 已提交
261
  tsem_t     canCommit;
262
  int64_t    sync;
263
  int32_t    blockCount;
wafwerar's avatar
wafwerar 已提交
264
  tsem_t     syncSem;
H
refact  
Hongze Cheng 已提交
265
  SQHandle*  pQuery;
H
save  
Hongze Cheng 已提交
266 267
};

C
Cary Xu 已提交
268
#define TD_VID(PVNODE) ((PVNODE)->config.vgId)
269

C
Cary Xu 已提交
270 271
#define VND_TSDB(vnd)       ((vnd)->pTsdb)
#define VND_RSMA0(vnd)      ((vnd)->pTsdb)
272 273
#define VND_RSMA1(vnd)      ((vnd)->pSma->pRSmaTsdb1)
#define VND_RSMA2(vnd)      ((vnd)->pSma->pRSmaTsdb2)
C
Cary Xu 已提交
274
#define VND_RETENTIONS(vnd) (&(vnd)->config.tsdbCfg.retentions)
C
Cary Xu 已提交
275 276
#define VND_IS_RSMA(v)      ((v)->config.isRsma == 1)
#define VND_IS_TSMA(v)      ((v)->config.isTsma == 1)
C
Cary Xu 已提交
277

H
Hongze Cheng 已提交
278 279 280 281 282 283
struct STbUidStore {
  tb_uid_t  suid;
  SArray*   tbUids;
  SHashObj* uidHash;
};

284 285 286 287 288 289 290 291 292 293 294 295
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)
H
Hongze Cheng 已提交
296
#define SMA_RETENTION(s)  ((SRetention*)&(s)->pVnode->config.tsdbCfg.retentions)
297 298 299 300 301 302 303 304 305
#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)
H
refact  
Hongze Cheng 已提交
306

L
Liu Jicong 已提交
307
// sma
L
Liu Jicong 已提交
308
void smaHandleRes(void* pVnode, int64_t smaId, const SArray* data);
L
Liu Jicong 已提交
309

H
Hongze Cheng 已提交
310 311 312 313 314 315
struct SSnapDataHdr {
  int8_t  type;
  int64_t size;
  uint8_t data[];
};

H
save  
Hongze Cheng 已提交
316 317 318 319
#ifdef __cplusplus
}
#endif

L
Liu Jicong 已提交
320
#endif /*_TD_VNODE_DEF_H_*/