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
Hongze Cheng 已提交
29
#include "tfs.h"
H
more  
Hongze Cheng 已提交
30

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

H
hzcheng 已提交
35 36
extern int tsdbDebugFlag;

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

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

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

51 52 53 54
// 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 已提交
55 56
// Definitions
// ------------------ tsdbMeta.c
H
TD-34  
hzcheng 已提交
57
typedef struct STable {
H
Haojun Liao 已提交
58
  STableId       tableId;
H
TD-353  
Hongze Cheng 已提交
59 60 61 62 63
  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 已提交
64
  STSchema*      schema[TSDB_MAX_TABLE_SCHEMAS];
H
TD-353  
Hongze Cheng 已提交
65 66
  STSchema*      tagSchema;
  SKVRow         tagVal;
H
TD-353  
Hongze Cheng 已提交
67
  SSkipList*     pIndex;         // For TSDB_SUPER_TABLE, it is the skiplist index
H
TD-353  
Hongze Cheng 已提交
68 69 70 71 72
  void*          eventHandler;   // TODO
  void*          streamHandler;  // TODO
  TSKEY          lastKey;        // lastkey inserted in this table, initialized as 0, TODO: make a structure
  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
Hongze Cheng 已提交
156 157 158 159 160 161 162 163

// minFid <= midFid <= maxFid
typedef struct {
  int minFid;  // >= minFid && < midFid, at level 2
  int midFid;  // >= midFid && < maxFid, at level 1
  int maxFid;  // >= maxFid, at level 0
} SFidGroup;

H
TD-353  
Hongze Cheng 已提交
164
typedef enum {
H
Hongze Cheng 已提交
165
  TSDB_FILE_TYPE_HEAD = 0,
H
TD-353  
Hongze Cheng 已提交
166 167
  TSDB_FILE_TYPE_DATA,
  TSDB_FILE_TYPE_LAST,
H
Hongze Cheng 已提交
168
  TSDB_FILE_TYPE_STAT,
H
TD-353  
Hongze Cheng 已提交
169
  TSDB_FILE_TYPE_NHEAD,
H
Hongze Cheng 已提交
170 171 172
  TSDB_FILE_TYPE_NDATA,
  TSDB_FILE_TYPE_NLAST,
  TSDB_FILE_TYPE_NSTAT
H
TD-353  
Hongze Cheng 已提交
173
} TSDB_FILE_TYPE;
H
Hongze Cheng 已提交
174

H
Hongze Cheng 已提交
175 176 177 178 179 180
#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 已提交
181
typedef struct {
H
Hongze Cheng 已提交
182
  uint32_t magic;
H
TD-353  
Hongze Cheng 已提交
183 184 185
  uint32_t len;
  uint32_t totalBlocks;
  uint32_t totalSubBlocks;
H
Hongze Cheng 已提交
186
  uint32_t offset;
H
Hongze Cheng 已提交
187 188
  uint64_t size;      // total size of the file
  uint64_t tombSize;  // unused file size
189
} STsdbFileInfo;
H
TD-34  
hzcheng 已提交
190 191

typedef struct {
H
Hongze Cheng 已提交
192 193
  // char  fname[TSDB_FILENAME_LEN];
  // int   fd;
H
TD-353  
Hongze Cheng 已提交
194

H
Hongze Cheng 已提交
195
  STfsFile tfile;
H
TD-353  
Hongze Cheng 已提交
196
  STsdbFileInfo info;
H
hzcheng 已提交
197
} SFile;
H
more  
Hongze Cheng 已提交
198

H
hzcheng 已提交
199
typedef struct {
H
TD-353  
Hongze Cheng 已提交
200
  int   fileId;
H
Hongze Cheng 已提交
201 202 203
  int   state;  // 0 for health, 1 for problem
  int   level;
  int   did;
H
TD-353  
Hongze Cheng 已提交
204
  SFile files[TSDB_FILE_TYPE_MAX];
H
hzcheng 已提交
205 206 207
} SFileGroup;

typedef struct {
H
TD-353  
Hongze Cheng 已提交
208 209
  pthread_rwlock_t fhlock;

H
TD-353  
Hongze Cheng 已提交
210 211 212
  int         maxFGroups;
  int         nFGroups;
  SFileGroup* pFGroup;
H
hzcheng 已提交
213
} STsdbFileH;
H
more  
Hongze Cheng 已提交
214

H
TD-34  
hzcheng 已提交
215 216
typedef struct {
  int         direction;
217 218 219
  STsdbFileH* pFileH;
  int         fileId;
  int         index;
H
TD-34  
hzcheng 已提交
220 221
} SFileGroupIter;

H
TD-353  
Hongze Cheng 已提交
222
// ------------------ tsdbMain.c
H
Hongze Cheng 已提交
223 224 225 226 227 228 229 230 231 232 233 234
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 已提交
235 236 237 238 239 240 241 242 243 244 245 246
typedef struct {
  int8_t state;

  char*           rootDir;
  STsdbCfg        config;
  STsdbAppH       appH;
  STsdbStat       stat;
  STsdbMeta*      tsdbMeta;
  STsdbBufPool*   pPool;
  SMemTable*      mem;
  SMemTable*      imem;
  STsdbFileH*     tsdbFileH;
H
Hongze Cheng 已提交
247
  sem_t           readyToCommit;
H
TD-353  
Hongze Cheng 已提交
248 249
  pthread_mutex_t mutex;
  bool            repoLocked;
H
Hongze Cheng 已提交
250
  int32_t         code; // Commit code
H
TD-353  
Hongze Cheng 已提交
251 252
} STsdbRepo;

H
TD-353  
Hongze Cheng 已提交
253
// ------------------ tsdbRWHelper.c
H
TD-34  
hzcheng 已提交
254
typedef struct {
H
Hongze Cheng 已提交
255
  int32_t  tid;
H
TD-185  
Hongze Cheng 已提交
256 257 258 259
  uint32_t len;
  uint32_t offset;
  uint32_t hasLast : 2;
  uint32_t numOfBlocks : 30;
260 261
  uint64_t uid;
  TSKEY    maxKey;
H
TD-353  
Hongze Cheng 已提交
262 263
} SCompIdx;

264
typedef struct {
H
TD-353  
Hongze Cheng 已提交
265 266 267 268 269
  int64_t last : 1;
  int64_t offset : 63;
  int32_t algorithm : 8;
  int32_t numOfRows : 24;
  int32_t len;
H
Hongze Cheng 已提交
270
  int32_t keyLen;     // key column length, keyOffset = offset+sizeof(SCompData)+sizeof(SCompCol)*numOfCols
H
TD-353  
Hongze Cheng 已提交
271
  int16_t numOfSubBlocks;
H
Hongze Cheng 已提交
272
  int16_t numOfCols; // not including timestamp column
273 274 275 276
  TSKEY   keyFirst;
  TSKEY   keyLast;
} SCompBlock;

H
TD-34  
hzcheng 已提交
277 278
typedef struct {
  int32_t    delimiter;  // For recovery usage
279
  int32_t    tid;
280
  uint64_t   uid;
H
TD-34  
hzcheng 已提交
281 282
  SCompBlock blocks[];
} SCompInfo;
H
hzcheng 已提交
283

H
TD-34  
hzcheng 已提交
284
typedef struct {
H
TD-353  
Hongze Cheng 已提交
285
  int16_t colId;
H
TD-541  
Hongze Cheng 已提交
286
  int32_t len;
H
TD-34  
hzcheng 已提交
287 288
  int32_t type : 8;
  int32_t offset : 24;
H
TD-321  
Hongze Cheng 已提交
289 290 291 292 293 294 295
  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 已提交
296 297 298 299 300
} SCompCol;

typedef struct {
  int32_t  delimiter;  // For recovery usage
  int32_t  numOfCols;  // For recovery usage
301
  uint64_t uid;        // For recovery usage
H
TD-34  
hzcheng 已提交
302 303
  SCompCol cols[];
} SCompData;
H
TD-34  
hzcheng 已提交
304

H
hzcheng 已提交
305 306 307
typedef enum { TSDB_WRITE_HELPER, TSDB_READ_HELPER } tsdb_rw_helper_t;

typedef struct {
H
Hongze Cheng 已提交
308 309 310 311 312
  TSKEY      minKey;
  TSKEY      maxKey;
  SFileGroup fGroup;
  SFile      nHeadF;
  SFile      nLastF;
H
hzcheng 已提交
313 314 315
} SHelperFile;

typedef struct {
316 317
  uint64_t uid;
  int32_t  tid;
H
hzcheng 已提交
318 319
} SHelperTable;

H
Hongze Cheng 已提交
320 321 322 323 324 325
typedef struct {
  SCompIdx* pIdxArray;
  int       numOfIdx;
  int       curIdx;
} SIdxH;

H
hzcheng 已提交
326
typedef struct {
H
TD-353  
Hongze Cheng 已提交
327
  tsdb_rw_helper_t type;
H
TD-100  
hzcheng 已提交
328

H
TD-353  
Hongze Cheng 已提交
329 330
  STsdbRepo* pRepo;
  int8_t     state;
H
TD-100  
hzcheng 已提交
331
  // For file set usage
H
hzcheng 已提交
332
  SHelperFile files;
H
Hongze Cheng 已提交
333 334 335
  SIdxH       idxH;
  SCompIdx    curCompIdx;
  void*       pWIdx;
H
TD-100  
hzcheng 已提交
336
  // For table set usage
H
hzcheng 已提交
337
  SHelperTable tableInfo;
H
TD-353  
Hongze Cheng 已提交
338
  SCompInfo*   pCompInfo;
H
TD-100  
hzcheng 已提交
339 340
  bool         hasOldLastBlock;
  // For block set usage
H
TD-353  
Hongze Cheng 已提交
341
  SCompData* pCompData;
H
TD-353  
Hongze Cheng 已提交
342 343 344
  SDataCols* pDataCols[2];
  void*      pBuffer;     // Buffer to hold the whole data block
  void*      compBuffer;  // Buffer for temperary compress/decompress purpose
H
hzcheng 已提交
345 346
} SRWHelper;

H
TD-1548  
Hongze Cheng 已提交
347 348 349 350 351 352 353 354 355
typedef struct {
  int   rowsInserted;
  int   rowsUpdated;
  int   rowsDeleteSucceed;
  int   rowsDeleteFailed;
  int   nOperations;
  TSKEY keyFirst;
  TSKEY keyLast;
} SMergeInfo;
H
TD-1027  
Hongze Cheng 已提交
356 357 358 359 360 361 362 363 364 365
// ------------------ tsdbScan.c
typedef struct {
  SFileGroup fGroup;
  int        numOfIdx;
  SCompIdx*  pCompIdx;
  SCompInfo* pCompInfo;
  void*      pBuf;
  FILE*      tLogStream;
} STsdbScanHandle;

H
TD-353  
Hongze Cheng 已提交
366 367
// Operations
// ------------------ tsdbMeta.c
H
TD-987  
Hongze Cheng 已提交
368
#define TSDB_INIT_NTABLES 1024
H
TD-353  
Hongze Cheng 已提交
369 370 371
#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 已提交
372
#define TABLE_UID(t) (t)->tableId.uid
H
TD-353  
Hongze Cheng 已提交
373
#define TABLE_TID(t) (t)->tableId.tid
H
TD-353  
Hongze Cheng 已提交
374
#define TABLE_SUID(t) (t)->suid
H
TD-353  
Hongze Cheng 已提交
375
#define TABLE_LASTKEY(t) (t)->lastKey
376
#define TSDB_META_FILE_MAGIC(m) KVSTORE_MAGIC((m)->pStore)
H
TD-353  
Hongze Cheng 已提交
377 378 379

STsdbMeta* tsdbNewMeta(STsdbCfg* pCfg);
void       tsdbFreeMeta(STsdbMeta* pMeta);
H
TD-353  
Hongze Cheng 已提交
380 381 382 383 384 385 386
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 已提交
387 388
void       tsdbRefTable(STable* pTable);
void       tsdbUnRefTable(STable* pTable);
H
Hongze Cheng 已提交
389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430
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;

  if (lock) taosRLockLatch(&(pDTable->latch));
  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:
  if (lock) taosRUnLockLatch(&(pDTable->latch));
  return pSchema;
}
H
TD-353  
Hongze Cheng 已提交
431

