#4986

Committed 15 Mar 2026 08:32AM UTC coverage: 37.305% (-31.3%) from 68.601%

Build # #4986

Build Type

push

travis-ci

Committed by

tomchon

Commit Message

test: keep docs and unit test

Coverage Stats

125478 of 336361 relevant lines covered (37.3%)

1134847.06 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

74.29

/source/dnode/mgmt/mgmt_mnode/src/mmInt.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 "mmInt.h"
#include "wal.h"

static bool mmDeployRequired(const SMgmtInputOpt *pInput) {
  if (pInput->pData->dnodeId > 0) return false;
  if (pInput->pData->clusterId > 0) return false;
  if (strcmp(tsLocalEp, tsFirst) != 0) return false;
  return true;
}

static int32_t mmRequire(const SMgmtInputOpt *pInput, bool *required) {
  int32_t   code = 0;
  SMnodeOpt option = {0};
  if ((code = mmReadFile(pInput->path, &option)) != 0) {
    return code;
  }

  if (!option.deploy) {
    *required = mmDeployRequired(pInput);
    if (*required) {
      dInfo("deploy mnode required. dnodeId:%d<=0, clusterId:0x%" PRIx64 "<=0, localEp:%s==firstEp",
            pInput->pData->dnodeId, pInput->pData->clusterId, tsLocalEp);
    }
  } else {
    *required = true;
    dInfo("deploy mnode required. option deploy:%d", option.deploy);
  }

  return code;
}

static void mmBuildOptionForDeploy(SMnodeMgmt *pMgmt, const SMgmtInputOpt *pInput, SMnodeOpt *pOption) {
  pOption->deploy = true;
  pOption->msgCb = pMgmt->msgCb;
  pOption->dnodeId = pMgmt->pData->dnodeId;
  pOption->selfIndex = 0;
  pOption->numOfReplicas = 1;
  pOption->numOfTotalReplicas = 1;
  pOption->replicas[0].id = 1;
  pOption->replicas[0].port = tsServerPort;
  tstrncpy(pOption->replicas[0].fqdn, tsLocalFqdn, TSDB_FQDN_LEN);
  pOption->lastIndex = SYNC_INDEX_INVALID;
}

static void mmBuildOptionForOpen(SMnodeMgmt *pMgmt, SMnodeOpt *pOption) {
  pOption->deploy = false;
  pOption->msgCb = pMgmt->msgCb;
  pOption->dnodeId = pMgmt->pData->dnodeId;
}

static void mmClose(SMnodeMgmt *pMgmt) {
  if (pMgmt->pMnode != NULL) {
    mmStopWorker(pMgmt);
    mndClose(pMgmt->pMnode);
    (void)taosThreadRwlockDestroy(&pMgmt->lock);
    pMgmt->pMnode = NULL;
  }

  taosMemoryFree(pMgmt);
}
static int32_t mndOpenWrapper(const char *path, SMnodeOpt *opt, SMnode **pMnode) {
  int32_t code = 0;
  *pMnode = mndOpen(path, opt);
  if (*pMnode == NULL) {
    code = terrno;
  }
  ///*pMnode = pNode;
  return code;
}
static int32_t mmOpen(SMgmtInputOpt *pInput, SMgmtOutputOpt *pOutput) {
  int32_t code = 0;
  if ((code = walInit(pInput->stopDnodeFp)) != 0) {
    dError("failed to init wal since %s", tstrerror(code));
    return code;
  }

  if ((code = syncInit()) != 0) {
    dError("failed to init sync since %s", tstrerror(code));
    return code;
  }

  SMnodeMgmt *pMgmt = taosMemoryCalloc(1, sizeof(SMnodeMgmt));
  if (pMgmt == NULL) {
    code = terrno;
    return code;
  }

  pMgmt->pData = pInput->pData;
  pMgmt->path = pInput->path;
  pMgmt->name = pInput->name;
  pMgmt->msgCb = pInput->msgCb;
  pMgmt->msgCb.putToQueueFp = (PutToQueueFp)mmPutMsgToQueue;
  pMgmt->msgCb.mgmt = pMgmt;
  (void)taosThreadRwlockInit(&pMgmt->lock, NULL);

  SMnodeOpt option = {0};
  if ((code = mmReadFile(pMgmt->path, &option)) != 0) {
    dError("failed to read file since %s", tstrerror(code));
    mmClose(pMgmt);
    return code;
  }

  if (!option.deploy) {
    dInfo("mnode start to deploy");
    pMgmt->pData->dnodeId = 1;
    mmBuildOptionForDeploy(pMgmt, pInput, &option);
  } else {
    dInfo("mnode start to open");
    mmBuildOptionForOpen(pMgmt, &option);
  }

  dInfo("mnode open with options: deploy:%d, dnodeId:%d, selfIndex:%d, numOfReplicas:%d, numOfTotalReplicas:%d",
        option.deploy, option.dnodeId, option.selfIndex, option.numOfReplicas, option.numOfTotalReplicas);
  code = mndOpenWrapper(pMgmt->path, &option, &pMgmt->pMnode);
  if (code != 0) {
    dError("failed to open mnode since %s", tstrerror(code));
    mmClose(pMgmt);
    return code;
  }
  tmsgReportStartup("mnode-impl", "initialized");

  if ((code = mmStartWorker(pMgmt)) != 0) {
    dError("failed to start mnode worker since %s", tstrerror(code));
    mmClose(pMgmt);
    return code;
  }
  tmsgReportStartup("mnode-worker", "initialized");

  if (option.numOfTotalReplicas > 0) {
    option.deploy = true;
    option.numOfReplicas = 0;
    option.numOfTotalReplicas = 0;
    if ((code = mmWriteFile(pMgmt->path, &option)) != 0) {
      dError("failed to write mnode file since %s", tstrerror(code));
      return code;
    }
  }

  pInput->pData->dnodeId = pMgmt->pData->dnodeId;
  pOutput->pMgmt = pMgmt;
  return 0;
}

