fix: fix Current Process CPU plugin (#85)

* fix cpu plugin

* fix cpu plugin

* fix license

* fix field

Author: innerpeacez

agent-plugins/agent-cpu-plugin/src/main/java/cloud/erda/agent/plugin/cpu/CPUDefaultProvider.java

@@ -1,3 +1,19 @@
+/*
+ * Copyright (c) 2021 Terminus, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package cloud.erda.agent.plugin.cpu;
 
 import cloud.erda.agent.core.config.AgentConfig;
@@ -14,6 +30,8 @@ public interface CPUDefaultProvider {
     AgentConfig agentConfig = ConfigAccessor.Default.getConfig(AgentConfig.class);
     ServiceConfig serviceConfig = ConfigAccessor.Default.getConfig(ServiceConfig.class);
 
+    Metric get();
+
     default void addDefaultTags(Metric metric) {
         metric.addTag("terminus_key", agentConfig.terminusKey()).
                 addTag("instance_id", serviceConfig.getServiceInstanceId()).

agent-plugins/agent-cpu-plugin/src/main/java/cloud/erda/agent/plugin/cpu/CPUInfo.java

@@ -1,6 +1,21 @@
+/*
+ * Copyright (c) 2021 Terminus, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package cloud.erda.agent.plugin.cpu;
 
-import cloud.erda.agent.core.metrics.Metric;
 import cloud.erda.agent.core.metrics.MetricDispatcher;
 import org.apache.skywalking.apm.agent.core.boot.ServiceManager;
 
@@ -10,14 +25,14 @@
  * @since 2022/2/21
  */
 public class CPUStatCollector {
-    private final CPUUsageProvider cpuUsageProvider;
+
+    private final ProcessCPUUsageProvider processCPUUsageProvider;
 
     public CPUStatCollector() {
-        this.cpuUsageProvider = new CPUUsageProvider();
+        this.processCPUUsageProvider = new ProcessCPUUsageProvider();
     }
 
     public void collect() {
-        Metric metric = cpuUsageProvider.get();
-        ServiceManager.INSTANCE.findService(MetricDispatcher.class).dispatch(metric);
+        ServiceManager.INSTANCE.findService(MetricDispatcher.class).dispatch(processCPUUsageProvider.get());
     }
 }

agent-plugins/agent-cpu-plugin/src/main/java/cloud/erda/agent/plugin/cpu/CPUStatCollector.java

@@ -1,3 +1,19 @@
+/*
+ * Copyright (c) 2021 Terminus, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package cloud.erda.agent.plugin.cpu;
 
 import cloud.erda.agent.core.utils.PluginConstants;

agent-plugins/agent-cpu-plugin/src/main/java/cloud/erda/agent/plugin/cpu/CPUStatService.java

@@ -0,0 +1,75 @@
+/*
+ * Copyright (c) 2021 Terminus, Inc.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+package cloud.erda.agent.plugin.cpu;
+
+import cloud.erda.agent.core.metrics.Metric;
+import cloud.erda.agent.core.utils.DateTime;
+
+import java.lang.management.ManagementFactory;
+import java.lang.management.ThreadMXBean;
+
+public class ProcessCPUUsageProvider implements CPUDefaultProvider {
+
+    private static final String METRIC_NAME = "process_cpu_usage";
+    private final ThreadMXBean threadBean;
+    boolean isFirst = true;
+    private long preNanoTime = System.nanoTime();
+    private long preUsedNanoTime = 0;
+
+    public ProcessCPUUsageProvider() {
+        threadBean = ManagementFactory.getThreadMXBean();
+    }
+
+    public Metric get() {
+        Metric metric = Metric.New(METRIC_NAME, DateTime.currentTimeNano()).
+                addField("cpu_usage", this.calculateProcessCpuUsage());
+        this.addDefaultTags(metric);
+        return metric;
+    }
+
+    /**
+     * Calculate the CPU usage of all threads in the current process
+     *
+     * @return calculateProcessCpuUsage
+     */
+    public double calculateProcessCpuUsage() {
+        long currentNanoTime = System.nanoTime();
+        long processUsedTotalNanoTime = 0;
+        for (long id : threadBean.getAllThreadIds()) {
+            processUsedTotalNanoTime += threadBean.getThreadCpuTime(id);
+        }
+        long usedNanoTime = processUsedTotalNanoTime - preUsedNanoTime;
+
+        if (isFirst) {
+            isFirst = false;
+            preNanoTime = currentNanoTime;
+            preUsedNanoTime = processUsedTotalNanoTime;
+            return 0;
+        }
+
+        long totalPassedNanoTime = currentNanoTime - preNanoTime;
+
+        preNanoTime = currentNanoTime;
+        preUsedNanoTime = processUsedTotalNanoTime;
+
+        double cpuUsage = ((double) usedNanoTime) / totalPassedNanoTime * 100;
+        if (cpuUsage < 0) {
+            return 0;
+        }
+        return (double) Math.round(cpuUsage * 100) / 100;
+    }
+}