H
Hongze Cheng 已提交
432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447
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
TD-353  
Hongze Cheng 已提交
448
// ------------------ tsdbBuffer.c
H
Hongze Cheng 已提交
449 450
#define TSDB_BUFFER_RESERVE 1024  // Reseve 1K as commit threshold

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

// ------------------ tsdbMemTable.c
H
TD-353  
Hongze Cheng 已提交
458 459 460
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 已提交
461
void  tsdbUnTakeMemSnapShot(STsdbRepo* pRepo, SMemTable* pMem, SMemTable* pIMem);
H
TD-353  
Hongze Cheng 已提交
462
void* tsdbAllocBytes(STsdbRepo* pRepo, int bytes);
H
TD-353  
Hongze Cheng 已提交
463
int   tsdbAsyncCommit(STsdbRepo* pRepo);
H
Hongze Cheng 已提交
464
int   tsdbLoadDataFromCache(STable* pTable, SSkipListIterator* pIter, TSKEY maxKey, int maxRowsToRead, SDataCols* pCols,
H
TD-1548  
Hongze Cheng 已提交
465
                            TKEY* filterKeys, int nFilterKeys, bool keepDup, SMergeInfo* pMergeInfo);
H
Hongze Cheng 已提交
466
void* tsdbCommitData(STsdbRepo* pRepo);
H
Hongze Cheng 已提交
467 468 469 470 471 472 473

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 已提交
474
  return (SDataRow)SL_GET_NODE_DATA(node);
