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.component.http;
018
019 import java.io.IOException;
020 import java.util.concurrent.ConcurrentHashMap;
021
022 import javax.servlet.ServletConfig;
023 import javax.servlet.ServletException;
024 import javax.servlet.http.HttpServlet;
025 import javax.servlet.http.HttpServletRequest;
026 import javax.servlet.http.HttpServletResponse;
027
028 import org.apache.camel.Exchange;
029 import org.apache.camel.ExchangePattern;
030 import org.apache.camel.component.http.helper.HttpHelper;
031 import org.apache.camel.impl.DefaultExchange;
032 import org.slf4j.Logger;
033 import org.slf4j.LoggerFactory;
034
035 /**
036 * @version
037 */
038 public class CamelServlet extends HttpServlet {
039 private static final long serialVersionUID = -7061982839117697829L;
040 protected final transient Logger log = LoggerFactory.getLogger(getClass());
041
042 /**
043 * We have to define this explicitly so the name can be set as we can not always be
044 * sure that it is already set via the init method
045 */
046 private String servletName;
047
048 private ConcurrentHashMap<String, HttpConsumer> consumers = new ConcurrentHashMap<String, HttpConsumer>();
049
050 @Override
051 public void init(ServletConfig config) throws ServletException {
052 super.init(config);
053 this.servletName = config.getServletName();
054 }
055
056 @Override
057 protected void service(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
058 log.trace("Service: {}", request);
059
060 // Is there a consumer registered for the request.
061 HttpConsumer consumer = resolve(request);
062 if (consumer == null) {
063 log.debug("No consumer to service request {}", request);
064 response.sendError(HttpServletResponse.SC_NOT_FOUND);
065 return;
066 }
067
068 // are we suspended?
069 if (consumer.isSuspended()) {
070 log.debug("Consumer suspended, cannot service request {}", request);
071 response.sendError(HttpServletResponse.SC_SERVICE_UNAVAILABLE);
072 return;
073 }
074
075 // create exchange and set data on it
076 Exchange exchange = new DefaultExchange(consumer.getEndpoint(), ExchangePattern.InOut);
077 if (consumer.getEndpoint().isBridgeEndpoint()) {
078 exchange.setProperty(Exchange.SKIP_GZIP_ENCODING, Boolean.TRUE);
079 }
080 if (consumer.getEndpoint().isDisableStreamCache()) {
081 exchange.setProperty(Exchange.DISABLE_HTTP_STREAM_CACHE, Boolean.TRUE);
082 }
083
084 HttpHelper.setCharsetFromContentType(request.getContentType(), exchange);
085 exchange.setIn(new HttpMessage(exchange, request, response));
086
087 try {
088 log.trace("Processing request for exchangeId: {}", exchange.getExchangeId());
089 // process the exchange
090 consumer.getProcessor().process(exchange);
091 } catch (Exception e) {
092 exchange.setException(e);
093 }
094
095 try {
096 log.trace("Writing response for exchangeId: {}", exchange.getExchangeId());
097
098 // now lets output to the response
099 consumer.getBinding().writeResponse(exchange, response);
100 } catch (IOException e) {
101 log.error("Error processing request", e);
102 throw e;
103 } catch (Exception e) {
104 log.error("Error processing request", e);
105 throw new ServletException(e);
106 }
107 }
108
109 protected HttpConsumer resolve(HttpServletRequest request) {
110 String path = request.getPathInfo();
111 if (path == null) {
112 return null;
113 }
114 HttpConsumer answer = consumers.get(path);
115
116 if (answer == null) {
117 for (String key : consumers.keySet()) {
118 if (consumers.get(key).getEndpoint().isMatchOnUriPrefix() && path.startsWith(key)) {
119 answer = consumers.get(key);
120 break;
121 }
122 }
123 }
124 return answer;
125 }
126
127 public void connect(HttpConsumer consumer) {
128 log.debug("Connecting consumer: {}", consumer);
129 consumers.put(consumer.getPath(), consumer);
130 }
131
132 public void disconnect(HttpConsumer consumer) {
133 log.debug("Disconnecting consumer: {}", consumer);
134 consumers.remove(consumer.getPath());
135 }
136
137 public String getServletName() {
138 return servletName;
139 }
140
141 public void setServletName(String servletName) {
142 this.servletName = servletName;
143 }
144
145 }