static int32_t mmStart(SMnodeMgmt *pMgmt) {
  int32_t code = 0;
  dDebug("mnode-mgmt start to run");
  SMnodeOpt option = {0};
  if ((code = mmReadFile(pMgmt->path, &option)) != 0) {
    dError("failed to read file since %s", tstrerror(code));
    mmClose(pMgmt);
    return code;
  }
  if ((code = mndStart(pMgmt->pMnode)) != 0) {
    return code;
  }
  if (mndNeedUpgrade(pMgmt->pMnode, option.version)) {
    option.version = mndGetVersion(pMgmt->pMnode);
    option.encrypted = mndGetEncryptedFlag(pMgmt->pMnode);
    if ((code = mmWriteFile(pMgmt->path, &option)) != 0) {
      dError("failed to write mnode file since %s", tstrerror(code));
      return code;
  }
  }
  return code;
}

static void mmStop(SMnodeMgmt *pMgmt) {
  dDebug("mnode-mgmt start to stop");
  mndPreClose(pMgmt->pMnode);
  (void)taosThreadRwlockWrlock(&pMgmt->lock);
  pMgmt->stopped = 1;
  (void)taosThreadRwlockUnlock(&pMgmt->lock);

  mndStop(pMgmt->pMnode);
}

static int32_t mmSyncIsCatchUp(SMnodeMgmt *pMgmt) { return mndIsCatchUp(pMgmt->pMnode); }

static ESyncRole mmSyncGetRole(SMnodeMgmt *pMgmt) { return mndGetRole(pMgmt->pMnode); }

SMgmtFunc mmGetMgmtFunc() {
  SMgmtFunc mgmtFunc = {0};
  mgmtFunc.openFp = mmOpen;
  mgmtFunc.closeFp = (NodeCloseFp)mmClose;
  mgmtFunc.startFp = (NodeStartFp)mmStart;
  mgmtFunc.stopFp = (NodeStopFp)mmStop;
  mgmtFunc.createFp = (NodeCreateFp)mmProcessCreateReq;
  mgmtFunc.dropFp = (NodeDropFp)mmProcessDropReq;
  mgmtFunc.requiredFp = mmRequire;
  mgmtFunc.getHandlesFp = mmGetMsgHandles;
  mgmtFunc.isCatchUpFp = (NodeIsCatchUpFp)mmSyncIsCatchUp;
  mgmtFunc.nodeRoleFp = (NodeRole)mmSyncGetRole;

  return mgmtFunc;
}