H
Hongze Cheng 已提交
475 476 477 478
}

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

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

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

  return dataRowTKey(row);
}

H
Hongze Cheng 已提交
491 492 493 494 495 496 497 498 499 500 501 502 503
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 已提交
504 505 506 507 508
// ------------------ 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 已提交
509
#define TSDB_IS_FILE_OPENED(f) ((f)->fd > 0)
H
TD-353  
Hongze Cheng 已提交
510 511 512
#define TSDB_FGROUP_ITER_FORWARD TSDB_ORDER_ASC
#define TSDB_FGROUP_ITER_BACKWARD TSDB_ORDER_DESC

H
TD-353  
Hongze Cheng 已提交
513 514
STsdbFileH* tsdbNewFileH(STsdbCfg* pCfg);
void        tsdbFreeFileH(STsdbFileH* pFileH);
H
TD-353  
Hongze Cheng 已提交
515
int         tsdbOpenFileH(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
516
void        tsdbCloseFileH(STsdbRepo* pRepo);
H
Hongze Cheng 已提交
517
SFileGroup* tsdbCreateFGroup(STsdbRepo* pRepo, int fid);
H
TD-353  
Hongze Cheng 已提交
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);
H
Hongze Cheng 已提交
523
int         tsdbCreateFile(SFile* pFile, STsdbRepo* pRepo, int fid, int type, SDisk* pDisk);
H
TD-353  
Hongze Cheng 已提交
524
SFileGroup* tsdbSearchFGroup(STsdbFileH* pFileH, int fid, int flags);
H
Hongze Cheng 已提交
525
void        tsdbRemoveFilesBeyondRetention(STsdbRepo* pRepo, SFidGroup* pFidGroup);
526
int         tsdbUpdateFileHeader(SFile* pFile);
H
TD-353  
Hongze Cheng 已提交
527
int         tsdbEncodeSFileInfo(void** buf, const STsdbFileInfo* pInfo);
H
TD-353  
Hongze Cheng 已提交
528
void*       tsdbDecodeSFileInfo(void* buf, STsdbFileInfo* pInfo);
H
TD-353  
Hongze Cheng 已提交
529
void        tsdbRemoveFileGroup(STsdbRepo* pRepo, SFileGroup* pFGroup);
H
Hongze Cheng 已提交
530
int         tsdbLoadFileHeader(SFile* pFile, uint32_t* version);
531
void        tsdbGetFileInfoImpl(char* fname, uint32_t* magic, int64_t* size);
H
Hongze Cheng 已提交
532
void        tsdbGetFidGroup(STsdbCfg* pCfg, SFidGroup* pFidGroup);
533
void        tsdbGetFidKeyRange(int daysPerFile, int8_t precision, int fileId, TSKEY *minKey, TSKEY *maxKey);
H
Hongze Cheng 已提交
534
int         tsdbGetBaseDirFromFile(char* fname, char* baseDir);
H
Hongze Cheng 已提交
535
int         tsdbApplyRetention(STsdbRepo* pRepo, SFidGroup *pFidGroup);
H
TD-353  
Hongze Cheng 已提交
536 537

