refact file directory
This commit is contained in:
Shengliang Guan
parent
f528d5a13d
commit
401cf95662
|
|
@ -13,8 +13,8 @@
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef _TD_DNODE_EPS_H_
|
#ifndef _TD_DNODE_CONFIG_H_
|
||||||
#define _TD_DNODE_EPS_H_
|
#define _TD_DNODE_CONFIG_H_
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
|
|
@ -39,4 +39,4 @@ void dnodeSendRedirectMsg(SRpcMsg *rpcMsg, bool forShell);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif /*_TD_DNODE_EPS_H_*/
|
#endif /*_TD_DNODE_CONFIG_H_*/
|
||||||
|
|
@ -0,0 +1,34 @@
|
||||||
|
/*
|
||||||
|
* 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_DNODE_H_
|
||||||
|
#define _TD_DNODE_DNODE_H_
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
#include "dnodeInt.h"
|
||||||
|
|
||||||
|
int32_t dnodeInitMsg();
|
||||||
|
void dnodeCleanupMsg();
|
||||||
|
void dnodeProcessStatusRsp(SRpcMsg *pMsg);
|
||||||
|
void dnodeProcessStartupReq(SRpcMsg *pMsg);
|
||||||
|
void dnodeProcessConfigDnodeReq(SRpcMsg *pMsg);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_DNODE_DNODE_H_*/
|
||||||
|
|
@ -0,0 +1,30 @@
|
||||||
|
/*
|
||||||
|
* 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_MNODE_H_
|
||||||
|
#define _TD_DNODE_MNODE_H_
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
extern "C" {
|
||||||
|
#endif
|
||||||
|
#include "dnodeInt.h"
|
||||||
|
|
||||||
|
void dnodeProcessCreateMnodeReq(SRpcMsg *pMsg);
|
||||||
|
|
||||||
|
#ifdef __cplusplus
|
||||||
|
}
|
||||||
|
#endif
|
||||||
|
|
||||||
|
#endif /*_TD_DNODE_MNODE_H_*/
|
||||||
|
|
@ -13,8 +13,8 @@
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef _TD_DNODE_TRANS_H_
|
#ifndef _TD_DNODE_TRANSPORT_H_
|
||||||
#define _TD_DNODE_TRANS_H_
|
#define _TD_DNODE_TRANSPORT_H_
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
|
|
@ -30,4 +30,4 @@ void dnodeSendMsgToDnode(SRpcEpSet *epSet, SRpcMsg *rpcMsg);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif /*_TD_DNODE_TRANS_H_*/
|
#endif /*_TD_DNODE_TRANSPORT_H_*/
|
||||||
|
|
@ -13,8 +13,8 @@
|
||||||
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
* along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#ifndef _TD_DNODE_STATUS_H_
|
#ifndef _TD_DNODE_VNODES_H_
|
||||||
#define _TD_DNODE_STATUS_H_
|
#define _TD_DNODE_VNODES_H_
|
||||||
|
|
||||||
#ifdef __cplusplus
|
#ifdef __cplusplus
|
||||||
extern "C" {
|
extern "C" {
|
||||||
|
|
@ -32,4 +32,4 @@ void dnodeProcessConfigDnodeReq(SRpcMsg *pMsg);
|
||||||
}
|
}
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#endif /*_TD_DNODE_STATUS_H_*/
|
#endif /*_TD_DNODE_VNODES_H_*/
|
||||||
|
|
@ -14,7 +14,7 @@
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#define _DEFAULT_SOURCE
|
#define _DEFAULT_SOURCE
|
||||||
#include "dnodeEps.h"
|
#include "dnodeConfig.h"
|
||||||
#include "cJSON.h"
|
#include "cJSON.h"
|
||||||
#include "thash.h"
|
#include "thash.h"
|
||||||
|
|
||||||
|
|
@ -0,0 +1,174 @@
|
||||||
|
/*
|
||||||
|
* 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 "dnodeDnode.h"
|
||||||
|
#include "dnodeConfig.h"
|
||||||
|
#include "mnode.h"
|
||||||
|
#include "tthread.h"
|
||||||
|
#include "ttime.h"
|
||||||
|
#include "vnode.h"
|
||||||
|
|
||||||
|
static struct {
|
||||||
|
pthread_t *threadId;
|
||||||
|
bool stop;
|
||||||
|
uint32_t rebootTime;
|
||||||
|
} tsMsg;
|
||||||
|
|
||||||
|
static void dnodeSendStatusMsg() {
|
||||||
|
int32_t contLen = sizeof(SStatusMsg) + TSDB_MAX_VNODES * sizeof(SVnodeLoad);
|
||||||
|
SStatusMsg *pStatus = rpcMallocCont(contLen);
|
||||||
|
if (pStatus == NULL) {
|
||||||
|
dError("failed to malloc status message");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
pStatus->version = htonl(tsVersion);
|
||||||
|
pStatus->dnodeId = htonl(dnodeGetDnodeId());
|
||||||
|
tstrncpy(pStatus->dnodeEp, tsLocalEp, TSDB_EP_LEN);
|
||||||
|
pStatus->clusterId = htobe64(dnodeGetClusterId());
|
||||||
|
pStatus->lastReboot = htonl(tsMsg.rebootTime);
|
||||||
|
pStatus->numOfCores = htonl(tsNumOfCores);
|
||||||
|
pStatus->diskAvailable = tsAvailDataDirGB;
|
||||||
|
|
||||||
|
// fill cluster cfg parameters
|
||||||
|
pStatus->clusterCfg.statusInterval = htonl(tsStatusInterval);
|
||||||
|
pStatus->clusterCfg.checkTime = 0;
|
||||||
|
tstrncpy(pStatus->clusterCfg.timezone, tsTimezone, 64);
|
||||||
|
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);
|
||||||
|
|
||||||
|
// vnodeGetStatus(NULL, pStatus);
|
||||||
|
// contLen = sizeof(SStatusMsg) + pStatus->openVnodes * sizeof(SVnodeLoad);
|
||||||
|
// pStatus->openVnodes = htons(pStatus->openVnodes);
|
||||||
|
|
||||||
|
SRpcMsg rpcMsg = {.ahandle = NULL, .pCont = pStatus, .contLen = contLen, .msgType = TSDB_MSG_TYPE_DM_STATUS};
|
||||||
|
|
||||||
|
dnodeSendMsgToMnode(&rpcMsg);
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeProcessStatusRsp(SRpcMsg *pMsg) {
|
||||||
|
dTrace("status rsp is received, code:%s", tstrerror(pMsg->code));
|
||||||
|
if (pMsg->code != TSDB_CODE_SUCCESS) return;
|
||||||
|
|
||||||
|
SStatusRsp *pStatusRsp = pMsg->pCont;
|
||||||
|
|
||||||
|
SDnodeCfg *pCfg = &pStatusRsp->dnodeCfg;
|
||||||
|
pCfg->dnodeId = htonl(pCfg->dnodeId);
|
||||||
|
pCfg->clusterId = htobe64(pCfg->clusterId);
|
||||||
|
pCfg->numOfVnodes = htonl(pCfg->numOfVnodes);
|
||||||
|
pCfg->numOfDnodes = htonl(pCfg->numOfDnodes);
|
||||||
|
dnodeUpdateCfg(pCfg);
|
||||||
|
|
||||||
|
if (pCfg->dropped) {
|
||||||
|
dError("status rsp is received, and set dnode to drop status");
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
// vnodeSetAccess(pStatusRsp->vgAccess, pCfg->numOfVnodes);
|
||||||
|
|
||||||
|
SDnodeEps *eps = (SDnodeEps *)((char *)pStatusRsp->vgAccess + pCfg->numOfVnodes * sizeof(SVgroupAccess));
|
||||||
|
eps->dnodeNum = htonl(eps->dnodeNum);
|
||||||
|
for (int32_t i = 0; i < eps->dnodeNum; ++i) {
|
||||||
|
eps->dnodeEps[i].dnodeId = htonl(eps->dnodeEps[i].dnodeId);
|
||||||
|
eps->dnodeEps[i].dnodePort = htons(eps->dnodeEps[i].dnodePort);
|
||||||
|
}
|
||||||
|
|
||||||
|
dnodeUpdateDnodeEps(eps);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void *dnodeThreadRoutine(void *param) {
|
||||||
|
int32_t ms = tsStatusInterval * 1000;
|
||||||
|
while (!tsMsg.stop) {
|
||||||
|
taosMsleep(ms);
|
||||||
|
dnodeSendStatusMsg();
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t dnodeInitMsg() {
|
||||||
|
tsMsg.stop = false;
|
||||||
|
tsMsg.rebootTime = taosGetTimestampSec();
|
||||||
|
tsMsg.threadId = taosCreateThread(dnodeThreadRoutine, NULL);
|
||||||
|
if (tsMsg.threadId == NULL) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
dInfo("dnode msg is initialized");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeCleanupMsg() {
|
||||||
|
if (tsMsg.threadId != NULL) {
|
||||||
|
tsMsg.stop = true;
|
||||||
|
taosDestoryThread(tsMsg.threadId);
|
||||||
|
tsMsg.threadId = NULL;
|
||||||
|
}
|
||||||
|
|
||||||
|
dInfo("dnode msg is cleanuped");
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeStartMnode(SRpcMsg *pMsg) {
|
||||||
|
SCreateMnodeMsg *pCfg = pMsg->pCont;
|
||||||
|
pCfg->dnodeId = htonl(pCfg->dnodeId);
|
||||||
|
pCfg->mnodeNum = htonl(pCfg->mnodeNum);
|
||||||
|
for (int32_t i = 0; i < pCfg->mnodeNum; ++i) {
|
||||||
|
pCfg->mnodeEps[i].dnodeId = htonl(pCfg->mnodeEps[i].dnodeId);
|
||||||
|
pCfg->mnodeEps[i].dnodePort = htons(pCfg->mnodeEps[i].dnodePort);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pCfg->dnodeId != dnodeGetDnodeId()) {
|
||||||
|
dDebug("dnode:%d, in create meps msg is not equal with saved dnodeId:%d", pCfg->dnodeId, dnodeGetDnodeId());
|
||||||
|
return TSDB_CODE_MND_DNODE_ID_NOT_CONFIGURED;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (mnodeGetStatus() == MN_STATUS_READY) return 0;
|
||||||
|
|
||||||
|
return mnodeDeploy();
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeProcessCreateMnodeReq(SRpcMsg *pMsg) {
|
||||||
|
int32_t code = dnodeStartMnode(pMsg);
|
||||||
|
|
||||||
|
SRpcMsg rspMsg = {.handle = pMsg->handle, .pCont = NULL, .contLen = 0, .code = code};
|
||||||
|
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeProcessConfigDnodeReq(SRpcMsg *pMsg) {
|
||||||
|
SCfgDnodeMsg *pCfg = pMsg->pCont;
|
||||||
|
|
||||||
|
int32_t code = taosCfgDynamicOptions(pCfg->config);
|
||||||
|
|
||||||
|
SRpcMsg rspMsg = {.handle = pMsg->handle, .pCont = NULL, .contLen = 0, .code = code};
|
||||||
|
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeProcessStartupReq(SRpcMsg *pMsg) {
|
||||||
|
dInfo("startup msg is received, cont:%s", (char *)pMsg->pCont);
|
||||||
|
|
||||||
|
SStartupStep *pStep = rpcMallocCont(sizeof(SStartupStep));
|
||||||
|
dnodeGetStartup(pStep);
|
||||||
|
|
||||||
|
dDebug("startup msg is sent, step:%s desc:%s finished:%d", pStep->name, pStep->desc, pStep->finished);
|
||||||
|
|
||||||
|
SRpcMsg rpcRsp = {.handle = pMsg->handle, .pCont = pStep, .contLen = sizeof(SStartupStep)};
|
||||||
|
rpcSendResponse(&rpcRsp);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
}
|
||||||
|
|
@ -15,9 +15,9 @@
|
||||||
|
|
||||||
#define _DEFAULT_SOURCE
|
#define _DEFAULT_SOURCE
|
||||||
#include "dnodeCheck.h"
|
#include "dnodeCheck.h"
|
||||||
#include "dnodeEps.h"
|
#include "dnodeConfig.h"
|
||||||
#include "dnodeMsg.h"
|
#include "dnodeDnode.h"
|
||||||
#include "dnodeTrans.h"
|
#include "dnodeTransport.h"
|
||||||
#include "mnode.h"
|
#include "mnode.h"
|
||||||
#include "sync.h"
|
#include "sync.h"
|
||||||
#include "tcache.h"
|
#include "tcache.h"
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,382 @@
|
||||||
|
/*
|
||||||
|
* 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/>.
|
||||||
|
*/
|
||||||
|
|
||||||
|
/* this file is mainly responsible for the communication between DNODEs. Each
|
||||||
|
* dnode works as both server and client. Dnode may send status, grant, config
|
||||||
|
* messages to mnode, mnode may send create/alter/drop table/vnode messages
|
||||||
|
* to dnode. All theses messages are handled from here
|
||||||
|
*/
|
||||||
|
|
||||||
|
#define _DEFAULT_SOURCE
|
||||||
|
#include "dnodeTransport.h"
|
||||||
|
#include "dnodeConfig.h"
|
||||||
|
#include "dnodeDnode.h"
|
||||||
|
#include "dnodeMnode.h"
|
||||||
|
#include "dnodeVnodes.h"
|
||||||
|
#include "mnode.h"
|
||||||
|
#include "vnode.h"
|
||||||
|
|
||||||
|
typedef void (*MsgFp)(SRpcMsg *pMsg);
|
||||||
|
|
||||||
|
static struct {
|
||||||
|
void *serverRpc;
|
||||||
|
void *clientRpc;
|
||||||
|
void *shellRpc;
|
||||||
|
MsgFp msgFp[TSDB_MSG_TYPE_MAX];
|
||||||
|
} tsTrans;
|
||||||
|
|
||||||
|
static void dnodeInitMsgFp() {
|
||||||
|
// msg from client to dnode
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SUBMIT] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_QUERY] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_FETCH] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_TABLE] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_TABLE] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_TABLE] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_UPDATE_TAG_VAL] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_TABLE_META] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_TABLES_META] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_STABLE_VGROUP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_MQ_QUERY] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_MQ_CONSUME] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_MQ_CONNECT] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_MQ_DISCONNECT] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_MQ_ACK] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_MQ_RESET] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_NETWORK_TEST] = dnodeProcessStartupReq;
|
||||||
|
|
||||||
|
// msg from client to mnode
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CONNECT] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_ACCT] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_ACCT] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_ACCT] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_USER] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_USER] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_USER] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_DNODE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CONFIG_DNODE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_DNODE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_DB] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_DB] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_USE_DB] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_DB] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SYNC_DB] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_TOPIC] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_TOPIC] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_TOPIC] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_FUNCTION] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_FUNCTION] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_FUNCTION] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_STABLE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_STABLE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_STABLE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_KILL_QUERY] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_KILL_CONN] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_HEARTBEAT] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SHOW] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SHOW_RETRIEVE] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SHOW_RETRIEVE_FUNC] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_COMPACT_VNODE] = mnodeProcessMsg;
|
||||||
|
|
||||||
|
// message from mnode to dnode
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_STABLE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_STABLE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_STABLE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_STABLE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_STABLE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_STABLE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_VNODE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_VNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_VNODE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_ALTER_VNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_VNODE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_VNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SYNC_VNODE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_SYNC_VNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_COMPACT_VNODE_IN] = vnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_COMPACT_VNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_MNODE_IN] = dnodeProcessCreateMnodeReq;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CREATE_MNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_MNODE_IN] = NULL;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DROP_MNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CONFIG_DNODE_IN] = NULL;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_CONFIG_DNODE_IN_RSP] = mnodeProcessMsg;
|
||||||
|
|
||||||
|
// message from dnode to mnode
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DM_AUTH] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DM_AUTH_RSP] = NULL;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DM_GRANT] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DM_GRANT_RSP] = NULL;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DM_STATUS] = mnodeProcessMsg;
|
||||||
|
tsTrans.msgFp[TSDB_MSG_TYPE_DM_STATUS_RSP] = dnodeProcessStatusRsp;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeProcessPeerReq(SRpcMsg *pMsg, SRpcEpSet *pEpSet) {
|
||||||
|
SRpcMsg rspMsg = {.handle = pMsg->handle, .pCont = NULL, .contLen = 0};
|
||||||
|
int32_t msgType = pMsg->msgType;
|
||||||
|
|
||||||
|
if (msgType == TSDB_MSG_TYPE_NETWORK_TEST) {
|
||||||
|
dnodeProcessStartupReq(pMsg);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dnodeGetRunStat() != DN_RUN_STAT_RUNNING) {
|
||||||
|
rspMsg.code = TSDB_CODE_APP_NOT_READY;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
dTrace("RPC %p, peer req:%s is ignored since dnode not running", pMsg->handle, taosMsg[msgType]);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pMsg->pCont == NULL) {
|
||||||
|
rspMsg.code = TSDB_CODE_DND_INVALID_MSG_LEN;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
MsgFp fp = tsTrans.msgFp[msgType];
|
||||||
|
if (fp != NULL) {
|
||||||
|
dTrace("RPC %p, peer req:%s will be processed", pMsg->handle, taosMsg[msgType]);
|
||||||
|
(*fp)(pMsg);
|
||||||
|
} else {
|
||||||
|
dError("RPC %p, peer req:%s not processed", pMsg->handle, taosMsg[msgType]);
|
||||||
|
rspMsg.code = TSDB_CODE_DND_MSG_NOT_PROCESSED;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeInitServer() {
|
||||||
|
SRpcInit rpcInit;
|
||||||
|
memset(&rpcInit, 0, sizeof(rpcInit));
|
||||||
|
rpcInit.localPort = tsDnodeDnodePort;
|
||||||
|
rpcInit.label = "DND-S";
|
||||||
|
rpcInit.numOfThreads = 1;
|
||||||
|
rpcInit.cfp = dnodeProcessPeerReq;
|
||||||
|
rpcInit.sessions = TSDB_MAX_VNODES << 4;
|
||||||
|
rpcInit.connType = TAOS_CONN_SERVER;
|
||||||
|
rpcInit.idleTime = tsShellActivityTimer * 1000;
|
||||||
|
|
||||||
|
tsTrans.serverRpc = rpcOpen(&rpcInit);
|
||||||
|
if (tsTrans.serverRpc == NULL) {
|
||||||
|
dError("failed to init peer rpc server");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
dInfo("dnode peer rpc server is initialized");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeCleanupServer() {
|
||||||
|
if (tsTrans.serverRpc) {
|
||||||
|
rpcClose(tsTrans.serverRpc);
|
||||||
|
tsTrans.serverRpc = NULL;
|
||||||
|
dInfo("dnode peer server is closed");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeProcessPeerRsp(SRpcMsg *pMsg, SRpcEpSet *pEpSet) {
|
||||||
|
int32_t msgType = pMsg->msgType;
|
||||||
|
|
||||||
|
if (dnodeGetRunStat() == DN_RUN_STAT_STOPPED) {
|
||||||
|
if (pMsg == NULL || pMsg->pCont == NULL) return;
|
||||||
|
dTrace("RPC %p, peer rsp:%s is ignored since dnode is stopping", pMsg->handle, taosMsg[msgType]);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (msgType == TSDB_MSG_TYPE_DM_STATUS_RSP && pEpSet) {
|
||||||
|
dnodeUpdateMnodeEps(pEpSet);
|
||||||
|
}
|
||||||
|
|
||||||
|
MsgFp fp = tsTrans.msgFp[msgType];
|
||||||
|
if (fp != NULL) {
|
||||||
|
dTrace("RPC %p, peer rsp:%s will be processed", pMsg->handle, taosMsg[msgType]);
|
||||||
|
(*fp)(pMsg);
|
||||||
|
} else {
|
||||||
|
dDebug("RPC %p, peer rsp:%s not processed", pMsg->handle, taosMsg[msgType]);
|
||||||
|
}
|
||||||
|
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeInitClient() {
|
||||||
|
char secret[TSDB_KEY_LEN] = "secret";
|
||||||
|
SRpcInit rpcInit;
|
||||||
|
memset(&rpcInit, 0, sizeof(rpcInit));
|
||||||
|
rpcInit.label = "DND-C";
|
||||||
|
rpcInit.numOfThreads = 1;
|
||||||
|
rpcInit.cfp = dnodeProcessPeerRsp;
|
||||||
|
rpcInit.sessions = TSDB_MAX_VNODES << 4;
|
||||||
|
rpcInit.connType = TAOS_CONN_CLIENT;
|
||||||
|
rpcInit.idleTime = tsShellActivityTimer * 1000;
|
||||||
|
rpcInit.user = "t";
|
||||||
|
rpcInit.ckey = "key";
|
||||||
|
rpcInit.secret = secret;
|
||||||
|
|
||||||
|
tsTrans.clientRpc = rpcOpen(&rpcInit);
|
||||||
|
if (tsTrans.clientRpc == NULL) {
|
||||||
|
dError("failed to init peer rpc client");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
dInfo("dnode peer rpc client is initialized");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeCleanupClient() {
|
||||||
|
if (tsTrans.clientRpc) {
|
||||||
|
rpcClose(tsTrans.clientRpc);
|
||||||
|
tsTrans.clientRpc = NULL;
|
||||||
|
dInfo("dnode peer rpc client is closed");
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeProcessShellReq(SRpcMsg *pMsg, SRpcEpSet *pEpSet) {
|
||||||
|
SRpcMsg rspMsg = {.handle = pMsg->handle, .pCont = NULL, .contLen = 0};
|
||||||
|
int32_t msgType = pMsg->msgType;
|
||||||
|
|
||||||
|
if (dnodeGetRunStat() == DN_RUN_STAT_STOPPED) {
|
||||||
|
dError("RPC %p, shell req:%s is ignored since dnode exiting", pMsg->handle, taosMsg[msgType]);
|
||||||
|
rspMsg.code = TSDB_CODE_DND_EXITING;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
return;
|
||||||
|
} else if (dnodeGetRunStat() != DN_RUN_STAT_RUNNING) {
|
||||||
|
dError("RPC %p, shell req:%s is ignored since dnode not running", pMsg->handle, taosMsg[msgType]);
|
||||||
|
rspMsg.code = TSDB_CODE_APP_NOT_READY;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (pMsg->pCont == NULL) {
|
||||||
|
rspMsg.code = TSDB_CODE_DND_INVALID_MSG_LEN;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
return;
|
||||||
|
}
|
||||||
|
|
||||||
|
MsgFp fp = tsTrans.msgFp[msgType];
|
||||||
|
if (fp != NULL) {
|
||||||
|
dTrace("RPC %p, shell req:%s will be processed", pMsg->handle, taosMsg[msgType]);
|
||||||
|
(*fp)(pMsg);
|
||||||
|
} else {
|
||||||
|
dError("RPC %p, shell req:%s is not processed", pMsg->handle, taosMsg[msgType]);
|
||||||
|
rspMsg.code = TSDB_CODE_DND_MSG_NOT_PROCESSED;
|
||||||
|
rpcSendResponse(&rspMsg);
|
||||||
|
rpcFreeCont(pMsg->pCont);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeSendMsgToDnode(SRpcEpSet *epSet, SRpcMsg *rpcMsg) { rpcSendRequest(tsTrans.clientRpc, epSet, rpcMsg, NULL); }
|
||||||
|
|
||||||
|
void dnodeSendMsgToMnode(SRpcMsg *rpcMsg) {
|
||||||
|
SRpcEpSet epSet = {0};
|
||||||
|
dnodeGetEpSetForPeer(&epSet);
|
||||||
|
dnodeSendMsgToDnode(&epSet, rpcMsg);
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeSendMsgToMnodeRecv(SRpcMsg *rpcMsg, SRpcMsg *rpcRsp) {
|
||||||
|
SRpcEpSet epSet = {0};
|
||||||
|
dnodeGetEpSetForPeer(&epSet);
|
||||||
|
rpcSendRecv(tsTrans.clientRpc, &epSet, rpcMsg, rpcRsp);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeRetrieveUserAuthInfo(char *user, char *spi, char *encrypt, char *secret, char *ckey) {
|
||||||
|
int32_t code = mnodeRetriveAuth(user, spi, encrypt, secret, ckey);
|
||||||
|
if (code != TSDB_CODE_APP_NOT_READY) return code;
|
||||||
|
|
||||||
|
SAuthMsg *pMsg = rpcMallocCont(sizeof(SAuthMsg));
|
||||||
|
tstrncpy(pMsg->user, user, sizeof(pMsg->user));
|
||||||
|
|
||||||
|
SRpcMsg rpcMsg = {0};
|
||||||
|
rpcMsg.pCont = pMsg;
|
||||||
|
rpcMsg.contLen = sizeof(SAuthMsg);
|
||||||
|
rpcMsg.msgType = TSDB_MSG_TYPE_DM_AUTH;
|
||||||
|
|
||||||
|
dDebug("user:%s, send auth msg to mnodes", user);
|
||||||
|
SRpcMsg rpcRsp = {0};
|
||||||
|
dnodeSendMsgToMnodeRecv(&rpcMsg, &rpcRsp);
|
||||||
|
|
||||||
|
if (rpcRsp.code != 0) {
|
||||||
|
dError("user:%s, auth msg received from mnodes, error:%s", user, tstrerror(rpcRsp.code));
|
||||||
|
} else {
|
||||||
|
SAuthRsp *pRsp = rpcRsp.pCont;
|
||||||
|
dDebug("user:%s, auth msg received from mnodes", user);
|
||||||
|
memcpy(secret, pRsp->secret, TSDB_KEY_LEN);
|
||||||
|
memcpy(ckey, pRsp->ckey, TSDB_KEY_LEN);
|
||||||
|
*spi = pRsp->spi;
|
||||||
|
*encrypt = pRsp->encrypt;
|
||||||
|
}
|
||||||
|
|
||||||
|
rpcFreeCont(rpcRsp.pCont);
|
||||||
|
return rpcRsp.code;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeInitShell() {
|
||||||
|
int32_t numOfThreads = (int32_t)((tsNumOfCores * tsNumOfThreadsPerCore) / 2.0);
|
||||||
|
if (numOfThreads < 1) {
|
||||||
|
numOfThreads = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
SRpcInit rpcInit;
|
||||||
|
memset(&rpcInit, 0, sizeof(rpcInit));
|
||||||
|
rpcInit.localPort = tsDnodeShellPort;
|
||||||
|
rpcInit.label = "SHELL";
|
||||||
|
rpcInit.numOfThreads = numOfThreads;
|
||||||
|
rpcInit.cfp = dnodeProcessShellReq;
|
||||||
|
rpcInit.sessions = tsMaxShellConns;
|
||||||
|
rpcInit.connType = TAOS_CONN_SERVER;
|
||||||
|
rpcInit.idleTime = tsShellActivityTimer * 1000;
|
||||||
|
rpcInit.afp = dnodeRetrieveUserAuthInfo;
|
||||||
|
|
||||||
|
tsTrans.shellRpc = rpcOpen(&rpcInit);
|
||||||
|
if (tsTrans.shellRpc == NULL) {
|
||||||
|
dError("failed to init shell rpc server");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
dInfo("dnode shell rpc server is initialized");
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeCleanupShell() {
|
||||||
|
if (tsTrans.shellRpc) {
|
||||||
|
rpcClose(tsTrans.shellRpc);
|
||||||
|
tsTrans.shellRpc = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t dnodeInitTrans() {
|
||||||
|
if (dnodeInitClient() != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dnodeInitServer() != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dnodeInitShell() != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeCleanupTrans() {
|
||||||
|
dnodeCleanupShell();
|
||||||
|
dnodeCleanupServer();
|
||||||
|
dnodeCleanupClient();
|
||||||
|
}
|
||||||
|
|
@ -0,0 +1,204 @@
|
||||||
|
/*
|
||||||
|
* 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 "dnodeCheck.h"
|
||||||
|
#include "dnodeConfig.h"
|
||||||
|
#include "dnodeDnode.h"
|
||||||
|
#include "dnodeTransport.h"
|
||||||
|
#include "mnode.h"
|
||||||
|
#include "sync.h"
|
||||||
|
#include "tcache.h"
|
||||||
|
#include "tconfig.h"
|
||||||
|
#include "tnote.h"
|
||||||
|
#include "tstep.h"
|
||||||
|
#include "vnode.h"
|
||||||
|
#include "wal.h"
|
||||||
|
|
||||||
|
static struct {
|
||||||
|
EDnStat runStatus;
|
||||||
|
SStartupStep startup;
|
||||||
|
SSteps *steps;
|
||||||
|
} tsDnode;
|
||||||
|
|
||||||
|
EDnStat dnodeGetRunStat() { return tsDnode.runStatus; }
|
||||||
|
|
||||||
|
void dnodeSetRunStat(EDnStat stat) { tsDnode.runStatus = stat; }
|
||||||
|
|
||||||
|
void dnodeReportStartup(char *name, char *desc) {
|
||||||
|
SStartupStep *startup = &tsDnode.startup;
|
||||||
|
tstrncpy(startup->name, name, strlen(startup->name));
|
||||||
|
tstrncpy(startup->desc, desc, strlen(startup->desc));
|
||||||
|
startup->finished = 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeReportStartupFinished(char *name, char *desc) {
|
||||||
|
SStartupStep *startup = &tsDnode.startup;
|
||||||
|
tstrncpy(startup->name, name, strlen(startup->name));
|
||||||
|
tstrncpy(startup->desc, desc, strlen(startup->desc));
|
||||||
|
startup->finished = 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeGetStartup(SStartupStep *pStep) { memcpy(pStep, &tsDnode.startup, sizeof(SStartupStep)); }
|
||||||
|
|
||||||
|
static int32_t dnodeInitVnode() {
|
||||||
|
return vnodeInit();
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeInitMnode() {
|
||||||
|
SMnodePara para;
|
||||||
|
para.fp.GetDnodeEp = dnodeGetEp;
|
||||||
|
para.fp.SendMsgToDnode = dnodeSendMsgToDnode;
|
||||||
|
para.fp.SendMsgToMnode = dnodeSendMsgToMnode;
|
||||||
|
para.fp.SendRedirectMsg = dnodeSendRedirectMsg;
|
||||||
|
para.dnodeId = dnodeGetDnodeId();
|
||||||
|
para.clusterId = dnodeGetClusterId();
|
||||||
|
|
||||||
|
return mnodeInit(para);
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeInitTfs() {}
|
||||||
|
|
||||||
|
static int32_t dnodeInitMain() {
|
||||||
|
tsDnode.runStatus = DN_RUN_STAT_STOPPED;
|
||||||
|
tscEmbedded = 1;
|
||||||
|
taosIgnSIGPIPE();
|
||||||
|
taosBlockSIGPIPE();
|
||||||
|
taosResolveCRC();
|
||||||
|
taosInitGlobalCfg();
|
||||||
|
taosReadGlobalLogCfg();
|
||||||
|
taosSetCoreDump(tsEnableCoreFile);
|
||||||
|
|
||||||
|
if (!taosMkDir(tsLogDir)) {
|
||||||
|
printf("failed to create dir: %s, reason: %s\n", tsLogDir, strerror(errno));
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
char temp[TSDB_FILENAME_LEN];
|
||||||
|
sprintf(temp, "%s/taosdlog", tsLogDir);
|
||||||
|
if (taosInitLog(temp, tsNumOfLogLines, 1) < 0) {
|
||||||
|
printf("failed to init log file\n");
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!taosReadGlobalCfg()) {
|
||||||
|
taosPrintGlobalCfg();
|
||||||
|
dError("TDengine read global config failed");
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
dInfo("start to initialize TDengine");
|
||||||
|
|
||||||
|
taosInitNotes();
|
||||||
|
|
||||||
|
return taosCheckGlobalCfg();
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeCleanupMain() {
|
||||||
|
taos_cleanup();
|
||||||
|
taosCloseLog();
|
||||||
|
taosStopCacheRefreshWorker();
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeCheckRunning(char *dir) {
|
||||||
|
char filepath[256] = {0};
|
||||||
|
snprintf(filepath, sizeof(filepath), "%s/.running", dir);
|
||||||
|
|
||||||
|
FileFd fd = taosOpenFileCreateWriteTrunc(filepath);
|
||||||
|
if (fd < 0) {
|
||||||
|
dError("failed to open lock file:%s since %s, quit", filepath, strerror(errno));
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
int32_t ret = taosLockFile(fd);
|
||||||
|
if (ret != 0) {
|
||||||
|
dError("failed to lock file:%s since %s, quit", filepath, strerror(errno));
|
||||||
|
taosCloseFile(fd);
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static int32_t dnodeInitDir() {
|
||||||
|
sprintf(tsMnodeDir, "%s/mnode", tsDataDir);
|
||||||
|
sprintf(tsVnodeDir, "%s/vnode", tsDataDir);
|
||||||
|
sprintf(tsDnodeDir, "%s/dnode", tsDataDir);
|
||||||
|
|
||||||
|
if (!taosMkDir(tsDnodeDir)) {
|
||||||
|
dError("failed to create dir:%s since %s", tsDnodeDir, strerror(errno));
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!taosMkDir(tsMnodeDir)) {
|
||||||
|
dError("failed to create dir:%s since %s", tsMnodeDir, strerror(errno));
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!taosMkDir(tsVnodeDir)) {
|
||||||
|
dError("failed to create dir:%s since %s", tsVnodeDir, strerror(errno));
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (dnodeCheckRunning(tsDnodeDir) != 0) {
|
||||||
|
return -1;
|
||||||
|
}
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
static void dnodeCleanupDir() {}
|
||||||
|
|
||||||
|
int32_t dnodeInit() {
|
||||||
|
SSteps *steps = taosStepInit(24, dnodeReportStartup);
|
||||||
|
if (steps == NULL) return -1;
|
||||||
|
|
||||||
|
taosStepAdd(steps, "dnode-main", dnodeInitMain, dnodeCleanupMain);
|
||||||
|
taosStepAdd(steps, "dnode-dir", dnodeInitDir, dnodeCleanupDir);
|
||||||
|
taosStepAdd(steps, "dnode-check", dnodeInitCheck, dnodeCleanupCheck);
|
||||||
|
taosStepAdd(steps, "dnode-rpc", rpcInit, rpcCleanup);
|
||||||
|
taosStepAdd(steps, "dnode-tfs", dnodeInitTfs, NULL);
|
||||||
|
taosStepAdd(steps, "dnode-wal", walInit, walCleanUp);
|
||||||
|
taosStepAdd(steps, "dnode-sync", syncInit, syncCleanUp);
|
||||||
|
taosStepAdd(steps, "dnode-eps", dnodeInitEps, dnodeCleanupEps);
|
||||||
|
taosStepAdd(steps, "dnode-vnode", dnodeInitVnode, vnodeCleanup);
|
||||||
|
taosStepAdd(steps, "dnode-mnode", dnodeInitMnode, mnodeCleanup);
|
||||||
|
taosStepAdd(steps, "dnode-trans", dnodeInitTrans, dnodeCleanupTrans);
|
||||||
|
taosStepAdd(steps, "dnode-msg", dnodeInitMsg, dnodeCleanupMsg);
|
||||||
|
|
||||||
|
tsDnode.steps = steps;
|
||||||
|
taosStepExec(tsDnode.steps);
|
||||||
|
|
||||||
|
dnodeSetRunStat(DN_RUN_STAT_RUNNING);
|
||||||
|
dnodeReportStartupFinished("TDengine", "initialized successfully");
|
||||||
|
dInfo("TDengine is initialized successfully");
|
||||||
|
|
||||||
|
return 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
void dnodeCleanup() {
|
||||||
|
if (dnodeGetRunStat() != DN_RUN_STAT_STOPPED) {
|
||||||
|
dnodeSetRunStat(DN_RUN_STAT_STOPPED);
|
||||||
|
taosStepCleanup(tsDnode.steps);
|
||||||
|
tsDnode.steps = NULL;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// tsVnode.msgFp[TSDB_MSG_TYPE_MD_CREATE_VNODE] = vnodeProcessMgmtMsg;
|
||||||
|
// tsVnode.msgFp[TSDB_MSG_TYPE_MD_ALTER_VNODE] = vnodeProcessMgmtMsg;
|
||||||
|
// tsVnode.msgFp[TSDB_MSG_TYPE_MD_SYNC_VNODE] = vnodeProcessMgmtMsg;
|
||||||
|
// tsVnode.msgFp[TSDB_MSG_TYPE_MD_COMPACT_VNODE] = vnodeProcessMgmtMsg;
|
||||||
|
// tsVnode.msgFp[TSDB_MSG_TYPE_MD_DROP_VNODE] = vnodeProcessMgmtMsg;
|
||||||
|
// tsVnode.msgFp[TSDB_MSG_TYPE_MD_ALTER_STREAM] = vnodeProcessMgmtMsg;
|
||||||
|
|
||||||
|
|
@ -14,8 +14,8 @@
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#define _DEFAULT_SOURCE
|
#define _DEFAULT_SOURCE
|
||||||
#include "dnodeMsg.h"
|
#include "dnodeDnode.h"
|
||||||
#include "dnodeEps.h"
|
#include "dnodeConfig.h"
|
||||||
#include "mnode.h"
|
#include "mnode.h"
|
||||||
#include "tthread.h"
|
#include "tthread.h"
|
||||||
#include "ttime.h"
|
#include "ttime.h"
|
||||||
|
|
@ -20,9 +20,9 @@
|
||||||
*/
|
*/
|
||||||
|
|
||||||
#define _DEFAULT_SOURCE
|
#define _DEFAULT_SOURCE
|
||||||
#include "dnodeTrans.h"
|
#include "dnodeTransport.h"
|
||||||
#include "dnodeEps.h"
|
#include "dnodeConfig.h"
|
||||||
#include "dnodeMsg.h"
|
#include "dnodeDnode.h"
|
||||||
#include "mnode.h"
|
#include "mnode.h"
|
||||||
#include "vnode.h"
|
#include "vnode.h"
|
||||||
|
|
||||||
Loading…
Reference in New Issue