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.Endpoint;
026    import org.apache.camel.Processor;
027    import org.apache.camel.processor.Enricher;
028    import org.apache.camel.processor.aggregate.AggregationStrategy;
029    import org.apache.camel.spi.RouteContext;
030    
031    /**
032     * Represents an XML <enrich/> element
033     *
034     * @see Enricher
035     */
036    @XmlRootElement(name = "enrich")
037    @XmlAccessorType(XmlAccessType.FIELD)
038    public class EnrichDefinition extends OutputDefinition<EnrichDefinition> {
039    
040        @XmlAttribute(name = "uri", required = true)
041        private String resourceUri;
042        
043        @XmlAttribute(name = "strategyRef", required = false)
044        private String aggregationStrategyRef;
045        
046        @XmlTransient
047        private AggregationStrategy aggregationStrategy;
048        
049        public EnrichDefinition() {
050            this(null, null);
051        }
052        
053        public EnrichDefinition(String resourceUri) {
054            this(null, resourceUri);
055        }
056        
057        public EnrichDefinition(AggregationStrategy aggregationStrategy, String resourceUri) {
058            this.aggregationStrategy = aggregationStrategy;
059            this.resourceUri = resourceUri;
060        }
061        
062        @Override
063        public String toString() {
064            return "Enrich[" + resourceUri + " " + aggregationStrategy + "]";
065        }
066    
067        @Override
068        public String getShortName() {
069            return "enrich";
070        }
071    
072        @Override
073        public Processor createProcessor(RouteContext routeContext) throws Exception {
074            Endpoint endpoint = routeContext.resolveEndpoint(resourceUri);
075            Enricher enricher = new Enricher(null, endpoint.createProducer());
076            if (aggregationStrategyRef != null) {
077                aggregationStrategy = routeContext.lookup(aggregationStrategyRef, AggregationStrategy.class);
078            }
079            if (aggregationStrategy == null) {
080                enricher.setDefaultAggregationStrategy();
081            } else {
082                enricher.setAggregationStrategy(aggregationStrategy);
083            }
084            return enricher;
085        }
086        
087    }