240 lines
8.4 KiB
C
240 lines
8.4 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_DM_INT_H_
|
|
#define _TD_DM_INT_H_
|
|
|
|
#include "cJSON.h"
|
|
#include "tcache.h"
|
|
#include "tcrc32c.h"
|
|
#include "tdatablock.h"
|
|
#include "tglobal.h"
|
|
#include "thash.h"
|
|
#include "tlockfree.h"
|
|
#include "tlog.h"
|
|
#include "tmsg.h"
|
|
#include "tmsgcb.h"
|
|
#include "tqueue.h"
|
|
#include "trpc.h"
|
|
#include "tthread.h"
|
|
#include "ttime.h"
|
|
#include "tworker.h"
|
|
|
|
#include "dnode.h"
|
|
#include "mnode.h"
|
|
#include "monitor.h"
|
|
#include "qnode.h"
|
|
#include "sync.h"
|
|
#include "tfs.h"
|
|
#include "wal.h"
|
|
|
|
#ifdef __cplusplus
|
|
extern "C" {
|
|
#endif
|
|
|
|
// clang-format off
|
|
|
|
#define dFatal(...) { if (dDebugFlag & DEBUG_FATAL) { taosPrintLog("DND FATAL ", DEBUG_FATAL, 255, __VA_ARGS__); }}
|
|
#define dError(...) { if (dDebugFlag & DEBUG_ERROR) { taosPrintLog("DND ERROR ", DEBUG_ERROR, 255, __VA_ARGS__); }}
|
|
#define dWarn(...) { if (dDebugFlag & DEBUG_WARN) { taosPrintLog("DND WARN ", DEBUG_WARN, 255, __VA_ARGS__); }}
|
|
#define dInfo(...) { if (dDebugFlag & DEBUG_INFO) { taosPrintLog("DND ", DEBUG_INFO, 255, __VA_ARGS__); }}
|
|
#define dDebug(...) { if (dDebugFlag & DEBUG_DEBUG) { taosPrintLog("DND ", DEBUG_DEBUG, dDebugFlag, __VA_ARGS__); }}
|
|
#define dTrace(...) { if (dDebugFlag & DEBUG_TRACE) { taosPrintLog("DND ", DEBUG_TRACE, dDebugFlag, __VA_ARGS__); }}
|
|
|
|
#define encryptDebug(...) { \
|
|
if (toLogFile) { \
|
|
if (dDebugFlag & DEBUG_DEBUG) {taosPrintLog("DND ", DEBUG_DEBUG, dDebugFlag, __VA_ARGS__);} \
|
|
} else { \
|
|
/*if (dDebugFlag & DEBUG_DEBUG) {taosPrintLog("DND ", DEBUG_SCREEN, dDebugFlag, __VA_ARGS__);}*/ \
|
|
if (dDebugFlag & DEBUG_DEBUG) {printf(__VA_ARGS__); printf("\n");} \
|
|
} \
|
|
}
|
|
|
|
#define encryptInfo(...) { \
|
|
if (toLogFile) { \
|
|
taosPrintLog("DND ", DEBUG_INFO, 255, __VA_ARGS__); \
|
|
} else { \
|
|
/*if (dDebugFlag & DEBUG_DEBUG) {taosPrintLog("DND ", DEBUG_SCREEN, dDebugFlag, __VA_ARGS__);}*/ \
|
|
printf(__VA_ARGS__); \
|
|
printf("\n"); \
|
|
} \
|
|
}
|
|
|
|
#define encryptError(...) { \
|
|
if (toLogFile) { \
|
|
taosPrintLog("DND ERROR ", DEBUG_ERROR, 255, __VA_ARGS__); \
|
|
}\
|
|
else{ \
|
|
/*taosPrintLog("DND ", DEBUG_SCREEN, 255, __VA_ARGS__); */\
|
|
printf("ERROR: " __VA_ARGS__); \
|
|
printf("\n"); \
|
|
}\
|
|
}
|
|
|
|
#define dGFatal(param, ...) {if (dDebugFlag & DEBUG_FATAL) { char buf[40] = {0}; TRACE_TO_STR(trace, buf); dFatal(param ",QID:%s", __VA_ARGS__, buf);}}
|
|
#define dGError(param, ...) {if (dDebugFlag & DEBUG_ERROR) { char buf[40] = {0}; TRACE_TO_STR(trace, buf); dError(param ",QID:%s", __VA_ARGS__, buf);}}
|
|
#define dGWarn(param, ...) {if (dDebugFlag & DEBUG_WARN) { char buf[40] = {0}; TRACE_TO_STR(trace, buf); dWarn(param ",QID:%s", __VA_ARGS__, buf);}}
|
|
#define dGInfo(param, ...) {if (dDebugFlag & DEBUG_INFO) { char buf[40] = {0}; TRACE_TO_STR(trace, buf); dInfo(param ",QID:%s", __VA_ARGS__, buf);}}
|
|
#define dGDebug(param, ...) {if (dDebugFlag & DEBUG_DEBUG) { char buf[40] = {0}; TRACE_TO_STR(trace, buf); dDebug(param ",QID:%s", __VA_ARGS__, buf);}}
|
|
#define dGTrace(param, ...) {if (dDebugFlag & DEBUG_TRACE) { char buf[40] = {0}; TRACE_TO_STR(trace, buf); dTrace(param ",QID:%s", __VA_ARGS__, buf);}}
|
|
|
|
// clang-format on
|
|
|
|
typedef enum {
|
|
DNODE = 0,
|
|
MNODE = 1,
|
|
VNODE = 2,
|
|
QNODE = 3,
|
|
SNODE = 4,
|
|
NODE_END = 5,
|
|
} EDndNodeType;
|
|
|
|
typedef enum {
|
|
DND_STAT_INIT,
|
|
DND_STAT_RUNNING,
|
|
DND_STAT_STOPPED,
|
|
} EDndRunStatus;
|
|
|
|
typedef enum {
|
|
DND_ENV_INIT,
|
|
DND_ENV_READY,
|
|
DND_ENV_CLEANUP,
|
|
} EDndEnvStatus;
|
|
|
|
typedef int32_t (*ProcessCreateNodeFp)(EDndNodeType ntype, SRpcMsg *pMsg);
|
|
typedef int32_t (*ProcessDropNodeFp)(EDndNodeType ntype, SRpcMsg *pMsg);
|
|
typedef void (*SendMonitorReportFp)();
|
|
typedef void (*MonitorCleanExpiredSamplesFp)();
|
|
typedef void (*SendAuditRecordsFp)();
|
|
typedef void (*GetVnodeLoadsFp)(SMonVloadInfo *pInfo);
|
|
typedef void (*GetMnodeLoadsFp)(SMonMloadInfo *pInfo);
|
|
typedef void (*GetQnodeLoadsFp)(SQnodeLoad *pInfo);
|
|
typedef int32_t (*ProcessAlterNodeTypeFp)(EDndNodeType ntype, SRpcMsg *pMsg);
|
|
typedef void (*StopDnodeFp)();
|
|
|
|
typedef struct {
|
|
int32_t dnodeId;
|
|
int32_t engineVer;
|
|
int64_t clusterId;
|
|
int64_t dnodeVer;
|
|
int64_t updateTime;
|
|
int64_t rebootTime;
|
|
bool dropped;
|
|
bool stopped;
|
|
SEpSet mnodeEps;
|
|
SArray *dnodeEps;
|
|
SArray *oldDnodeEps;
|
|
SHashObj *dnodeHash;
|
|
TdThreadRwlock lock;
|
|
SMsgCb msgCb;
|
|
bool validMnodeEps;
|
|
int64_t ipWhiteVer;
|
|
char machineId[TSDB_MACHINE_ID_LEN + 1];
|
|
EEncryptAlgor encryptAlgorigthm;
|
|
EEncryptScope encryptScope;
|
|
TdThreadMutex statusInfolock;
|
|
} SDnodeData;
|
|
|
|
typedef struct {
|
|
const char *path;
|
|
const char *name;
|
|
STfs *pTfs;
|
|
SDnodeData *pData;
|
|
SMsgCb msgCb;
|
|
ProcessCreateNodeFp processCreateNodeFp;
|
|
ProcessAlterNodeTypeFp processAlterNodeTypeFp;
|
|
ProcessDropNodeFp processDropNodeFp;
|
|
SendMonitorReportFp sendMonitorReportFp;
|
|
MonitorCleanExpiredSamplesFp monitorCleanExpiredSamplesFp;
|
|
SendAuditRecordsFp sendAuditRecordFp;
|
|
GetVnodeLoadsFp getVnodeLoadsFp;
|
|
GetVnodeLoadsFp getVnodeLoadsLiteFp;
|
|
GetMnodeLoadsFp getMnodeLoadsFp;
|
|
GetQnodeLoadsFp getQnodeLoadsFp;
|
|
StopDnodeFp stopDnodeFp;
|
|
} SMgmtInputOpt;
|
|
|
|
typedef struct {
|
|
void *pMgmt;
|
|
} SMgmtOutputOpt;
|
|
|
|
typedef int32_t (*NodeMsgFp)(void *pMgmt, SRpcMsg *pMsg);
|
|
typedef int32_t (*NodeOpenFp)(SMgmtInputOpt *pInput, SMgmtOutputOpt *pOutput);
|
|
typedef void (*NodeCloseFp)(void *pMgmt);
|
|
typedef int32_t (*NodeStartFp)(void *pMgmt);
|
|
typedef void (*NodeStopFp)(void *pMgmt);
|
|
typedef int32_t (*NodeCreateFp)(const SMgmtInputOpt *pInput, SRpcMsg *pMsg);
|
|
typedef int32_t (*NodeDropFp)(const SMgmtInputOpt *pInput, SRpcMsg *pMsg);
|
|
typedef int32_t (*NodeRequireFp)(const SMgmtInputOpt *pInput, bool *required);
|
|
typedef SArray *(*NodeGetHandlesFp)(); // array of SMgmtHandle
|
|
typedef bool (*NodeIsCatchUpFp)(void *pMgmt);
|
|
typedef bool (*NodeRole)(void *pMgmt);
|
|
|
|
typedef struct {
|
|
NodeOpenFp openFp;
|
|
NodeCloseFp closeFp;
|
|
NodeStartFp startFp;
|
|
NodeStopFp stopFp;
|
|
NodeCreateFp createFp;
|
|
NodeDropFp dropFp;
|
|
NodeRequireFp requiredFp;
|
|
NodeGetHandlesFp getHandlesFp;
|
|
NodeIsCatchUpFp isCatchUpFp;
|
|
NodeRole nodeRoleFp;
|
|
} SMgmtFunc;
|
|
|
|
typedef struct {
|
|
tmsg_t msgType;
|
|
bool needCheckVgId;
|
|
NodeMsgFp msgFp;
|
|
} SMgmtHandle;
|
|
|
|
// dmUtil.c
|
|
const char *dmStatStr(EDndRunStatus stype);
|
|
const char *dmNodeName(EDndNodeType ntype);
|
|
void *dmSetMgmtHandle(SArray *pArray, tmsg_t msgType, void *nodeMsgFp, bool needCheckVgId);
|
|
void dmGetMonitorSystemInfo(SMonSysInfo *pInfo);
|
|
|
|
// dmFile.c
|
|
int32_t dmReadFile(const char *path, const char *name, bool *pDeployed);
|
|
int32_t dmWriteFile(const char *path, const char *name, bool deployed);
|
|
int32_t dmCheckRunning(const char *dataDir, TdFilePtr *pFile);
|
|
// int32_t dmCheckRunningWrapper(const char *dataDir, TdFilePtr *pFile);
|
|
|
|
// dmodule.c
|
|
int32_t dmInitDndInfo(SDnodeData *pData);
|
|
|
|
// dmEps.c
|
|
int32_t dmGetDnodeSize(SDnodeData *pData);
|
|
int32_t dmGetDnodeId(SDnodeData *pData);
|
|
int32_t dmReadEps(SDnodeData *pData);
|
|
int32_t dmWriteEps(SDnodeData *pData);
|
|
void dmUpdateEps(SDnodeData *pData, SArray *pDnodeEps);
|
|
void dmGetMnodeEpSet(SDnodeData *pData, SEpSet *pEpSet);
|
|
void dmEpSetToStr(char *buf, int32_t len, SEpSet *epSet);
|
|
void dmRotateMnodeEpSet(SDnodeData *pData);
|
|
void dmGetMnodeEpSetForRedirect(SDnodeData *pData, SRpcMsg *pMsg, SEpSet *pEpSet);
|
|
void dmSetMnodeEpSet(SDnodeData *pData, SEpSet *pEpSet);
|
|
bool dmUpdateDnodeInfo(void *pData, int32_t *dnodeId, int64_t *clusterId, char *fqdn, uint16_t *port);
|
|
void dmRemoveDnodePairs(SDnodeData *pData);
|
|
void dmGetDnodeEp(void *pData, int32_t dnodeId, char *pEp, char *pFqdn, uint16_t *pPort);
|
|
int32_t dmUpdateEncryptKey(char *key, bool toLogFile);
|
|
int32_t dmGetEncryptKey();
|
|
#ifdef __cplusplus
|
|
}
|
|
#endif
|
|
|
|
#endif /*_TD_DM_INT_H_*/
|