#3961

Committed 29 Apr 2025 09:40AM UTC coverage: 62.291% (-0.2%) from 62.445%

Build # #3961

Build Type

push

travis-ci

Committed by

web-flow

Commit Message

Merge pull request #30938 from taosdata/fix/taosdCmdArgs

fix: taosd cmd args

Run Details

154880 of 317399 branches covered (48.8%)

Branch coverage included in aggregate %.

1 of 4 new or added lines in 1 file covered. (25.0%)

1209 existing lines in 154 files now uncovered.

239982 of 316504 relevant lines covered (75.82%)

6306221.95 hits per line

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

74.03

/source/dnode/mgmt/node_mgmt/src/dmMonitor.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 "dmMgmt.h"
#include "dmNodes.h"
#include "audit.h"

static void dmGetMonitorBasicInfo(SDnode *pDnode, SMonBasicInfo *pInfo) {
  pInfo->protocol = 1;
  pInfo->dnode_id = pDnode->data.dnodeId;
  pInfo->cluster_id = pDnode->data.clusterId;
  tstrncpy(pInfo->dnode_ep, tsLocalEp, TSDB_EP_LEN);
}

static void dmGetMonitorBasicInfoBasic(SDnode *pDnode, SMonBasicInfo *pInfo) {
  pInfo->protocol = 2;
  pInfo->dnode_id = pDnode->data.dnodeId;
  pInfo->cluster_id = pDnode->data.clusterId;
  tstrncpy(pInfo->dnode_ep, tsLocalEp, TSDB_EP_LEN);
}

static void dmGetMonitorDnodeInfo(SDnode *pDnode, SMonDnodeInfo *pInfo) {
  // pInfo->uptime = (taosGetTimestampMs() - pDnode->data.rebootTime) / (86400000.0f);
  pInfo->uptime = (taosGetTimestampMs() - pDnode->data.rebootTime) / 1000.0f;
  pInfo->has_mnode = pDnode->wrappers[MNODE].required;
  pInfo->has_qnode = pDnode->wrappers[QNODE].required;
  pInfo->has_snode = pDnode->wrappers[SNODE].required;
  tstrncpy(pInfo->logdir.name, tsLogDir, sizeof(pInfo->logdir.name));
  pInfo->logdir.size = tsLogSpace.size;
  tstrncpy(pInfo->tempdir.name, tsTempDir, sizeof(pInfo->tempdir.name));
  pInfo->tempdir.size = tsTempSpace.size;
}

static void dmGetDmMonitorInfo(SDnode *pDnode) {
  SMonDmInfo dmInfo = {0};
  dmGetMonitorBasicInfo(pDnode, &dmInfo.basic);
  dmGetMonitorDnodeInfo(pDnode, &dmInfo.dnode);
  dmGetMonitorSystemInfo(&dmInfo.sys);
  monSetDmInfo(&dmInfo);
}

void dmCleanExpriedSamples(SDnode *pDnode) {
  SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    if (pWrapper->pMgmt != NULL) {
      vmCleanExpriedSamples(pWrapper->pMgmt);
    }
  }
  dmReleaseWrapper(pWrapper);
  return;
}

static void dmGetDmMonitorInfoBasic(SDnode *pDnode) {
  SMonDmInfo dmInfo = {0};
  dmGetMonitorBasicInfoBasic(pDnode, &dmInfo.basic);
  monSetDmInfo(&dmInfo);
}

static void dmGetMmMonitorInfo(SDnode *pDnode) {
  SMgmtWrapper *pWrapper = &pDnode->wrappers[MNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    SMonMmInfo mmInfo = {0};
    if (pWrapper->pMgmt != NULL) {
      mmGetMonitorInfo(pWrapper->pMgmt, &mmInfo);
    }
    dmReleaseWrapper(pWrapper);
    monSetMmInfo(&mmInfo);
    tFreeSMonMmInfo(&mmInfo);
  }
}

static void dmGetVmMonitorInfo(SDnode *pDnode) {
  SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    SMonVmInfo vmInfo = {0};
    if (pWrapper->pMgmt != NULL) {
      vmGetMonitorInfo(pWrapper->pMgmt, &vmInfo);
    }
    dmReleaseWrapper(pWrapper);
    monSetVmInfo(&vmInfo);
    tFreeSMonVmInfo(&vmInfo);
  }
}

