org.glowroot.jvm.HeapDumps Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of glowroot-test-harness Show documentation
Glowroot Test Harness
The newest version!
/*
 * Copyright 2013-2015 the original author or authors.
 *
 * 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 org.glowroot.jvm;

import javax.management.InstanceNotFoundException;
import javax.management.MalformedObjectNameException;
import javax.management.ObjectName;

import org.glowroot.shaded.google.common.base.Supplier;
import org.glowroot.shaded.slf4j.Logger;
import org.glowroot.shaded.slf4j.LoggerFactory;

public class HeapDumps {

    private static final String MBEAN_NAME = "com.sun.management:type=HotSpotDiagnostic";

    private static final Logger logger = LoggerFactory.getLogger(HeapDumps.class);

    private final LazyPlatformMBeanServer lazyPlatformMBeanServer;
    private final ObjectName objectName;

    private HeapDumps(LazyPlatformMBeanServer lazyPlatformMBeanServer, ObjectName objectName) {
        this.lazyPlatformMBeanServer = lazyPlatformMBeanServer;
        this.objectName = objectName;
    }

    public void dumpHeap(String path) throws Exception {
        lazyPlatformMBeanServer.invoke(objectName, "dumpHeap", new Object[] {path, false},
                new String[] {"java.lang.String", "boolean"});
    }

    static OptionalService create(
            final LazyPlatformMBeanServer lazyPlatformMBeanServer) {
        final ObjectName objectName;
        try {
            objectName = ObjectName.getInstance(MBEAN_NAME);
        } catch (MalformedObjectNameException e) {
            logger.error(e.getMessage(), e);
            return OptionalService.unavailable("");
        }

        return OptionalService.lazy(new Supplier>() {
            @Override
            public OptionalService get() {
                // verify that mbean exists
                try {
                    lazyPlatformMBeanServer.getObjectInstance(objectName);
                } catch (InstanceNotFoundException e) {
                    // log exception at debug level
                    logger.debug(e.getMessage(), e);
                    return OptionalService.unavailable(
                            "No such MBean " + MBEAN_NAME + " (introduced in Oracle Java SE 6)");
                } catch (Exception e) {
                    logger.error(e.getMessage(), e);
                    return OptionalService.unavailable("");
                }
                return OptionalService.available(
                        new HeapDumps(lazyPlatformMBeanServer, objectName));
            }
        });

    }
}