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    
018    package org.apache.camel.component.cxf.jaxrs;
019    
020    import java.util.Arrays;
021    import java.util.List;
022    import java.util.Map;
023    import java.util.concurrent.atomic.AtomicBoolean;
024    
025    import org.apache.camel.CamelContext;
026    import org.apache.camel.Component;
027    import org.apache.camel.Consumer;
028    import org.apache.camel.Processor;
029    import org.apache.camel.Producer;
030    import org.apache.camel.component.cxf.spring.CxfRsClientFactoryBeanDefinitionParser.SpringJAXRSClientFactoryBean;
031    import org.apache.camel.component.cxf.spring.CxfRsServerFactoryBeanDefinitionParser.SpringJAXRSServerFactoryBean;
032    import org.apache.camel.impl.DefaultEndpoint;
033    import org.apache.camel.spi.HeaderFilterStrategy;
034    import org.apache.camel.spi.HeaderFilterStrategyAware;
035    import org.apache.commons.logging.Log;
036    import org.apache.commons.logging.LogFactory;
037    import org.apache.cxf.helpers.CastUtils;
038    import org.apache.cxf.jaxrs.JAXRSServerFactoryBean;
039    import org.apache.cxf.jaxrs.client.JAXRSClientFactoryBean;
040    
041    /**
042     * 
043     */
044    public class CxfRsEndpoint extends DefaultEndpoint implements HeaderFilterStrategyAware {
045        private static final Log LOG = LogFactory.getLog(CxfRsEndpoint.class);
046    
047        private Map<String, String> parameters;
048        private List<Class<?>> resourceClasses;    
049        private HeaderFilterStrategy headerFilterStrategy;
050        private CxfRsBinding binding;
051        private boolean httpClientAPI = true;
052    
053        private AtomicBoolean bindingInitialized = new AtomicBoolean(false);
054        
055        public CxfRsEndpoint(String endpointUri, CamelContext camelContext) {
056           super(endpointUri, camelContext);
057        }
058        
059        public CxfRsEndpoint(String endpointUri, Component component) {
060            super(endpointUri, component);
061        }
062        
063        public void setParameters(Map<String, String> param) {
064            parameters = param;
065        }
066        
067        public Map<String, String> getParameters() {
068            return parameters;
069        }
070        
071        public void setHttpClientAPI(boolean clientAPI) {
072            httpClientAPI = clientAPI;
073        }
074        
075        public boolean isHttpClientAPI() {
076            return httpClientAPI;
077        }
078        
079        @Override
080        public boolean isLenientProperties() {
081            return true;
082        }
083        
084        public HeaderFilterStrategy getHeaderFilterStrategy() {    
085            if (headerFilterStrategy == null) {
086                headerFilterStrategy = new CxfRsHeaderFilterStrategy();
087                if (LOG.isDebugEnabled()) {
088                    LOG.debug("Create default header filter strategy " + headerFilterStrategy);
089                }
090            }
091            return headerFilterStrategy;
092        }
093    
094        public void setHeaderFilterStrategy(HeaderFilterStrategy strategy) {
095            headerFilterStrategy = strategy;
096            if (binding instanceof HeaderFilterStrategyAware) {
097                ((HeaderFilterStrategyAware)binding)
098                    .setHeaderFilterStrategy(headerFilterStrategy);
099            }
100        }
101    
102        public Consumer createConsumer(Processor processor) throws Exception {        
103            return new CxfRsConsumer(this, processor);
104        }
105    
106        public Producer createProducer() throws Exception {        
107            return new CxfRsProducer(this);
108        }
109    
110        public boolean isSingleton() {        
111            return false;
112        }
113        
114        public void setBinding(CxfRsBinding binding) {
115            this.binding = binding;
116            bindingInitialized.set(false);
117    
118        }
119        
120        public CxfRsBinding getBinding() {
121            if (binding == null) {
122                binding = new DefaultCxfRsBinding();
123                if (LOG.isDebugEnabled()) {
124                    LOG.debug("Create default CXF Binding " + binding);
125                }
126            } 
127            
128            if (!bindingInitialized.getAndSet(true) && binding instanceof HeaderFilterStrategyAware) {
129                ((HeaderFilterStrategyAware)binding).setHeaderFilterStrategy(getHeaderFilterStrategy());
130            }
131            
132            return binding;
133        }
134        
135        protected void setupJAXRSServerFactoryBean(JAXRSServerFactoryBean sfb) {        
136            // address
137            sfb.setAddress(getEndpointUri());
138            sfb.setResourceClasses(CastUtils.cast(getResourceClasses(), Class.class));
139            sfb.setStart(false);
140        }
141        
142        protected void setupJAXRSClientFactoryBean(JAXRSClientFactoryBean cfb) {        
143            // address
144            cfb.setAddress(getEndpointUri());
145            if (getResourceClasses() != null) {
146                cfb.setResourceClass(getResourceClasses().get(0));
147            }    
148        }
149       
150        public JAXRSServerFactoryBean createJAXRSServerFactoryBean() {
151            JAXRSServerFactoryBean answer = new SpringJAXRSServerFactoryBean();        
152            setupJAXRSServerFactoryBean(answer);
153            return answer;
154        }
155        
156        public JAXRSClientFactoryBean createJAXRSClientFactoryBean() {
157            JAXRSClientFactoryBean answer = new SpringJAXRSClientFactoryBean();
158            setupJAXRSClientFactoryBean(answer);
159            return answer;
160        }
161           
162        public List<Class<?>> getResourceClasses() {
163            return resourceClasses;
164        }
165    
166        public void setResourceClasses(List<Class<?>> classes) {
167            resourceClasses = classes;
168        }
169    
170        public void setResourceClasses(Class<?>... classes) {
171            setResourceClasses(Arrays.asList(classes));
172        }
173    
174    }