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;
018    
019    import javax.xml.bind.annotation.XmlAccessType;
020    import javax.xml.bind.annotation.XmlAccessorType;
021    import javax.xml.bind.annotation.XmlAttribute;
022    import javax.xml.bind.annotation.XmlRootElement;
023    import javax.xml.bind.annotation.XmlTransient;
024    
025    import org.apache.camel.Processor;
026    import org.apache.camel.impl.RouteContext;
027    import org.apache.camel.spi.Policy;
028    
029    /**
030     * @version $Revision: 36321 $
031     */
032    @XmlRootElement(name = "policy")
033    @XmlAccessorType(XmlAccessType.FIELD)
034    public class PolicyRef extends OutputType<ProcessorType> {
035        @XmlAttribute(required = true)
036        private String ref;
037        @XmlTransient
038        private Policy policy;
039    
040        public PolicyRef() {
041        }
042    
043        public PolicyRef(Policy policy) {
044            this.policy = policy;
045        }
046    
047        @Override
048        public String toString() {
049            return "Policy[" + description() + "]";
050        }
051    
052        @Override
053        public String getLabel() {
054            if (ref != null) {
055                return "ref:  " + ref;
056            } else if (policy != null) {
057                return policy.toString();
058            } else {
059                return "";
060            }
061        }
062    
063        public String getRef() {
064            return ref;
065        }
066    
067        public void setRef(String ref) {
068            this.ref = ref;
069        }
070    
071        @Override
072        public Processor createProcessor(RouteContext routeContext) throws Exception {
073            Processor childProcessor = createOutputsProcessor(routeContext);
074    
075            Policy policy = resolvePolicy(routeContext);
076            if (policy == null) {
077                throw new IllegalArgumentException("No policy configured: " + this);
078            }
079            return policy.wrap(childProcessor);
080        }
081    
082        protected Policy resolvePolicy(RouteContext routeContext) {
083            if (policy == null) {
084                policy = routeContext.lookup(getRef(), Policy.class);
085            }
086            return policy;
087        }
088    
089        protected String description() {
090            if (policy != null) {
091                return policy.toString();
092            } else {
093                return "ref:  " + ref;
094            }
095        }
096    }