tsdb.h 19.0 KB
Newer Older
L
Liu Jicong 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/*
 * 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
Hongze Cheng 已提交
15 16 17 18

#ifndef _TD_VNODE_TSDB_H_
#define _TD_VNODE_TSDB_H_

H
Hongze Cheng 已提交
19 20
#include "vnodeInt.h"

H
Hongze Cheng 已提交
21 22 23 24
#ifdef __cplusplus
extern "C" {
#endif

H
Hongze Cheng 已提交
25 26 27 28 29 30 31 32 33
// tsdbDebug ================
// clang-format off
#define tsdbFatal(...) do { if (tsdbDebugFlag & DEBUG_FATAL) { taosPrintLog("TSDB FATAL ", DEBUG_FATAL, 255, __VA_ARGS__); }}     while(0)
#define tsdbError(...) do { if (tsdbDebugFlag & DEBUG_ERROR) { taosPrintLog("TSDB ERROR ", DEBUG_ERROR, 255, __VA_ARGS__); }}     while(0)
#define tsdbWarn(...)  do { if (tsdbDebugFlag & DEBUG_WARN)  { taosPrintLog("TSDB WARN ", DEBUG_WARN, 255, __VA_ARGS__); }}       while(0)
#define tsdbInfo(...)  do { if (tsdbDebugFlag & DEBUG_INFO)  { taosPrintLog("TSDB ", DEBUG_INFO, 255, __VA_ARGS__); }}            while(0)
#define tsdbDebug(...) do { if (tsdbDebugFlag & DEBUG_DEBUG) { taosPrintLog("TSDB ", DEBUG_DEBUG, tsdbDebugFlag, __VA_ARGS__); }} while(0)
#define tsdbTrace(...) do { if (tsdbDebugFlag & DEBUG_TRACE) { taosPrintLog("TSDB ", DEBUG_TRACE, tsdbDebugFlag, __VA_ARGS__); }} while(0)
// clang-format on
H
Hongze Cheng 已提交
34

H
Hongze Cheng 已提交
35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53
typedef struct TSDBROW        TSDBROW;
typedef struct TABLEID        TABLEID;
typedef struct TSDBKEY        TSDBKEY;
typedef struct KEYINFO        KEYINFO;
typedef struct SDelData       SDelData;
typedef struct SDelIdx        SDelIdx;
typedef struct STbData        STbData;
typedef struct SMemTable      SMemTable;
typedef struct STbDataIter    STbDataIter;
typedef struct STable         STable;
typedef struct SMapData       SMapData;
typedef struct SBlockSMA      SBlockSMA;
typedef struct SBlockIdx      SBlockIdx;
typedef struct SBlock         SBlock;
typedef struct SBlockStatis   SBlockStatis;
typedef struct SAggrBlkCol    SAggrBlkCol;
typedef struct SColData       SColData;
typedef struct SBlockDataHdr  SBlockDataHdr;
typedef struct SBlockData     SBlockData;
H
Hongze Cheng 已提交
54 55
typedef struct SDelFile       SDelFile;
typedef struct STsdbCacheFile STsdbCacheFile;
H
Hongze Cheng 已提交
56 57 58 59
typedef struct SHeadFile      SHeadFile;
typedef struct SDataFile      SDataFile;
typedef struct SLastFile      SLastFile;
typedef struct SSmaFile       SSmaFile;
H
Hongze Cheng 已提交
60
typedef struct SDFileSet      SDFileSet;
H
Hongze Cheng 已提交
61 62 63 64
typedef struct SDataFWriter   SDataFWriter;
typedef struct SDataFReader   SDataFReader;
typedef struct SDelFWriter    SDelFWriter;
typedef struct SDelFReader    SDelFReader;
H
Hongze Cheng 已提交
65 66
typedef struct SRowIter       SRowIter;
typedef struct STsdbFS        STsdbFS;
H
Hongze Cheng 已提交
67
typedef struct SRowMerger     SRowMerger;
H
Hongze Cheng 已提交
68
typedef struct STsdbFSState   STsdbFSState;
H
Hongze Cheng 已提交
69

H
Hongze Cheng 已提交
70
#define TSDB_MAX_SUBBLOCKS 8
H
Hongze Cheng 已提交
71
#define TSDB_FHDR_SIZE     512
H
refact  
Hongze Cheng 已提交
72

H
Hongze Cheng 已提交
73 74 75
#define HAS_NONE  ((int8_t)0x1)
#define HAS_NULL  ((int8_t)0x2)
#define HAS_VALUE ((int8_t)0x4)
H
Hongze Cheng 已提交
76

H
Hongze Cheng 已提交
77 78 79
#define VERSION_MIN 0
#define VERSION_MAX INT64_MAX

H
Hongze Cheng 已提交
80 81 82
#define TSDBKEY_MIN ((TSDBKEY){.ts = TSKEY_MIN, .version = VERSION_MIN})
#define TSDBKEY_MAX ((TSDBKEY){.ts = TSKEY_MAX, .version = VERSION_MAX})

H
Hongze Cheng 已提交
83
// tsdbUtil.c ==============================================================================================
H
Hongze Cheng 已提交
84
// TSDBROW
H
Hongze Cheng 已提交
85 86
#define TSDBROW_TS(ROW)                       (((ROW)->type == 0) ? (ROW)->pTSRow->ts : (ROW)->pBlockData->aTSKEY[(ROW)->iRow])
#define TSDBROW_VERSION(ROW)                  (((ROW)->type == 0) ? (ROW)->version : (ROW)->pBlockData->aVersion[(ROW)->iRow])
H
Hongze Cheng 已提交
87
#define TSDBROW_SVERSION(ROW)                 TD_ROW_SVER((ROW)->pTSRow)
H
Hongze Cheng 已提交
88
#define TSDBROW_KEY(ROW)                      ((TSDBKEY){.version = TSDBROW_VERSION(ROW), .ts = TSDBROW_TS(ROW)})
H
Hongze Cheng 已提交
89 90
#define tsdbRowFromTSRow(VERSION, TSROW)      ((TSDBROW){.type = 0, .version = (VERSION), .pTSRow = (TSROW)})
#define tsdbRowFromBlockData(BLOCKDATA, IROW) ((TSDBROW){.type = 1, .pBlockData = (BLOCKDATA), .iRow = (IROW)})
H
Hongze Cheng 已提交
91 92 93
void    tsdbRowGetColVal(TSDBROW *pRow, STSchema *pTSchema, int32_t iCol, SColVal *pColVal);
int32_t tPutTSDBRow(uint8_t *p, TSDBROW *pRow);
int32_t tGetTSDBRow(uint8_t *p, TSDBROW *pRow);
H
Hongze Cheng 已提交
94
int32_t tsdbRowCmprFn(const void *p1, const void *p2);
H
Hongze Cheng 已提交
95
// SRowIter
H
Hongze Cheng 已提交
96
void     tRowIterInit(SRowIter *pIter, TSDBROW *pRow, STSchema *pTSchema);
H
Hongze Cheng 已提交
97
SColVal *tRowIterNext(SRowIter *pIter);
H
Hongze Cheng 已提交
98 99 100 101 102
// SRowMerger
int32_t tRowMergerInit(SRowMerger *pMerger, TSDBROW *pRow, STSchema *pTSchema);
void    tRowMergerClear(SRowMerger *pMerger);
int32_t tRowMerge(SRowMerger *pMerger, TSDBROW *pRow);
int32_t tRowMergerGetRow(SRowMerger *pMerger, STSRow **ppRow);
H
Hongze Cheng 已提交
103
// TABLEID
H
Hongze Cheng 已提交
104
int32_t tTABLEIDCmprFn(const void *p1, const void *p2);
H
Hongze Cheng 已提交
105
// TSDBKEY
H
Hongze Cheng 已提交
106
int32_t tsdbKeyCmprFn(const void *p1, const void *p2);
H
Hongze Cheng 已提交
107 108
#define MIN_TSDBKEY(KEY1, KEY2) ((tsdbKeyCmprFn(&(KEY1), &(KEY2)) < 0) ? (KEY1) : (KEY2))
#define MAX_TSDBKEY(KEY1, KEY2) ((tsdbKeyCmprFn(&(KEY1), &(KEY2)) > 0) ? (KEY1) : (KEY2))
H
Hongze Cheng 已提交
109
// KEYINFO
H
Hongze Cheng 已提交
110 111 112 113 114
#define tKEYINFOInit()                                          \
  ((KEYINFO){.maxKey = {.ts = TSKEY_MIN, .version = -1},        \
             .minKey = {.ts = TSKEY_MAX, .version = INT64_MAX}, \
             .minVerion = INT64_MAX,                            \
             .maxVersion = -1})
H
Hongze Cheng 已提交
115 116
int32_t tPutKEYINFO(uint8_t *p, KEYINFO *pKeyInfo);
int32_t tGetKEYINFO(uint8_t *p, KEYINFO *pKeyInfo);
H
Hongze Cheng 已提交
117 118 119
// SBlockCol
int32_t tPutBlockCol(uint8_t *p, void *ph);
int32_t tGetBlockCol(uint8_t *p, void *ph);
H
Hongze Cheng 已提交
120
int32_t tBlockColCmprFn(const void *p1, const void *p2);
H
Hongze Cheng 已提交
121
// SBlock
H
Hongze Cheng 已提交
122
#define tBlockInit() ((SBlock){0})
H
Hongze Cheng 已提交
123 124
void    tBlockReset(SBlock *pBlock);
void    tBlockClear(SBlock *pBlock);
H
Hongze Cheng 已提交
125
int32_t tBlockCopy(SBlock *pBlockSrc, SBlock *pBlockDest);
H
Hongze Cheng 已提交
126 127
int32_t tPutBlock(uint8_t *p, void *ph);
int32_t tGetBlock(uint8_t *p, void *ph);
H
Hongze Cheng 已提交
128
int32_t tBlockCmprFn(const void *p1, const void *p2);
H
Hongze Cheng 已提交
129
// SBlockIdx
H
Hongze Cheng 已提交
130 131
// #define tBlockIdxInit(SUID, UID) ((SBlockIdx){.suid = (SUID), .uid = (UID), .info = tKEYINFOInit()})
void    tBlockIdxReset(SBlockIdx *pBlockIdx);
H
Hongze Cheng 已提交
132 133
int32_t tPutBlockIdx(uint8_t *p, void *ph);
int32_t tGetBlockIdx(uint8_t *p, void *ph);
134
int32_t tCmprBlockIdx(void const *lhs, void const *rhs);
H
Hongze Cheng 已提交
135 136
// SColdata
#define tColDataInit() ((SColData){0})
H
Hongze Cheng 已提交
137
void    tColDataReset(SColData *pColData, int16_t cid, int8_t type);
H
Hongze Cheng 已提交
138
void    tColDataClear(void *ph);
H
Hongze Cheng 已提交
139
int32_t tColDataAppendValue(SColData *pColData, SColVal *pColVal);
H
Hongze Cheng 已提交
140
int32_t tColDataCopy(SColData *pColDataSrc, SColData *pColDataDest);
H
Hongze Cheng 已提交
141
int32_t tColDataGetValue(SColData *pColData, int32_t iRow, SColVal *pColVal);
H
Hongze Cheng 已提交
142
// SBlockData
H
Hongze Cheng 已提交
143 144
#define tBlockDataFirstRow(PBLOCKDATA) tsdbRowFromBlockData(PBLOCKDATA, 0)
#define tBlockDataLastRow(PBLOCKDATA)  tsdbRowFromBlockData(PBLOCKDATA, (PBLOCKDATA)->nRow - 1)
H
Hongze Cheng 已提交
145 146
#define tBlockDataFirstKey(PBLOCKDATA) TSDBROW_KEY(&tBlockDataFirstRow(PBLOCKDATA))
#define tBlockDataLastKey(PBLOCKDATA)  TSDBROW_KEY(&tBlockDataLastRow(PBLOCKDATA))
H
Hongze Cheng 已提交
147
int32_t tBlockDataInit(SBlockData *pBlockData);
H
Hongze Cheng 已提交
148 149
void    tBlockDataReset(SBlockData *pBlockData);
void    tBlockDataClear(SBlockData *pBlockData);
H
Hongze Cheng 已提交
150
int32_t tBlockDataAddColData(SBlockData *pBlockData, int32_t iColData, SColData **ppColData);
H
Hongze Cheng 已提交
151
int32_t tBlockDataAppendRow(SBlockData *pBlockData, TSDBROW *pRow, STSchema *pTSchema);
H
Hongze Cheng 已提交
152 153
int32_t tBlockDataMerge(SBlockData *pBlockData1, SBlockData *pBlockData2, SBlockData *pBlockData);
int32_t tBlockDataCopy(SBlockData *pBlockDataSrc, SBlockData *pBlockDataDest);
H
Hongze Cheng 已提交
154
void    tBlockDataGetColData(SBlockData *pBlockData, int16_t cid, SColData **ppColData);
H
Hongze Cheng 已提交
155
// SDelIdx
H
Hongze Cheng 已提交
156 157
int32_t tPutDelIdx(uint8_t *p, void *ph);
int32_t tGetDelIdx(uint8_t *p, void *ph);
158
int32_t tCmprDelIdx(void const *lhs, void const *rhs);
H
Hongze Cheng 已提交
159
// SDelData
H
Hongze Cheng 已提交
160 161
int32_t tPutDelData(uint8_t *p, void *ph);
int32_t tGetDelData(uint8_t *p, void *ph);
H
Hongze Cheng 已提交
162 163 164 165
// memory
int32_t tsdbRealloc(uint8_t **ppBuf, int64_t size);
void    tsdbFree(uint8_t *pBuf);
// SMapData
H
Hongze Cheng 已提交
166
#define tMapDataInit() ((SMapData){0})
H
Hongze Cheng 已提交
167 168
void    tMapDataReset(SMapData *pMapData);
void    tMapDataClear(SMapData *pMapData);
H
Hongze Cheng 已提交
169
int32_t tMapDataCopy(SMapData *pMapDataSrc, SMapData *pMapDataDest);
H
Hongze Cheng 已提交
170 171
int32_t tMapDataPutItem(SMapData *pMapData, void *pItem, int32_t (*tPutItemFn)(uint8_t *, void *));
int32_t tMapDataGetItemByIdx(SMapData *pMapData, int32_t idx, void *pItem, int32_t (*tGetItemFn)(uint8_t *, void *));
H
Hongze Cheng 已提交
172 173
int32_t tMapDataSearch(SMapData *pMapData, void *pSearchItem, int32_t (*tGetItemFn)(uint8_t *, void *),
                       int32_t (*tItemCmprFn)(const void *, const void *), void *pItem);
H
Hongze Cheng 已提交
174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191
int32_t tPutMapData(uint8_t *p, SMapData *pMapData);
int32_t tGetMapData(uint8_t *p, SMapData *pMapData);
// other
int32_t tsdbKeyFid(TSKEY key, int32_t minutes, int8_t precision);
void    tsdbFidKeyRange(int32_t fid, int32_t minutes, int8_t precision, TSKEY *minKey, TSKEY *maxKey);
int32_t tsdbBuildDeleteSkyline(SArray *aDelData, int32_t sidx, int32_t eidx, SArray *aSkyline);
// tsdbMemTable ==============================================================================================
// SMemTable
int32_t tsdbMemTableCreate(STsdb *pTsdb, SMemTable **ppMemTable);
void    tsdbMemTableDestroy(SMemTable *pMemTable);
void    tsdbGetTbDataFromMemTable(SMemTable *pMemTable, tb_uid_t suid, tb_uid_t uid, STbData **ppTbData);
// STbDataIter
int32_t  tsdbTbDataIterCreate(STbData *pTbData, TSDBKEY *pFrom, int8_t backward, STbDataIter **ppIter);
void    *tsdbTbDataIterDestroy(STbDataIter *pIter);
void     tsdbTbDataIterOpen(STbData *pTbData, TSDBKEY *pFrom, int8_t backward, STbDataIter *pIter);
TSDBROW *tsdbTbDataIterGet(STbDataIter *pIter);
bool     tsdbTbDataIterNext(STbDataIter *pIter);
// tsdbFile.c ==============================================================================================
H
Hongze Cheng 已提交
192
typedef enum { TSDB_HEAD_FILE = 0, TSDB_DATA_FILE, TSDB_LAST_FILE, TSDB_SMA_FILE } EDataFileT;
H
Hongze Cheng 已提交
193
void    tsdbDataFileName(STsdb *pTsdb, SDFileSet *pDFileSet, EDataFileT ftype, char fname[]);
H
Hongze Cheng 已提交
194 195 196
bool    tsdbFileIsSame(SDFileSet *pDFileSet1, SDFileSet *pDFileSet2, EDataFileT ftype);
int32_t tsdbUpdateDFileHdr(TdFilePtr pFD, SDFileSet *pSet, EDataFileT ftype);
int32_t tsdbDFileRollback(STsdb *pTsdb, SDFileSet *pSet, EDataFileT ftype);
H
Hongze Cheng 已提交
197
int32_t tPutDataFileHdr(uint8_t *p, SDFileSet *pSet, EDataFileT ftype);
H
Hongze Cheng 已提交
198 199 200 201
int32_t tPutDelFile(uint8_t *p, SDelFile *pDelFile);
int32_t tGetDelFile(uint8_t *p, SDelFile *pDelFile);
int32_t tPutDFileSet(uint8_t *p, SDFileSet *pSet);
int32_t tGetDFileSet(uint8_t *p, SDFileSet *pSet);
H
Hongze Cheng 已提交
202
// SDelFile
H
Hongze Cheng 已提交
203
void tsdbDelFileName(STsdb *pTsdb, SDelFile *pFile, char fname[]);
H
Hongze Cheng 已提交
204
// tsdbFS.c ==============================================================================================
H
Hongze Cheng 已提交
205 206
int32_t tsdbFSOpen(STsdb *pTsdb, STsdbFS **ppFS);
int32_t tsdbFSClose(STsdbFS *pFS);
H
Hongze Cheng 已提交
207 208 209
int32_t tsdbFSBegin(STsdbFS *pFS);
int32_t tsdbFSCommit(STsdbFS *pFS);
int32_t tsdbFSRollback(STsdbFS *pFS);
H
Hongze Cheng 已提交
210 211 212 213 214

int32_t    tsdbFSStateUpsertDelFile(STsdbFSState *pState, SDelFile *pDelFile);
int32_t    tsdbFSStateUpsertDFileSet(STsdbFSState *pState, SDFileSet *pSet);
SDelFile  *tsdbFSStateGetDelFile(STsdbFSState *pState);
SDFileSet *tsdbFSStateGetDFileSet(STsdbFSState *pState, int32_t fid);
H
Hongze Cheng 已提交
215 216 217
// tsdbReaderWriter.c ==============================================================================================
// SDataFWriter
int32_t tsdbDataFWriterOpen(SDataFWriter **ppWriter, STsdb *pTsdb, SDFileSet *pSet);
H
Hongze Cheng 已提交
218
int32_t tsdbDataFWriterClose(SDataFWriter **ppWriter, int8_t sync);
H
Hongze Cheng 已提交
219
int32_t tsdbUpdateDFileSetHeader(SDataFWriter *pWriter);
H
Hongze Cheng 已提交
220 221 222
int32_t tsdbWriteBlockIdx(SDataFWriter *pWriter, SMapData *pMapData, uint8_t **ppBuf);
int32_t tsdbWriteBlock(SDataFWriter *pWriter, SMapData *pMapData, uint8_t **ppBuf, SBlockIdx *pBlockIdx);
int32_t tsdbWriteBlockData(SDataFWriter *pWriter, SBlockData *pBlockData, uint8_t **ppBuf1, uint8_t **ppBuf2,
H
Hongze Cheng 已提交
223
                           SBlockIdx *pBlockIdx, SBlock *pBlock, int8_t cmprAlg);
H
Hongze Cheng 已提交
224
int32_t tsdbWriteBlockSMA(SDataFWriter *pWriter, SBlockSMA *pBlockSMA, int64_t *rOffset, int64_t *rSize);
H
Hongze Cheng 已提交
225 226

SDFileSet *tsdbDataFWriterGetWSet(SDataFWriter *pWriter);
H
Hongze Cheng 已提交
227 228
// SDataFReader
int32_t tsdbDataFReaderOpen(SDataFReader **ppReader, STsdb *pTsdb, SDFileSet *pSet);
H
Hongze Cheng 已提交
229
int32_t tsdbDataFReaderClose(SDataFReader **ppReader);
H
Hongze Cheng 已提交
230 231
int32_t tsdbReadBlockIdx(SDataFReader *pReader, SMapData *pMapData, uint8_t **ppBuf);
int32_t tsdbReadBlock(SDataFReader *pReader, SBlockIdx *pBlockIdx, SMapData *pMapData, uint8_t **ppBuf);
H
Hongze Cheng 已提交
232
int32_t tsdbReadColData(SDataFReader *pReader, SBlockIdx *pBlockIdx, SBlock *pBlock, int16_t *aColId, int32_t nCol,
H
Hongze Cheng 已提交
233
                        SBlockData *pBlockData, uint8_t **ppBuf1, uint8_t **ppBuf2);
H
Hongze Cheng 已提交
234
int32_t tsdbReadBlockData(SDataFReader *pReader, SBlockIdx *pBlockIdx, SBlock *pBlock, SBlockData *pBlockData,
H
Hongze Cheng 已提交
235
                          uint8_t **ppBuf1, uint8_t **ppBuf2);
H
Hongze Cheng 已提交
236 237 238 239
int32_t tsdbReadBlockSMA(SDataFReader *pReader, SBlockSMA *pBlkSMA);
// SDelFWriter
int32_t tsdbDelFWriterOpen(SDelFWriter **ppWriter, SDelFile *pFile, STsdb *pTsdb);
int32_t tsdbDelFWriterClose(SDelFWriter *pWriter, int8_t sync);
H
Hongze Cheng 已提交
240 241 242
int32_t tsdbWriteDelData(SDelFWriter *pWriter, SArray *aDelData, uint8_t **ppBuf, SDelIdx *pDelIdx);
int32_t tsdbWriteDelIdx(SDelFWriter *pWriter, SArray *aDelIdx, uint8_t **ppBuf);
int32_t tsdbUpdateDelFileHdr(SDelFWriter *pWriter);
H
Hongze Cheng 已提交
243 244 245
// SDelFReader
int32_t tsdbDelFReaderOpen(SDelFReader **ppReader, SDelFile *pFile, STsdb *pTsdb, uint8_t **ppBuf);
int32_t tsdbDelFReaderClose(SDelFReader *pReader);
H
Hongze Cheng 已提交
246 247
int32_t tsdbReadDelData(SDelFReader *pReader, SDelIdx *pDelIdx, SArray *aDelData, uint8_t **ppBuf);
int32_t tsdbReadDelIdx(SDelFReader *pReader, SArray *aDelIdx, uint8_t **ppBuf);
H
Hongze Cheng 已提交
248

249 250
// tsdbCache
int32_t tsdbOpenCache(STsdb *pTsdb);
H
Hongze Cheng 已提交
251
void    tsdbCloseCache(SLRUCache *pCache);
252
int32_t tsdbCacheInsertLastrow(SLRUCache *pCache, tb_uid_t uid, STSRow *row);
253 254 255

int32_t tsdbCacheGetLastH(SLRUCache *pCache, tb_uid_t uid, STsdb *pTsdb, LRUHandle **h);
int32_t tsdbCacheGetLastrowH(SLRUCache *pCache, tb_uid_t uid, STsdb *pTsdb, LRUHandle **h);
M
Minglei Jin 已提交
256
int32_t tsdbCacheRelease(SLRUCache *pCache, LRUHandle *h);
257

258
int32_t tsdbCacheDelete(SLRUCache *pCache, tb_uid_t uid, TSKEY eKey);
259

H
Hongze Cheng 已提交
260
// structs =======================
H
Hongze Cheng 已提交
261 262 263 264 265 266 267 268
typedef struct {
  int   minFid;
  int   midFid;
  int   maxFid;
  TSKEY minKey;
} SRtn;

struct STsdb {
H
refact  
Hongze Cheng 已提交
269 270 271 272 273 274 275 276 277
  char         *path;
  SVnode       *pVnode;
  TdThreadMutex mutex;
  bool          repoLocked;
  STsdbKeepCfg  keepCfg;
  SMemTable    *mem;
  SMemTable    *imem;
  SRtn          rtn;
  STsdbFS      *fs;
278
  SLRUCache    *lruCache;
H
Hongze Cheng 已提交
279 280
};

H
Hongze Cheng 已提交
281
struct STable {
H
Hongze Cheng 已提交
282
  uint64_t  suid;
H
Hongze Cheng 已提交
283
  uint64_t  uid;
284 285
  STSchema *pSchema;       // latest schema
  STSchema *pCacheSchema;  // cached cache
H
Hongze Cheng 已提交
286
};
H
Hongze Cheng 已提交
287

H
refact  
Hongze Cheng 已提交
288 289 290 291 292
struct TSDBKEY {
  int64_t version;
  TSKEY   ts;
};

H
Hongze Cheng 已提交
293 294 295 296 297 298 299
struct KEYINFO {
  TSDBKEY minKey;
  TSDBKEY maxKey;
  int64_t minVerion;
  int64_t maxVersion;
};

H
Hongze Cheng 已提交
300 301 302 303 304 305 306 307 308 309 310 311 312 313
typedef struct SMemSkipListNode SMemSkipListNode;
struct SMemSkipListNode {
  int8_t            level;
  SMemSkipListNode *forwards[0];
};
typedef struct SMemSkipList {
  uint32_t          seed;
  int64_t           size;
  int8_t            maxLevel;
  int8_t            level;
  SMemSkipListNode *pHead;
  SMemSkipListNode *pTail;
} SMemSkipList;

H
Hongze Cheng 已提交
314 315 316 317 318
struct SDelDataInfo {
  tb_uid_t suid;
  tb_uid_t uid;
};

H
Hongze Cheng 已提交
319
struct STbData {
H
Hongze Cheng 已提交
320 321
  tb_uid_t     suid;
  tb_uid_t     uid;
H
Hongze Cheng 已提交
322 323 324 325
  TSKEY        minKey;
  TSKEY        maxKey;
  int64_t      minVersion;
  int64_t      maxVersion;
H
Hongze Cheng 已提交
326 327
  SDelData    *pHead;
  SDelData    *pTail;
H
Hongze Cheng 已提交
328
  SMemSkipList sl;
H
Hongze Cheng 已提交
329
};
H
Hongze Cheng 已提交
330

H
refact  
Hongze Cheng 已提交
331
struct SMemTable {
H
Hongze Cheng 已提交
332 333 334
  SRWLatch latch;
  STsdb   *pTsdb;
  int32_t  nRef;
H
Hongze Cheng 已提交
335 336 337 338
  TSKEY    minKey;
  TSKEY    maxKey;
  int64_t  minVersion;
  int64_t  maxVersion;
H
Hongze Cheng 已提交
339
  int64_t  nRow;
H
Hongze Cheng 已提交
340
  int64_t  nDel;
H
Hongze Cheng 已提交
341
  SArray  *aTbData;  // SArray<STbData*>
H
Hongze Cheng 已提交
342
};
H
Hongze Cheng 已提交
343 344 345 346

int tsdbLockRepo(STsdb *pTsdb);
int tsdbUnlockRepo(STsdb *pTsdb);

H
Hongze Cheng 已提交
347
struct TSDBROW {
H
Hongze Cheng 已提交
348
  int8_t type;  // 0 for row from tsRow, 1 for row from block data
H
Hongze Cheng 已提交
349 350 351 352 353 354
  union {
    struct {
      int64_t version;
      STSRow *pTSRow;
    };
    struct {
H
Hongze Cheng 已提交
355 356
      SBlockData *pBlockData;
      int32_t     iRow;
H
Hongze Cheng 已提交
357 358
    };
  };
H
Hongze Cheng 已提交
359 360
};

H
Hongze Cheng 已提交
361
struct SBlockIdx {
H
Hongze Cheng 已提交
362 363
  int64_t suid;
  int64_t uid;
H
Hongze Cheng 已提交
364 365 366 367
  TSKEY   minKey;
  TSKEY   maxKey;
  int64_t minVersion;
  int64_t maxVersion;
H
Hongze Cheng 已提交
368 369 370 371
  int64_t offset;
  int64_t size;
};

H
Hongze Cheng 已提交
372 373 374 375 376 377 378 379 380
struct SMapData {
  int32_t  nItem;
  uint8_t  flag;
  uint8_t *pOfst;
  uint32_t nData;
  uint8_t *pData;
  uint8_t *pBuf;
};

H
Hongze Cheng 已提交
381
typedef struct {
H
Hongze Cheng 已提交
382 383
  int16_t cid;
  int8_t  type;
H
Hongze Cheng 已提交
384
  int8_t  flag;  // HAS_NONE|HAS_NULL|HAS_VALUE
H
Hongze Cheng 已提交
385
  int64_t offset;
H
Hongze Cheng 已提交
386 387 388
  int64_t bsize;  // bitmap size
  int64_t csize;  // compressed column value size
  int64_t osize;  // original column value size (only save for variant data type)
H
Hongze Cheng 已提交
389 390 391
} SBlockCol;

typedef struct {
H
Hongze Cheng 已提交
392 393
  int64_t  nRow;
  int8_t   cmprAlg;
H
Hongze Cheng 已提交
394
  int64_t  offset;
H
Hongze Cheng 已提交
395 396 397
  int64_t  vsize;      // VERSION size
  int64_t  ksize;      // TSKEY size
  int64_t  bsize;      // total block size
H
Hongze Cheng 已提交
398
  SMapData mBlockCol;  // SMapData<SBlockCol>
H
Hongze Cheng 已提交
399 400
} SSubBlock;

H
Hongze Cheng 已提交
401
struct SBlock {
H
Hongze Cheng 已提交
402 403 404 405
  TSDBKEY   minKey;
  TSDBKEY   maxKey;
  int64_t   minVersion;
  int64_t   maxVersion;
H
Hongze Cheng 已提交
406
  int32_t   nRow;
H
Hongze Cheng 已提交
407 408 409
  int8_t    last;
  int8_t    hasDup;
  int8_t    nSubBlock;
H
Hongze Cheng 已提交
410
  SSubBlock aSubBlock[TSDB_MAX_SUBBLOCKS];
H
refact  
Hongze Cheng 已提交
411
};
H
Hongze Cheng 已提交
412

H
refact  
Hongze Cheng 已提交
413
struct SAggrBlkCol {
H
Hongze Cheng 已提交
414 415 416 417 418 419 420
  int16_t colId;
  int16_t maxIndex;
  int16_t minIndex;
  int16_t numOfNull;
  int64_t sum;
  int64_t max;
  int64_t min;
H
refact  
Hongze Cheng 已提交
421
};
H
Hongze Cheng 已提交
422

H
Hongze Cheng 已提交
423 424 425
struct SColData {
  int16_t  cid;
  int8_t   type;
H
Hongze Cheng 已提交
426 427 428
  int8_t   offsetValid;
  int32_t  nVal;
  uint8_t  flag;
H
Hongze Cheng 已提交
429
  uint8_t *pBitMap;
H
Hongze Cheng 已提交
430 431
  int32_t *aOffset;
  int32_t  nData;
H
Hongze Cheng 已提交
432
  uint8_t *pData;
H
refact  
Hongze Cheng 已提交
433
};
H
Hongze Cheng 已提交
434

H
Hongze Cheng 已提交
435
struct SBlockData {
H
Hongze Cheng 已提交
436 437 438
  int32_t  nRow;
  int64_t *aVersion;
  TSKEY   *aTSKEY;
H
Hongze Cheng 已提交
439 440
  SArray  *aColDataP;  // SArray<SColData *>
  SArray  *aColData;   // SArray<SColData>
H
Hongze Cheng 已提交
441
};
H
Hongze Cheng 已提交
442 443 444 445

// ================== TSDB global config
extern bool tsdbForceKeepFile;

H
Hongze Cheng 已提交
446
#define TSDB_FS_ITER_FORWARD  TSDB_ORDER_ASC
H
Hongze Cheng 已提交
447 448
#define TSDB_FS_ITER_BACKWARD TSDB_ORDER_DESC

H
Hongze Cheng 已提交
449 450 451 452 453
struct TABLEID {
  tb_uid_t suid;
  tb_uid_t uid;
};

H
Hongze Cheng 已提交
454 455 456 457
struct STbDataIter {
  STbData          *pTbData;
  int8_t            backward;
  SMemSkipListNode *pNode;
H
Hongze Cheng 已提交
458 459
  TSDBROW          *pRow;
  TSDBROW           row;
H
Hongze Cheng 已提交
460 461
};

H
Hongze Cheng 已提交
462
struct SDelData {
H
Hongze Cheng 已提交
463 464 465 466
  int64_t   version;
  TSKEY     sKey;
  TSKEY     eKey;
  SDelData *pNext;
H
Hongze Cheng 已提交
467
};
H
Hongze Cheng 已提交
468

H
Hongze Cheng 已提交
469
struct SDelIdx {
H
Hongze Cheng 已提交
470 471 472 473
  tb_uid_t suid;
  tb_uid_t uid;
  int64_t  offset;
  int64_t  size;
H
Hongze Cheng 已提交
474
};
H
Hongze Cheng 已提交
475

H
Hongze Cheng 已提交
476
struct SDelFile {
H
Hongze Cheng 已提交
477
  int64_t commitID;
H
Hongze Cheng 已提交
478 479 480
  int64_t size;
  int64_t offset;
};
H
Hongze Cheng 已提交
481

H
Hongze Cheng 已提交
482 483 484 485 486 487 488 489 490 491 492 493 494 495 496
typedef struct {
  int16_t colId;
  int16_t maxIndex;
  int16_t minIndex;
  int16_t numOfNull;
  int64_t sum;
  int64_t max;
  int64_t min;
} SColSMA;

struct SBlockSMA {
  int32_t  nCol;
  SColSMA *aColSMA;
};

H
Hongze Cheng 已提交
497
#pragma pack(push, 1)
H
Hongze Cheng 已提交
498
struct SBlockDataHdr {
H
more  
Hongze Cheng 已提交
499 500 501
  uint32_t delimiter;
  int64_t  suid;
  int64_t  uid;
H
Hongze Cheng 已提交
502
};
H
Hongze Cheng 已提交
503
#pragma pack(pop)
H
Hongze Cheng 已提交
504

H
Hongze Cheng 已提交
505
struct SHeadFile {
H
Hongze Cheng 已提交
506
  int64_t commitID;
H
Hongze Cheng 已提交
507 508 509 510 511
  int64_t size;
  int64_t offset;
};

struct SDataFile {
H
Hongze Cheng 已提交
512
  int64_t commitID;
H
Hongze Cheng 已提交
513 514 515 516
  int64_t size;
};

struct SLastFile {
H
Hongze Cheng 已提交
517
  int64_t commitID;
H
Hongze Cheng 已提交
518 519 520 521
  int64_t size;
};

struct SSmaFile {
H
Hongze Cheng 已提交
522
  int64_t commitID;
H
Hongze Cheng 已提交
523 524 525 526
  int64_t size;
};

struct SDFileSet {
H
Hongze Cheng 已提交
527 528 529 530 531 532
  SDiskID   diskId;
  int32_t   fid;
  SHeadFile fHead;
  SDataFile fData;
  SLastFile fLast;
  SSmaFile  fSma;
H
Hongze Cheng 已提交
533 534 535 536
};

struct SRowIter {
  TSDBROW  *pRow;
H
Hongze Cheng 已提交
537
  STSchema *pTSchema;
H
Hongze Cheng 已提交
538 539
  SColVal   colVal;
  int32_t   i;
H
Hongze Cheng 已提交
540
};
H
Hongze Cheng 已提交
541 542 543 544 545
struct SRowMerger {
  STSchema *pTSchema;
  int64_t   version;
  SArray   *pArray;  // SArray<SColVal>
};
H
Hongze Cheng 已提交
546

H
more  
Hongze Cheng 已提交
547 548 549 550 551 552
struct STsdbFSState {
  SDelFile *pDelFile;
  SArray   *aDFileSet;  // SArray<aDFileSet>
  SDelFile  delFile;
};

H
Hongze Cheng 已提交
553 554 555 556 557 558 559 560
struct STsdbFS {
  STsdb         *pTsdb;
  TdThreadRwlock lock;
  int8_t         inTxn;
  STsdbFSState  *cState;
  STsdbFSState  *nState;
};

H
Hongze Cheng 已提交
561 562 563 564 565 566
struct SDelFWriter {
  STsdb    *pTsdb;
  SDelFile  fDel;
  TdFilePtr pWriteH;
};

H
Hongze Cheng 已提交
567 568 569 570
#ifdef __cplusplus
}
#endif

571
#endif /*_TD_VNODE_TSDB_H_*/