static void dmGetQmMonitorInfo(SDnode *pDnode) {
  SMgmtWrapper *pWrapper = &pDnode->wrappers[QNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    SMonQmInfo qmInfo = {0};
    if (pWrapper->pMgmt != NULL) {
      qmGetMonitorInfo(pWrapper->pMgmt, &qmInfo);
    }
    dmReleaseWrapper(pWrapper);
    monSetQmInfo(&qmInfo);
    tFreeSMonQmInfo(&qmInfo);
  }
}

static void dmGetSmMonitorInfo(SDnode *pDnode) {
  SMgmtWrapper *pWrapper = &pDnode->wrappers[SNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    SMonSmInfo smInfo = {0};
    if (pWrapper->pMgmt != NULL) {
      smGetMonitorInfo(pWrapper->pMgmt, &smInfo);
    }
    dmReleaseWrapper(pWrapper);
    monSetSmInfo(&smInfo);
    tFreeSMonSmInfo(&smInfo);
  }
}

void dmSendMonitorReport() {
  if (!tsEnableMonitor || tsMonitorFqdn[0] == 0 || tsMonitorPort == 0) return;
  dTrace("send monitor report to %s:%u", tsMonitorFqdn, tsMonitorPort);

  SDnode *pDnode = dmInstance();
  dmGetDmMonitorInfo(pDnode);
  dmGetMmMonitorInfo(pDnode);
  dmGetVmMonitorInfo(pDnode);
  dmGetQmMonitorInfo(pDnode);
  dmGetSmMonitorInfo(pDnode);
  monGenAndSendReport();
}

void dmMonitorCleanExpiredSamples() {
  if (!tsEnableMonitor || tsMonitorFqdn[0] == 0 || tsMonitorPort == 0) return;
  dTrace("clean monitor expired samples");

  SDnode *pDnode = dmInstance();
  (void)dmCleanExpriedSamples(pDnode);
}

// Todo: put this in seperate file in the future
void dmSendAuditRecords() { auditSendRecordsInBatch(); }

void dmGetVnodeLoads(SMonVloadInfo *pInfo) {
  SDnode       *pDnode = dmInstance();
  SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    if (pWrapper->pMgmt != NULL) {
      vmGetVnodeLoads(pWrapper->pMgmt, pInfo, false);
    }
    dmReleaseWrapper(pWrapper);
  }
}

void dmGetVnodeLoadsLite(SMonVloadInfo *pInfo) {
  SDnode       *pDnode = dmInstance();
  SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    if (pWrapper->pMgmt != NULL) {
      vmGetVnodeLoadsLite(pWrapper->pMgmt, pInfo);
    }
    dmReleaseWrapper(pWrapper);
  }
}

void dmGetMnodeLoads(SMonMloadInfo *pInfo) {
  SDnode       *pDnode = dmInstance();
  SMgmtWrapper *pWrapper = &pDnode->wrappers[MNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    if (pWrapper->pMgmt != NULL) {
      mmGetMnodeLoads(pWrapper->pMgmt, pInfo);
    }
    dmReleaseWrapper(pWrapper);
  }
}

