tsdbMain.h 19.3 KB
Newer Older
H
more  
hzcheng 已提交
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
TD-34  
hzcheng 已提交
15 16
#ifndef _TD_TSDB_MAIN_H_
#define _TD_TSDB_MAIN_H_
H
more  
Hongze Cheng 已提交
17

S
TD-1057  
Shengliang Guan 已提交
18
#include "os.h"
H
TD-353  
Hongze Cheng 已提交
19 20
#include "hash.h"
#include "tcoding.h"
S
slguan 已提交
21
#include "tglobal.h"
H
TD-353  
Hongze Cheng 已提交
22
#include "tkvstore.h"
H
hzcheng 已提交
23
#include "tlist.h"
H
TD-353  
Hongze Cheng 已提交
24
#include "tlog.h"
B
Bomin Zhang 已提交
25
#include "tlockfree.h"
H
hzcheng 已提交
26
#include "tsdb.h"
H
TD-34  
hzcheng 已提交
27 28
#include "tskiplist.h"
#include "tutil.h"
H
more  
Hongze Cheng 已提交
29

H
more  
hzcheng 已提交
30 31 32 33
#ifdef __cplusplus
extern "C" {
#endif

S
Shengliang Guan 已提交
34
extern int32_t tsdbDebugFlag;
H
hzcheng 已提交
35

H
Haojun Liao 已提交
36 37 38 39 40 41
#define tsdbFatal(...) do { if (tsdbDebugFlag & DEBUG_FATAL) { taosPrintLog("TDB FATAL ", 255, __VA_ARGS__); }}     while(0)
#define tsdbError(...) do { if (tsdbDebugFlag & DEBUG_ERROR) { taosPrintLog("TDB ERROR ", 255, __VA_ARGS__); }}     while(0)
#define tsdbWarn(...)  do { if (tsdbDebugFlag & DEBUG_WARN)  { taosPrintLog("TDB WARN ", 255, __VA_ARGS__); }}      while(0)
#define tsdbInfo(...)  do { if (tsdbDebugFlag & DEBUG_INFO)  { taosPrintLog("TDB ", 255, __VA_ARGS__); }}           while(0)
#define tsdbDebug(...) do { if (tsdbDebugFlag & DEBUG_DEBUG) { taosPrintLog("TDB ", tsdbDebugFlag, __VA_ARGS__); }} while(0)
#define tsdbTrace(...) do { if (tsdbDebugFlag & DEBUG_TRACE) { taosPrintLog("TDB ", tsdbDebugFlag, __VA_ARGS__); }} while(0)
H
hzcheng 已提交
42

H
TD-90  
Hongze Cheng 已提交
43
#define TSDB_MAX_TABLE_SCHEMAS 16
H
Haojun Liao 已提交
44 45 46
#define TSDB_FILE_HEAD_SIZE    512
#define TSDB_FILE_DELIMITER    0xF00AFA0F
#define TSDB_FILE_INIT_MAGIC   0xFFFFFFFF
H
TD-353  
Hongze Cheng 已提交
47

H
Hongze Cheng 已提交
48 49
#define TAOS_IN_RANGE(key, keyMin, keyLast) (((key) >= (keyMin)) && ((key) <= (keyMax)))

50 51 52 53
// NOTE: Any file format change must increase this version number by 1
//       Also, implement the convert function
#define TSDB_FILE_VERSION ((uint32_t)0)

H
TD-353  
Hongze Cheng 已提交
54 55
// Definitions
// ------------------ tsdbMeta.c
H
TD-34  
hzcheng 已提交
56
typedef struct STable {
H
Haojun Liao 已提交
57
  STableId       tableId;
H
TD-353  
Hongze Cheng 已提交
58 59 60 61 62
  ETableType     type;
  tstr*          name;  // NOTE: there a flexible string here
  uint64_t       suid;
  struct STable* pSuper;  // super table pointer
  uint8_t        numOfSchemas;
H
TD-353  
Hongze Cheng 已提交
63
  STSchema*      schema[TSDB_MAX_TABLE_SCHEMAS];
H
TD-353  
Hongze Cheng 已提交
64 65
  STSchema*      tagSchema;
  SKVRow         tagVal;
H
TD-353  
Hongze Cheng 已提交
66
  SSkipList*     pIndex;         // For TSDB_SUPER_TABLE, it is the skiplist index
H
TD-353  
Hongze Cheng 已提交
67 68
  void*          eventHandler;   // TODO
  void*          streamHandler;  // TODO
H
Hongze Cheng 已提交
69 70
  TSKEY          lastKey;
  SDataRow       lastRow;
H
TD-353  
Hongze Cheng 已提交
71 72
  char*          sql;
  void*          cqhandle;
H
Hongze Cheng 已提交
73
  SRWLatch       latch;  // TODO: implementa latch functions
S
TD-1057  
Shengliang Guan 已提交
74
  T_REF_DECLARE()
H
TD-34  
hzcheng 已提交
75 76 77
} STable;

typedef struct {
H
TD-353  
Hongze Cheng 已提交
78 79 80
  pthread_rwlock_t rwLock;

  int32_t   nTables;
H
TD-987  
Hongze Cheng 已提交
81
  int32_t   maxTables;
H
TD-353  
Hongze Cheng 已提交
82 83 84 85
  STable**  tables;
  SList*    superList;
  SHashObj* uidMap;
  SKVStore* pStore;
H
TD-353  
Hongze Cheng 已提交
86 87
  int       maxRowBytes;
  int       maxCols;
H
TD-34  
hzcheng 已提交
88 89
} STsdbMeta;

H
TD-353  
Hongze Cheng 已提交
90
// ------------------ tsdbBuffer.c
H
TD-34  
hzcheng 已提交
91
typedef struct {
H
TD-353  
Hongze Cheng 已提交
92 93 94 95 96
  int64_t blockId;
  int     offset;
  int     remain;
  char    data[];
} STsdbBufBlock;
H
TD-34  
hzcheng 已提交
97 98

typedef struct {
H
TD-353  
Hongze Cheng 已提交
99 100 101 102 103 104 105 106 107
  pthread_cond_t poolNotEmpty;
  int            bufBlockSize;
  int            tBufBlocks;
  int            nBufBlocks;
  int64_t        index;
  SList*         bufBlockList;
} STsdbBufPool;

// ------------------ tsdbMemTable.c
H
Hongze Cheng 已提交
108 109 110 111 112
typedef struct {
  STable *           pTable;
  SSkipListIterator *pIter;
} SCommitIter;

H
TD-34  
hzcheng 已提交
113
typedef struct {
H
TD-353  
Hongze Cheng 已提交
114 115 116 117 118 119
  uint64_t   uid;
  TSKEY      keyFirst;
  TSKEY      keyLast;
  int64_t    numOfRows;
  SSkipList* pData;
} STableData;
H
TD-34  
hzcheng 已提交
120 121

typedef struct {
S
TD-1057  
Shengliang Guan 已提交
122
  T_REF_DECLARE()
H
TD-987  
Hongze Cheng 已提交
123
  SRWLatch     latch;
H
TD-353  
Hongze Cheng 已提交
124 125 126
  TSKEY        keyFirst;
  TSKEY        keyLast;
  int64_t      numOfRows;
H
TD-987  
Hongze Cheng 已提交
127
  int32_t      maxTables;
H
TD-353  
Hongze Cheng 已提交
128 129
  STableData** tData;
  SList*       actList;
H
Hongze Cheng 已提交
130
  SList*       extraBuffList;
H
TD-353  
Hongze Cheng 已提交
131 132
  SList*       bufBlockList;
} SMemTable;
H
TD-34  
hzcheng 已提交
133

H
TD-353  
Hongze Cheng 已提交
134
enum { TSDB_UPDATE_META, TSDB_DROP_META };
S
TD-1057  
Shengliang Guan 已提交
135 136 137 138 139

#ifdef WINDOWS
#pragma pack(push ,1) 
typedef struct {
#else
H
TD-353  
Hongze Cheng 已提交
140
typedef struct __attribute__((packed)){
S
TD-1057  
Shengliang Guan 已提交
141
#endif
H
TD-353  
Hongze Cheng 已提交
142 143 144
  char     act;
  uint64_t uid;
} SActObj;
S
TD-1057  
Shengliang Guan 已提交
145 146 147
#ifdef WINDOWS
#pragma pack(pop) 
#endif
H
TD-353  
Hongze Cheng 已提交
148 149 150 151 152 153

typedef struct {
  int  len;
  char cont[];
} SActCont;

H
TD-353  
Hongze Cheng 已提交
154
// ------------------ tsdbFile.c
H
TD-353  
Hongze Cheng 已提交
155
extern const char* tsdbFileSuffix[];
H
TD-353  
Hongze Cheng 已提交
156
typedef enum {
H
Hongze Cheng 已提交
157
  TSDB_FILE_TYPE_HEAD = 0,
H
TD-353  
Hongze Cheng 已提交
158 159
  TSDB_FILE_TYPE_DATA,
  TSDB_FILE_TYPE_LAST,
H
Hongze Cheng 已提交
160
  TSDB_FILE_TYPE_STAT,
H
TD-353  
Hongze Cheng 已提交
161
  TSDB_FILE_TYPE_NHEAD,
H
Hongze Cheng 已提交
162 163 164
  TSDB_FILE_TYPE_NDATA,
  TSDB_FILE_TYPE_NLAST,
  TSDB_FILE_TYPE_NSTAT
H
TD-353  
Hongze Cheng 已提交
165
} TSDB_FILE_TYPE;
H
Hongze Cheng 已提交
166

H
Hongze Cheng 已提交
167 168 169 170 171 172
#ifndef TDINTERNAL
#define TSDB_FILE_TYPE_MAX (TSDB_FILE_TYPE_LAST+1)
#else
#define TSDB_FILE_TYPE_MAX (TSDB_FILE_TYPE_STAT+1)
#endif

H
more  
Hongze Cheng 已提交
173
typedef struct {
H
Hongze Cheng 已提交
174
  uint32_t magic;
H
TD-353  
Hongze Cheng 已提交
175 176 177
  uint32_t len;
  uint32_t totalBlocks;
  uint32_t totalSubBlocks;
H
Hongze Cheng 已提交
178
  uint32_t offset;
H
Hongze Cheng 已提交
179 180
  uint64_t size;      // total size of the file
  uint64_t tombSize;  // unused file size
181
} STsdbFileInfo;
H
TD-34  
hzcheng 已提交
182 183

typedef struct {
H
TD-353  
Hongze Cheng 已提交
184
  char  fname[TSDB_FILENAME_LEN];
H
TD-353  
Hongze Cheng 已提交
185 186
  int   fd;

H
TD-353  
Hongze Cheng 已提交
187
  STsdbFileInfo info;
H
hzcheng 已提交
188
} SFile;
H
more  
Hongze Cheng 已提交
189

H
hzcheng 已提交
190
typedef struct {
H
TD-353  
Hongze Cheng 已提交
191
  int   fileId;
H
TD-1102  
Hongze Cheng 已提交
192
  int   state; // 0 for health, 1 for problem
H
TD-353  
Hongze Cheng 已提交
193
  SFile files[TSDB_FILE_TYPE_MAX];
H
hzcheng 已提交
194 195 196
} SFileGroup;

typedef struct {
H
TD-353  
Hongze Cheng 已提交
197 198
  pthread_rwlock_t fhlock;

H
TD-353  
Hongze Cheng 已提交
199 200 201
  int         maxFGroups;
  int         nFGroups;
  SFileGroup* pFGroup;
H
hzcheng 已提交
202
} STsdbFileH;
H
more  
Hongze Cheng 已提交
203

H
TD-34  
hzcheng 已提交
204 205
typedef struct {
  int         direction;
206 207 208
  STsdbFileH* pFileH;
  int         fileId;
  int         index;
H
TD-34  
hzcheng 已提交
209 210
} SFileGroupIter;

H
TD-353  
Hongze Cheng 已提交
211
// ------------------ tsdbMain.c
H
Hongze Cheng 已提交
212 213 214 215 216 217 218 219 220 221 222 223
typedef struct {
  int32_t  totalLen;
  int32_t  len;
  SDataRow row;
} SSubmitBlkIter;

typedef struct {
  int32_t totalLen;
  int32_t len;
  void *  pMsg;
} SSubmitMsgIter;

H
TD-353  
Hongze Cheng 已提交
224 225 226 227 228 229 230 231 232 233 234 235
typedef struct {
  int8_t state;

  char*           rootDir;
  STsdbCfg        config;
  STsdbAppH       appH;
  STsdbStat       stat;
  STsdbMeta*      tsdbMeta;
  STsdbBufPool*   pPool;
  SMemTable*      mem;
  SMemTable*      imem;
  STsdbFileH*     tsdbFileH;
F
freemine 已提交
236
  tsem_t          readyToCommit;
H
TD-353  
Hongze Cheng 已提交
237 238
  pthread_mutex_t mutex;
  bool            repoLocked;
H
Hongze Cheng 已提交
239
  int32_t         code; // Commit code
H
TD-353  
Hongze Cheng 已提交
240 241
} STsdbRepo;

H
TD-353  
Hongze Cheng 已提交
242
// ------------------ tsdbRWHelper.c
H
TD-34  
hzcheng 已提交
243
typedef struct {
H
Hongze Cheng 已提交
244
  int32_t  tid;
H
TD-185  
Hongze Cheng 已提交
245 246 247 248
  uint32_t len;
  uint32_t offset;
  uint32_t hasLast : 2;
  uint32_t numOfBlocks : 30;
249 250
  uint64_t uid;
  TSKEY    maxKey;
H
TD-353  
Hongze Cheng 已提交
251 252
} SCompIdx;

253
typedef struct {
H
TD-353  
Hongze Cheng 已提交
254 255 256 257 258
  int64_t last : 1;
  int64_t offset : 63;
  int32_t algorithm : 8;
  int32_t numOfRows : 24;
  int32_t len;
H
Hongze Cheng 已提交
259
  int32_t keyLen;     // key column length, keyOffset = offset+sizeof(SCompData)+sizeof(SCompCol)*numOfCols
H
TD-353  
Hongze Cheng 已提交
260
  int16_t numOfSubBlocks;
H
Hongze Cheng 已提交
261
  int16_t numOfCols; // not including timestamp column
262 263 264 265
  TSKEY   keyFirst;
  TSKEY   keyLast;
} SCompBlock;

H
TD-34  
hzcheng 已提交
266 267
typedef struct {
  int32_t    delimiter;  // For recovery usage
268
  int32_t    tid;
269
  uint64_t   uid;
H
TD-34  
hzcheng 已提交
270 271
  SCompBlock blocks[];
} SCompInfo;
H
hzcheng 已提交
272

H
TD-34  
hzcheng 已提交
273
typedef struct {
H
TD-353  
Hongze Cheng 已提交
274
  int16_t colId;
H
TD-541  
Hongze Cheng 已提交
275
  int32_t len;
H
TD-34  
hzcheng 已提交
276 277
  int32_t type : 8;
  int32_t offset : 24;
H
TD-321  
Hongze Cheng 已提交
278 279 280 281 282 283 284
  int64_t sum;
  int64_t max;
  int64_t min;
  int16_t maxIndex;
  int16_t minIndex;
  int16_t numOfNull;
  char    padding[2];
H
TD-34  
hzcheng 已提交
285 286 287 288 289
} SCompCol;

typedef struct {
  int32_t  delimiter;  // For recovery usage
  int32_t  numOfCols;  // For recovery usage
290
  uint64_t uid;        // For recovery usage
H
TD-34  
hzcheng 已提交
291 292
  SCompCol cols[];
} SCompData;
H
TD-34  
hzcheng 已提交
293

H
hzcheng 已提交
294 295 296
typedef enum { TSDB_WRITE_HELPER, TSDB_READ_HELPER } tsdb_rw_helper_t;

typedef struct {
H
Hongze Cheng 已提交
297 298 299 300 301
  TSKEY      minKey;
  TSKEY      maxKey;
  SFileGroup fGroup;
  SFile      nHeadF;
  SFile      nLastF;
H
hzcheng 已提交
302 303 304
} SHelperFile;

typedef struct {
305 306
  uint64_t uid;
  int32_t  tid;
H
hzcheng 已提交
307 308
} SHelperTable;

H
Hongze Cheng 已提交
309 310 311 312 313 314
typedef struct {
  SCompIdx* pIdxArray;
  int       numOfIdx;
  int       curIdx;
} SIdxH;

H
hzcheng 已提交
315
typedef struct {
H
TD-353  
Hongze Cheng 已提交
316
  tsdb_rw_helper_t type;
H
TD-100  
hzcheng 已提交
317

H
TD-353  
Hongze Cheng 已提交
318 319
  STsdbRepo* pRepo;
  int8_t     state;
H
TD-100  
hzcheng 已提交
320
  // For file set usage
H
hzcheng 已提交
321
  SHelperFile files;
H
Hongze Cheng 已提交
322 323 324
  SIdxH       idxH;
  SCompIdx    curCompIdx;
  void*       pWIdx;
H
TD-100  
hzcheng 已提交
325
  // For table set usage
H
hzcheng 已提交
326
  SHelperTable tableInfo;
H
TD-353  
Hongze Cheng 已提交
327
  SCompInfo*   pCompInfo;
H
TD-100  
hzcheng 已提交
328 329
  bool         hasOldLastBlock;
  // For block set usage
H
TD-353  
Hongze Cheng 已提交
330
  SCompData* pCompData;
H
TD-353  
Hongze Cheng 已提交
331 332 333
  SDataCols* pDataCols[2];
  void*      pBuffer;     // Buffer to hold the whole data block
  void*      compBuffer;  // Buffer for temperary compress/decompress purpose
H
hzcheng 已提交
334 335
} SRWHelper;

H
TD-1548  
Hongze Cheng 已提交
336 337 338 339 340 341 342 343 344
typedef struct {
  int   rowsInserted;
  int   rowsUpdated;
  int   rowsDeleteSucceed;
  int   rowsDeleteFailed;
  int   nOperations;
  TSKEY keyFirst;
  TSKEY keyLast;
} SMergeInfo;
H
TD-1027  
Hongze Cheng 已提交
345 346 347 348 349 350 351 352 353 354
// ------------------ tsdbScan.c
typedef struct {
  SFileGroup fGroup;
  int        numOfIdx;
  SCompIdx*  pCompIdx;
  SCompInfo* pCompInfo;
  void*      pBuf;
  FILE*      tLogStream;
} STsdbScanHandle;

H
TD-353  
Hongze Cheng 已提交
355 356
// Operations
// ------------------ tsdbMeta.c
H
TD-987  
Hongze Cheng 已提交
357
#define TSDB_INIT_NTABLES 1024
H
TD-353  
Hongze Cheng 已提交
358 359 360
#define TABLE_TYPE(t) (t)->type
#define TABLE_NAME(t) (t)->name
#define TABLE_CHAR_NAME(t) TABLE_NAME(t)->data
H
TD-353  
Hongze Cheng 已提交
361
#define TABLE_UID(t) (t)->tableId.uid
H
TD-353  
Hongze Cheng 已提交
362
#define TABLE_TID(t) (t)->tableId.tid
H
TD-353  
Hongze Cheng 已提交
363
#define TABLE_SUID(t) (t)->suid
364
#define TSDB_META_FILE_MAGIC(m) KVSTORE_MAGIC((m)->pStore)
365 366 367 368
#define TSDB_RLOCK_TABLE(t) taosRLockLatch(&((t)->latch))
#define TSDB_RUNLOCK_TABLE(t) taosRUnLockLatch(&((t)->latch))
#define TSDB_WLOCK_TABLE(t) taosWLockLatch(&((t)->latch))
#define TSDB_WUNLOCK_TABLE(t) taosWUnLockLatch(&((t)->latch))
H
TD-353  
Hongze Cheng 已提交
369 370 371

STsdbMeta* tsdbNewMeta(STsdbCfg* pCfg);
void       tsdbFreeMeta(STsdbMeta* pMeta);
H
TD-353  
Hongze Cheng 已提交
372 373 374 375 376 377 378
int        tsdbOpenMeta(STsdbRepo* pRepo);
int        tsdbCloseMeta(STsdbRepo* pRepo);
STable*    tsdbGetTableByUid(STsdbMeta* pMeta, uint64_t uid);
STSchema*  tsdbGetTableSchemaByVersion(STable* pTable, int16_t version);
int        tsdbWLockRepoMeta(STsdbRepo* pRepo);
int        tsdbRLockRepoMeta(STsdbRepo* pRepo);
int        tsdbUnlockRepoMeta(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
379 380
void       tsdbRefTable(STable* pTable);
void       tsdbUnRefTable(STable* pTable);
H
Hongze Cheng 已提交
381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397
void       tsdbUpdateTableSchema(STsdbRepo* pRepo, STable* pTable, STSchema* pSchema, bool insertAct);

static FORCE_INLINE int tsdbCompareSchemaVersion(const void *key1, const void *key2) {
  if (*(int16_t *)key1 < schemaVersion(*(STSchema **)key2)) {
    return -1;
  } else if (*(int16_t *)key1 > schemaVersion(*(STSchema **)key2)) {
    return 1;
  } else {
    return 0;
  }
}

static FORCE_INLINE STSchema* tsdbGetTableSchemaImpl(STable* pTable, bool lock, bool copy, int16_t version) {
  STable*   pDTable = (TABLE_TYPE(pTable) == TSDB_CHILD_TABLE) ? pTable->pSuper : pTable;
  STSchema* pSchema = NULL;
  STSchema* pTSchema = NULL;

398
  if (lock) TSDB_RLOCK_TABLE(pDTable);
H
Hongze Cheng 已提交
399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419
  if (version < 0) {  // get the latest version of schema
    pTSchema = pDTable->schema[pDTable->numOfSchemas - 1];
  } else {  // get the schema with version
    void* ptr = taosbsearch(&version, pDTable->schema, pDTable->numOfSchemas, sizeof(STSchema*),
                            tsdbCompareSchemaVersion, TD_EQ);
    if (ptr == NULL) {
      terrno = TSDB_CODE_TDB_IVD_TB_SCHEMA_VERSION;
      goto _exit;
    }
    pTSchema = *(STSchema**)ptr;
  }

  ASSERT(pTSchema != NULL);

  if (copy) {
    if ((pSchema = tdDupSchema(pTSchema)) == NULL) terrno = TSDB_CODE_TDB_OUT_OF_MEMORY;
  } else {
    pSchema = pTSchema;
  }

_exit:
420
  if (lock) TSDB_RUNLOCK_TABLE(pDTable);
H
Hongze Cheng 已提交
421 422
  return pSchema;
}
H
TD-353  
Hongze Cheng 已提交
423

H
Hongze Cheng 已提交
424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439
static FORCE_INLINE STSchema* tsdbGetTableSchema(STable* pTable) {
  return tsdbGetTableSchemaImpl(pTable, false, false, -1);
}

static FORCE_INLINE STSchema *tsdbGetTableTagSchema(STable *pTable) {
  if (pTable->type == TSDB_CHILD_TABLE) {  // check child table first
    STable *pSuper = pTable->pSuper;
    if (pSuper == NULL) return NULL;
    return pSuper->tagSchema;
  } else if (pTable->type == TSDB_SUPER_TABLE) {
    return pTable->tagSchema;
  } else {
    return NULL;
  }
}

H
Hongze Cheng 已提交
440 441 442
static FORCE_INLINE TSKEY tsdbGetTableLastKeyImpl(STable* pTable) {
  ASSERT(pTable->lastRow == NULL || pTable->lastKey == dataRowKey(pTable->lastRow));
  return pTable->lastKey;
443 444
}

H
TD-353  
Hongze Cheng 已提交
445
// ------------------ tsdbBuffer.c
H
Hongze Cheng 已提交
446 447
#define TSDB_BUFFER_RESERVE 1024  // Reseve 1K as commit threshold

H
TD-353  
Hongze Cheng 已提交
448 449 450
STsdbBufPool* tsdbNewBufPool();
void          tsdbFreeBufPool(STsdbBufPool* pBufPool);
int           tsdbOpenBufPool(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
451
void          tsdbCloseBufPool(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
452 453 454
SListNode*    tsdbAllocBufBlockFromPool(STsdbRepo* pRepo);

// ------------------ tsdbMemTable.c
H
TD-353  
Hongze Cheng 已提交
455 456 457
int   tsdbRefMemTable(STsdbRepo* pRepo, SMemTable* pMemTable);
int   tsdbUnRefMemTable(STsdbRepo* pRepo, SMemTable* pMemTable);
int   tsdbTakeMemSnapshot(STsdbRepo* pRepo, SMemTable** pMem, SMemTable** pIMem);
H
TD-987  
Hongze Cheng 已提交
458
void  tsdbUnTakeMemSnapShot(STsdbRepo* pRepo, SMemTable* pMem, SMemTable* pIMem);
H
TD-353  
Hongze Cheng 已提交
459
void* tsdbAllocBytes(STsdbRepo* pRepo, int bytes);
H
TD-353  
Hongze Cheng 已提交
460
int   tsdbAsyncCommit(STsdbRepo* pRepo);
H
Hongze Cheng 已提交
461
int   tsdbLoadDataFromCache(STable* pTable, SSkipListIterator* pIter, TSKEY maxKey, int maxRowsToRead, SDataCols* pCols,
H
TD-1548  
Hongze Cheng 已提交
462
                            TKEY* filterKeys, int nFilterKeys, bool keepDup, SMergeInfo* pMergeInfo);
H
Hongze Cheng 已提交
463
void* tsdbCommitData(STsdbRepo* pRepo);
H
Hongze Cheng 已提交
464 465 466 467 468 469 470

static FORCE_INLINE SDataRow tsdbNextIterRow(SSkipListIterator* pIter) {
  if (pIter == NULL) return NULL;

  SSkipListNode* node = tSkipListIterGet(pIter);
  if (node == NULL) return NULL;

H
TD-1194  
Hongze Cheng 已提交
471
  return (SDataRow)SL_GET_NODE_DATA(node);
H
Hongze Cheng 已提交
472 473 474 475
}

static FORCE_INLINE TSKEY tsdbNextIterKey(SSkipListIterator* pIter) {
  SDataRow row = tsdbNextIterRow(pIter);
H
TD-1548  
Hongze Cheng 已提交
476
  if (row == NULL) return TSDB_DATA_TIMESTAMP_NULL;
H
Hongze Cheng 已提交
477 478 479

  return dataRowKey(row);
}
H
TD-353  
Hongze Cheng 已提交
480

H
TD-1548  
Hongze Cheng 已提交
481 482 483 484 485 486 487
static FORCE_INLINE TKEY tsdbNextIterTKey(SSkipListIterator* pIter) {
  SDataRow row = tsdbNextIterRow(pIter);
  if (row == NULL) return TKEY_NULL;

  return dataRowTKey(row);
}

H
Hongze Cheng 已提交
488 489 490 491 492 493 494 495 496 497 498 499 500
static FORCE_INLINE STsdbBufBlock* tsdbGetCurrBufBlock(STsdbRepo* pRepo) {
  ASSERT(pRepo != NULL);
  if (pRepo->mem == NULL) return NULL;

  SListNode* pNode = listTail(pRepo->mem->bufBlockList);
  if (pNode == NULL) return NULL;

  STsdbBufBlock* pBufBlock = NULL;
  tdListNodeGetData(pRepo->mem->bufBlockList, pNode, (void*)(&pBufBlock));

  return pBufBlock;
}

H
TD-353  
Hongze Cheng 已提交
501 502 503 504 505
// ------------------ tsdbFile.c
#define TSDB_KEY_FILEID(key, daysPerFile, precision) ((key) / tsMsPerDay[(precision)] / (daysPerFile))
#define TSDB_MAX_FILE(keep, daysPerFile) ((keep) / (daysPerFile) + 3)
#define TSDB_MIN_FILE_ID(fh) (fh)->pFGroup[0].fileId
#define TSDB_MAX_FILE_ID(fh) (fh)->pFGroup[(fh)->nFGroups - 1].fileId
H
TD-353  
Hongze Cheng 已提交
506
#define TSDB_IS_FILE_OPENED(f) ((f)->fd > 0)
H
TD-353  
Hongze Cheng 已提交
507 508 509
#define TSDB_FGROUP_ITER_FORWARD TSDB_ORDER_ASC
#define TSDB_FGROUP_ITER_BACKWARD TSDB_ORDER_DESC

H
TD-353  
Hongze Cheng 已提交
510 511
STsdbFileH* tsdbNewFileH(STsdbCfg* pCfg);
void        tsdbFreeFileH(STsdbFileH* pFileH);
H
TD-353  
Hongze Cheng 已提交
512
int         tsdbOpenFileH(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
513
void        tsdbCloseFileH(STsdbRepo* pRepo);
H
TD-987  
Hongze Cheng 已提交
514
SFileGroup* tsdbCreateFGroupIfNeed(STsdbRepo* pRepo, char* dataDir, int fid);
H
TD-353  
Hongze Cheng 已提交
515 516 517 518 519 520 521 522
void        tsdbInitFileGroupIter(STsdbFileH* pFileH, SFileGroupIter* pIter, int direction);
void        tsdbSeekFileGroupIter(SFileGroupIter* pIter, int fid);
SFileGroup* tsdbGetFileGroupNext(SFileGroupIter* pIter);
int         tsdbOpenFile(SFile* pFile, int oflag);
void        tsdbCloseFile(SFile* pFile);
int         tsdbCreateFile(SFile* pFile, STsdbRepo* pRepo, int fid, int type);
SFileGroup* tsdbSearchFGroup(STsdbFileH* pFileH, int fid, int flags);
void        tsdbFitRetention(STsdbRepo* pRepo);
523
int         tsdbUpdateFileHeader(SFile* pFile);
H
TD-353  
Hongze Cheng 已提交
524
int         tsdbEncodeSFileInfo(void** buf, const STsdbFileInfo* pInfo);
H
TD-353  
Hongze Cheng 已提交
525
void*       tsdbDecodeSFileInfo(void* buf, STsdbFileInfo* pInfo);
H
TD-353  
Hongze Cheng 已提交
526
void        tsdbRemoveFileGroup(STsdbRepo* pRepo, SFileGroup* pFGroup);
H
Hongze Cheng 已提交
527
int         tsdbLoadFileHeader(SFile* pFile, uint32_t* version);
528
void        tsdbGetFileInfoImpl(char* fname, uint32_t* magic, int64_t* size);
529
void        tsdbGetFidKeyRange(int daysPerFile, int8_t precision, int fileId, TSKEY *minKey, TSKEY *maxKey);
H
TD-353  
Hongze Cheng 已提交
530 531

// ------------------ tsdbRWHelper.c
H
TD-353  
Hongze Cheng 已提交
532 533 534 535 536 537
#define TSDB_HELPER_CLEAR_STATE 0x0        // Clear state
#define TSDB_HELPER_FILE_SET_AND_OPEN 0x1  // File is set
#define TSDB_HELPER_IDX_LOAD 0x2           // SCompIdx part is loaded
#define TSDB_HELPER_TABLE_SET 0x4          // Table is set
#define TSDB_HELPER_INFO_LOAD 0x8          // SCompInfo part is loaded
#define TSDB_HELPER_FILE_DATA_LOAD 0x10    // SCompData part is loaded
H
TD-353  
Hongze Cheng 已提交
538 539 540 541
#define helperSetState(h, s) (((h)->state) |= (s))
#define helperClearState(h, s) ((h)->state &= (~(s)))
#define helperHasState(h, s) ((((h)->state) & (s)) == (s))
#define blockAtIdx(h, idx) ((h)->pCompInfo->blocks + idx)
H
TD-353  
Hongze Cheng 已提交
542 543
#define TSDB_MAX_SUBBLOCKS 8
#define IS_SUB_BLOCK(pBlock) ((pBlock)->numOfSubBlocks == 0)
H
TD-353  
Hongze Cheng 已提交
544 545 546
#define helperType(h) (h)->type
#define helperRepo(h) (h)->pRepo
#define helperState(h) (h)->state
H
Hongze Cheng 已提交
547
#define TSDB_NLAST_FILE_OPENED(h) ((h)->files.nLastF.fd > 0)
H
Hongze Cheng 已提交
548 549 550 551 552 553
#define helperFileId(h) ((h)->files.fGroup.fileId)
#define helperHeadF(h) (&((h)->files.fGroup.files[TSDB_FILE_TYPE_HEAD]))
#define helperDataF(h) (&((h)->files.fGroup.files[TSDB_FILE_TYPE_DATA]))
#define helperLastF(h) (&((h)->files.fGroup.files[TSDB_FILE_TYPE_LAST]))
#define helperNewHeadF(h) (&((h)->files.nHeadF))
#define helperNewLastF(h) (&((h)->files.nLastF))
H
Hongze Cheng 已提交
554 555 556 557 558 559

int  tsdbInitReadHelper(SRWHelper* pHelper, STsdbRepo* pRepo);
int  tsdbInitWriteHelper(SRWHelper* pHelper, STsdbRepo* pRepo);
void tsdbDestroyHelper(SRWHelper* pHelper);
void tsdbResetHelper(SRWHelper* pHelper);
int  tsdbSetAndOpenHelperFile(SRWHelper* pHelper, SFileGroup* pGroup);
H
Hongze Cheng 已提交
560
int  tsdbCloseHelperFile(SRWHelper* pHelper, bool hasError, SFileGroup* pGroup);
H
Hongze Cheng 已提交
561
int  tsdbSetHelperTable(SRWHelper* pHelper, STable* pTable, STsdbRepo* pRepo);
H
Hongze Cheng 已提交
562 563 564 565
int  tsdbCommitTableData(SRWHelper* pHelper, SCommitIter* pCommitIter, SDataCols* pDataCols, TSKEY maxKey);
int  tsdbMoveLastBlockIfNeccessary(SRWHelper* pHelper);
int  tsdbWriteCompInfo(SRWHelper* pHelper);
int  tsdbWriteCompIdx(SRWHelper* pHelper);
H
Hongze Cheng 已提交
566 567
int  tsdbLoadCompIdxImpl(SFile* pFile, uint32_t offset, uint32_t len, void* buffer);
int  tsdbDecodeSCompIdxImpl(void* buffer, uint32_t len, SCompIdx** ppCompIdx, int* numOfIdx);
H
Hongze Cheng 已提交
568
int  tsdbLoadCompIdx(SRWHelper* pHelper, void* target);
H
Hongze Cheng 已提交
569
int  tsdbLoadCompInfoImpl(SFile* pFile, SCompIdx* pIdx, SCompInfo** ppCompInfo);
H
Hongze Cheng 已提交
570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585
int  tsdbLoadCompInfo(SRWHelper* pHelper, void* target);
int  tsdbLoadCompData(SRWHelper* phelper, SCompBlock* pcompblock, void* target);
void tsdbGetDataStatis(SRWHelper* pHelper, SDataStatis* pStatis, int numOfCols);
int  tsdbLoadBlockDataCols(SRWHelper* pHelper, SCompBlock* pCompBlock, SCompInfo* pCompInfo, int16_t* colIds,
                           int numOfColIds);
int  tsdbLoadBlockData(SRWHelper* pHelper, SCompBlock* pCompBlock, SCompInfo* pCompInfo);

static FORCE_INLINE int compTSKEY(const void* key1, const void* key2) {
  if (*(TSKEY*)key1 > *(TSKEY*)key2) {
    return 1;
  } else if (*(TSKEY*)key1 == *(TSKEY*)key2) {
    return 0;
  } else {
    return -1;
  }
}
H
TD-353  
Hongze Cheng 已提交
586

H
TD-353  
Hongze Cheng 已提交
587 588
// ------------------ tsdbMain.c
#define REPO_ID(r) (r)->config.tsdbId
H
TD-353  
Hongze Cheng 已提交
589
#define IS_REPO_LOCKED(r) (r)->repoLocked
H
TD-353  
Hongze Cheng 已提交
590
#define TSDB_SUBMIT_MSG_HEAD_SIZE sizeof(SSubmitMsg)
H
TD-353  
Hongze Cheng 已提交
591

H
TD-353  
Hongze Cheng 已提交
592
char*       tsdbGetMetaFileName(char* rootDir);
H
Hongze Cheng 已提交
593
void        tsdbGetDataFileName(char* rootDir, int vid, int fid, int type, char* fname);
H
TD-353  
Hongze Cheng 已提交
594 595 596
int         tsdbLockRepo(STsdbRepo* pRepo);
int         tsdbUnlockRepo(STsdbRepo* pRepo);
char*       tsdbGetDataDirName(char* rootDir);
H
TD-987  
Hongze Cheng 已提交
597
int         tsdbGetNextMaxTables(int tid);
H
TD-353  
Hongze Cheng 已提交
598 599
STsdbMeta*  tsdbGetMeta(TSDB_REPO_T* pRepo);
STsdbFileH* tsdbGetFile(TSDB_REPO_T* pRepo);
H
Hongze Cheng 已提交
600
int         tsdbCheckCommit(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
601

H
TD-1027  
Hongze Cheng 已提交
602 603 604 605 606 607 608 609 610 611
// ------------------ tsdbScan.c
int              tsdbScanFGroup(STsdbScanHandle* pScanHandle, char* rootDir, int fid);
STsdbScanHandle* tsdbNewScanHandle();
void             tsdbSetScanLogStream(STsdbScanHandle* pScanHandle, FILE* fLogStream);
int              tsdbSetAndOpenScanFile(STsdbScanHandle* pScanHandle, char* rootDir, int fid);
int              tsdbScanSCompIdx(STsdbScanHandle* pScanHandle);
int              tsdbScanSCompBlock(STsdbScanHandle* pScanHandle, int idx);
int              tsdbCloseScanFile(STsdbScanHandle* pScanHandle);
void             tsdbFreeScanHandle(STsdbScanHandle* pScanHandle);

H
Hongze Cheng 已提交
612 613 614
// ------------------ tsdbCommitQueue.c
int tsdbScheduleCommit(STsdbRepo *pRepo);

H
more  
hzcheng 已提交
615 616 617 618
#ifdef __cplusplus
}
#endif

F
eok  
freemine 已提交
619
#endif