Apache/nifi
1
0
Fork 0
mirror of https://github.com/apache/nifi.git synced 2025-02-06 10:08:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

NIFI-11476: Populate resource consumitions

Browse Source 
Signed-off-by: Ferenc Erdei <ferdei@cloudera.com>
This commit is contained in:
Kalman Jantner 2023-04-10 19:13:11 +02:00 committed by Ferenc Erdei
parent 51956a4426
commit 1fa5da9a55
No known key found for this signature in database
GPG Key ID: F8FAAD09EA246008
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
c2/c2-client-bundle/c2-client-service/src/main/java/org/apache/nifi/c2/client/service/C2HeartbeatFactory.java
View File

@ -36,6 +36,7 @@ import org.apache.nifi.c2.client.service.model.RuntimeInfoWrapper;
import org.apache.nifi.c2.protocol.api.AgentInfo;
import org.apache.nifi.c2.protocol.api.AgentManifest;
import org.apache.nifi.c2.protocol.api.AgentRepositories;
import org.apache.nifi.c2.protocol.api.AgentResourceConsumption;
import org.apache.nifi.c2.protocol.api.AgentStatus;
import org.apache.nifi.c2.protocol.api.C2Heartbeat;
import org.apache.nifi.c2.protocol.api.DeviceInfo;
@ -95,7 +96,9 @@ public class C2HeartbeatFactory {
AgentStatus agentStatus = new AgentStatus();
agentStatus.setUptime(ManagementFactory.getRuntimeMXBean().getUptime());
agentStatus.setRepositories(repos);
AgentResourceConsumption agentResourceConsumption = new AgentResourceConsumption();
agentResourceConsumption.setMemoryUsage(Runtime.getRuntime().maxMemory() - Runtime.getRuntime().freeMemory());
agentStatus.setResourceConsumption(agentResourceConsumption);
agentInfo.setStatus(agentStatus);
agentInfo.setAgentManifestHash(manifestHashProvider.calculateManifestHash(manifest.getBundles(), getSupportedOperations(manifest)));
@ -215,15 +218,14 @@ public class C2HeartbeatFactory {
private SystemInfo generateSystemInfo() {
SystemInfo systemInfo = new SystemInfo();
systemInfo.setPhysicalMem(Runtime.getRuntime().maxMemory());
systemInfo.setMemoryUsage(Runtime.getRuntime().maxMemory() - Runtime.getRuntime().freeMemory());
systemInfo.setvCores(Runtime.getRuntime().availableProcessors());
OperatingSystemMXBean osMXBean = ManagementFactory.getOperatingSystemMXBean();
systemInfo.setMachineArch(osMXBean.getArch());
systemInfo.setOperatingSystem(osMXBean.getName());
double systemLoadAverage = osMXBean.getSystemLoadAverage();
systemInfo.setvCores(osMXBean.getAvailableProcessors());
// getSystemLoadAverage is not available in Windows, so we need to prevent to send invalid data
if (systemLoadAverage >= 0) {
systemInfo.setCpuUtilization(systemLoadAverage / (double) osMXBean.getAvailableProcessors());