#3796

Committed 31 Mar 2025 10:39AM UTC coverage: 30.372% (-7.1%) from 37.443%

Build # #3796

Build Type

push

travis-ci

Committed by

happyguoxy

Commit Message

test:add test cases

Run Details

69287 of 309062 branches covered (22.42%)

Branch coverage included in aggregate %.

118044 of 307720 relevant lines covered (38.36%)

278592.15 hits per line

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

14.36

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

taosdata / TDengine / #3796

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