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.region.virtual;
  
  import java.util.Set;
  
  import org.apache.activemq.broker.ConnectionContext;
  import org.apache.activemq.broker.region.BaseDestination;
  import org.apache.activemq.broker.region.Destination;
  import org.apache.activemq.broker.region.IndirectMessageReference;
  import org.apache.activemq.broker.region.RegionBroker;
  import org.apache.activemq.broker.region.Subscription;
  import org.apache.activemq.broker.region.Topic;
  import org.apache.activemq.command.ActiveMQDestination;
  import org.apache.activemq.command.Message;
  import org.apache.activemq.util.SubscriptionKey;
  
  /**
   * Creates a mapped Queue that can recover messages from subscription recovery
   * policy of its Virtual Topic.
   */
  public class MappedQueueFilter extends BaseVirtualDestinationFilter {
  
      private final ActiveMQDestination virtualDestination;
  
      public MappedQueueFilter(ActiveMQDestination virtualDestination, Destination destination) {
          super(destination);
          this.virtualDestination = virtualDestination;
      }
  
      @Override
      public synchronized void addSubscription(ConnectionContext context, Subscription sub) throws Exception {
          // recover messages for first consumer only
          boolean noSubs = getConsumers().isEmpty();
  
          // for virtual consumer wildcard dests, only subscribe to exact match or non wildcard dests to ensure no duplicates
          int match = sub.getActiveMQDestination().compareTo(next.getActiveMQDestination());
          if (match == 0 || (!next.getActiveMQDestination().isPattern() && match == 1)) {
              super.addSubscription(context, sub);
          }
          if (noSubs && !getConsumers().isEmpty()) {
              // new subscription added, recover retroactive messages
              final RegionBroker regionBroker = (RegionBroker) context.getBroker().getAdaptor(RegionBroker.class);
              final Set<Destination> virtualDests = regionBroker.getDestinations(virtualDestination);
  
              final ActiveMQDestination newDestination = sub.getActiveMQDestination();
              BaseDestination regionDest = null;
  
              for (Destination virtualDest : virtualDests) {
                  if (virtualDest.getActiveMQDestination().isTopic() &&
                          (virtualDest.isAlwaysRetroactive() || sub.getConsumerInfo().isRetroactive())) {
  
                      Topic topic = (Topic) getBaseDestination(virtualDest);
                      if (topic != null) {
                          // re-use browse() to get recovered messages
                          final Message[] messages = topic.getSubscriptionRecoveryPolicy().browse(topic.getActiveMQDestination());
  
                          // add recovered messages to subscription
                          for (Message message : messages) {
                              final Message copy = message.copy();
                              copy.setOriginalDestination(message.getDestination());
                              copy.setDestination(newDestination);
                              if (regionDest == null) {
                                  regionDest = getBaseDestination((Destination) regionBroker.getDestinations(newDestination).toArray()[0]);
                              }
                              copy.setRegionDestination(regionDest);
                              sub.addRecoveredMessage(context, newDestination.isQueue() ? new IndirectMessageReference(copy) : copy);
                          }
                      }
                  }
              }
          }
      }
  
      @Override
      public synchronized void removeSubscription(ConnectionContext context, Subscription sub, long lastDeliveredSequenceId) throws Exception {
          super.removeSubscription(context, sub, lastDeliveredSequenceId);
      }
  
      @Override
      public synchronized void deleteSubscription(ConnectionContext context, SubscriptionKey key) throws Exception {
          super.deleteSubscription(context, key);
      }
  
      @Override
      public String toString() {
          return "MappedQueueFilter[" + virtualDestination + ", " + next + "]";
      }
  }
  