/*
* 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.api.MuleContext;
import org.mule.construct.AbstractFlowConstruct;
import org.mule.management.stats.FlowConstructStatistics;
import org.mule.module.management.support.AutoDiscoveryJmxSupportFactory;
import org.mule.module.management.support.JmxSupport;
import org.mule.module.management.support.JmxSupportFactory;
import javax.management.MBeanRegistration;
import javax.management.MBeanServer;
import javax.management.ObjectName;
import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
/**
* <code>FlowConstructService</code> exposes service information about a Mule Managed
* flow construct.
*/
public class FlowConstructService implements FlowConstructServiceMBean, MBeanRegistration, FlowConstructStatsMBean
{
private static Log LOGGER = LogFactory.getLog(FlowConstructService.class);
protected FlowConstructStatistics statistics;
protected MBeanServer server;
protected String name;
protected String type;
protected ObjectName statsName;
protected ObjectName objectName;
protected MuleContext muleContext;
// JmxSupport in order to build MBean's ObjectNames properly.
protected JmxSupportFactory jmxSupportFactory = AutoDiscoveryJmxSupportFactory.getInstance();
protected JmxSupport jmxSupport = jmxSupportFactory.getJmxSupport();
public FlowConstructService(String type, String name, MuleContext muleContext, FlowConstructStatistics statistics)
{
this.muleContext = muleContext;
this.type = type;
this.name = name;
this.statistics = statistics;
}
protected FlowConstructService(String type, String name, MuleContext muleContext)
{
this.muleContext = muleContext;
this.type = type;
this.name = name;
}
public String getName()
{
return name;
}
public String getType()
{
return type;
}
public ObjectName getStatistics()
{
return statsName;
}
public void clearStatistics()
{
statistics.clear();
}
public long getAsyncEventsReceived()
{
return statistics.getAsyncEventsReceived();
}
public long getSyncEventsReceived()
{
return statistics.getSyncEventsReceived();
}
public long getTotalEventsReceived()
{
return statistics.getTotalEventsReceived();
}
public long getAverageProcessingTime()
{
return statistics.getAverageProcessingTime();
}
public long getProcessedEvents()
{
return statistics.getProcessedEvents();
}
public long getMaxProcessingTime()
{
return statistics.getMaxProcessingTime();
}
public long getMinProcessingTime()
{
return statistics.getMinProcessingTime();
}
public long getTotalProcessingTime()
{
return statistics.getTotalProcessingTime();
}
public long getExecutionErrors()
{
return statistics.getExecutionErrors();
}
public long getFatalErrors()
{
return statistics.getFatalErrors();
}
public ObjectName preRegister(MBeanServer server, ObjectName name) throws Exception
{
this.server = server;
this.objectName = name;
return name;
}
public void postRegister(Boolean registrationDone)
{
AbstractFlowConstruct flow = muleContext.getRegistry().lookupObject(getName());
try
{
if (flow.getStatistics() != null)
{
statsName = jmxSupport.getObjectName(String.format("%s:type=org.mule.Statistics,%s=%s", objectName.getDomain(),
flow.getConstructType(), jmxSupport.escape(getName())));
// unregister old version if exists
if (this.server.isRegistered(statsName))
{
this.server.unregisterMBean(statsName);
}
this.server.registerMBean(new FlowConstructStats(flow.getStatistics()), this.statsName);
}
}
catch (Exception e)
{
LOGGER.error("Error post-registering the MBean", e);
}
}
public void preDeregister() throws Exception
{
try
{
if (this.server.isRegistered(statsName))
{
this.server.unregisterMBean(statsName);
}
}
catch (Exception ex)
{
LOGGER.error("Error unregistering ServiceService child " + statsName.getCanonicalName(), ex);
}
}
public void postDeregister()
{
// nothing to do
}
}