MemoryDumpHelper -> community

2026-09-27 10:03:11 +07:00 · 2015-01-14 19:28:00 +01:00
parent ba5cf21e7f
commit fbec1a153c
1 changed files with 76 additions and 0 deletions
@@ -0,0 +1,76 @@
+package com.intellij.util;
+
+import com.intellij.openapi.diagnostic.Logger;
+import org.jetbrains.annotations.NotNull;
+
+import javax.management.MBeanServer;
+import javax.management.ObjectName;
+import java.lang.management.ManagementFactory;
+import java.lang.reflect.Method;
+import java.security.AccessController;
+import java.security.PrivilegedExceptionAction;
+import java.util.Iterator;
+import java.util.Set;
+
+/**
+ * @author Pavel.Sher
+ */
+public class MemoryDumpHelper {
+  private static final String HOT_SPOT_BEAN_NAME = "com.sun.management:type=HotSpotDiagnostic";
+
+  private static final Object ourMXBean;
+  private static final Method ourDumpHeap;
+
+  static {
+    Object mxBean;
+    Method dumpHeap;
+
+    try {
+      final Class hotSpotMxBeanClass = Class.forName("com.sun.management.HotSpotDiagnosticMXBean");
+
+      mxBean = AccessController.doPrivileged(new PrivilegedExceptionAction<Object>() {
+        public Object run() throws Exception {
+          MBeanServer server = ManagementFactory.getPlatformMBeanServer();
+          Set<ObjectName> s = server.queryNames(new ObjectName(HOT_SPOT_BEAN_NAME), null);
+          Iterator<ObjectName> itr = s.iterator();
+          if (itr.hasNext()) {
+            ObjectName name = itr.next();
+            return ManagementFactory.newPlatformMXBeanProxy(server, name.toString(), hotSpotMxBeanClass);
+          }
+          else {
+            return null;
+          }
+        }
+      });
+
+      dumpHeap = mxBean.getClass().getMethod("dumpHeap", String.class, boolean.class);
+    }
+    catch (Throwable t) {
+      Logger.getInstance("#com.intellij.util.MemoryDumpHelper").info(t.getMessage());
+      mxBean = null;
+      dumpHeap = null;
+    }
+
+    ourMXBean = mxBean;
+    ourDumpHeap = dumpHeap;
+  }
+
+  public static boolean memoryDumpAvailable() {
+    try {
+      return ourMXBean != null;
+    }
+    catch (UnsupportedOperationException e) {
+      return false;
+    }
+  }
+
+  public static synchronized void captureMemoryDump(@NotNull String dumpPath) throws Exception {
+    ourDumpHeap.invoke(ourMXBean, dumpPath, true);
+  }
+
+  /** @deprecated use {@link #captureMemoryDump(String)} (to be removed in IDEA 15) */
+  @SuppressWarnings({"UnusedDeclaration", "SpellCheckingInspection"})
+  public static Object getHotspotMBean() {
+    return ourMXBean;
+  }
+}