001/** 002 * Licensed to the Apache Software Foundation (ASF) under one or more 003 * contributor license agreements. See the NOTICE file distributed with 004 * this work for additional information regarding copyright ownership. 005 * The ASF licenses this file to You under the Apache License, Version 2.0 006 * (the "License"); you may not use this file except in compliance with 007 * the License. You may obtain a copy of the License at 008 * 009 * http://www.apache.org/licenses/LICENSE-2.0 010 * 011 * Unless required by applicable law or agreed to in writing, software 012 * distributed under the License is distributed on an "AS IS" BASIS, 013 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 014 * See the License for the specific language governing permissions and 015 * limitations under the License. 016 */ 017package org.apache.camel.management.mbean; 018 019import java.util.Map; 020import javax.management.openmbean.CompositeData; 021import javax.management.openmbean.CompositeDataSupport; 022import javax.management.openmbean.CompositeType; 023import javax.management.openmbean.TabularData; 024import javax.management.openmbean.TabularDataSupport; 025 026import org.apache.camel.CamelContext; 027import org.apache.camel.api.management.ManagedResource; 028import org.apache.camel.api.management.mbean.CamelOpenMBeanTypes; 029import org.apache.camel.api.management.mbean.ManagedPollEnricherMBean; 030import org.apache.camel.model.PollEnrichDefinition; 031import org.apache.camel.processor.PollEnricher; 032import org.apache.camel.spi.EndpointUtilizationStatistics; 033import org.apache.camel.spi.ManagementStrategy; 034import org.apache.camel.util.ObjectHelper; 035import org.apache.camel.util.URISupport; 036 037/** 038 * @version 039 */ 040@ManagedResource(description = "Managed PollEnricher") 041public class ManagedPollEnricher extends ManagedProcessor implements ManagedPollEnricherMBean { 042 private final PollEnricher processor; 043 private String uri; 044 private boolean sanitize; 045 046 public ManagedPollEnricher(CamelContext context, PollEnricher processor, PollEnrichDefinition definition) { 047 super(context, processor, definition); 048 this.processor = processor; 049 } 050 051 public void init(ManagementStrategy strategy) { 052 super.init(strategy); 053 sanitize = strategy.getManagementAgent().getMask() != null ? strategy.getManagementAgent().getMask() : false; 054 uri = getDefinition().getExpression().getExpression(); 055 if (sanitize) { 056 uri = URISupport.sanitizeUri(uri); 057 } 058 } 059 060 @Override 061 public void reset() { 062 super.reset(); 063 if (processor.getEndpointUtilizationStatistics() != null) { 064 processor.getEndpointUtilizationStatistics().clear(); 065 } 066 } 067 068 @Override 069 public PollEnrichDefinition getDefinition() { 070 return (PollEnrichDefinition) super.getDefinition(); 071 } 072 073 @Override 074 public PollEnricher getProcessor() { 075 return processor; 076 } 077 078 @Override 079 public String getExpressionLanguage() { 080 return getDefinition().getExpression().getLanguage(); 081 } 082 083 @Override 084 public String getExpression() { 085 return uri; 086 } 087 088 @Override 089 public Long getTimeout() { 090 return processor.getTimeout(); 091 } 092 093 @Override 094 public Integer getCacheSize() { 095 return processor.getCacheSize(); 096 } 097 098 @Override 099 public Boolean isIgnoreInvalidEndpoint() { 100 return processor.isIgnoreInvalidEndpoint(); 101 } 102 103 @Override 104 public Boolean isAggregateOnException() { 105 return processor.isAggregateOnException(); 106 } 107 108 @Override 109 public TabularData extendedInformation() { 110 try { 111 TabularData answer = new TabularDataSupport(CamelOpenMBeanTypes.endpointsUtilizationTabularType()); 112 113 EndpointUtilizationStatistics stats = processor.getEndpointUtilizationStatistics(); 114 if (stats != null) { 115 for (Map.Entry<String, Long> entry : stats.getStatistics().entrySet()) { 116 CompositeType ct = CamelOpenMBeanTypes.endpointsUtilizationCompositeType(); 117 String url = entry.getKey(); 118 if (sanitize) { 119 url = URISupport.sanitizeUri(url); 120 } 121 122 Long hits = entry.getValue(); 123 if (hits == null) { 124 hits = 0L; 125 } 126 127 CompositeData data = new CompositeDataSupport(ct, new String[]{"url", "hits"}, new Object[]{url, hits}); 128 answer.put(data); 129 } 130 } 131 return answer; 132 } catch (Exception e) { 133 throw ObjectHelper.wrapRuntimeCamelException(e); 134 } 135 } 136 137}