/*
*
* 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.qpid.server.exchange;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.atomic.AtomicReference;
import org.apache.log4j.Logger;
import org.apache.qpid.AMQInvalidArgumentException;
import org.apache.qpid.framing.AMQShortString;
import org.apache.qpid.framing.FieldTable;
import org.apache.qpid.server.binding.Binding;
import org.apache.qpid.server.filter.MessageFilter;
import org.apache.qpid.server.message.InboundMessage;
import org.apache.qpid.server.plugin.ExchangeType;
import org.apache.qpid.server.queue.AMQQueue;
import org.apache.qpid.server.queue.BaseQueue;
import java.util.ArrayList;
import java.util.concurrent.ConcurrentHashMap;
public class FanoutExchange extends AbstractExchange
{
private static final Logger _logger = Logger.getLogger(FanoutExchange.class);
private static final Integer ONE = Integer.valueOf(1);
/**
* Maps from queue name to queue instances
*/
private final Map<AMQQueue,Integer> _queues = new HashMap<AMQQueue,Integer>();
private final CopyOnWriteArrayList<AMQQueue> _unfilteredQueues = new CopyOnWriteArrayList<AMQQueue>();
private final CopyOnWriteArrayList<AMQQueue> _filteredQueues = new CopyOnWriteArrayList<AMQQueue>();
private final AtomicReference<Map<AMQQueue,Map<Binding, MessageFilter>>> _filteredBindings =
new AtomicReference<Map<AMQQueue,Map<Binding, MessageFilter>>>();
{
Map<AMQQueue,Map<Binding, MessageFilter>> emptyMap = Collections.emptyMap();
_filteredBindings.set(emptyMap);
}
public static final ExchangeType<FanoutExchange> TYPE = new FanoutExchangeType();
public FanoutExchange()
{
super(TYPE);
}
public ArrayList<BaseQueue> doRoute(InboundMessage payload)
{
for(Binding b : getBindings())
{
b.incrementMatches();
}
final ArrayList<BaseQueue> result = new ArrayList<BaseQueue>(_unfilteredQueues);
final Map<AMQQueue, Map<Binding, MessageFilter>> filteredBindings = _filteredBindings.get();
if(!_filteredQueues.isEmpty())
{
for(AMQQueue q : _filteredQueues)
{
final Map<Binding, MessageFilter> bindingMessageFilterMap = filteredBindings.get(q);
if(!(bindingMessageFilterMap == null || result.contains(q)))
{
for(MessageFilter filter : bindingMessageFilterMap.values())
{
if(filter.matches(payload))
{
result.add(q);
break;
}
}
}
}
}
if (_logger.isDebugEnabled())
{
_logger.debug("Publishing message to queue " + result);
}
return result;
}
protected synchronized void onBind(final Binding binding)
{
AMQQueue queue = binding.getQueue();
assert queue != null;
if(binding.getArguments() == null || binding.getArguments().isEmpty() || !FilterSupport.argumentsContainFilter(binding.getArguments()))
{
Integer oldVal;
if(_queues.containsKey(queue))
{
_queues.put(queue,_queues.get(queue)+1);
}
else
{
_queues.put(queue, ONE);
_unfilteredQueues.add(queue);
// No longer any reason to check filters for this queue
_filteredQueues.remove(queue);
}
}
else
{
try
{
HashMap<AMQQueue,Map<Binding, MessageFilter>> filteredBindings =
new HashMap<AMQQueue,Map<Binding, MessageFilter>>(_filteredBindings.get());
Map<Binding, MessageFilter> bindingsForQueue = filteredBindings.remove(binding.getQueue());
final
MessageFilter messageFilter =
FilterSupport.createMessageFilter(binding.getArguments(), binding.getQueue());
if(bindingsForQueue != null)
{
bindingsForQueue = new HashMap<Binding,MessageFilter>(bindingsForQueue);
bindingsForQueue.put(binding, messageFilter);
}
else
{
bindingsForQueue = Collections.singletonMap(binding, messageFilter);
if(!_unfilteredQueues.contains(queue))
{
_filteredQueues.add(queue);
}
}
filteredBindings.put(binding.getQueue(), bindingsForQueue);
_filteredBindings.set(filteredBindings);
}
catch (AMQInvalidArgumentException e)
{
_logger.warn("Cannoy bind queue " + queue + " to exchange this " + this + " beacuse selector cannot be parsed.", e);
return;
}
}
if (_logger.isDebugEnabled())
{
_logger.debug("Binding queue " + queue
+ " with routing key " + binding.getBindingKey() + " to exchange " + this);
}
}
protected synchronized void onUnbind(final Binding binding)
{
AMQQueue queue = binding.getQueue();
if(binding.getArguments() == null || binding.getArguments().isEmpty() || !FilterSupport.argumentsContainFilter(binding.getArguments()))
{
Integer oldValue = _queues.remove(queue);
if(ONE.equals(oldValue))
{
// should start checking filters for this queue
if(_filteredBindings.get().containsKey(queue))
{
_filteredQueues.add(queue);
}
_unfilteredQueues.remove(queue);
}
else
{
_queues.put(queue,oldValue-1);
}
}
else // we are removing a binding with filters
{
HashMap<AMQQueue,Map<Binding, MessageFilter>> filteredBindings =
new HashMap<AMQQueue,Map<Binding, MessageFilter>>(_filteredBindings.get());
Map<Binding,MessageFilter> bindingsForQueue = filteredBindings.remove(binding.getQueue());
if(bindingsForQueue.size()>1)
{
bindingsForQueue = new HashMap<Binding,MessageFilter>(bindingsForQueue);
bindingsForQueue.remove(binding);
filteredBindings.put(binding.getQueue(),bindingsForQueue);
}
else
{
_filteredQueues.remove(queue);
}
_filteredBindings.set(filteredBindings);
}
}
}