View Javadoc

    1   /***
    2    *
    3    * Copyright 2004 Protique Ltd
    4    *
    5    * Licensed under the Apache License, Version 2.0 (the "License");
    6    * you may not use this file except in compliance with the License.
    7    * You may obtain a copy of the License at
    8    *
    9    * http://www.apache.org/licenses/LICENSE-2.0
    10   *
    11   * Unless required by applicable law or agreed to in writing, software
    12   * distributed under the License is distributed on an "AS IS" BASIS,
    13   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
    14   * See the License for the specific language governing permissions and
    15   * limitations under the License.
    16   *
    17   **/
    18  package org.codehaus.activemq.service.impl;
    19
    20  import EDU.oswego.cs.dl.util.concurrent.ConcurrentHashMap;
    21  import org.codehaus.activemq.broker.BrokerClient;
    22  import org.codehaus.activemq.message.ActiveMQDestination;
    23  import org.codehaus.activemq.message.ActiveMQMessage;
    24  import org.codehaus.activemq.service.Dispatcher;
    25  import org.codehaus.activemq.service.MessageContainer;
    26  import org.codehaus.activemq.service.MessageContainerManager;
    27  import org.codehaus.activemq.service.Subscription;
    28
    29  import javax.jms.Destination;
    30  import javax.jms.JMSException;
    31  import java.util.Collections;
    32  import java.util.Iterator;
    33  import java.util.Map;
    34
    35  /***
    36   * @version $Revision: 1.5 $
    37   */
    38  public abstract class MessageContainerManagerSupport implements MessageContainerManager {
    39      protected Dispatcher dispatcher;
    40      protected Map messageContainers = new ConcurrentHashMap();
    41      private Map destinations = new ConcurrentHashMap();
    42      private boolean maintainDestinationStats = true;
    43
    44      public MessageContainerManagerSupport(Dispatcher dispatcher) {
    45          this.dispatcher = dispatcher;
    46          dispatcher.register(this);
    47      }
    48
    49      public Map getDestinations() {
    50          return Collections.unmodifiableMap(destinations);
    51      }
    52
    53      public void start() throws JMSException {
    54          dispatcher.start();
    55      }
    56
    57      public void stop() throws JMSException {
    58          dispatcher.stop();
    59          JMSException firstException = null;
    60          try {
    61              dispatcher.stop();
    62          }
    63          catch (JMSException e) {
    64              firstException = e;
    65          }
    66
    67          // lets stop all the containers
    68          for (Iterator iter = messageContainers.values().iterator(); iter.hasNext();) {
    69              MessageContainer container = (MessageContainer) iter.next();
    70              try {
    71                  container.stop();
    72              }
    73              catch (JMSException e) {
    74                  if (firstException == null) {
    75                      firstException = e;
    76                  }
    77              }
    78          }
    79          if (firstException != null) {
    80              throw firstException;
    81          }
    82
    83      }
    84
    85      public synchronized MessageContainer getContainer(String destinationName) throws JMSException {
    86          MessageContainer container = (MessageContainer) messageContainers.get(destinationName);
    87          if (container == null) {
    88              container = createContainer(destinationName);
    89              container.start();
    90              messageContainers.put(destinationName, container);
    91
    92              destinations.put(destinationName, createDestination(destinationName));
    93          }
    94          return container;
    95      }
    96
    97
    98      // Properties
    99      //-------------------------------------------------------------------------
    100     public boolean isMaintainDestinationStats() {
    101         return maintainDestinationStats;
    102     }
    103
    104     public void setMaintainDestinationStats(boolean maintainDestinationStats) {
    105         this.maintainDestinationStats = maintainDestinationStats;
    106     }
    107
    108     // Implementation methods
    109     //-------------------------------------------------------------------------
    110
    111     /***
    112      * Factory method to create a new {@link Destination}
    113      */
    114     protected abstract Destination createDestination(String destinationName);
    115
    116     /***
    117      * Factory method to create a new {@link MessageContainer}
    118      */
    119     protected abstract MessageContainer createContainer(String destinationName) throws JMSException;
    120
    121     /***
    122      * Loads the container for the given name and destination on startup
    123      */
    124     protected void loadContainer(String destinationName, Destination destination) throws JMSException {
    125         destinations.put(destinationName, destination);
    126
    127         MessageContainer container = createContainer(destinationName);
    128         container.start();
    129         messageContainers.put(destinationName, container);
    130     }
    131
    132     /***
    133      * Updates the message acknowledgement stats
    134      *
    135      * @param client
    136      * @param subscription
    137      */
    138     protected void updateAcknowledgeStats(BrokerClient client, Subscription subscription) {
    139         if (isMaintainDestinationStats()) {
    140             // lets lookup the destination which has the stats hanging off it
    141             String name = subscription.getDestination().getPhysicalName();
    142             ActiveMQDestination destination = (ActiveMQDestination) destinations.get(name);
    143             destination.getStats().onMessageAck();
    144         }
    145     }
    146
    147     /***
    148      * Updates the message sending stats
    149      *
    150      * @param client
    151      * @param message
    152      * @throws JMSException
    153      */
    154     protected void updateSendStats(BrokerClient client, ActiveMQMessage message) throws JMSException {
    155         if (isMaintainDestinationStats()) {
    156             // lets lookup the destination which has the stats hanging off it
    157             String name = message.getJMSActiveMQDestination().getPhysicalName();
    158             ActiveMQDestination destination = (ActiveMQDestination) destinations.get(name);
    159             if (destination != null){
    160                 destination.getStats().onMessageSend(message);
    161             }
    162         }
    163     }
    164 }