68 lines
1.6 KiB
C
68 lines
1.6 KiB
C
/*
|
|
* 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"
|
|
|
|
void *tdbRealloc(void *ptr, size_t size) {
|
|
void *nPtr;
|
|
if ((ptr) == NULL || ((int *)(ptr))[-1] < (size)) {
|
|
nPtr = tdbOsRealloc((ptr) ? (char *)(ptr) - sizeof(int) : NULL, (size) + sizeof(int));
|
|
if (nPtr) {
|
|
((int *)nPtr)[0] = (size);
|
|
nPtr = (char *)nPtr + sizeof(int);
|
|
}
|
|
} else {
|
|
nPtr = (ptr);
|
|
}
|
|
return nPtr;
|
|
}
|
|
|
|
void tdbFree(void *p) {
|
|
if (p) {
|
|
tdbOsFree((char *)(p) - sizeof(int));
|
|
}
|
|
}
|
|
|
|
int tdbGnrtFileID(tdb_fd_t fd, uint8_t *fileid, bool unique) {
|
|
int64_t stDev = 0, stIno = 0;
|
|
|
|
if (taosDevInoFile(fd, &stDev, &stIno) < 0) {
|
|
return -1;
|
|
}
|
|
|
|
memset(fileid, 0, TDB_FILE_ID_LEN);
|
|
|
|
((uint64_t *)fileid)[0] = stDev;
|
|
((uint64_t *)fileid)[1] = stIno;
|
|
if (unique) {
|
|
((uint64_t *)fileid)[2] = taosRand();
|
|
}
|
|
|
|
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;
|
|
return 0;
|
|
}
|