/**
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.apache.camel.component.cxf.jaxrs;
import java.util.Arrays;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicBoolean;
import org.apache.camel.CamelContext;
import org.apache.camel.Component;
import org.apache.camel.Consumer;
import org.apache.camel.Processor;
import org.apache.camel.Producer;
import org.apache.camel.Service;
import org.apache.camel.component.cxf.CxfEndpointUtils;
import org.apache.camel.impl.DefaultEndpoint;
import org.apache.camel.spi.HeaderFilterStrategy;
import org.apache.camel.spi.HeaderFilterStrategyAware;
import org.apache.camel.util.ObjectHelper;
import org.apache.cxf.Bus;
import org.apache.cxf.BusFactory;
import org.apache.cxf.feature.LoggingFeature;
import org.apache.cxf.helpers.CastUtils;
import org.apache.cxf.jaxrs.JAXRSServerFactoryBean;
import org.apache.cxf.jaxrs.client.JAXRSClientFactoryBean;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class CxfRsEndpoint extends DefaultEndpoint implements HeaderFilterStrategyAware, Service {
private static final Logger LOG = LoggerFactory.getLogger(CxfRsEndpoint.class);
protected Bus bus;
private Map<String, String> parameters;
private List<Class<?>> resourceClasses;
private HeaderFilterStrategy headerFilterStrategy;
private CxfRsBinding binding;
private boolean httpClientAPI = true;
private String address;
private boolean throwExceptionOnFailure = true;
private int maxClientCacheSize = 10;
private boolean loggingFeatureEnabled;
private int loggingSizeLimit;
private AtomicBoolean getBusHasBeenCalled = new AtomicBoolean(false);
private boolean isSetDefaultBus;
@Deprecated
public CxfRsEndpoint(String endpointUri, CamelContext camelContext) {
super(endpointUri, camelContext);
setAddress(endpointUri);
}
public CxfRsEndpoint(String endpointUri, Component component) {
super(endpointUri, component);
setAddress(endpointUri);
}
// This method is for CxfRsComponent setting the EndpointUri
protected void updateEndpointUri(String endpointUri) {
super.setEndpointUri(endpointUri);
}
public void setParameters(Map<String, String> param) {
parameters = param;
}
public Map<String, String> getParameters() {
return parameters;
}
public void setHttpClientAPI(boolean clientAPI) {
httpClientAPI = clientAPI;
}
public boolean isHttpClientAPI() {
return httpClientAPI;
}
@Override
public boolean isLenientProperties() {
return true;
}
public HeaderFilterStrategy getHeaderFilterStrategy() {
return headerFilterStrategy;
}
public void setHeaderFilterStrategy(HeaderFilterStrategy strategy) {
headerFilterStrategy = strategy;
}
public Consumer createConsumer(Processor processor) throws Exception {
return new CxfRsConsumer(this, processor);
}
public Producer createProducer() throws Exception {
return new CxfRsProducer(this);
}
public boolean isSingleton() {
return false;
}
public void setBinding(CxfRsBinding binding) {
this.binding = binding;
}
public CxfRsBinding getBinding() {
return binding;
}
protected void checkBeanType(Object object, Class<?> clazz) {
if (!clazz.isAssignableFrom(object.getClass())) {
throw new IllegalArgumentException("The configure bean is not the instance of " + clazz.getName());
}
}
@SuppressWarnings({"rawtypes", "unchecked"})
protected void setupJAXRSServerFactoryBean(JAXRSServerFactoryBean sfb) {
// address
if (getAddress() != null) {
sfb.setAddress(getAddress());
}
if (getResourceClasses() != null) {
List res = CastUtils.cast(getResourceClasses());
sfb.setResourceClasses(res);
}
sfb.setStart(false);
}
protected void setupJAXRSClientFactoryBean(JAXRSClientFactoryBean cfb, String address) {
// address
if (address != null) {
cfb.setAddress(address);
}
if (getResourceClasses() != null) {
cfb.setResourceClass(getResourceClasses().get(0));
}
if (isLoggingFeatureEnabled()) {
if (getLoggingSizeLimit() > 0) {
cfb.getFeatures().add(new LoggingFeature(getLoggingSizeLimit()));
} else {
cfb.getFeatures().add(new LoggingFeature());
}
}
cfb.setThreadSafe(true);
}
protected JAXRSServerFactoryBean newJAXRSServerFactoryBean() {
return new JAXRSServerFactoryBean();
}
protected JAXRSClientFactoryBean newJAXRSClientFactoryBean() {
return new JAXRSClientFactoryBean();
}
protected String resolvePropertyPlaceholders(String str) {
try {
if (getCamelContext() != null) {
return getCamelContext().resolvePropertyPlaceholders(str);
} else {
return str;
}
} catch (Exception ex) {
throw ObjectHelper.wrapRuntimeCamelException(ex);
}
}
public JAXRSServerFactoryBean createJAXRSServerFactoryBean() {
JAXRSServerFactoryBean answer = newJAXRSServerFactoryBean();
setupJAXRSServerFactoryBean(answer);
if (isLoggingFeatureEnabled()) {
if (getLoggingSizeLimit() > 0) {
answer.getFeatures().add(new LoggingFeature(getLoggingSizeLimit()));
} else {
answer.getFeatures().add(new LoggingFeature());
}
}
return answer;
}
public JAXRSClientFactoryBean createJAXRSClientFactoryBean() {
return createJAXRSClientFactoryBean(getAddress());
}
public JAXRSClientFactoryBean createJAXRSClientFactoryBean(String address) {
JAXRSClientFactoryBean answer = newJAXRSClientFactoryBean();
setupJAXRSClientFactoryBean(answer, address);
if (isLoggingFeatureEnabled()) {
if (getLoggingSizeLimit() > 0) {
answer.getFeatures().add(new LoggingFeature(getLoggingSizeLimit()));
} else {
answer.getFeatures().add(new LoggingFeature());
}
}
return answer;
}
public List<Class<?>> getResourceClasses() {
return resourceClasses;
}
public void setResourceClasses(List<Class<?>> classes) {
resourceClasses = classes;
}
public void setResourceClasses(Class<?>... classes) {
setResourceClasses(Arrays.asList(classes));
}
public void setResourceClass(Class<?> clazz) {
setResourceClasses(clazz);
}
public void setAddress(String address) {
this.address = address;
}
public String getAddress() {
return resolvePropertyPlaceholders(address);
}
public boolean isLoggingFeatureEnabled() {
return loggingFeatureEnabled;
}
public void setLoggingFeatureEnabled(boolean loggingFeatureEnabled) {
this.loggingFeatureEnabled = loggingFeatureEnabled;
}
public int getLoggingSizeLimit() {
return loggingSizeLimit;
}
public void setLoggingSizeLimit(int loggingSizeLimit) {
this.loggingSizeLimit = loggingSizeLimit;
}
public boolean isThrowExceptionOnFailure() {
return throwExceptionOnFailure;
}
public void setThrowExceptionOnFailure(boolean throwExceptionOnFailure) {
this.throwExceptionOnFailure = throwExceptionOnFailure;
}
/**
* @param maxClientCacheSize the maxClientCacheSize to set
*/
public void setMaxClientCacheSize(int maxClientCacheSize) {
this.maxClientCacheSize = maxClientCacheSize;
}
/**
* @return the maxClientCacheSize
*/
public int getMaxClientCacheSize() {
return maxClientCacheSize;
}
public void setBus(Bus bus) {
this.bus = bus;
}
public Bus getBus() {
if (bus == null) {
bus = CxfEndpointUtils.createBus(getCamelContext());
LOG.debug("Using DefaultBus {}", bus);
}
if (!getBusHasBeenCalled.getAndSet(true) && isSetDefaultBus) {
BusFactory.setDefaultBus(bus);
LOG.debug("Set bus {} as thread default bus", bus);
}
return bus;
}
public void setSetDefaultBus(boolean isSetDefaultBus) {
this.isSetDefaultBus = isSetDefaultBus;
}
public boolean isSetDefaultBus() {
return isSetDefaultBus;
}
@Override
protected void doStart() throws Exception {
if (headerFilterStrategy == null) {
headerFilterStrategy = new CxfRsHeaderFilterStrategy();
}
if (binding == null) {
binding = new DefaultCxfRsBinding();
}
if (binding instanceof HeaderFilterStrategyAware) {
((HeaderFilterStrategyAware) binding).setHeaderFilterStrategy(getHeaderFilterStrategy());
}
}
@Override
protected void doStop() throws Exception {
// noop
}
}