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.spi.Policy;
027    import org.apache.camel.spi.RouteContext;
028    import org.apache.camel.util.ObjectHelper;
029    
030    /**
031     * Represents an XML <policy/> element
032     *
033     * @version $Revision: 751357 $
034     */
035    @XmlRootElement(name = "policy")
036    @XmlAccessorType(XmlAccessType.FIELD)
037    public class PolicyDefinition extends OutputDefinition<ProcessorDefinition> {
038        @XmlAttribute(required = true)
039        private String ref;
040        @XmlTransient
041        private Policy policy;
042    
043        public PolicyDefinition() {
044        }
045    
046        public PolicyDefinition(Policy policy) {
047            this.policy = policy;
048        }
049    
050        @Override
051        public String toString() {
052            return "Policy[" + description() + "]";
053        }
054    
055        @Override
056        public String getShortName() {
057            return "policy";
058        }
059    
060        @Override
061        public String getLabel() {
062            if (ref != null) {
063                return "ref: " + ref;
064            } else if (policy != null) {
065                return policy.toString();
066            } else {
067                return "";
068            }
069        }
070    
071        public String getRef() {
072            return ref;
073        }
074    
075        public void setRef(String ref) {
076            this.ref = ref;
077        }
078    
079        @Override
080        public Processor createProcessor(RouteContext routeContext) throws Exception {
081            Processor childProcessor = createOutputsProcessor(routeContext);
082    
083            Policy policy = resolvePolicy(routeContext);
084            ObjectHelper.notNull(policy, "policy", this);
085            return policy.wrap(childProcessor);
086        }
087    
088        protected Policy resolvePolicy(RouteContext routeContext) {
089            if (policy == null) {
090                policy = routeContext.lookup(getRef(), Policy.class);
091            }
092            return policy;
093        }
094    
095        protected String description() {
096            if (policy != null) {
097                return policy.toString();
098            } else {
099                return "ref: " + ref;
100            }
101        }
102    }