/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You 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.apache.camel.management.mbean;
import java.io.InputStream;
import java.util.Collection;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;
import java.util.concurrent.TimeUnit;
import javax.management.MBeanServer;
import javax.management.MBeanServerInvocationHandler;
import javax.management.ObjectName;
import org.apache.camel.CamelContext;
import org.apache.camel.Endpoint;
import org.apache.camel.ManagementStatisticsLevel;
import org.apache.camel.ProducerTemplate;
import org.apache.camel.TimerListener;
import org.apache.camel.api.management.ManagedResource;
import org.apache.camel.api.management.mbean.ManagedCamelContextMBean;
import org.apache.camel.api.management.mbean.ManagedProcessorMBean;
import org.apache.camel.api.management.mbean.ManagedRouteMBean;
import org.apache.camel.model.ModelCamelContext;
import org.apache.camel.model.ModelHelper;
import org.apache.camel.model.RouteDefinition;
import org.apache.camel.model.RoutesDefinition;
import org.apache.camel.util.CamelContextHelper;
/**
* @version
*/
@ManagedResource(description = "Managed CamelContext")
public class ManagedCamelContext extends ManagedPerformanceCounter implements TimerListener, ManagedCamelContextMBean {
private final ModelCamelContext context;
private final LoadTriplet load = new LoadTriplet();
public ManagedCamelContext(ModelCamelContext context) {
this.context = context;
boolean enabled = context.getManagementStrategy().getStatisticsLevel() != ManagementStatisticsLevel.Off;
setStatisticsEnabled(enabled);
}
public CamelContext getContext() {
return context;
}
public String getCamelId() {
return context.getName();
}
public String getManagementName() {
return context.getManagementName();
}
public String getCamelVersion() {
return context.getVersion();
}
public String getState() {
return context.getStatus().name();
}
public String getUptime() {
return context.getUptime();
}
public Map<String, String> getProperties() {
if (context.getProperties().isEmpty()) {
return null;
}
return context.getProperties();
}
public String getProperty(String name) throws Exception {
return context.getProperty(name);
}
public void setProperty(String name, String value) throws Exception {
context.getProperties().put(name, value);
}
public Boolean getTracing() {
return context.isTracing();
}
public void setTracing(Boolean tracing) {
context.setTracing(tracing);
}
public Integer getInflightExchanges() {
return context.getInflightRepository().size();
}
public void setTimeout(long timeout) {
context.getShutdownStrategy().setTimeout(timeout);
}
public long getTimeout() {
return context.getShutdownStrategy().getTimeout();
}
public void setTimeUnit(TimeUnit timeUnit) {
context.getShutdownStrategy().setTimeUnit(timeUnit);
}
public TimeUnit getTimeUnit() {
return context.getShutdownStrategy().getTimeUnit();
}
public void setShutdownNowOnTimeout(boolean shutdownNowOnTimeout) {
context.getShutdownStrategy().setShutdownNowOnTimeout(shutdownNowOnTimeout);
}
public boolean isShutdownNowOnTimeout() {
return context.getShutdownStrategy().isShutdownNowOnTimeout();
}
public String getLoad01() {
return String.format("%.2f", load.getLoad1());
}
public String getLoad05() {
return String.format("%.2f", load.getLoad5());
}
public String getLoad15() {
return String.format("%.2f", load.getLoad15());
}
public void onTimer() {
load.update(getInflightExchanges());
}
public void start() throws Exception {
if (context.isSuspended()) {
context.resume();
} else {
context.start();
}
}
public void stop() throws Exception {
context.stop();
}
public void restart() throws Exception {
context.stop();
context.start();
}
public void suspend() throws Exception {
context.suspend();
}
public void resume() throws Exception {
if (context.isSuspended()) {
context.resume();
} else {
throw new IllegalStateException("CamelContext is not suspended");
}
}
public void sendBody(String endpointUri, Object body) throws Exception {
ProducerTemplate template = context.createProducerTemplate();
try {
template.sendBody(endpointUri, body);
} finally {
template.stop();
}
}
public void sendStringBody(String endpointUri, String body) throws Exception {
sendBody(endpointUri, body);
}
public void sendBodyAndHeaders(String endpointUri, Object body, Map<String, Object> headers) throws Exception {
ProducerTemplate template = context.createProducerTemplate();
try {
template.sendBodyAndHeaders(endpointUri, body, headers);
} finally {
template.stop();
}
}
public Object requestBody(String endpointUri, Object body) throws Exception {
ProducerTemplate template = context.createProducerTemplate();
Object answer = null;
try {
answer = template.requestBody(endpointUri, body);
} finally {
template.stop();
}
return answer;
}
public Object requestStringBody(String endpointUri, String body) throws Exception {
return requestBody(endpointUri, body);
}
public Object requestBodyAndHeaders(String endpointUri, Object body, Map<String, Object> headers) throws Exception {
ProducerTemplate template = context.createProducerTemplate();
Object answer = null;
try {
answer = template.requestBodyAndHeaders(endpointUri, body, headers);
} finally {
template.stop();
}
return answer;
}
public String dumpRoutesAsXml() throws Exception {
List<RouteDefinition> routes = context.getRouteDefinitions();
if (routes.isEmpty()) {
return null;
}
// use a routes definition to dump the routes
RoutesDefinition def = new RoutesDefinition();
def.setRoutes(routes);
return ModelHelper.dumpModelAsXml(def);
}
public void addOrUpdateRoutesFromXml(String xml) throws Exception {
// convert to model from xml
InputStream is = context.getTypeConverter().mandatoryConvertTo(InputStream.class, xml);
RoutesDefinition def = context.loadRoutesDefinition(is);
if (def == null) {
return;
}
// add will remove existing route first
context.addRouteDefinitions(def.getRoutes());
}
public String dumpRoutesStatsAsXml(boolean fullStats, boolean includeProcessors) throws Exception {
StringBuilder sb = new StringBuilder();
sb.append("<camelContextStat").append(String.format(" id=\"%s\"", getCamelId()));
// use substring as we only want the attributes
String stat = dumpStatsAsXml(fullStats);
sb.append(" ").append(stat.substring(7, stat.length() - 2)).append(">\n");
MBeanServer server = getContext().getManagementStrategy().getManagementAgent().getMBeanServer();
if (server != null) {
// gather all the routes for this CamelContext, which requires JMX
ObjectName query = ObjectName.getInstance("org.apache.camel:context=*/" + getContext().getManagementName() + ",type=routes,*");
Set<ObjectName> routes = server.queryNames(query, null);
Set<ManagedProcessorMBean> processors = new LinkedHashSet<ManagedProcessorMBean>();
if (includeProcessors) {
// gather all the processors for this CamelContext, which requires JMX
query = ObjectName.getInstance("org.apache.camel:context=*/" + getContext().getManagementName() + ",type=processors,*");
Set<ObjectName> names = server.queryNames(query, null);
for (ObjectName on : names) {
ManagedProcessorMBean processor = MBeanServerInvocationHandler.newProxyInstance(server, on, ManagedProcessorMBean.class, true);
processors.add(processor);
}
}
// loop the routes, and append the processor stats if needed
sb.append(" <routeStats>\n");
for (ObjectName on : routes) {
ManagedRouteMBean route = MBeanServerInvocationHandler.newProxyInstance(server, on, ManagedRouteMBean.class, true);
sb.append(" <routeStat").append(String.format(" id=\"%s\"", route.getRouteId()));
// use substring as we only want the attributes
stat = route.dumpStatsAsXml(fullStats);
sb.append(" ").append(stat.substring(7, stat.length() - 2)).append(">\n");
// add processor details if needed
if (includeProcessors) {
sb.append(" <processorStats>\n");
for (ManagedProcessorMBean processor : processors) {
// the processor must belong to this route
if (route.getRouteId().equals(processor.getRouteId())) {
sb.append(" <processorStat").append(String.format(" id=\"%s\"", processor.getProcessorId()));
// use substring as we only want the attributes
sb.append(" ").append(processor.dumpStatsAsXml(fullStats).substring(7)).append("\n");
}
}
sb.append(" </processorStats>\n");
}
sb.append(" </routeStat>\n");
}
sb.append(" </routeStats>\n");
}
sb.append("</camelContextStat>");
return sb.toString();
}
public boolean createEndpoint(String uri) throws Exception {
if (context.hasEndpoint(uri) != null) {
// endpoint already exists
return false;
}
Endpoint endpoint = context.getEndpoint(uri);
if (endpoint != null) {
// ensure endpoint is registered, as the management strategy could have been configured to not always
// register new endpoints in JMX, so we need to check if its registered, and if not register it manually
ObjectName on = context.getManagementStrategy().getManagementNamingStrategy().getObjectNameForEndpoint(endpoint);
if (on != null && !context.getManagementStrategy().getManagementAgent().isRegistered(on)) {
// register endpoint as mbean
Object me = context.getManagementStrategy().getManagementObjectStrategy().getManagedObjectForEndpoint(context, endpoint);
context.getManagementStrategy().getManagementAgent().register(me, on);
}
return true;
} else {
return false;
}
}
public int removeEndpoints(String pattern) throws Exception {
// endpoints is always removed from JMX if removed from context
Collection<Endpoint> removed = context.removeEndpoints(pattern);
return removed.size();
}
public Map<String, Properties> findComponents() throws Exception {
return CamelContextHelper.findComponents(context);
}
}