tdbOs.h 2.6 KB
Newer Older
H
Hongze Cheng 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
/*
 * 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/>.
 */

#ifndef _TDB_OS_H_
#define _TDB_OS_H_

#ifdef __cplusplus
extern "C" {
#endif

// TODO: kmake
#define TDB_FOR_TDENGINE

// For memor
#ifdef TDB_FOR_TDENGINE
#define tdbOsMalloc  taosMemoryMalloc
#define tdbOsCalloc  taosMemoryCalloc
#define tdbOsRealloc taosMemoryRealloc
#define tdbOsFree    taosMemoryFree
#else
#define tdbOsMalloc  malloc
#define tdbOsCalloc  calloc
#define tdbOsRealloc realloc
#define tdbOsFree    free
#endif

// For file
#ifdef TDB_FOR_TDENGINE
typedef TdFilePtr tdb_fd_t;

#define tdbOsOpen  taosOpenFile
#define tdbOsClose taosCloseFile
#define tdbOsRead  taosReadFile
#define tdbOsPRead taosPReadFile
#define tdbOsWrite taosWriteFile
H
Hongze Cheng 已提交
48
#define tdbOsFSync taosFsyncFile
H
Hongze Cheng 已提交
49 50 51
#else
#define tdbOsOpen  open
#define tdbOsClose close
H
Hongze Cheng 已提交
52 53 54
#define tdbOsRead  read   // TODO
#define tdbOsPRead pread  // TODO
#define tdbOsWrite write  // TODO
H
Hongze Cheng 已提交
55
#define tdbOsFSync fsync
H
Hongze Cheng 已提交
56 57 58 59 60 61 62 63 64 65 66 67
#endif

// For threads and lock
#ifdef TDB_FOR_TDENGINE

// spin lock
typedef TdThreadSpinlock tdb_spinlock_t;

#define tdbSpinlockInit    taosThreadSpinInit
#define tdbSpinlockDestroy taosThreadSpinDestroy
#define tdbSpinlockLock    taosThreadSpinLock
#define tdbSpinlockUnlock  taosThreadSpinUnlock
H
Hongze Cheng 已提交
68
#define tdbSpinlockTrylock pthread_spin_trylock  // TODO
H
Hongze Cheng 已提交
69

H
Hongze Cheng 已提交
70 71 72 73 74 75 76 77
// mutex lock
typedef TdThreadMutex tdb_mutex_t;

#define tdbMutexInit    taosThreadMutexInit
#define tdbMutexDestroy taosThreadMutexDestroy
#define tdbMutexLock    taosThreadMutexLock
#define tdbMutexUnlock  taosThreadMutexUnlock

H
Hongze Cheng 已提交
78 79 80 81 82 83 84 85 86 87 88
#else

// spin lock
typedef pthread_spinlock_t tdb_spinlock_t;

#define tdbSpinlockInit    pthread_spin_init
#define tdbSpinlockDestroy pthread_spin_destroy
#define tdbSpinlockLock    pthread_spin_lock
#define tdbSpinlockUnlock  pthread_spin_unlock
#define tdbSpinlockTrylock pthread_spin_trylock

H
Hongze Cheng 已提交
89 90 91 92 93 94 95 96
// mutex lock
typedef pthread_mutex_t tdb_mutex_t;

#define tdbMutexInit    pthread_mutex_init
#define tdbMutexDestroy pthread_mutex_destroy
#define tdbMutexLock    pthread_mutex_lock
#define tdbMutexUnlock  pthread_mutex_unlock

H
Hongze Cheng 已提交
97 98 99 100 101 102 103
#endif

#ifdef __cplusplus
}
#endif

#endif /*_TDB_OS_H_*/