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.blueprint;
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.XmlElement;
023    import javax.xml.bind.annotation.XmlRootElement;
024    import javax.xml.bind.annotation.XmlTransient;
025    
026    import org.apache.aries.blueprint.ExtendedBlueprintContainer;
027    import org.apache.camel.CamelContext;
028    import org.apache.camel.LoggingLevel;
029    import org.apache.camel.Processor;
030    import org.apache.camel.builder.DefaultErrorHandlerBuilder;
031    import org.apache.camel.builder.ErrorHandlerBuilder;
032    import org.apache.camel.builder.LoggingErrorHandlerBuilder;
033    import org.apache.camel.core.xml.AbstractCamelFactoryBean;
034    import org.apache.camel.model.RedeliveryPolicyDefinition;
035    import org.osgi.service.blueprint.container.BlueprintContainer;
036    import org.osgi.service.blueprint.container.NoSuchComponentException;
037    
038    @XmlRootElement(name = "errorHandler")
039    @XmlAccessorType(XmlAccessType.FIELD)
040    public class CamelErrorHandlerFactoryBean extends AbstractCamelFactoryBean<ErrorHandlerBuilder> {
041    
042        @XmlAttribute
043        private ErrorHandlerType type = ErrorHandlerType.DefaultErrorHandler;
044        @XmlAttribute
045        private String deadLetterUri;
046        @XmlAttribute
047        private LoggingLevel level = LoggingLevel.ERROR;
048        @XmlAttribute
049        private Boolean useOriginalMessage;
050        @XmlAttribute
051        private String onRedeliveryRef;
052        @XmlAttribute
053        private String retryWhileRef;
054        @XmlElement
055        private RedeliveryPolicyDefinition redeliveryPolicy;
056        @XmlTransient
057        private BlueprintContainer blueprintContainer;
058    
059        @Override
060        public ErrorHandlerBuilder getObject() throws Exception {
061            ErrorHandlerBuilder errorHandler = getObjectType().newInstance();
062            if (errorHandler instanceof DefaultErrorHandlerBuilder) {
063                DefaultErrorHandlerBuilder handler = (DefaultErrorHandlerBuilder) errorHandler;
064                if (deadLetterUri != null) {
065                    handler.setDeadLetterUri(deadLetterUri);
066                }
067                if (useOriginalMessage != null) {
068                    handler.setUseOriginalMessage(useOriginalMessage);
069                }
070                if (redeliveryPolicy != null) {
071                    handler.setRedeliveryPolicy(redeliveryPolicy.createRedeliveryPolicy(getCamelContext(), null));
072                }
073                if (onRedeliveryRef != null) {
074                    handler.setOnRedelivery(lookup(onRedeliveryRef, Processor.class));
075                }
076                if (retryWhileRef != null) {
077                    handler.setRetryWhileRef(retryWhileRef);
078                }
079            } else if (errorHandler instanceof LoggingErrorHandlerBuilder) {
080                LoggingErrorHandlerBuilder handler = (LoggingErrorHandlerBuilder) errorHandler;
081                if (level != null) {
082                    handler.setLevel(level);
083                }
084            }
085            return errorHandler;
086        }
087    
088        @Override
089        @SuppressWarnings("unchecked")
090        public Class<? extends ErrorHandlerBuilder> getObjectType() {
091            return (Class<ErrorHandlerBuilder>) type.getTypeAsClass();
092        }
093    
094        public void setBlueprintContainer(BlueprintContainer blueprintContainer) {
095            this.blueprintContainer = blueprintContainer;
096        }
097    
098        protected CamelContext getCamelContextWithId(String camelContextId) {
099            if (blueprintContainer != null) {
100                return (CamelContext) blueprintContainer.getComponentInstance(camelContextId);
101            }
102            return null;
103        }
104    
105        protected <T> T lookup(String name, Class<T> type) {
106            try {
107                return type.cast(blueprintContainer.getComponentInstance(name));
108            } catch (NoSuchComponentException e) {
109                return null;
110            }
111        }
112    
113    }