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 */
017 package org.apache.activemq;
018
019 import javax.jms.JMSException;
020
021 import org.apache.activemq.advisory.AdvisorySupport;
022 import org.apache.activemq.command.ActiveMQDestination;
023 import org.apache.activemq.command.ActiveMQTempDestination;
024 import org.apache.activemq.command.ConsumerId;
025 import org.apache.activemq.command.ConsumerInfo;
026 import org.apache.activemq.command.DataStructure;
027 import org.apache.activemq.command.DestinationInfo;
028 import org.apache.activemq.command.MessageAck;
029 import org.apache.activemq.command.MessageDispatch;
030 import org.slf4j.Logger;
031 import org.slf4j.LoggerFactory;
032
033 public class AdvisoryConsumer implements ActiveMQDispatcher {
034 private static final transient Logger LOG = LoggerFactory.getLogger(AdvisoryConsumer.class);
035
036 int deliveredCounter;
037
038 private final ActiveMQConnection connection;
039 private ConsumerInfo info;
040 private boolean closed;
041
042 public AdvisoryConsumer(ActiveMQConnection connection, ConsumerId consumerId) throws JMSException {
043 this.connection = connection;
044 info = new ConsumerInfo(consumerId);
045 info.setDestination(AdvisorySupport.TEMP_DESTINATION_COMPOSITE_ADVISORY_TOPIC);
046 info.setPrefetchSize(1000);
047 info.setNoLocal(true);
048 info.setDispatchAsync(true);
049
050 this.connection.addDispatcher(info.getConsumerId(), this);
051 this.connection.syncSendPacket(this.info);
052 }
053
054 public synchronized void dispose() {
055 if (!closed) {
056 try {
057 this.connection.asyncSendPacket(info.createRemoveCommand());
058 } catch (JMSException e) {
059 LOG.debug("Failed to send remove command: " + e, e);
060 }
061 this.connection.removeDispatcher(info.getConsumerId());
062 closed = true;
063 }
064 }
065
066 public void dispatch(MessageDispatch md) {
067
068 // Auto ack messages when we reach 75% of the prefetch
069 deliveredCounter++;
070 if (deliveredCounter > (0.75 * info.getPrefetchSize())) {
071 try {
072 MessageAck ack = new MessageAck(md, MessageAck.STANDARD_ACK_TYPE, deliveredCounter);
073 connection.asyncSendPacket(ack);
074 deliveredCounter = 0;
075 } catch (JMSException e) {
076 connection.onClientInternalException(e);
077 }
078 }
079
080 DataStructure o = md.getMessage().getDataStructure();
081 if (o != null && o.getClass() == DestinationInfo.class) {
082 processDestinationInfo((DestinationInfo)o);
083 } else {
084 //This can happen across networks
085 if (LOG.isDebugEnabled()) {
086 LOG.debug("Unexpected message was dispatched to the AdvisoryConsumer: "+md);
087 }
088 }
089
090 }
091
092 private void processDestinationInfo(DestinationInfo dinfo) {
093 ActiveMQDestination dest = dinfo.getDestination();
094 if (!dest.isTemporary()) {
095 return;
096 }
097
098 ActiveMQTempDestination tempDest = (ActiveMQTempDestination)dest;
099 if (dinfo.getOperationType() == DestinationInfo.ADD_OPERATION_TYPE) {
100 if (tempDest.getConnection() != null) {
101 tempDest = (ActiveMQTempDestination) tempDest.createDestination(tempDest.getPhysicalName());
102 }
103 connection.activeTempDestinations.put(tempDest, tempDest);
104 } else if (dinfo.getOperationType() == DestinationInfo.REMOVE_OPERATION_TYPE) {
105 connection.activeTempDestinations.remove(tempDest);
106 }
107 }
108
109 }