// ------------------ tsdbRWHelper.c
H
TD-353  
Hongze Cheng 已提交
538 539 540 541 542 543
#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 已提交
544 545 546 547
#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 已提交
548 549
#define TSDB_MAX_SUBBLOCKS 8
#define IS_SUB_BLOCK(pBlock) ((pBlock)->numOfSubBlocks == 0)
H
TD-353  
Hongze Cheng 已提交
550 551 552
#define helperType(h) (h)->type
#define helperRepo(h) (h)->pRepo
#define helperState(h) (h)->state
H
Hongze Cheng 已提交
553
#define TSDB_NLAST_FILE_OPENED(h) ((h)->files.nLastF.fd > 0)
H
Hongze Cheng 已提交
554 555 556 557 558 559
#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 已提交
560 561 562 563 564 565

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 已提交
566
int  tsdbCloseHelperFile(SRWHelper* pHelper, bool hasError, SFileGroup* pGroup);
H
Hongze Cheng 已提交
567
int  tsdbSetHelperTable(SRWHelper* pHelper, STable* pTable, STsdbRepo* pRepo);
H
Hongze Cheng 已提交
568 569 570 571
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 已提交
572 573
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 已提交
574
int  tsdbLoadCompIdx(SRWHelper* pHelper, void* target);
H
Hongze Cheng 已提交
575
int  tsdbLoadCompInfoImpl(SFile* pFile, SCompIdx* pIdx, SCompInfo** ppCompInfo);
H
Hongze Cheng 已提交
576 577 578 579 580 581 582 583 584 585 586 587 588 589 590 591
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 已提交
592