1	/*
2	* Copyright (c) 2019 TAOS Data, Inc. <jhtao@taosdata.com>
3	*
4	* This program is free software: you can use, redistribute, and/or modify
5	* it under the terms of the GNU Affero General Public License, version 3
6	* or later ("AGPL"), as published by the Free Software Foundation.
7	*
8	* This program is distributed in the hope that it will be useful, but WITHOUT
9	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10	* FITNESS FOR A PARTICULAR PURPOSE.
11	*
12	* You should have received a copy of the GNU Affero General Public License
13	* along with this program. If not, see <http:www.gnu.org/licenses/>.
14	*/
15
16	#define _DEFAULT_SOURCE
17	#include "mmInt.h"
18	#include "wal.h"
19
20	static bool mmDeployRequired(const SMgmtInputOpt *pInput) {	16✔
21	if (pInput->pData->dnodeId > 0) return false;	16✔
22	if (pInput->pData->clusterId > 0) return false;	16✔
23	if (strcmp(tsLocalEp, tsFirst) != 0) return false;	16✔
24	return true;	16✔
25	}
26
27	static int32_t mmRequire(const SMgmtInputOpt pInput, bool required) {	16✔
28	int32_t code = 0;	16✔
29	SMnodeOpt option = {0};	16✔
30	if ((code = mmReadFile(pInput->path, &option)) != 0) {	16✔
31	return code;	×
32	}
33
34	if (!option.deploy) {	16✔
35	*required = mmDeployRequired(pInput);	16✔
36	if (*required) {	16✔
37	dInfo("deploy mnode required. dnodeId:%d<=0, clusterId:0x%" PRIx64 "<=0, localEp:%s==firstEp",	16✔
38	pInput->pData->dnodeId, pInput->pData->clusterId, tsLocalEp);
39	}
40	} else {
41	*required = true;	×
42	dInfo("deploy mnode required. option deploy:%d", option.deploy);	×
43	}
44
45	return code;	16✔
46	}
47
48	static void mmBuildOptionForDeploy(SMnodeMgmt pMgmt, const SMgmtInputOpt pInput, SMnodeOpt *pOption) {	16✔
49	pOption->deploy = true;	16✔
50	pOption->msgCb = pMgmt->msgCb;	16✔
51	pOption->dnodeId = pMgmt->pData->dnodeId;	16✔
52	pOption->selfIndex = 0;	16✔
53	pOption->numOfReplicas = 1;	16✔
54	pOption->numOfTotalReplicas = 1;	16✔
55	pOption->replicas[0].id = 1;	16✔
56	pOption->replicas[0].port = tsServerPort;	16✔
57	tstrncpy(pOption->replicas[0].fqdn, tsLocalFqdn, TSDB_FQDN_LEN);	16✔
58	pOption->lastIndex = SYNC_INDEX_INVALID;	16✔
59	}	16✔
60
61	static void mmBuildOptionForOpen(SMnodeMgmt pMgmt, SMnodeOpt pOption) {	×
62	pOption->deploy = false;	×
63	pOption->msgCb = pMgmt->msgCb;	×
64	pOption->dnodeId = pMgmt->pData->dnodeId;	×
65	}	×
66
67	static void mmClose(SMnodeMgmt *pMgmt) {	16✔
68	if (pMgmt->pMnode != NULL) {	16✔
69	mmStopWorker(pMgmt);	16✔
70	mndClose(pMgmt->pMnode);	16✔
71	(void)taosThreadRwlockDestroy(&pMgmt->lock);	16✔
72	pMgmt->pMnode = NULL;	16✔
73	}
74
75	taosMemoryFree(pMgmt);	16✔
76	}	16✔
77	static int32_t mndOpenWrapper(const char path, SMnodeOpt opt, SMnode **pMnode) {	16✔
78	int32_t code = 0;	16✔
79	*pMnode = mndOpen(path, opt);	16✔
80	if (*pMnode == NULL) {	16✔
81	code = terrno;	×
82	}
83	///*pMnode = pNode;
84	return code;	16✔
85	}
86	static int32_t mmOpen(SMgmtInputOpt pInput, SMgmtOutputOpt pOutput) {	16✔
87	int32_t code = 0;	16✔
88	if ((code = walInit(pInput->stopDnodeFp)) != 0) {	16✔
89	dError("failed to init wal since %s", tstrerror(code));	×
90	return code;	×
91	}
92
93	if ((code = syncInit()) != 0) {	16✔
94	dError("failed to init sync since %s", tstrerror(code));	×
95	return code;	×
96	}
97
98	SMnodeMgmt *pMgmt = taosMemoryCalloc(1, sizeof(SMnodeMgmt));	16✔
99	if (pMgmt == NULL) {	16✔
100	code = terrno;	×
101	return code;	×
102	}
103
104	pMgmt->pData = pInput->pData;	16✔
105	pMgmt->path = pInput->path;	16✔
106	pMgmt->name = pInput->name;	16✔
107	pMgmt->msgCb = pInput->msgCb;	16✔
108	pMgmt->msgCb.putToQueueFp = (PutToQueueFp)mmPutMsgToQueue;	16✔
109	pMgmt->msgCb.mgmt = pMgmt;	16✔
110	(void)taosThreadRwlockInit(&pMgmt->lock, NULL);	16✔
111
112	SMnodeOpt option = {0};	16✔
113	if ((code = mmReadFile(pMgmt->path, &option)) != 0) {	16✔
114	dError("failed to read file since %s", tstrerror(code));	×
115	mmClose(pMgmt);	×
116	return code;	×
117	}
118
119	if (!option.deploy) {	16✔
120	dInfo("mnode start to deploy");	16✔
121	pMgmt->pData->dnodeId = 1;	16✔
122	mmBuildOptionForDeploy(pMgmt, pInput, &option);	16✔
123	} else {
124	dInfo("mnode start to open");	×
125	mmBuildOptionForOpen(pMgmt, &option);	×
126	}
127
128	dInfo("mnode open with options: deploy:%d, dnodeId:%d, selfIndex:%d, numOfReplicas:%d, numOfTotalReplicas:%d",	16✔
129	option.deploy, option.dnodeId, option.selfIndex, option.numOfReplicas, option.numOfTotalReplicas);
130	code = mndOpenWrapper(pMgmt->path, &option, &pMgmt->pMnode);	16✔
131	if (code != 0) {	16✔
132	dError("failed to open mnode since %s", tstrerror(code));	×
133	mmClose(pMgmt);	×
134	return code;	×
135	}
136	tmsgReportStartup("mnode-impl", "initialized");	16✔
137
138	if ((code = mmStartWorker(pMgmt)) != 0) {	16✔
139	dError("failed to start mnode worker since %s", tstrerror(code));	×
140	mmClose(pMgmt);	×
141	return code;	×
142	}
143	tmsgReportStartup("mnode-worker", "initialized");	16✔
144
145	if (option.numOfTotalReplicas > 0) {	16✔
146	option.deploy = true;	16✔
147	option.numOfReplicas = 0;	16✔
148	option.numOfTotalReplicas = 0;	16✔
149	if ((code = mmWriteFile(pMgmt->path, &option)) != 0) {	16✔
150	dError("failed to write mnode file since %s", tstrerror(code));	×
151	return code;	×
152	}
153	}
154
155	pInput->pData->dnodeId = pMgmt->pData->dnodeId;	16✔
156	pOutput->pMgmt = pMgmt;	16✔
157	return 0;	16✔
158	}
159
160	static int32_t mmStart(SMnodeMgmt *pMgmt) {	16✔
161	int32_t code = 0;	16✔
162	dDebug("mnode-mgmt start to run");	16✔
163	SMnodeOpt option = {0};	16✔
164	if ((code = mmReadFile(pMgmt->path, &option)) != 0) {	16✔
165	dError("failed to read file since %s", tstrerror(code));	×
166	mmClose(pMgmt);	×
167	return code;	×
168	}
169	if ((code = mndStart(pMgmt->pMnode)) != 0) {	16✔
170	return code;	×
171	}
172	if (mndNeedUpgrade(pMgmt->pMnode, option.version)) {	16✔
173	option.version = mndGetVersion(pMgmt->pMnode);	16✔
174	option.encrypted = mndGetEncryptedFlag(pMgmt->pMnode);	16✔
175	if ((code = mmWriteFile(pMgmt->path, &option)) != 0) {	16✔
176	dError("failed to write mnode file since %s", tstrerror(code));	×
177	return code;	×
178	}
179	}
180	return code;	16✔
181	}
182
183	static void mmStop(SMnodeMgmt *pMgmt) {	16✔
184	dDebug("mnode-mgmt start to stop");	16✔
185	mndPreClose(pMgmt->pMnode);	16✔
186	(void)taosThreadRwlockWrlock(&pMgmt->lock);	16✔
187	pMgmt->stopped = 1;	16✔
188	(void)taosThreadRwlockUnlock(&pMgmt->lock);	16✔
189
190	mndStop(pMgmt->pMnode);	16✔
191	}	16✔
192
193	static int32_t mmSyncIsCatchUp(SMnodeMgmt *pMgmt) { return mndIsCatchUp(pMgmt->pMnode); }	×
194
195	static ESyncRole mmSyncGetRole(SMnodeMgmt *pMgmt) { return mndGetRole(pMgmt->pMnode); }	×
196
197	SMgmtFunc mmGetMgmtFunc() {	16✔
198	SMgmtFunc mgmtFunc = {0};	16✔
199	mgmtFunc.openFp = mmOpen;	16✔
200	mgmtFunc.closeFp = (NodeCloseFp)mmClose;	16✔
201	mgmtFunc.startFp = (NodeStartFp)mmStart;	16✔
202	mgmtFunc.stopFp = (NodeStopFp)mmStop;	16✔
203	mgmtFunc.createFp = (NodeCreateFp)mmProcessCreateReq;	16✔
204	mgmtFunc.dropFp = (NodeDropFp)mmProcessDropReq;	16✔
205	mgmtFunc.requiredFp = mmRequire;	16✔
206	mgmtFunc.getHandlesFp = mmGetMsgHandles;	16✔
207	mgmtFunc.isCatchUpFp = (NodeIsCatchUpFp)mmSyncIsCatchUp;	16✔
208	mgmtFunc.nodeRoleFp = (NodeRole)mmSyncGetRole;	16✔
209
210	return mgmtFunc;	16✔
211	}

taosdata / TDengine / #4986

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous