qpid/java/broker/src/main/java/org/apache/qpid/server/exchange/HeadersExchangeMBean.java

/*
 *
 * 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 org.apache.qpid.AMQException;
import org.apache.qpid.framing.AMQShortString;
import org.apache.qpid.management.common.mbeans.annotations.MBeanConstructor;
import org.apache.qpid.management.common.mbeans.annotations.MBeanDescription;
import org.apache.qpid.server.binding.Binding;
import org.apache.qpid.server.logging.actors.CurrentActor;
import org.apache.qpid.server.logging.actors.ManagementActor;
import org.apache.qpid.server.queue.AMQQueue;
import org.apache.qpid.server.virtualhost.VirtualHost;

import javax.management.JMException;
import javax.management.MBeanException;
import javax.management.openmbean.ArrayType;
import javax.management.openmbean.CompositeData;
import javax.management.openmbean.CompositeDataSupport;
import javax.management.openmbean.CompositeType;
import javax.management.openmbean.OpenDataException;
import javax.management.openmbean.OpenType;
import javax.management.openmbean.SimpleType;
import javax.management.openmbean.TabularData;
import javax.management.openmbean.TabularDataSupport;
import javax.management.openmbean.TabularType;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
     * HeadersExchangeMBean class implements the management interface for the
 * Header Exchanges.
 */
@MBeanDescription("Management Bean for Headers Exchange")
final class HeadersExchangeMBean extends AbstractExchangeMBean<HeadersExchange>
{

    @MBeanConstructor("Creates an MBean for AMQ Headers exchange")
    public HeadersExchangeMBean(final HeadersExchange headersExchange) throws JMException
    {
        super(headersExchange);
        init();
    }

    /**
     * initialises the OpenType objects.
     */
    protected void init() throws OpenDataException
    {

        setBindingItemTypes(new OpenType[3]);
        getBindingItemTypes()[0] = SimpleType.INTEGER;
        getBindingItemTypes()[1] = SimpleType.STRING;
        getBindingItemTypes()[2] = new ArrayType(1, SimpleType.STRING);
        setBindingDataType(new CompositeType("Exchange Binding", "Queue name and header bindings",
                HEADERS_COMPOSITE_ITEM_NAMES.toArray(new String[HEADERS_COMPOSITE_ITEM_NAMES.size()]),
                HEADERS_COMPOSITE_ITEM_DESC.toArray(new String[HEADERS_COMPOSITE_ITEM_DESC.size()]), getBindingItemTypes()));
        setBindinglistDataType(new TabularType("Exchange Bindings", "List of exchange bindings for " + getName(),
                getBindingDataType(), HEADERS_TABULAR_UNIQUE_INDEX.toArray(new String[HEADERS_TABULAR_UNIQUE_INDEX.size()])));
    }

    public TabularData bindings() throws OpenDataException
    {
        TabularDataSupport bindingList = new TabularDataSupport(getBindinglistDataType());
        int count = 1;
        for (Binding binding : getExchange().getBindings())
        {

            String queueName = binding.getQueue().getNameShortString().toString();


            Map<String,Object> headerMappings = binding.getArguments();
            final List<String> mappingList = new ArrayList<String>();

            if(headerMappings != null)
            {
                for(Map.Entry<String,Object> entry : headerMappings.entrySet())
                {

                    mappingList.add(entry.getKey() + "=" + entry.getValue());
                }
            }


            Object[] bindingItemValues = {count++, queueName, mappingList.toArray(new String[0])};
            CompositeData bindingData = new CompositeDataSupport(getBindingDataType(),
                    HEADERS_COMPOSITE_ITEM_NAMES.toArray(new String[HEADERS_COMPOSITE_ITEM_NAMES.size()]), bindingItemValues);
            bindingList.put(bindingData);
        }

        return bindingList;
    }

    @Override
    public void createNewBinding(String queueName, String binding) throws JMException
    {
        VirtualHost vhost = getExchange().getVirtualHost();
        AMQQueue queue = vhost.getQueueRegistry().getQueue(new AMQShortString(queueName));
        if (queue == null)
        {
            throw new JMException("Queue \"" + queueName + "\" is not registered with the virtualhost.");
        }

        CurrentActor.set(new ManagementActor(getLogActor().getRootMessageLogger()));

        final Map<String,Object> arguments = new HashMap<String, Object>();
        final String[] bindings = binding.split(",");
        for (int i = 0; i < bindings.length; i++)
        {
            final String[] keyAndValue = bindings[i].split("=");
            if (keyAndValue == null || keyAndValue.length == 0 || keyAndValue.length > 2 || keyAndValue[0].length() == 0)
            {
                throw new JMException("Format for headers binding should be \"<attribute1>=<value1>,<attribute2>=<value2>\" ");
            }

            if(keyAndValue.length == 1)
            {
                //no value was given, only a key. Use an empty value to signal match on key presence alone
                arguments.put(keyAndValue[0], "");
            }
            else
            {
                arguments.put(keyAndValue[0], keyAndValue[1]);
            }
        }
        try
        {
            vhost.getBindingFactory().addBinding(binding,queue,getExchange(),arguments);
        }
        catch (AMQException ex)
        {
            JMException jme = new JMException(ex.toString());
            throw new MBeanException(jme, "Error creating new binding " + binding);
        }
        CurrentActor.remove();
    }

} // End of MBean class