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.camel.model.loadbalancer; 018 019import javax.xml.bind.annotation.XmlAccessType; 020import javax.xml.bind.annotation.XmlAccessorType; 021import javax.xml.bind.annotation.XmlAttribute; 022import javax.xml.bind.annotation.XmlRootElement; 023import javax.xml.bind.annotation.XmlTransient; 024 025import org.apache.camel.model.LoadBalancerDefinition; 026import org.apache.camel.processor.loadbalancer.LoadBalancer; 027import org.apache.camel.spi.Metadata; 028import org.apache.camel.spi.RouteContext; 029import org.apache.camel.util.CamelContextHelper; 030import org.apache.camel.util.ObjectHelper; 031 032/** 033 * Custom load balancer 034 */ 035@Metadata(label = "eip,routing,loadbalance") 036@XmlRootElement(name = "customLoadBalancer") 037@XmlAccessorType(XmlAccessType.FIELD) 038public class CustomLoadBalancerDefinition extends LoadBalancerDefinition { 039 040 @XmlTransient 041 private LoadBalancer loadBalancer; 042 @XmlAttribute(required = true) 043 private String ref; 044 045 public CustomLoadBalancerDefinition() { 046 } 047 048 public String getRef() { 049 return ref; 050 } 051 052 /** 053 * Refers to the custom load balancer to lookup from the registry 054 */ 055 public void setRef(String ref) { 056 this.ref = ref; 057 } 058 059 public LoadBalancer getLoadBalancer() { 060 return loadBalancer; 061 } 062 063 /** 064 * The custom load balancer to use. 065 */ 066 public void setLoadBalancer(LoadBalancer loadBalancer) { 067 this.loadBalancer = loadBalancer; 068 } 069 070 @Override 071 protected LoadBalancer createLoadBalancer(RouteContext routeContext) { 072 if (loadBalancer != null) { 073 return loadBalancer; 074 } 075 076 ObjectHelper.notEmpty(ref, "ref", this); 077 return CamelContextHelper.mandatoryLookup(routeContext.getCamelContext(), ref, LoadBalancer.class); 078 } 079 080 @Override 081 public String toString() { 082 if (loadBalancer != null) { 083 return "CustomLoadBalancer[" + loadBalancer + "]"; 084 } else { 085 return "CustomLoadBalancer[" + ref + "]"; 086 } 087 } 088 089}