vnodeInt.h 17.9 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
#include "filter.h"
#include "qworker.h"
22
#include "rocksdb/c.h"
H
Hongze Cheng 已提交
23
#include "sync.h"
H
Hongze Cheng 已提交
24
#include "tRealloc.h"
H
Hongze Cheng 已提交
25
#include "tchecksum.h"
H
more  
Hongze Cheng 已提交
26
#include "tcoding.h"
H
Hongze Cheng 已提交
27
#include "tcompare.h"
H
Hongze Cheng 已提交
28
#include "tcompression.h"
L
Liu Jicong 已提交
29
#include "tdatablock.h"
H
Hongze Cheng 已提交
30
#include "tdb.h"
H
Hongze Cheng 已提交
31
#include "tencode.h"
H
refact  
Hongze Cheng 已提交
32
#include "tfs.h"
H
Hongze Cheng 已提交
33
#include "tglobal.h"
H
Hongze Cheng 已提交
34
#include "tjson.h"
H
Hongze Cheng 已提交
35
#include "tlist.h"
H
refact  
Hongze Cheng 已提交
36
#include "tlockfree.h"
H
Hongze Cheng 已提交
37
#include "tlosertree.h"
H
Hongze Cheng 已提交
38
#include "tlrucache.h"
39
#include "tmsgcb.h"
H
Hongze Cheng 已提交
40
#include "trbtree.h"
H
Hongze Cheng 已提交
41
#include "tref.h"
H
Hongze Cheng 已提交
42
#include "tskiplist.h"
H
Hongze Cheng 已提交
43
#include "tstream.h"
H
Hongze Cheng 已提交
44
#include "ttime.h"
H
Hongze Cheng 已提交
45 46
#include "ttimer.h"
#include "wal.h"
H
save  
Hongze Cheng 已提交
47

H
Hongze Cheng 已提交
48 49
#include "vnode.h"

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

L
Liu Jicong 已提交
54 55 56 57 58 59
typedef struct SVnodeInfo         SVnodeInfo;
typedef struct SMeta              SMeta;
typedef struct SSma               SSma;
typedef struct STsdb              STsdb;
typedef struct STQ                STQ;
typedef struct SVState            SVState;
C
Cary Xu 已提交
60
typedef struct SVStatis           SVStatis;
L
Liu Jicong 已提交
61
typedef struct SVBufPool          SVBufPool;
62
typedef struct SQueueWorker       SQHandle;
L
Liu Jicong 已提交
63 64 65 66 67 68 69 70 71 72 73 74 75
typedef struct STsdbKeepCfg       STsdbKeepCfg;
typedef struct SMetaSnapReader    SMetaSnapReader;
typedef struct SMetaSnapWriter    SMetaSnapWriter;
typedef struct STsdbSnapReader    STsdbSnapReader;
typedef struct STsdbSnapWriter    STsdbSnapWriter;
typedef struct STqSnapReader      STqSnapReader;
typedef struct STqSnapWriter      STqSnapWriter;
typedef struct STqOffsetReader    STqOffsetReader;
typedef struct STqOffsetWriter    STqOffsetWriter;
typedef struct SStreamTaskReader  SStreamTaskReader;
typedef struct SStreamTaskWriter  SStreamTaskWriter;
typedef struct SStreamStateReader SStreamStateReader;
typedef struct SStreamStateWriter SStreamStateWriter;
76 77
typedef struct SRSmaSnapReader    SRSmaSnapReader;
typedef struct SRSmaSnapWriter    SRSmaSnapWriter;
L
Liu Jicong 已提交
78
typedef struct SSnapDataHdr       SSnapDataHdr;
H
Hongze Cheng 已提交
79
typedef struct SCommitInfo        SCommitInfo;
H
Hongze Cheng 已提交
80
typedef struct SCompactInfo       SCompactInfo;
H
Hongze Cheng 已提交
81
typedef struct SQueryNode         SQueryNode;
H
refact  
Hongze Cheng 已提交
82

C
Cary Xu 已提交
83 84 85 86 87
#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 已提交
88
#define VNODE_RSMA_DIR  "rsma"
C
Cary Xu 已提交
89
#define VNODE_RSMA0_DIR "tsdb"
C
Cary Xu 已提交
90 91
#define VNODE_RSMA1_DIR "rsma1"
#define VNODE_RSMA2_DIR "rsma2"
H
Hongze Cheng 已提交
92

