tdbEnv.c 1.7 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17
/*
 * 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/>.
 */

#include "tdbInt.h"

H
Hongze Cheng 已提交
18 19
struct STEnv {
  char *   rootDir;
H
more  
Hongze Cheng 已提交
20
  char *   jfname;
H
Hongze Cheng 已提交
21
  int      jfd;
H
Hongze Cheng 已提交
22
  SPCache *pCache;
H
Hongze Cheng 已提交
23 24
};

H
Hongze Cheng 已提交
25
int tdbEnvOpen(const char *rootDir, int pageSize, int cacheSize, STEnv **ppEnv) {
H
more  
Hongze Cheng 已提交
26 27 28 29 30
  STEnv *pEnv;
  int    dsize;
  int    zsize;
  u8 *   pPtr;
  int    ret;
H
Hongze Cheng 已提交
31 32 33 34

  *ppEnv = NULL;

  dsize = strlen(rootDir);
H
more  
Hongze Cheng 已提交
35
  zsize = sizeof(*pEnv) + dsize * 2 + strlen(TDB_JOURNAL_NAME) + 3;
H
Hongze Cheng 已提交
36 37 38 39 40 41 42 43

  pPtr = (uint8_t *)calloc(1, zsize);
  if (pPtr == NULL) {
    return -1;
  }

  pEnv = (STEnv *)pPtr;
  pPtr += sizeof(*pEnv);
H
more  
Hongze Cheng 已提交
44
  // pEnv->rootDir
H
Hongze Cheng 已提交
45 46 47
  pEnv->rootDir = pPtr;
  memcpy(pEnv->rootDir, rootDir, dsize);
  pEnv->rootDir[dsize] = '\0';
H
more  
Hongze Cheng 已提交
48 49 50 51 52 53 54
  pPtr = pPtr + dsize + 1;
  // pEnv->jfname
  pEnv->jfname = pPtr;
  memcpy(pEnv->jfname, rootDir, dsize);
  pEnv->jfname[dsize] = '/';
  memcpy(pEnv->jfname + dsize + 1, TDB_JOURNAL_NAME, strlen(TDB_JOURNAL_NAME));
  pEnv->jfname[dsize + 1 + strlen(TDB_JOURNAL_NAME)] = '\0';
H
Hongze Cheng 已提交
55

H
Hongze Cheng 已提交
56 57
  pEnv->jfd = -1;

H
more  
Hongze Cheng 已提交
58 59 60 61 62 63
  ret = tdbPCacheOpen(pageSize, cacheSize, 0, &(pEnv->pCache));
  if (ret < 0) {
    return -1;
  }

  mkdir(rootDir, 0755);
H
Hongze Cheng 已提交
64

H
Hongze Cheng 已提交
65 66 67 68 69 70 71
  *ppEnv = pEnv;
  return 0;
}

int tdbEnvClose(STEnv *pEnv) {
  // TODO
  return 0;
H
Hongze Cheng 已提交
72
}