add perfcounter command. #1029

core/src/main/java/com/taobao/arthas/core/command/BuiltinCommandPack.java

@@ -34,6 +34,7 @@ import com.taobao.arthas.core.command.monitor200.HeapDumpCommand;
 import com.taobao.arthas.core.command.monitor200.JvmCommand;
 import com.taobao.arthas.core.command.monitor200.MBeanCommand;
 import com.taobao.arthas.core.command.monitor200.MonitorCommand;
+import com.taobao.arthas.core.command.monitor200.PerfCounterCommand;
 import com.taobao.arthas.core.command.monitor200.ProfilerCommand;
 import com.taobao.arthas.core.command.monitor200.StackCommand;
 import com.taobao.arthas.core.command.monitor200.ThreadCommand;
@@ -78,6 +79,7 @@ public class BuiltinCommandPack implements CommandResolver {
         commands.add(Command.create(WatchCommand.class));
         commands.add(Command.create(TimeTunnelCommand.class));
         commands.add(Command.create(JvmCommand.class));
+        commands.add(Command.create(PerfCounterCommand.class));
         // commands.add(Command.create(GroovyScriptCommand.class));
         commands.add(Command.create(OgnlCommand.class));
         commands.add(Command.create(MemoryCompilerCommand.class));

core/src/main/java/com/taobao/arthas/core/command/monitor200/PerfCounterCommand.java

@@ -0,0 +1,120 @@
+package com.taobao.arthas.core.command.monitor200;
+
+import static com.taobao.text.ui.Element.label;
+
+import java.lang.reflect.Method;
+import java.nio.ByteBuffer;
+import java.util.Collections;
+import java.util.List;
+
+import com.taobao.arthas.common.JavaVersionUtils;
+import com.taobao.arthas.common.PidUtils;
+import com.taobao.arthas.core.command.Constants;
+import com.taobao.arthas.core.shell.command.AnnotatedCommand;
+import com.taobao.arthas.core.shell.command.CommandProcess;
+import com.taobao.arthas.core.util.LogUtil;
+import com.taobao.middleware.cli.annotations.Description;
+import com.taobao.middleware.cli.annotations.Name;
+import com.taobao.middleware.cli.annotations.Option;
+import com.taobao.middleware.cli.annotations.Summary;
+import com.taobao.middleware.logger.Logger;
+import com.taobao.text.Decoration;
+import com.taobao.text.ui.TableElement;
+import com.taobao.text.util.RenderUtil;
+
+import sun.management.counter.Counter;
+import sun.management.counter.perf.PerfInstrumentation;
+
+/**
+ * @see sun.misc.Perf
+ * @see sun.management.counter.perf.PerfInstrumentation
+ * @author hengyunabc 2020-02-16
+ */
+@Name("perfcounter")
+@Summary("Display the perf counter infornation.")
+@Description(Constants.WIKI + Constants.WIKI_HOME + "perf")
+public class PerfCounterCommand extends AnnotatedCommand {
+    private static final Logger logger = LogUtil.getArthasLogger();
+    private static Object perfObject;
+    private static Method attachMethod;
+
+    private boolean details;
+
+    @Option(shortName = "d", longName = "details", flag = true)
+    @Description("print all perf counter details")
+    public void setDetails(boolean details) {
+        this.details = details;
+    }
+
+    @Override
+    public void process(CommandProcess process) {
+        try {
+            TableElement table = new TableElement().leftCellPadding(1).rightCellPadding(1);
+            if (this.details) {
+                table = new TableElement(3, 1, 1, 10).leftCellPadding(1).rightCellPadding(1);
+                table.row(true, label("Name").style(Decoration.bold.bold()),
+                        label("Variability").style(Decoration.bold.bold()),
+                        label("Units").style(Decoration.bold.bold()), label("Value").style(Decoration.bold.bold()));
+            }
+
+            List<Counter> perfCounters = getPerfCounters();
+            if (perfCounters.isEmpty()) {
+                process.write(
+                        "please check arthas log. if java version >=9 , try to add jvm options when start your process: "
+                                + "--add-opens java.base/jdk.internal.perf=ALL-UNNAMED "
+                                + "--add-exports java.base/jdk.internal.perf=ALL-UNNAMED\n");
+            } else {
+                for (Counter counter : perfCounters) {
+                    if (details) {
+                        table.row(counter.getName(), String.valueOf(counter.getVariability()),
+                                String.valueOf(counter.getUnits()), String.valueOf(counter.getValue()));
+                    } else {
+                        table.row(counter.getName(), String.valueOf(counter.getValue()));
+                    }
+                }
+            }
+
+            process.write(RenderUtil.render(table, process.width()));
+        } finally {
+            process.end();
+        }
+    }
+
+    private static List<Counter> getPerfCounters() {
+
+        /**
+         * <pre>
+         * Perf p = Perf.getPerf();
+         * ByteBuffer buffer = p.attach(pid, "r");
+         * </pre>
+         */
+        try {
+            if (perfObject == null) {
+                // jdk8
+                String perfClassName = "sun.misc.Perf";
+                // jdk 11
+                if (!JavaVersionUtils.isLessThanJava9()) {
+                    perfClassName = "jdk.internal.perf.Perf";
+                }
+
+                Class<?> perfClass = ClassLoader.getSystemClassLoader().loadClass(perfClassName);
+                Method getPerfMethod = perfClass.getDeclaredMethod("getPerf");
+                perfObject = getPerfMethod.invoke(null);
+            }
+
+            if (attachMethod == null) {
+                attachMethod = perfObject.getClass().getDeclaredMethod("attach",
+                        new Class<?>[] { int.class, String.class });
+            }
+
+            ByteBuffer buffer = (ByteBuffer) attachMethod.invoke(perfObject,
+                    new Object[] { (int) PidUtils.currentLongPid(), "r" });
+
+            PerfInstrumentation perfInstrumentation = new PerfInstrumentation(buffer);
+            return perfInstrumentation.getAllCounters();
+        } catch (Throwable e) {
+            logger.error("arthas", "get perf counter error", e);
+        }
+        return Collections.emptyList();
+    }
+}