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.activemq.network; 018 019import java.util.Map; 020import java.util.concurrent.ConcurrentHashMap; 021 022import javax.management.MalformedObjectNameException; 023import javax.management.ObjectName; 024import org.apache.activemq.broker.BrokerService; 025import org.apache.activemq.broker.jmx.AnnotatedMBean; 026import org.apache.activemq.broker.jmx.BrokerMBeanSupport; 027import org.apache.activemq.broker.jmx.NetworkBridgeView; 028import org.apache.activemq.command.Message; 029import org.slf4j.Logger; 030import org.slf4j.LoggerFactory; 031 032public class MBeanNetworkListener implements NetworkBridgeListener { 033 034 private static final Logger LOG = LoggerFactory.getLogger(MBeanNetworkListener.class); 035 036 private final BrokerService brokerService; 037 private final ObjectName connectorName; 038 private final NetworkBridgeConfiguration networkBridgeConfiguration; 039 private boolean createdByDuplex = false; 040 private Map<NetworkBridge,MBeanBridgeDestination> destinationObjectNameMap = new ConcurrentHashMap<NetworkBridge,MBeanBridgeDestination>(); 041 042 public MBeanNetworkListener(BrokerService brokerService, NetworkBridgeConfiguration networkBridgeConfiguration, ObjectName connectorName) { 043 this.brokerService = brokerService; 044 this.networkBridgeConfiguration = networkBridgeConfiguration; 045 this.connectorName = connectorName; 046 } 047 048 @Override 049 public void bridgeFailed() { 050 } 051 052 @Override 053 public void onStart(NetworkBridge bridge) { 054 if (!brokerService.isUseJmx()) { 055 return; 056 } 057 NetworkBridgeView view = new NetworkBridgeView(bridge); 058 view.setCreateByDuplex(createdByDuplex); 059 try { 060 ObjectName objectName = createNetworkBridgeObjectName(bridge); 061 AnnotatedMBean.registerMBean(brokerService.getManagementContext(), view, objectName); 062 bridge.setMbeanObjectName(objectName); 063 MBeanBridgeDestination mBeanBridgeDestination = new MBeanBridgeDestination(brokerService,networkBridgeConfiguration,bridge,view); 064 destinationObjectNameMap.put(bridge,mBeanBridgeDestination); 065 mBeanBridgeDestination.start(); 066 LOG.debug("registered: {} as: {}", bridge, objectName); 067 } catch (Throwable e) { 068 LOG.debug("Network bridge could not be registered in JMX: {}", e.getMessage(), e); 069 } 070 } 071 072 @Override 073 public void onStop(NetworkBridge bridge) { 074 if (!brokerService.isUseJmx()) { 075 return; 076 } 077 try { 078 ObjectName objectName = bridge.getMbeanObjectName(); 079 if (objectName != null) { 080 brokerService.getManagementContext().unregisterMBean(objectName); 081 } 082 MBeanBridgeDestination mBeanBridgeDestination = destinationObjectNameMap.remove(bridge); 083 if (mBeanBridgeDestination != null){ 084 mBeanBridgeDestination.stop(); 085 } 086 } catch (Throwable e) { 087 LOG.debug("Network bridge could not be unregistered in JMX: {}", e.getMessage(), e); 088 } 089 } 090 091 092 093 protected ObjectName createNetworkBridgeObjectName(NetworkBridge bridge) throws MalformedObjectNameException { 094 return BrokerMBeanSupport.createNetworkBridgeObjectName(connectorName, bridge.getRemoteAddress()); 095 } 096 097 public void setCreatedByDuplex(boolean createdByDuplex) { 098 this.createdByDuplex = createdByDuplex; 099 } 100 101 102 103 @Override 104 public void onOutboundMessage(NetworkBridge bridge,Message message) { 105 MBeanBridgeDestination mBeanBridgeDestination = destinationObjectNameMap.get(bridge); 106 if (mBeanBridgeDestination != null){ 107 mBeanBridgeDestination.onOutboundMessage(message); 108 } 109 } 110 111 @Override 112 public void onInboundMessage(NetworkBridge bridge,Message message) { 113 MBeanBridgeDestination mBeanBridgeDestination = destinationObjectNameMap.get(bridge); 114 if (mBeanBridgeDestination != null){ 115 mBeanBridgeDestination.onInboundMessage(message); 116 } 117 } 118 119}