void dmGetQnodeLoads(SQnodeLoad *pInfo) {
  SDnode       *pDnode = dmInstance();
  SMgmtWrapper *pWrapper = &pDnode->wrappers[QNODE];
  if (dmMarkWrapper(pWrapper) == 0) {
    if (pWrapper->pMgmt != NULL) {
      qmGetQnodeLoads(pWrapper->pMgmt, pInfo);
    }
    dmReleaseWrapper(pWrapper);
  }
}

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 "dmMgmt.h"
18	#include "dmNodes.h"
19	#include "audit.h"
20
21	static void dmGetMonitorBasicInfo(SDnode pDnode, SMonBasicInfo pInfo) {	13✔
22	pInfo->protocol = 1;	13✔
23	pInfo->dnode_id = pDnode->data.dnodeId;	13✔
24	pInfo->cluster_id = pDnode->data.clusterId;	13✔
25	tstrncpy(pInfo->dnode_ep, tsLocalEp, TSDB_EP_LEN);	13✔
26	}	13✔
27
28	static void dmGetMonitorBasicInfoBasic(SDnode pDnode, SMonBasicInfo pInfo) {	×
29	pInfo->protocol = 2;	×
30	pInfo->dnode_id = pDnode->data.dnodeId;	×
31	pInfo->cluster_id = pDnode->data.clusterId;	×
32	tstrncpy(pInfo->dnode_ep, tsLocalEp, TSDB_EP_LEN);	×
33	}	×
34
35	static void dmGetMonitorDnodeInfo(SDnode pDnode, SMonDnodeInfo pInfo) {	13✔
36	// pInfo->uptime = (taosGetTimestampMs() - pDnode->data.rebootTime) / (86400000.0f);
37	pInfo->uptime = (taosGetTimestampMs() - pDnode->data.rebootTime) / 1000.0f;	13✔
38	pInfo->has_mnode = pDnode->wrappers[MNODE].required;	13✔
39	pInfo->has_qnode = pDnode->wrappers[QNODE].required;	13✔
40	pInfo->has_snode = pDnode->wrappers[SNODE].required;	13✔
41	tstrncpy(pInfo->logdir.name, tsLogDir, sizeof(pInfo->logdir.name));	13✔
42	pInfo->logdir.size = tsLogSpace.size;	13✔
43	tstrncpy(pInfo->tempdir.name, tsTempDir, sizeof(pInfo->tempdir.name));	13✔
44	pInfo->tempdir.size = tsTempSpace.size;	13✔
45	}	13✔
46
47	static void dmGetDmMonitorInfo(SDnode *pDnode) {	13✔
48	SMonDmInfo dmInfo = {0};	13✔
49	dmGetMonitorBasicInfo(pDnode, &dmInfo.basic);	13✔
50	dmGetMonitorDnodeInfo(pDnode, &dmInfo.dnode);	13✔
51	dmGetMonitorSystemInfo(&dmInfo.sys);	13✔
52	monSetDmInfo(&dmInfo);	13✔
53	}	13✔
54
55	void dmCleanExpriedSamples(SDnode *pDnode) {	13✔
56	SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];	13✔
57	if (dmMarkWrapper(pWrapper) == 0) {	13!
58	if (pWrapper->pMgmt != NULL) {	13!
59	vmCleanExpriedSamples(pWrapper->pMgmt);	13✔
60	}
61	}
62	dmReleaseWrapper(pWrapper);	13✔
63	return;	13✔
64	}
65
66	static void dmGetDmMonitorInfoBasic(SDnode *pDnode) {	×
67	SMonDmInfo dmInfo = {0};	×
68	dmGetMonitorBasicInfoBasic(pDnode, &dmInfo.basic);	×
69	monSetDmInfo(&dmInfo);	×
70	}	×
71
72	static void dmGetMmMonitorInfo(SDnode *pDnode) {	13✔
73	SMgmtWrapper *pWrapper = &pDnode->wrappers[MNODE];	13✔
74	if (dmMarkWrapper(pWrapper) == 0) {	13!
75	SMonMmInfo mmInfo = {0};	13✔
76	if (pWrapper->pMgmt != NULL) {	13!
77	mmGetMonitorInfo(pWrapper->pMgmt, &mmInfo);	13✔
78	}
79	dmReleaseWrapper(pWrapper);	13✔
80	monSetMmInfo(&mmInfo);	13✔
81	tFreeSMonMmInfo(&mmInfo);	13✔
82	}
83	}	13✔
84
85	static void dmGetVmMonitorInfo(SDnode *pDnode) {	13✔
86	SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];	13✔
87	if (dmMarkWrapper(pWrapper) == 0) {	13!
88	SMonVmInfo vmInfo = {0};	13✔
89	if (pWrapper->pMgmt != NULL) {	13!
90	vmGetMonitorInfo(pWrapper->pMgmt, &vmInfo);	13✔
91	}
92	dmReleaseWrapper(pWrapper);	13✔
93	monSetVmInfo(&vmInfo);	13✔
94	tFreeSMonVmInfo(&vmInfo);	13✔
95	}
96	}	13✔
97
98	static void dmGetQmMonitorInfo(SDnode *pDnode) {	13✔
99	SMgmtWrapper *pWrapper = &pDnode->wrappers[QNODE];	13✔
100	if (dmMarkWrapper(pWrapper) == 0) {	13✔
101	SMonQmInfo qmInfo = {0};	10✔
102	if (pWrapper->pMgmt != NULL) {	10!
103	qmGetMonitorInfo(pWrapper->pMgmt, &qmInfo);	10✔
104	}
105	dmReleaseWrapper(pWrapper);	10✔
106	monSetQmInfo(&qmInfo);	10✔
107	tFreeSMonQmInfo(&qmInfo);	10✔
108	}
109	}	13✔
110
111	static void dmGetSmMonitorInfo(SDnode *pDnode) {	13✔
112	SMgmtWrapper *pWrapper = &pDnode->wrappers[SNODE];	13✔
113	if (dmMarkWrapper(pWrapper) == 0) {	13!
114	SMonSmInfo smInfo = {0};	×
115	if (pWrapper->pMgmt != NULL) {	×
116	smGetMonitorInfo(pWrapper->pMgmt, &smInfo);	×
117	}
118	dmReleaseWrapper(pWrapper);	×
119	monSetSmInfo(&smInfo);	×
120	tFreeSMonSmInfo(&smInfo);	×
121	}
122	}	13✔
123
124	void dmSendMonitorReport() {	1,885✔
125	if (!tsEnableMonitor \|\| tsMonitorFqdn[0] == 0 \|\| tsMonitorPort == 0) return;	1,885!
126	dTrace("send monitor report to %s:%u", tsMonitorFqdn, tsMonitorPort);	13!
127
128	SDnode *pDnode = dmInstance();	13✔
129	dmGetDmMonitorInfo(pDnode);	13✔
130	dmGetMmMonitorInfo(pDnode);	13✔
131	dmGetVmMonitorInfo(pDnode);	13✔
132	dmGetQmMonitorInfo(pDnode);	13✔
133	dmGetSmMonitorInfo(pDnode);	13✔
134	monGenAndSendReport();	13✔
135	}
136
137	void dmMonitorCleanExpiredSamples() {	1,885✔
138	if (!tsEnableMonitor \|\| tsMonitorFqdn[0] == 0 \|\| tsMonitorPort == 0) return;	1,885!
139	dTrace("clean monitor expired samples");	13!
140
141	SDnode *pDnode = dmInstance();	13✔
142	(void)dmCleanExpriedSamples(pDnode);	13✔
143	}
144
145	// Todo: put this in seperate file in the future
146	void dmSendAuditRecords() { auditSendRecordsInBatch(); }	×
147
148	void dmGetVnodeLoads(SMonVloadInfo *pInfo) {	79,755✔
149	SDnode *pDnode = dmInstance();	79,755✔
150	SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];	79,755✔
151	if (dmMarkWrapper(pWrapper) == 0) {	79,755✔
152	if (pWrapper->pMgmt != NULL) {	77,534!
153	vmGetVnodeLoads(pWrapper->pMgmt, pInfo, false);	77,534✔
154	}
155	dmReleaseWrapper(pWrapper);	77,534✔
156	}
157	}	79,755✔
158
UNCOV 159	void dmGetVnodeLoadsLite(SMonVloadInfo *pInfo) {	×
UNCOV 160	SDnode *pDnode = dmInstance();	×
UNCOV 161	SMgmtWrapper *pWrapper = &pDnode->wrappers[VNODE];	×
UNCOV 162	if (dmMarkWrapper(pWrapper) == 0) {	×
UNCOV 163	if (pWrapper->pMgmt != NULL) {	×
UNCOV 164	vmGetVnodeLoadsLite(pWrapper->pMgmt, pInfo);	×
165	}
UNCOV 166	dmReleaseWrapper(pWrapper);	×
167	}
UNCOV 168	}	×
169
170	void dmGetMnodeLoads(SMonMloadInfo *pInfo) {	79,755✔
171	SDnode *pDnode = dmInstance();	79,755✔
172	SMgmtWrapper *pWrapper = &pDnode->wrappers[MNODE];	79,755✔
173	if (dmMarkWrapper(pWrapper) == 0) {	79,755✔
174	if (pWrapper->pMgmt != NULL) {	50,947!
175	mmGetMnodeLoads(pWrapper->pMgmt, pInfo);	50,947✔
176	}
177	dmReleaseWrapper(pWrapper);	50,947✔
178	}
179	}	79,755✔
180
181	void dmGetQnodeLoads(SQnodeLoad *pInfo) {	77,331✔
182	SDnode *pDnode = dmInstance();	77,331✔
183	SMgmtWrapper *pWrapper = &pDnode->wrappers[QNODE];	77,331✔
184	if (dmMarkWrapper(pWrapper) == 0) {	77,331✔
185	if (pWrapper->pMgmt != NULL) {	14,633!
186	qmGetQnodeLoads(pWrapper->pMgmt, pInfo);	14,633✔
187	}
188	dmReleaseWrapper(pWrapper);	14,633✔
189	}
190	}	77,331✔

taosdata / TDengine / #3961

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