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 */
017package org.apache.activemq.partition;
018
019import org.apache.activemq.broker.Broker;
020import org.apache.activemq.broker.BrokerPlugin;
021import org.apache.activemq.partition.dto.Partitioning;
022
023import java.io.IOException;
024
025/**
026 * A BrokerPlugin which partitions client connections over a cluster of brokers.
027 *
028 * @org.apache.xbean.XBean element="partitionBrokerPlugin"
029 */
030public class PartitionBrokerPlugin implements BrokerPlugin {
031
032    protected int minTransferCount;
033    protected Partitioning config;
034
035    @Override
036    public Broker installPlugin(Broker broker) throws Exception {
037        return new PartitionBroker(broker, this);
038    }
039
040    public int getMinTransferCount() {
041        return minTransferCount;
042    }
043
044    public void setMinTransferCount(int minTransferCount) {
045        this.minTransferCount = minTransferCount;
046    }
047
048    public Partitioning getConfig() {
049        return config;
050    }
051
052    public void setConfig(Partitioning config) {
053        this.config = config;
054    }
055
056    public void setConfigAsJson(String config) throws IOException {
057        this.config = Partitioning.MAPPER.readValue(config, Partitioning.class);
058    }
059
060    public String getBrokerURL(PartitionBroker partitionBroker, String id) {
061        if( config!=null && config.brokers!=null ) {
062            return config.brokers.get(id);
063        }
064        return null;
065    }
066}