117 lines
3.8 KiB
C
117 lines
3.8 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/>.
|
|
*/
|
|
|
|
#ifndef _TD_TFS_INT_H_
|
|
#define _TD_TFS_INT_H_
|
|
|
|
#include "os.h"
|
|
|
|
#include "taosdef.h"
|
|
#include "taoserror.h"
|
|
#include "tcoding.h"
|
|
#include "tfs.h"
|
|
#include "tglobal.h"
|
|
#include "thash.h"
|
|
#include "tlog.h"
|
|
|
|
extern int32_t fsDebugFlag;
|
|
|
|
// For debug purpose
|
|
#define fFatal(...) { if (fsDebugFlag & DEBUG_FATAL) { taosPrintLog("TFS FATAL ", 255, __VA_ARGS__); }}
|
|
#define fError(...) { if (fsDebugFlag & DEBUG_ERROR) { taosPrintLog("TFS ERROR ", 255, __VA_ARGS__); }}
|
|
#define fWarn(...) { if (fsDebugFlag & DEBUG_WARN) { taosPrintLog("TFS WARN ", 255, __VA_ARGS__); }}
|
|
#define fInfo(...) { if (fsDebugFlag & DEBUG_INFO) { taosPrintLog("TFS ", 255, __VA_ARGS__); }}
|
|
#define fDebug(...) { if (fsDebugFlag & DEBUG_DEBUG) { taosPrintLog("TFS ", fsDebugFlag, __VA_ARGS__); }}
|
|
#define fTrace(...) { if (fsDebugFlag & DEBUG_TRACE) { taosPrintLog("TFS ", fsDebugFlag, __VA_ARGS__); }}
|
|
|
|
typedef struct {
|
|
int32_t level;
|
|
int32_t id;
|
|
char *path;
|
|
SDiskSize size;
|
|
} STfsDisk;
|
|
|
|
typedef struct {
|
|
pthread_spinlock_t lock;
|
|
int32_t level;
|
|
int32_t nextid; // next disk id to allocate
|
|
int32_t ndisk; // # of disks mounted to this tier
|
|
int32_t nAvailDisks; // # of Available disks
|
|
STfsDisk *disks[TFS_MAX_DISKS_PER_TIER];
|
|
SDiskSize size;
|
|
} STfsTier;
|
|
|
|
typedef struct {
|
|
STfsDisk *pDisk;
|
|
} SDiskIter;
|
|
|
|
typedef struct STfsDir {
|
|
SDiskIter *iter;
|
|
SDiskID did;
|
|
char dirname[TSDB_FILENAME_LEN];
|
|
STfsFile tfile;
|
|
DIR *dir;
|
|
STfs *pTfs;
|
|
} STfsDir;
|
|
|
|
typedef struct STfs {
|
|
pthread_spinlock_t lock;
|
|
SDiskSize size;
|
|
int32_t nlevel;
|
|
STfsTier tiers[TFS_MAX_TIERS];
|
|
SHashObj *hash; // name to did map
|
|
} STfs;
|
|
|
|
STfsDisk *tfsNewDisk(int32_t level, int32_t id, const char *dir);
|
|
STfsDisk *tfsFreeDisk(STfsDisk *pDisk);
|
|
int32_t tfsUpdateDiskSize(STfsDisk *pDisk);
|
|
|
|
int32_t tfsInitTier(STfsTier *pTier, int32_t level);
|
|
void tfsDestroyTier(STfsTier *pTier);
|
|
STfsDisk *tfsMountDiskToTier(STfsTier *pTier, SDiskCfg *pCfg);
|
|
void tfsUpdateTierSize(STfsTier *pTier);
|
|
int32_t tfsAllocDiskOnTier(STfsTier *pTier);
|
|
void tfsPosNextId(STfsTier *pTier);
|
|
|
|
#define tfsLockTier(pTier) pthread_spin_lock(&(pTier)->lock)
|
|
#define tfsUnLockTier(pTier) pthread_spin_unlock(&(pTier)->lock)
|
|
|
|
#define TMPNAME_LEN (TSDB_FILENAME_LEN * 2 + 32)
|
|
|
|
#define tfsLock(pTfs) pthread_spin_lock(&(pTfs)->lock)
|
|
#define tfsUnLock(pTfs) pthread_spin_unlock(&(pTfs)->lock)
|
|
|
|
#define TFS_TIER_AT(pTfs, level) (&pTfs->tiers[level])
|
|
#define TFS_DISK_AT(pTfs, did) (pTfs->tiers[(did).level].disks[(did).id])
|
|
#define TFS_PRIMARY_DISK(pTfs) (pTfs->tiers[0].disks[0])
|
|
|
|
#define TFS_IS_VALID_LEVEL(level) (((level) >= 0) && ((level) < TFS_NLEVEL()))
|
|
#define TFS_IS_VALID_ID(level, id) (((id) >= 0) && ((id) < TIER_NDISKS(TFS_TIER_AT(level))))
|
|
#define TFS_IS_VALID_DISK(level, id) (TFS_IS_VALID_LEVEL(level) && TFS_IS_VALID_ID(level, id))
|
|
|
|
#define TIER_LEVEL(pt) ((pt)->level)
|
|
#define TIER_NDISKS(pt) ((pt)->ndisk)
|
|
#define TIER_SIZE(pt) ((pt)->tmeta.size)
|
|
#define TIER_FREE_SIZE(pt) ((pt)->tmeta.free)
|
|
|
|
#define DISK_AT_TIER(pt, id) ((pt)->disks[id])
|
|
#define DISK_DIR(pd) ((pd)->path)
|
|
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_TFS_INT_H_*/
|