H
Hongze Cheng 已提交
93 94
#define VNODE_BUFPOOL_SEGMENTS 3

H
Hongze Cheng 已提交
95
#define VND_INFO_FNAME "vnode.json"
H
Hongze Cheng 已提交
96

H
Hongze Cheng 已提交
97
// vnd.h
H
Hongze Cheng 已提交
98 99 100 101 102 103 104
typedef int32_t (*_query_reseek_func_t)(void* pQHandle);
struct SQueryNode {
  SQueryNode*          pNext;
  SQueryNode**         ppNext;
  void*                pQHandle;
  _query_reseek_func_t reseek;
};
105

H
Hongze Cheng 已提交
106
void* vnodeBufPoolMalloc(SVBufPool* pPool, int size);
107
void* vnodeBufPoolMallocAligned(SVBufPool* pPool, int size);
H
Hongze Cheng 已提交
108
void  vnodeBufPoolFree(SVBufPool* pPool, void* p);
H
Hongze Cheng 已提交
109
void  vnodeBufPoolRef(SVBufPool* pPool);
H
Hongze Cheng 已提交
110
void  vnodeBufPoolUnRef(SVBufPool* pPool, bool proactive);
H
Hongze Cheng 已提交
111
int   vnodeDecodeInfo(uint8_t* pData, SVnodeInfo* pInfo);
H
Hongze Cheng 已提交
112

H
Hongze Cheng 已提交
113
int32_t vnodeBufPoolRegisterQuery(SVBufPool* pPool, SQueryNode* pQNode);
H
Hongze Cheng 已提交
114
void    vnodeBufPoolDeregisterQuery(SVBufPool* pPool, SQueryNode* pQNode, bool proactive);
H
Hongze Cheng 已提交
115

H
Hongze Cheng 已提交
116 117
// meta
typedef struct SMCtbCursor SMCtbCursor;
C
Cary Xu 已提交
118
typedef struct SMStbCursor SMStbCursor;
H
Hongze Cheng 已提交
119 120
typedef struct STbUidStore STbUidStore;

121 122 123 124
#define META_BEGIN_HEAP_BUFFERPOOL 0
#define META_BEGIN_HEAP_OS         1
#define META_BEGIN_HEAP_NIL        2

H
Hongze Cheng 已提交
125
int             metaOpen(SVnode* pVnode, SMeta** ppMeta, int8_t rollback);
126
int             metaClose(SMeta** pMeta);
127
int             metaBegin(SMeta* pMeta, int8_t fromSys);
128 129 130
TXN*            metaGetTxn(SMeta* pMeta);
int             metaCommit(SMeta* pMeta, TXN* txn);
int             metaFinishCommit(SMeta* pMeta, TXN* txn);
131
int             metaPrepareAsyncCommit(SMeta* pMeta);
132
int             metaAbort(SMeta* pMeta);
H
Hongze Cheng 已提交
133
int             metaCreateSTable(SMeta* pMeta, int64_t version, SVCreateStbReq* pReq);
H
Hongze Cheng 已提交
134
int             metaAlterSTable(SMeta* pMeta, int64_t version, SVCreateStbReq* pReq);
135
int             metaDropSTable(SMeta* pMeta, int64_t verison, SVDropStbReq* pReq, SArray* tbUidList);
L
Liu Jicong 已提交
136
int             metaCreateTable(SMeta* pMeta, int64_t version, SVCreateTbReq* pReq, STableMetaRsp** pMetaRsp);
H
Hongze Cheng 已提交
137
int             metaDropTable(SMeta* pMeta, int64_t version, SVDropTbReq* pReq, SArray* tbUids, int64_t* tbUid);
H
Hongze Cheng 已提交
138
int             metaTtlDropTable(SMeta* pMeta, int64_t ttl, SArray* tbUids);
H
more  
Hongze Cheng 已提交
139
int             metaAlterTable(SMeta* pMeta, int64_t version, SVAlterTbReq* pReq, STableMetaRsp* pMetaRsp);
140 141
SSchemaWrapper* metaGetTableSchema(SMeta* pMeta, tb_uid_t uid, int32_t sver, int lock);
STSchema*       metaGetTbTSchema(SMeta* pMeta, tb_uid_t uid, int32_t sver, int lock);
H
Hongze Cheng 已提交
142
int32_t         metaGetTbTSchemaEx(SMeta* pMeta, tb_uid_t suid, tb_uid_t uid, int32_t sver, STSchema** ppTSchema);
H
Hongze Cheng 已提交
143
int             metaGetTableEntryByName(SMetaReader* pReader, const char* name);
H
Hongze Cheng 已提交
144
int             metaAlterCache(SMeta* pMeta, int32_t nPage);
dengyihao's avatar
dengyihao 已提交
145

