/*
* $Id: YourKitProfilerService.java 21939 2011-05-18 13:32:09Z aperepel $
* --------------------------------------------------------------------------------------
* Copyright (c) MuleSoft, Inc. All rights reserved. http://www.mulesoft.com
*
* The software in this package is published under the terms of the CPAL v1.0
* license, a copy of which has been included with this distribution in the
* LICENSE.txt file.
*/
package org.mule.module.management.mbean;
import org.mule.module.management.i18n.ManagementMessages;
import com.yourkit.api.Controller;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
public class YourKitProfilerService implements YourKitProfilerServiceMBean
{
/**
* logger used by this class
*/
protected transient Log logger = LogFactory.getLog(getClass());
private final Controller controller;
private final AtomicBoolean capturing = new AtomicBoolean(false);
public YourKitProfilerService() throws Exception
{
controller = new Controller();
}
/**
* {@inheritDoc}
*/
public String getHost()
{
return controller.getHost();
}
/**
* {@inheritDoc}
*/
public int getPort()
{
return controller.getPort();
}
/**
* {@inheritDoc}
*/
public String captureMemorySnapshot() throws Exception
{
return controller.captureMemorySnapshot();
}
/**
* {@inheritDoc}
*/
public String captureSnapshot(long snapshotFlags) throws Exception
{
return controller.captureSnapshot(snapshotFlags);
}
/**
* {@inheritDoc}
*/
public void startAllocationRecording(long mode) throws Exception
{
if (ALLOCATION_RECORDING_ADAPTIVE != mode && ALLOCATION_RECORDING_ALL != mode)
{
throw new IllegalArgumentException("Invalid allocation recording mode requested: " + mode);
}
if (mode == ALLOCATION_RECORDING_ALL)
{
controller.startAllocationRecording(true, 1, false, 0);
}
else
{
// adaptive, record every 10th object OR above 1MB in size
controller.startAllocationRecording(true, 10, true, 1048576);
}
}
/**
* {@inheritDoc}
*/
public void stopAllocationRecording() throws Exception
{
controller.stopAllocationRecording();
}
/**
* {@inheritDoc}
*/
public void startCPUProfiling(long mode, String filters) throws Exception
{
controller.startCPUProfiling(mode, filters);
}
/**
* {@inheritDoc}
*/
public void stopCPUProfiling() throws Exception
{
controller.stopCPUProfiling();
}
/**
* {@inheritDoc}
*/
public void startMonitorProfiling() throws Exception
{
controller.startMonitorProfiling();
}
/**
* {@inheritDoc}
*/
public void stopMonitorProfiling() throws Exception
{
controller.stopMonitorProfiling();
}
/**
* {@inheritDoc}
*/
public void advanceGeneration(String description) throws Exception
{
controller.advanceGeneration(description);
}
/**
* {@inheritDoc}
*/
public String forceGC() throws Exception
{
final long[] heapSizes = controller.forceGC();
return ManagementMessages.forceGC(heapSizes).getMessage();
}
/**
* {@inheritDoc}
*/
public void startCapturingMemSnapshot(final int seconds)
{
if (!this.capturing.compareAndSet(false, true))
{
return;
}
final Thread thread = new Thread(new Runnable()
{
public void run()
{
try
{
while (capturing.get())
{
controller.captureMemorySnapshot();
Thread.sleep(seconds * 1000 /* millis in second */);
}
}
catch (final Exception e)
{
logger.error("Failed to capture memory snapshot", e);
}
}
});
thread.setDaemon(true); // let the application normally terminate
thread.start();
}
/**
* {@inheritDoc}
*/
public void stopCapturingMemSnapshot()
{
this.capturing.set(false);
}
/**
* {@inheritDoc}
*/
public long getStatus() throws java.lang.Exception
{
return (this.capturing.get())
? (controller.getStatus() | SNAPSHOT_CAPTURING)
: controller.getStatus();
}
}