[TD-10430] add dnode main for dnode module

source/server/dnode/inc/dnodeStatus.h

+/*
+ * 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_*/
\ No newline at end of file

source/server/dnode/src/dnodeStatus.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/>.
+ */
+
+#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);
+}