/*
*
* 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.ArrayList;
import java.util.List;
import java.util.Map;
import javax.management.JMException;
import javax.management.MBeanException;
import javax.management.openmbean.CompositeData;
import javax.management.openmbean.CompositeDataSupport;
import javax.management.openmbean.OpenDataException;
import javax.management.openmbean.TabularData;
import javax.management.openmbean.TabularDataSupport;
import org.apache.log4j.Logger;
import org.apache.qpid.AMQException;
import org.apache.qpid.management.common.mbeans.annotations.MBeanConstructor;
import org.apache.qpid.management.common.mbeans.annotations.MBeanDescription;
import org.apache.qpid.protocol.AMQConstant;
import org.apache.qpid.exchange.ExchangeDefaults;
import org.apache.qpid.framing.AMQShortString;
import org.apache.qpid.framing.FieldTable;
import org.apache.qpid.server.queue.AMQQueue;
import org.apache.qpid.server.virtualhost.VirtualHost;
import org.apache.qpid.server.message.InboundMessage;
import org.apache.qpid.server.logging.actors.CurrentActor;
import org.apache.qpid.server.logging.actors.ManagementActor;
public class DirectExchange extends AbstractExchange
{
private static final Logger _logger = Logger.getLogger(DirectExchange.class);
/**
* Maps from queue name to queue instances
*/
private final Index _index = new Index();
public static final ExchangeType<DirectExchange> TYPE = new ExchangeType<DirectExchange>()
{
public AMQShortString getName()
{
return ExchangeDefaults.DIRECT_EXCHANGE_CLASS;
}
public Class<DirectExchange> getExchangeClass()
{
return DirectExchange.class;
}
public DirectExchange newInstance(VirtualHost host,
AMQShortString name,
boolean durable,
int ticket,
boolean autoDelete) throws AMQException
{
DirectExchange exch = new DirectExchange();
exch.initialise(host,name,durable,ticket,autoDelete);
return exch;
}
public AMQShortString getDefaultExchangeName()
{
return ExchangeDefaults.DIRECT_EXCHANGE_NAME;
}
};
/**
* MBean class implementing the management interfaces.
*/
@MBeanDescription("Management Bean for Direct Exchange")
private final class DirectExchangeMBean extends ExchangeMBean
{
@MBeanConstructor("Creates an MBean for AMQ direct exchange")
public DirectExchangeMBean() throws JMException
{
super();
_exchangeType = "direct";
init();
}
public TabularData bindings() throws OpenDataException
{
Map<AMQShortString, List<AMQQueue>> bindings = _index.getBindingsMap();
_bindingList = new TabularDataSupport(_bindinglistDataType);
for (Map.Entry<AMQShortString, List<AMQQueue>> entry : bindings.entrySet())
{
AMQShortString key = entry.getKey();
List<String> queueList = new ArrayList<String>();
List<AMQQueue> queues = entry.getValue();
for (AMQQueue q : queues)
{
queueList.add(q.getName().toString());
}
Object[] bindingItemValues = {key.toString(), queueList.toArray(new String[0])};
CompositeData bindingData = new CompositeDataSupport(_bindingDataType, COMPOSITE_ITEM_NAMES, bindingItemValues);
_bindingList.put(bindingData);
}
return _bindingList;
}
public void createNewBinding(String queueName, String binding) throws JMException
{
AMQQueue queue = getQueueRegistry().getQueue(new AMQShortString(queueName));
if (queue == null)
{
throw new JMException("Queue \"" + queueName + "\" is not registered with the exchange.");
}
CurrentActor.set(new ManagementActor(_logActor.getRootMessageLogger()));
try
{
queue.bind(DirectExchange.this, new AMQShortString(binding), null);
}
catch (AMQException ex)
{
throw new MBeanException(ex);
}
finally
{
CurrentActor.remove();
}
}
}// End of MBean class
protected ExchangeMBean createMBean() throws JMException
{
return new DirectExchangeMBean();
}
public Logger getLogger()
{
return _logger;
}
public AMQShortString getType()
{
return ExchangeDefaults.DIRECT_EXCHANGE_CLASS;
}
public void registerQueue(String routingKey, AMQQueue queue, Map<String,Object> args) throws AMQException
{
registerQueue(new AMQShortString(routingKey), queue);
}
public void registerQueue(AMQShortString routingKey, AMQQueue queue, FieldTable args) throws AMQException
{
registerQueue(routingKey, queue);
}
private void registerQueue(AMQShortString routingKey, AMQQueue queue) throws AMQException
{
assert queue != null;
assert routingKey != null;
if (!_index.add(routingKey, queue))
{
if (_logger.isDebugEnabled())
{
_logger.debug("Queue (" + queue + ") is already registered with routing key " + routingKey);
}
}
else
{
if (_logger.isDebugEnabled())
{
_logger.debug("Binding queue:" + queue + " with routing key '" + routingKey +"' to exchange:" + this);
}
}
}
public void deregisterQueue(AMQShortString routingKey, AMQQueue queue, FieldTable args) throws AMQException
{
assert queue != null;
assert routingKey != null;
if (!_index.remove(routingKey, queue))
{
throw new AMQException(AMQConstant.NOT_FOUND, "Queue " + queue + " was not registered with exchange " + this.getName() +
" with routing key " + routingKey + ". No queue was registered with that _routing key");
}
}
public ArrayList<AMQQueue> route(InboundMessage payload)
{
final String routingKey = payload.getRoutingKey();
final ArrayList<AMQQueue> queues = (routingKey == null) ? _index.get("") : _index.get(routingKey);
if (_logger.isDebugEnabled())
{
_logger.debug("Publishing message to queue " + queues);
}
return queues;
}
public boolean isBound(AMQShortString routingKey, FieldTable arguments, AMQQueue queue)
{
return isBound(routingKey,queue);
}
public boolean isBound(AMQShortString routingKey, AMQQueue queue)
{
final List<AMQQueue> queues = _index.get(routingKey);
return queues != null && queues.contains(queue);
}
public boolean isBound(AMQShortString routingKey)
{
final List<AMQQueue> queues = _index.get(routingKey);
return queues != null && !queues.isEmpty();
}
public boolean isBound(AMQQueue queue)
{
Map<AMQShortString, List<AMQQueue>> bindings = _index.getBindingsMap();
for (List<AMQQueue> queues : bindings.values())
{
if (queues.contains(queue))
{
return true;
}
}
return false;
}
public boolean hasBindings()
{
return !_index.getBindingsMap().isEmpty();
}
public Map<AMQShortString, List<AMQQueue>> getBindings()
{
return _index.getBindingsMap();
}
}