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     */
017    package org.apache.camel.model.loadbalancer;
018    
019    import java.util.ArrayList;
020    import java.util.List;
021    import javax.xml.bind.annotation.XmlAccessType;
022    import javax.xml.bind.annotation.XmlAccessorType;
023    import javax.xml.bind.annotation.XmlElement;
024    import javax.xml.bind.annotation.XmlRootElement;
025    
026    import org.apache.camel.model.LoadBalancerDefinition;
027    import org.apache.camel.processor.loadbalancer.FailOverLoadBalancer;
028    import org.apache.camel.processor.loadbalancer.LoadBalancer;
029    import org.apache.camel.spi.RouteContext;
030    
031    @XmlRootElement(name = "failover")
032    @XmlAccessorType(XmlAccessType.FIELD)
033    public class FailoverLoadBalancerDefinition extends LoadBalancerDefinition {
034    
035        @XmlElement(name = "exception")
036        private List<String> exceptions = new ArrayList<String>();
037    
038        @Override
039        protected LoadBalancer createLoadBalancer(RouteContext routeContext) {
040            if (!exceptions.isEmpty()) {
041                List<Class> classes = new ArrayList<Class>();
042                for (String name : exceptions) {
043                    Class type = routeContext.getCamelContext().getClassResolver().resolveClass(name);
044                    if (type == null) {
045                        throw new IllegalArgumentException("Cannot find class: " + name + " in the classpath");
046                    }
047                    classes.add(type);
048                }
049                return new FailOverLoadBalancer(classes);
050            } else {
051                return new FailOverLoadBalancer();
052            }
053        }
054    
055        public List<String> getExceptions() {
056            return exceptions;
057        }
058    
059        public void setExceptions(List<String> exceptions) {
060            this.exceptions = exceptions;
061        }
062    
063        @Override
064        public String toString() {
065            return "FailoverLoadBalancer";
066        }
067    }