H
TD-353  
Hongze Cheng 已提交
593 594
// ------------------ tsdbMain.c
#define REPO_ID(r) (r)->config.tsdbId
H
TD-353  
Hongze Cheng 已提交
595
#define IS_REPO_LOCKED(r) (r)->repoLocked
H
TD-353  
Hongze Cheng 已提交
596
#define TSDB_SUBMIT_MSG_HEAD_SIZE sizeof(SSubmitMsg)
H
TD-353  
Hongze Cheng 已提交
597

H
TD-353  
Hongze Cheng 已提交
598
char*       tsdbGetMetaFileName(char* rootDir);
H
Hongze Cheng 已提交
599
void        tsdbGetDataFileName(char* rootDir, int vid, int fid, int type, char* fname);
H
TD-353  
Hongze Cheng 已提交
600 601 602
int         tsdbLockRepo(STsdbRepo* pRepo);
int         tsdbUnlockRepo(STsdbRepo* pRepo);
char*       tsdbGetDataDirName(char* rootDir);
H
TD-987  
Hongze Cheng 已提交
603
int         tsdbGetNextMaxTables(int tid);
H
TD-353  
Hongze Cheng 已提交
604 605
STsdbMeta*  tsdbGetMeta(TSDB_REPO_T* pRepo);
STsdbFileH* tsdbGetFile(TSDB_REPO_T* pRepo);
H
Hongze Cheng 已提交
606
int         tsdbCheckCommit(STsdbRepo* pRepo);
H
TD-353  
Hongze Cheng 已提交
607

H
TD-1027  
Hongze Cheng 已提交
608 609 610 611 612 613 614 615 616 617
// ------------------ 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 已提交
618 619 620
// ------------------ tsdbCommitQueue.c
int tsdbScheduleCommit(STsdbRepo *pRepo);

H
more  
hzcheng 已提交
621 622 623 624
#ifdef __cplusplus
}
#endif

H
TD-34  
hzcheng 已提交
625
#endif