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; 020 021import javax.management.openmbean.CompositeData; 022import javax.management.openmbean.CompositeDataSupport; 023import javax.management.openmbean.CompositeType; 024import javax.management.openmbean.TabularData; 025import javax.management.openmbean.TabularDataSupport; 026 027import org.apache.camel.CamelContext; 028import org.apache.camel.RuntimeCamelException; 029import org.apache.camel.api.management.ManagedResource; 030import org.apache.camel.api.management.mbean.CamelOpenMBeanTypes; 031import org.apache.camel.api.management.mbean.ManagedPollEnricherMBean; 032import org.apache.camel.model.PollEnrichDefinition; 033import org.apache.camel.processor.PollEnricher; 034import org.apache.camel.spi.EndpointUtilizationStatistics; 035import org.apache.camel.spi.ManagementStrategy; 036import org.apache.camel.util.URISupport; 037 038@ManagedResource(description = "Managed PollEnricher") 039public class ManagedPollEnricher extends ManagedProcessor implements ManagedPollEnricherMBean { 040 private final PollEnricher processor; 041 private String uri; 042 private boolean sanitize; 043 044 public ManagedPollEnricher(CamelContext context, PollEnricher processor, PollEnrichDefinition definition) { 045 super(context, processor, definition); 046 this.processor = processor; 047 } 048 049 @Override 050 public void init(ManagementStrategy strategy) { 051 super.init(strategy); 052 sanitize = strategy.getManagementAgent().getMask() != null ? strategy.getManagementAgent().getMask() : false; 053 uri = getDefinition().getExpression().getExpression(); 054 if (sanitize) { 055 uri = URISupport.sanitizeUri(uri); 056 } 057 } 058 059 @Override 060 public void reset() { 061 super.reset(); 062 if (processor.getEndpointUtilizationStatistics() != null) { 063 processor.getEndpointUtilizationStatistics().clear(); 064 } 065 } 066 067 @Override 068 public PollEnrichDefinition getDefinition() { 069 return (PollEnrichDefinition) super.getDefinition(); 070 } 071 072 @Override 073 public PollEnricher getProcessor() { 074 return processor; 075 } 076 077 @Override 078 public String getExpressionLanguage() { 079 return getDefinition().getExpression().getLanguage(); 080 } 081 082 @Override 083 public String getExpression() { 084 return uri; 085 } 086 087 @Override 088 public Long getTimeout() { 089 return processor.getTimeout(); 090 } 091 092 @Override 093 public Integer getCacheSize() { 094 return processor.getCacheSize(); 095 } 096 097 @Override 098 public Boolean isIgnoreInvalidEndpoint() { 099 return processor.isIgnoreInvalidEndpoint(); 100 } 101 102 @Override 103 public Boolean isAggregateOnException() { 104 return processor.isAggregateOnException(); 105 } 106 107 @Override 108 public TabularData extendedInformation() { 109 try { 110 TabularData answer = new TabularDataSupport(CamelOpenMBeanTypes.endpointsUtilizationTabularType()); 111 112 EndpointUtilizationStatistics stats = processor.getEndpointUtilizationStatistics(); 113 if (stats != null) { 114 for (Map.Entry<String, Long> entry : stats.getStatistics().entrySet()) { 115 CompositeType ct = CamelOpenMBeanTypes.endpointsUtilizationCompositeType(); 116 String url = entry.getKey(); 117 if (sanitize) { 118 url = URISupport.sanitizeUri(url); 119 } 120 121 Long hits = entry.getValue(); 122 if (hits == null) { 123 hits = 0L; 124 } 125 126 CompositeData data 127 = 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 RuntimeCamelException.wrapRuntimeCamelException(e); 134 } 135 } 136 137}