tdbUtil.c 1.2 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
  if (unique) {
wafwerar's avatar
wafwerar 已提交
30
    ((uint64_t *)fileid)[2] = taosRand();
H
Hongze Cheng 已提交
31
  }
H
Hongze Cheng 已提交
32

H
Hongze Cheng 已提交
33 34 35 36 37 38 39 40 41 42 43 44 45
  return 0;
}

int tdbGetFileSize(tdb_fd_t fd, int szPage, SPgno *size) {
  int     ret;
  int64_t szBytes;

  ret = tdbOsFileSize(fd, &szBytes);
  if (ret < 0) {
    return -1;
  }

  *size = szBytes / szPage;
H
Hongze Cheng 已提交
46
  return 0;
H
Hongze Cheng 已提交
47
}