[TD-10430] add dnode main for dnode module
This commit is contained in:
Shengliang Guan
parent
89005c3b14
commit
7dc1fb79c4
|
|
@ -0,0 +1,39 @@
|
||||||
|
/*
|
||||||
|
* 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_DNODE_STATUS_H_
|
||||||
|
#define _TD_DNODE_STATUS_H_
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
#include "dnodeInt.h"
|
||||||
|
|
||||||
|
typedef struct DnStatus {
|
||||||
|
Dnode * dnode;
|
||||||
|
void * dnodeTimer;
|
||||||
|
void * statusTimer;
|
||||||
|
uint32_t rebootTime;
|
||||||
|
} DnStatus;
|
||||||
|
|
||||||
|
int32_t dnodeInitStatus(Dnode *dnode, DnStatus **status);
|
||||||
|
void dnodeCleanupStatus(DnStatus **status);
|
||||||
|
void dnodeProcessStatusRsp(Dnode *dnode, SRpcMsg *pMsg);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_DNODE_STATUS_H_*/
|
||||||
|
|
@ -0,0 +1,149 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define _DEFAULT_SOURCE
|
||||||
|
#include "os.h"
|
||||||
|
#include "ttimer.h"
|
||||||
|
#include "tglobal.h"
|
||||||
|
#include "dnodeCfg.h"
|
||||||
|
#include "dnodeEps.h"
|
||||||
|
#include "dnodeMnodeEps.h"
|
||||||
|
#include "dnodeStatus.h"
|
||||||
|
#include "dnodeMain.h"
|
||||||
|
#include "vnode.h"
|
||||||
|
|
||||||
|
static void dnodeSendStatusMsg(void *handle, void *tmrId) {
|
||||||
|
DnStatus *status = handle;
|
||||||
|
if (status->dnodeTimer == NULL) {
|
||||||
|
dError("dnode timer is already released");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (status->statusTimer == NULL) {
|
||||||
|
taosTmrReset(dnodeSendStatusMsg, tsStatusInterval * 1000, status, status->dnodeTimer, &status->statusTimer);
|
||||||
|
dError("failed to start status timer");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t contLen = sizeof(SStatusMsg) + TSDB_MAX_VNODES * sizeof(SVnodeLoad);
|
||||||
|
SStatusMsg *pStatus = rpcMallocCont(contLen);
|
||||||
|
if (pStatus == NULL) {
|
||||||
|
taosTmrReset(dnodeSendStatusMsg, tsStatusInterval * 1000, status, status->dnodeTimer, &status->statusTimer);
|
||||||
|
dError("failed to malloc status message");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
Dnode *dnode = status->dnode;
|
||||||
|
dnodeGetCfg(dnode->cfg, &pStatus->dnodeId, pStatus->clusterId);
|
||||||
|
pStatus->dnodeId = htonl(dnodeGetDnodeId(dnode->cfg));
|
||||||
|
pStatus->version = htonl(tsVersion);
|
||||||
|
pStatus->lastReboot = htonl(status->rebootTime);
|
||||||
|
pStatus->numOfCores = htons((uint16_t)tsNumOfCores);
|
||||||
|
pStatus->diskAvailable = tsAvailDataDirGB;
|
||||||
|
pStatus->alternativeRole = tsAlternativeRole;
|
||||||
|
tstrncpy(pStatus->dnodeEp, tsLocalEp, TSDB_EP_LEN);
|
||||||
|
|
||||||
|
// fill cluster cfg parameters
|
||||||
|
pStatus->clusterCfg.numOfMnodes = htonl(tsNumOfMnodes);
|
||||||
|
pStatus->clusterCfg.mnodeEqualVnodeNum = htonl(tsMnodeEqualVnodeNum);
|
||||||
|
pStatus->clusterCfg.offlineThreshold = htonl(tsOfflineThreshold);
|
||||||
|
pStatus->clusterCfg.statusInterval = htonl(tsStatusInterval);
|
||||||
|
pStatus->clusterCfg.maxtablesPerVnode = htonl(tsMaxTablePerVnode);
|
||||||
|
pStatus->clusterCfg.maxVgroupsPerDb = htonl(tsMaxVgroupsPerDb);
|
||||||
|
tstrncpy(pStatus->clusterCfg.arbitrator, tsArbitrator, TSDB_EP_LEN);
|
||||||
|
tstrncpy(pStatus->clusterCfg.timezone, tsTimezone, 64);
|
||||||
|
pStatus->clusterCfg.checkTime = 0;
|
||||||
|
char timestr[32] = "1970-01-01 00:00:00.00";
|
||||||
|
(void)taosParseTime(timestr, &pStatus->clusterCfg.checkTime, (int32_t)strlen(timestr), TSDB_TIME_PRECISION_MILLI, 0);
|
||||||
|
tstrncpy(pStatus->clusterCfg.locale, tsLocale, TSDB_LOCALE_LEN);
|
||||||
|
tstrncpy(pStatus->clusterCfg.charset, tsCharset, TSDB_LOCALE_LEN);
|
||||||
|
|
||||||
|
pStatus->clusterCfg.enableBalance = tsEnableBalance;
|
||||||
|
pStatus->clusterCfg.flowCtrl = tsEnableFlowCtrl;
|
||||||
|
pStatus->clusterCfg.slaveQuery = tsEnableSlaveQuery;
|
||||||
|
pStatus->clusterCfg.adjustMaster = tsEnableAdjustMaster;
|
||||||
|
|
||||||
|
vnodeGetStatus(dnode->vnode, pStatus);
|
||||||
|
contLen = sizeof(SStatusMsg) + pStatus->openVnodes * sizeof(SVnodeLoad);
|
||||||
|
pStatus->openVnodes = htons(pStatus->openVnodes);
|
||||||
|
|
||||||
|
SRpcMsg rpcMsg = {.ahandle = status, .pCont = pStatus, .contLen = contLen, .msgType = TSDB_MSG_TYPE_DM_STATUS};
|
||||||
|
|
||||||
|
dnodeSendMsgToMnode(status->dnode, &rpcMsg);
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeProcessStatusRsp(Dnode *dnode, SRpcMsg *pMsg) {
|
||||||
|
DnStatus *status = pMsg->ahandle;
|
||||||
|
|
||||||
|
if (pMsg->code != TSDB_CODE_SUCCESS) {
|
||||||
|
dError("status rsp is received, error:%s", tstrerror(pMsg->code));
|
||||||
|
if (pMsg->code == TSDB_CODE_MND_DNODE_NOT_EXIST) {
|
||||||
|
char clusterId[TSDB_CLUSTER_ID_LEN];
|
||||||
|
dnodeGetClusterId(dnode->cfg, clusterId);
|
||||||
|
if (clusterId[0] != '\0') {
|
||||||
|
dnodeSetDropped(dnode->cfg);
|
||||||
|
dError("exit zombie dropped dnode");
|
||||||
|
exit(EXIT_FAILURE);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
taosTmrReset(dnodeSendStatusMsg, tsStatusInterval * 1000, status, status->dnodeTimer, &status->statusTimer);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
SStatusRsp *pStatusRsp = pMsg->pCont;
|
||||||
|
SMInfos * minfos = &pStatusRsp->mnodes;
|
||||||
|
dnodeUpdateMnodeFromStatus(dnode->meps, minfos);
|
||||||
|
|
||||||
|
SDnodeCfg *pCfg = &pStatusRsp->dnodeCfg;
|
||||||
|
pCfg->numOfVnodes = htonl(pCfg->numOfVnodes);
|
||||||
|
pCfg->moduleStatus = htonl(pCfg->moduleStatus);
|
||||||
|
pCfg->dnodeId = htonl(pCfg->dnodeId);
|
||||||
|
dnodeUpdateCfg(dnode->cfg, pCfg);
|
||||||
|
|
||||||
|
vnodeSetAccess(dnode->vnode, pStatusRsp->vgAccess, pCfg->numOfVnodes);
|
||||||
|
|
||||||
|
SDnodeEps *pEps = (SDnodeEps *)((char *)pStatusRsp->vgAccess + pCfg->numOfVnodes * sizeof(SVgroupAccess));
|
||||||
|
dnodeUpdateEps(dnode->eps, pEps);
|
||||||
|
|
||||||
|
taosTmrReset(dnodeSendStatusMsg, tsStatusInterval * 1000, status, status->dnodeTimer, &status->statusTimer);
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t dnodeInitStatus(Dnode *dnode, DnStatus **out) {
|
||||||
|
DnStatus *status = calloc(1, sizeof(DnStatus));
|
||||||
|
if (status == NULL) return -1;
|
||||||
|
|
||||||
|
status->dnode = dnode;
|
||||||
|
status->statusTimer = NULL;
|
||||||
|
status->dnodeTimer = dnode->main->dnodeTimer;
|
||||||
|
status->rebootTime = taosGetTimestampSec();
|
||||||
|
taosTmrReset(dnodeSendStatusMsg, 500, status, status->dnodeTimer, &status->statusTimer);
|
||||||
|
*out = status;
|
||||||
|
|
||||||
|
dInfo("dnode status timer is initialized");
|
||||||
|
return TSDB_CODE_SUCCESS;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeCleanupStatus(DnStatus **out) {
|
||||||
|
DnStatus *status = *out;
|
||||||
|
*out = NULL;
|
||||||
|
|
||||||
|
if (status->statusTimer != NULL) {
|
||||||
|
taosTmrStopA(&status->statusTimer);
|
||||||
|
status->statusTimer = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
free(status);
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue