content: /**
   * 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.activemq.broker.inteceptor;
  
  import java.util.Set;
  
  import org.apache.activemq.broker.Broker;
  import org.apache.activemq.broker.BrokerFilter;
  import org.apache.activemq.broker.ConnectionContext;
  import org.apache.activemq.broker.ProducerBrokerExchange;
  import org.apache.activemq.command.ActiveMQDestination;
  import org.apache.activemq.command.Message;
  import org.apache.activemq.command.ProducerInfo;
  import org.apache.activemq.filter.DestinationMap;
  import org.apache.activemq.state.ProducerState;
  
  class MessageInterceptorFilter extends BrokerFilter {
      private DestinationMap interceptorMap = new DestinationMap();
  
      MessageInterceptorFilter(Broker next) {
          super(next);
      }
  
  
      MessageInterceptor addMessageInterceptor(String destinationName, MessageInterceptor messageInterceptor) {
          ActiveMQDestination activeMQDestination = ActiveMQDestination.createDestination(destinationName, ActiveMQDestination.QUEUE_TYPE);
          interceptorMap.put(activeMQDestination, messageInterceptor);
          return messageInterceptor;
      }
  
      void removeMessageInterceptor(String destinationName, MessageInterceptor interceptor) {
          ActiveMQDestination activeMQDestination = ActiveMQDestination.createDestination(destinationName, ActiveMQDestination.QUEUE_TYPE);
          interceptorMap.remove(activeMQDestination, interceptor);
      }
  
  
      MessageInterceptor addMessageInterceptorForQueue(String destinationName, MessageInterceptor messageInterceptor) {
          ActiveMQDestination activeMQDestination = ActiveMQDestination.createDestination(destinationName, ActiveMQDestination.QUEUE_TYPE);
          interceptorMap.put(activeMQDestination, messageInterceptor);
          return messageInterceptor;
      }
  
      void removeMessageInterceptorForQueue(String destinationName, MessageInterceptor interceptor) {
          ActiveMQDestination activeMQDestination = ActiveMQDestination.createDestination(destinationName, ActiveMQDestination.QUEUE_TYPE);
          interceptorMap.remove(activeMQDestination, interceptor);
      }
  
  
      MessageInterceptor addMessageInterceptorForTopic(String destinationName, MessageInterceptor messageInterceptor) {
          ActiveMQDestination activeMQDestination = ActiveMQDestination.createDestination(destinationName, ActiveMQDestination.TOPIC_TYPE);
          interceptorMap.put(activeMQDestination, messageInterceptor);
          return messageInterceptor;
      }
  
      void removeMessageInterceptorForTopic(String destinationName, MessageInterceptor interceptor) {
          ActiveMQDestination activeMQDestination = ActiveMQDestination.createDestination(destinationName, ActiveMQDestination.TOPIC_TYPE);
          interceptorMap.remove(activeMQDestination, interceptor);
      }
  
      MessageInterceptor addMessageInterceptor(ActiveMQDestination activeMQDestination, MessageInterceptor messageInterceptor) {
          interceptorMap.put(activeMQDestination, messageInterceptor);
          return messageInterceptor;
      }
  
      void removeMessageInterceptor(ActiveMQDestination activeMQDestination, MessageInterceptor interceptor) {
          interceptorMap.remove(activeMQDestination, interceptor);
      }
  
  
      /**
       * Re-inject into the Broker chain
       */
  
      void injectMessage(ProducerBrokerExchange producerExchange, final Message messageSend) throws Exception {
          ProducerBrokerExchange pe = producerExchange;
          if (pe == null) {
              pe = new ProducerBrokerExchange();
              ConnectionContext cc = new ConnectionContext();
              cc.setBroker(this.getRoot());
              pe.setConnectionContext(cc);
              pe.setMutable(true);
              pe.setProducerState(new ProducerState(new ProducerInfo()));
          }
          super.send(pe, messageSend);
      }
  
  
      @Override
      public void send(ProducerBrokerExchange producerExchange, final Message messageSend) throws Exception {
          ActiveMQDestination activeMQDestination = messageSend.getDestination();
          if (!interceptorMap.isEmpty() && activeMQDestination != null) {
              Set<MessageInterceptor> set = interceptorMap.get(activeMQDestination);
              if (set != null && !set.isEmpty()) {
                  for (MessageInterceptor mi : set) {
                      mi.intercept(producerExchange, messageSend);
                  }
              } else {
                  super.send(producerExchange, messageSend);
              }
  
          } else {
              super.send(producerExchange, messageSend);
          }
      }
  }
  