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.builder.ProcessorBuilder;
027    import org.apache.camel.spi.RouteContext;
028    import org.apache.camel.util.ObjectHelper;
029    
030    /**
031     * Represents an XML <removeHeaders/> element
032     */
033    @XmlRootElement(name = "removeHeaders")
034    @XmlAccessorType(XmlAccessType.FIELD)
035    public class RemoveHeadersDefinition extends NoOutputDefinition<RemoveHeadersDefinition> {
036        @XmlAttribute(required = true)
037        private String pattern;
038        @XmlAttribute
039        private String excludePattern;
040        // in XML we cannot use String[] for attributes, so we provide a single attribute instead
041        @XmlTransient
042        private String[] excludePatterns;
043    
044        public RemoveHeadersDefinition() {
045        }
046    
047        public RemoveHeadersDefinition(String pattern) {
048            setPattern(pattern);
049        }
050        
051        public RemoveHeadersDefinition(String pattern, String... excludePatterns) {
052            setPattern(pattern);
053            setExcludePatterns(excludePatterns);
054        }
055    
056        @Override
057        public String toString() {
058            return "RemoveHeaders[" + getPattern() + "]";
059        }
060    
061        @Override
062        public String getShortName() {
063            return "removeHeaders";
064        }
065    
066        @Override
067        public String getLabel() {
068            return "removeHeaders[" + getPattern() + "]";
069        }
070    
071        @Override
072        public Processor createProcessor(RouteContext routeContext) throws Exception {
073            ObjectHelper.notNull(getPattern(), "patterns", this);
074            if (getExcludePatterns() != null) {
075                return ProcessorBuilder.removeHeaders(getPattern(), getExcludePatterns());
076            } else if (getExcludePattern() != null) {
077                return ProcessorBuilder.removeHeaders(getPattern(), getExcludePattern());
078            } else {
079                return ProcessorBuilder.removeHeaders(getPattern());
080            }
081        }
082    
083        public void setPattern(String pattern) {
084            this.pattern = pattern;
085        }
086    
087        public String getPattern() {
088            return pattern;
089        }
090    
091        public String[] getExcludePatterns() {
092            return excludePatterns;
093        }
094    
095        public void setExcludePatterns(String[] excludePatterns) {
096            this.excludePatterns = excludePatterns;
097        }
098    
099        public String getExcludePattern() {
100            return excludePattern;
101        }
102    
103        public void setExcludePattern(String excludePattern) {
104            this.excludePattern = excludePattern;
105        }
106    }