dengyihao's avatar
dengyihao 已提交
146
int metaAddIndexToSTable(SMeta* pMeta, int64_t version, SVCreateStbReq* pReq);
dengyihao's avatar
dengyihao 已提交
147
int metaDropIndexFromSTable(SMeta* pMeta, int64_t version, SDropIndexReq* pReq);
dengyihao's avatar
dengyihao 已提交
148

dengyihao's avatar
dengyihao 已提交
149
int64_t       metaGetTimeSeriesNum(SMeta* pMeta);
150 151
SMCtbCursor*  metaOpenCtbCursor(SMeta* pMeta, tb_uid_t uid, int lock);
void          metaCloseCtbCursor(SMCtbCursor* pCtbCur, int lock);
dengyihao's avatar
dengyihao 已提交
152 153 154 155 156 157 158 159 160 161 162
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);
C
Cary Xu 已提交
163

164 165
int32_t metaCreateTSma(SMeta* pMeta, int64_t version, SSmaCfg* pCfg);
int32_t metaDropTSma(SMeta* pMeta, int64_t indexUid);
H
Hongze Cheng 已提交
166

H
Hongze Cheng 已提交
167 168 169 170 171 172
typedef struct SMetaInfo {
  int64_t uid;
  int64_t suid;
  int64_t version;
  int32_t skmVer;
} SMetaInfo;
173
int32_t metaGetInfo(SMeta* pMeta, int64_t uid, SMetaInfo* pInfo, SMetaReader* pReader);
H
Hongze Cheng 已提交
174

H
Hongze Cheng 已提交
175
// tsdb
176 177 178
int     tsdbOpen(SVnode* pVnode, STsdb** ppTsdb, const char* dir, STsdbKeepCfg* pKeepCfg, int8_t rollback);
int     tsdbClose(STsdb** pTsdb);
int32_t tsdbBegin(STsdb* pTsdb);
H
Hongze Cheng 已提交
179
int32_t tsdbPrepareCommit(STsdb* pTsdb);
H
Hongze Cheng 已提交
180
int32_t tsdbCommit(STsdb* pTsdb, SCommitInfo* pInfo);
181
int32_t tsdbCacheCommit(STsdb* pTsdb);
H
Hongze Cheng 已提交
182
int32_t tsdbCompact(STsdb* pTsdb, SCompactInfo* pInfo);
183 184
int32_t tsdbFinishCommit(STsdb* pTsdb);
int32_t tsdbRollbackCommit(STsdb* pTsdb);
185 186
int     tsdbScanAndConvertSubmitMsg(STsdb* pTsdb, SSubmitReq2* pMsg);
int     tsdbInsertData(STsdb* pTsdb, int64_t version, SSubmitReq2* pMsg, SSubmitRsp2* pRsp);
H
Hongze Cheng 已提交
187
int32_t tsdbInsertTableData(STsdb* pTsdb, int64_t version, SSubmitTbData* pSubmitTbData, int32_t* affectedRows);
188 189
int32_t tsdbDeleteTableData(STsdb* pTsdb, int64_t version, tb_uid_t suid, tb_uid_t uid, TSKEY sKey, TSKEY eKey);
int32_t tsdbSetKeepCfg(STsdb* pTsdb, STsdbCfg* pCfg);
H
Hongze Cheng 已提交
190 191

