tdbUtil.c 1.5 KB
Newer Older
H
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
Hongze Cheng 已提交
16
#include "tdbInt.h"
H
Hongze Cheng 已提交
17

H
Hongze Cheng 已提交
18
int tdbGnrtFileID(const char *fname, uint8_t *fileid, bool unique) {
19
  int64_t stDev = 0, stIno = 0;
H
Hongze Cheng 已提交
20

21
  if (taosDevInoFile(fname, &stDev, &stIno) < 0) {
H
Hongze Cheng 已提交
22 23 24 25 26
    return -1;
  }

  memset(fileid, 0, TDB_FILE_ID_LEN);

27 28
  ((uint64_t *)fileid)[0] = stDev;
  ((uint64_t *)fileid)[1] = stIno;
H
Hongze Cheng 已提交
29 30 31
  if (unique) {
    ((uint64_t *)fileid)[2] = rand();
  }
H
Hongze Cheng 已提交
32 33 34 35

  return 0;
}

36 37
// int tdbCheckFileAccess(const char *pathname, int mode) {
//   int flags = 0;
H
Hongze Cheng 已提交
38

39 40 41
//   if (mode & TDB_F_OK) {
//     flags |= F_OK;
//   }
H
Hongze Cheng 已提交
42

43 44 45
//   if (mode & TDB_R_OK) {
//     flags |= R_OK;
//   }
H
Hongze Cheng 已提交
46

47 48 49
//   if (mode & TDB_W_OK) {
//     flags |= W_OK;
//   }
H
Hongze Cheng 已提交
50

51 52
//   return access(pathname, flags);
// }
H
Hongze Cheng 已提交
53

H
Hongze Cheng 已提交
54 55
int tdbGetFileSize(const char *fname, pgsz_t pgSize, pgno_t *pSize) {
  int         ret;
56 57
  int64_t file_size = 0;
  ret = taosStatFile(fname, &file_size, NULL);
H
Hongze Cheng 已提交
58 59 60 61
  if (ret != 0) {
    return -1;
  }

62
  ASSERT(file_size % pgSize == 0);
H
Hongze Cheng 已提交
63

64
  *pSize = file_size / pgSize;
H
Hongze Cheng 已提交
65
  return 0;
H
Hongze Cheng 已提交
66
}