/**
* 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.camel.model.loadbalancer;
import java.util.ArrayList;
import java.util.List;
import javax.xml.bind.annotation.XmlAccessType;
import javax.xml.bind.annotation.XmlAccessorType;
import javax.xml.bind.annotation.XmlAttribute;
import javax.xml.bind.annotation.XmlElement;
import javax.xml.bind.annotation.XmlElementWrapper;
import javax.xml.bind.annotation.XmlRootElement;
import org.apache.camel.model.LoadBalancerDefinition;
import org.apache.camel.processor.loadbalancer.LoadBalancer;
import org.apache.camel.processor.loadbalancer.WeightedLoadBalancer;
import org.apache.camel.processor.loadbalancer.WeightedRandomLoadBalancer;
import org.apache.camel.processor.loadbalancer.WeightedRoundRobinLoadBalancer;
import org.apache.camel.spi.RouteContext;
/**
* Represents an XML <weighted/> element
*/
@XmlRootElement(name = "weighted")
@XmlAccessorType(XmlAccessType.FIELD)
public class WeightedLoadBalancerDefinition extends LoadBalancerDefinition {
@XmlAttribute(name = "roundRobin", required = false)
private Boolean roundRobin = Boolean.FALSE;
@XmlAttribute(name = "distributionRatio", required = true)
private String distributionRatio;
@XmlAttribute(name = "distributionRatioDelimiter", required = false)
private String distributionRatioDelimiter;
@Override
protected LoadBalancer createLoadBalancer(RouteContext routeContext) {
WeightedLoadBalancer loadBalancer = null;
List<Integer> distributionRatioList = new ArrayList<Integer>();
try {
if (distributionRatioDelimiter == null) {
distributionRatioDelimiter = ",";
}
String[] ratios = distributionRatio.split(distributionRatioDelimiter);
for (String ratio : ratios) {
distributionRatioList.add(new Integer(ratio.trim()));
}
if (!roundRobin) {
loadBalancer = new WeightedRandomLoadBalancer(distributionRatioList);
} else {
loadBalancer = new WeightedRoundRobinLoadBalancer(distributionRatioList);
}
} catch (Exception e) {
}
return loadBalancer;
}
public Boolean isRoundRobin() {
return roundRobin;
}
public void setRoundRobin(Boolean roundRobin) {
this.roundRobin = roundRobin;
}
public String getDistributionRatio() {
return distributionRatio;
}
public void setDistributionRatioList(String distributionRatio) {
this.distributionRatio = distributionRatio;
}
@Override
public String toString() {
if (!roundRobin) {
return "WeightedRandomLoadBalancer[" + distributionRatio + "]";
} else {
return "WeightedRoundRobinLoadBalancer[" + distributionRatio + "]";
}
}
}