// tq
X
Xiaoyu Wang 已提交
192 193 194
int  tqInit();
void tqCleanUp();
STQ* tqOpen(const char* path, SVnode* pVnode);
H
Haojun Liao 已提交
195
void tqNotifyClose(STQ*);
X
Xiaoyu Wang 已提交
196 197
void tqClose(STQ*);
int  tqPushMsg(STQ*, void* msg, int32_t msgLen, tmsg_t msgType, int64_t ver);
198
int  tqRegisterPushHandle(STQ* pTq, void* handle, SRpcMsg* pMsg);
199
int  tqUnregisterPushHandle(STQ* pTq, void* pHandle);
H
Haojun Liao 已提交
200
int  tqStartStreamTasks(STQ* pTq);    // restore all stream tasks after vnode launching completed.
201

H
Hongze Cheng 已提交
202
int     tqCommit(STQ*);
203
int32_t tqUpdateTbUidList(STQ* pTq, const SArray* tbUidList, bool isAdd);
L
Liu Jicong 已提交
204
int32_t tqCheckColModifiable(STQ* pTq, int64_t tbUid, int32_t colId);
205 206 207
// tq-mq
int32_t tqProcessAddCheckInfoReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
int32_t tqProcessDelCheckInfoReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
208 209
int32_t tqProcessSubscribeReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
int32_t tqProcessDeleteSubReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
210
int32_t tqProcessOffsetCommitReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
211
int32_t tqProcessSeekReq(STQ* pTq, int64_t sversion, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
212
int32_t tqProcessPollReq(STQ* pTq, SRpcMsg* pMsg);
213 214
int32_t tqProcessVgWalInfoReq(STQ* pTq, SRpcMsg* pMsg);

215 216 217
// tq-stream
int32_t tqProcessTaskDeployReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
int32_t tqProcessTaskDropReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
5
54liuyao 已提交
218 219
int32_t tqProcessTaskPauseReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
int32_t tqProcessTaskResumeReq(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
220 221
int32_t tqProcessStreamTaskCheckReq(STQ* pTq, SRpcMsg* pMsg);
int32_t tqProcessStreamTaskCheckRsp(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
L
Liu Jicong 已提交
222
int32_t tqProcessTaskRunReq(STQ* pTq, SRpcMsg* pMsg);
L
Liu Jicong 已提交
223
int32_t tqProcessTaskDispatchReq(STQ* pTq, SRpcMsg* pMsg, bool exec);
L
Liu Jicong 已提交
224
int32_t tqProcessTaskDispatchRsp(STQ* pTq, SRpcMsg* pMsg);
L
Liu Jicong 已提交
225 226
int32_t tqProcessTaskRetrieveReq(STQ* pTq, SRpcMsg* pMsg);
int32_t tqProcessTaskRetrieveRsp(STQ* pTq, SRpcMsg* pMsg);
L
Liu Jicong 已提交
227 228 229 230
int32_t tqProcessTaskRecover1Req(STQ* pTq, SRpcMsg* pMsg);
int32_t tqProcessTaskRecover2Req(STQ* pTq, int64_t version, char* msg, int32_t msgLen);
int32_t tqProcessTaskRecoverFinishReq(STQ* pTq, SRpcMsg* pMsg);
int32_t tqProcessTaskRecoverFinishRsp(STQ* pTq, SRpcMsg* pMsg);
L
Liu Jicong 已提交
231
int32_t tqCheckLogInWal(STQ* pTq, int64_t version);
H
Hongze Cheng 已提交
232

C
Cary Xu 已提交
233
// sma
C
Cary Xu 已提交
234 235
int32_t smaInit();
void    smaCleanUp();
H
Hongze Cheng 已提交
236
int32_t smaOpen(SVnode* pVnode, int8_t rollback);
237
int32_t smaClose(SSma* pSma);
C
Cary Xu 已提交
238
int32_t smaBegin(SSma* pSma);
K
kailixu 已提交
239
int32_t smaPrepareAsyncCommit(SSma* pSma);
H
Hongze Cheng 已提交
240
int32_t smaCommit(SSma* pSma, SCommitInfo* pInfo);
C
Cary Xu 已提交
241
int32_t smaFinishCommit(SSma* pSma);
C
Cary Xu 已提交
242
int32_t smaPostCommit(SSma* pSma);
C
Cary Xu 已提交
243
int32_t smaDoRetention(SSma* pSma, int64_t now);
C
Cary Xu 已提交
244

C
Cary Xu 已提交
245
int32_t tdProcessTSmaCreate(SSma* pSma, int64_t version, const char* msg);
246 247
int32_t tdProcessTSmaInsert(SSma* pSma, int64_t indexUid, const char* msg);

248
int32_t tdProcessRSmaCreate(SSma* pSma, SVCreateStbReq* pReq);
K
kailixu 已提交
249
int32_t tdProcessRSmaSubmit(SSma* pSma, int64_t version, void* pReq, void* pMsg, int32_t len, int32_t inputType);
250
int32_t tdProcessRSmaDrop(SSma* pSma, SVDropStbReq* pReq);
251
int32_t tdFetchTbUidList(SSma* pSma, STbUidStore** ppStore, tb_uid_t suid, tb_uid_t uid);
252
int32_t tdUpdateTbUidList(SSma* pSma, STbUidStore* pUidStore, bool isAdd);
253 254
void*   tdUidStoreFree(STbUidStore* pStore);

H
Hongze Cheng 已提交
255 256 257
// SMetaSnapReader ========================================
int32_t metaSnapReaderOpen(SMeta* pMeta, int64_t sver, int64_t ever, SMetaSnapReader** ppReader);
int32_t metaSnapReaderClose(SMetaSnapReader** ppReader);
H
Hongze Cheng 已提交
258
int32_t metaSnapRead(SMetaSnapReader* pReader, uint8_t** ppData);
H
Hongze Cheng 已提交
259
// SMetaSnapWriter ========================================
H
Hongze Cheng 已提交
260 261 262
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 已提交
263
// STsdbSnapReader ========================================
C
Cary Xu 已提交
264
int32_t tsdbSnapReaderOpen(STsdb* pTsdb, int64_t sver, int64_t ever, int8_t type, STsdbSnapReader** ppReader);
H
Hongze Cheng 已提交
265
int32_t tsdbSnapReaderClose(STsdbSnapReader** ppReader);
H
Hongze Cheng 已提交
266
int32_t tsdbSnapRead(STsdbSnapReader* pReader, uint8_t** ppData);
H
Hongze Cheng 已提交
267
// STsdbSnapWriter ========================================
H
Hongze Cheng 已提交
268
int32_t tsdbSnapWriterOpen(STsdb* pTsdb, int64_t sver, int64_t ever, STsdbSnapWriter** ppWriter);
H
Hongze Cheng 已提交
269
int32_t tsdbSnapWrite(STsdbSnapWriter* pWriter, SSnapDataHdr* pHdr);
H
Hongze Cheng 已提交
270
int32_t tsdbSnapWriterPrepareClose(STsdbSnapWriter* pWriter);
H
Hongze Cheng 已提交
271
int32_t tsdbSnapWriterClose(STsdbSnapWriter** ppWriter, int8_t rollback);
L
Liu Jicong 已提交
272 273 274 275 276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291
// STqSnapshotReader ==
int32_t tqSnapReaderOpen(STQ* pTq, int64_t sver, int64_t ever, STqSnapReader** ppReader);
int32_t tqSnapReaderClose(STqSnapReader** ppReader);
int32_t tqSnapRead(STqSnapReader* pReader, uint8_t** ppData);
// STqSnapshotWriter ======================================
int32_t tqSnapWriterOpen(STQ* pTq, int64_t sver, int64_t ever, STqSnapWriter** ppWriter);
int32_t tqSnapWriterClose(STqSnapWriter** ppWriter, int8_t rollback);
int32_t tqSnapWrite(STqSnapWriter* pWriter, uint8_t* pData, uint32_t nData);
// STqOffsetReader ========================================
int32_t tqOffsetReaderOpen(STQ* pTq, int64_t sver, int64_t ever, STqOffsetReader** ppReader);
int32_t tqOffsetReaderClose(STqOffsetReader** ppReader);
int32_t tqOffsetSnapRead(STqOffsetReader* pReader, uint8_t** ppData);
// STqOffsetWriter ========================================
int32_t tqOffsetWriterOpen(STQ* pTq, int64_t sver, int64_t ever, STqOffsetWriter** ppWriter);
int32_t tqOffsetWriterClose(STqOffsetWriter** ppWriter, int8_t rollback);
int32_t tqOffsetSnapWrite(STqOffsetWriter* pWriter, uint8_t* pData, uint32_t nData);
// SStreamTaskWriter ======================================
// SStreamTaskReader ======================================
// SStreamStateWriter =====================================
// SStreamStateReader =====================================
292 293 294 295 296 297 298
// SRSmaSnapReader ========================================
int32_t rsmaSnapReaderOpen(SSma* pSma, int64_t sver, int64_t ever, SRSmaSnapReader** ppReader);
int32_t rsmaSnapReaderClose(SRSmaSnapReader** ppReader);
int32_t rsmaSnapRead(SRSmaSnapReader* pReader, uint8_t** ppData);
// SRSmaSnapWriter ========================================
int32_t rsmaSnapWriterOpen(SSma* pSma, int64_t sver, int64_t ever, SRSmaSnapWriter** ppWriter);
int32_t rsmaSnapWrite(SRSmaSnapWriter* pWriter, uint8_t* pData, uint32_t nData);
K
kailixu 已提交
299
int32_t rsmaSnapWriterPrepareClose(SRSmaSnapWriter* pWriter);
300
int32_t rsmaSnapWriterClose(SRSmaSnapWriter** ppWriter, int8_t rollback);
H
Hongze Cheng 已提交
301

L
Liu Jicong 已提交
302 303 304 305 306 307 308 309 310 311 312 313 314 315 316
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 已提交
317 318 319 320
// SVState
struct SVState {
  int64_t committed;
  int64_t applied;
H
Hongze Cheng 已提交
321
  int64_t applyTerm;
H
Hongze Cheng 已提交
322
  int64_t commitID;
H
Hongze Cheng 已提交
323
  int64_t commitTerm;
H
refact  
Hongze Cheng 已提交
324 325
};

C
Cary Xu 已提交
326
struct SVStatis {
C
Cary Xu 已提交
327 328 329 330
  int64_t nInsert;              // delta
  int64_t nInsertSuccess;       // delta
  int64_t nBatchInsert;         // delta
  int64_t nBatchInsertSuccess;  // delta
C
Cary Xu 已提交
331 332
};

H
Hongze Cheng 已提交
333 334 335
struct SVnodeInfo {
  SVnodeCfg config;
  SVState   state;
C
Cary Xu 已提交
336
  SVStatis  statis;
H
Hongze Cheng 已提交
337 338
};

C
Cary Xu 已提交
339 340 341 342 343 344 345 346
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 已提交
347
struct STsdbKeepCfg {
C
Cary Xu 已提交
348 349 350 351 352
  int8_t  precision;  // precision always be used with below keep cfgs
  int32_t days;
  int32_t keep0;
  int32_t keep1;
  int32_t keep2;
353
};
C
Cary Xu 已提交
354

355 356 357 358 359
typedef struct SVCommitSched {
  int64_t commitMs;
  int64_t maxWaitMs;
} SVCommitSched;

H
save  
Hongze Cheng 已提交
360
struct SVnode {
H
Hongze Cheng 已提交
361 362 363 364 365 366 367 368
  char*     path;
  SVnodeCfg config;
  SVState   state;
  SVStatis  statis;
  STfs*     pTfs;
  SMsgCb    msgCb;

  // Buffer Pool
H
Hongze Cheng 已提交
369 370
  TdThreadMutex mutex;
  TdThreadCond  poolNotEmpty;
H
Hongze Cheng 已提交
371 372
  SVBufPool*    aBufPool[VNODE_BUFPOOL_SEGMENTS];
  SVBufPool*    freeList;
H
Hongze Cheng 已提交
373
  SVBufPool*    inUse;
H
Hongze Cheng 已提交
374 375 376
  SVBufPool*    onCommit;
  SVBufPool*    recycleHead;
  SVBufPool*    recycleTail;
H
Hongze Cheng 已提交
377
  SVBufPool*    onRecycle;
H
Hongze Cheng 已提交
378

H
Hongze Cheng 已提交
379 380 381 382 383 384 385 386
  SMeta*        pMeta;
  SSma*         pSma;
  STsdb*        pTsdb;
  SWal*         pWal;
  STQ*          pTq;
  SSink*        pSink;
  tsem_t        canCommit;
  int64_t       sync;
387
  TdThreadMutex lock;
388
  bool          blocked;
389
  bool          restored;
H
Hongze Cheng 已提交
390
  tsem_t        syncSem;
391 392
  int32_t       blockSec;
  int64_t       blockSeq;
H
Hongze Cheng 已提交
393
  SQHandle*     pQuery;
H
save  
Hongze Cheng 已提交
394 395
};

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

C
Cary Xu 已提交
398 399
#define VND_TSDB(vnd)       ((vnd)->pTsdb)
#define VND_RSMA0(vnd)      ((vnd)->pTsdb)
C
Cary Xu 已提交
400 401
#define VND_RSMA1(vnd)      ((vnd)->pSma->pRSmaTsdb[TSDB_RETENTION_L0])
#define VND_RSMA2(vnd)      ((vnd)->pSma->pRSmaTsdb[TSDB_RETENTION_L1])
C
Cary Xu 已提交
402
#define VND_RETENTIONS(vnd) (&(vnd)->config.tsdbCfg.retentions)
C
Cary Xu 已提交
403 404
#define VND_IS_RSMA(v)      ((v)->config.isRsma == 1)
#define VND_IS_TSMA(v)      ((v)->config.isTsma == 1)
C
Cary Xu 已提交
405

H
Hongze Cheng 已提交
406 407 408 409 410 411
struct STbUidStore {
  tb_uid_t  suid;
  SArray*   tbUids;
  SHashObj* uidHash;
};

412 413 414 415
struct SSma {
  bool          locked;
  TdThreadMutex mutex;
  SVnode*       pVnode;
C
Cary Xu 已提交
416
  STsdb*        pRSmaTsdb[TSDB_RETENTION_L2];
417 418 419 420 421 422
  void*         pTSmaEnv;
  void*         pRSmaEnv;
};

#define SMA_CFG(s)        (&(s)->pVnode->config)
#define SMA_TSDB_CFG(s)   (&(s)->pVnode->config.tsdbCfg)
H
Hongze Cheng 已提交
423
#define SMA_RETENTION(s)  ((SRetention*)&(s)->pVnode->config.tsdbCfg.retentions)
424 425 426 427 428 429 430
#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)
C
Cary Xu 已提交
431 432
#define SMA_RSMA_TSDB1(s) ((s)->pRSmaTsdb[TSDB_RETENTION_L0])
#define SMA_RSMA_TSDB2(s) ((s)->pRSmaTsdb[TSDB_RETENTION_L1])
H
refact  
Hongze Cheng 已提交
433

L
Liu Jicong 已提交
434
// sma
L
Liu Jicong 已提交
435
void smaHandleRes(void* pVnode, int64_t smaId, const SArray* data);
L
Liu Jicong 已提交
436

C
Cary Xu 已提交
437
enum {
H
Hongze Cheng 已提交
438
  SNAP_DATA_CFG = 0,
C
Cary Xu 已提交
439 440 441 442 443 444 445 446 447 448
  SNAP_DATA_META = 1,
  SNAP_DATA_TSDB = 2,
  SNAP_DATA_DEL = 3,
  SNAP_DATA_RSMA1 = 4,
  SNAP_DATA_RSMA2 = 5,
  SNAP_DATA_QTASK = 6,
  SNAP_DATA_TQ_HANDLE = 7,
  SNAP_DATA_TQ_OFFSET = 8,
  SNAP_DATA_STREAM_TASK = 9,
  SNAP_DATA_STREAM_STATE = 10,
C
Cary Xu 已提交
449 450
};

H
Hongze Cheng 已提交
451 452
struct SSnapDataHdr {
  int8_t  type;
K
kailixu 已提交
453
  int8_t  flag;
H
Hongze Cheng 已提交
454
  int64_t index;
H
Hongze Cheng 已提交
455 456 457 458
  int64_t size;
  uint8_t data[];
};

H
Hongze Cheng 已提交
459 460 461
struct SCommitInfo {
  SVnodeInfo info;
  SVnode*    pVnode;
462
  TXN*       txn;
H
Hongze Cheng 已提交
463
};
H
Hongze Cheng 已提交
464

H
Hongze Cheng 已提交
465
struct SCompactInfo {
H
Hongze Cheng 已提交
466 467 468 469
  SVnode*     pVnode;
  int32_t     flag;
  int64_t     commitID;
  STimeWindow tw;
H
Hongze Cheng 已提交
470
};
H
Hongze Cheng 已提交
471

H
save  
Hongze Cheng 已提交
472 473 474 475
#ifdef __cplusplus
}
#endif

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