View Javadoc
1   /*
2    * ====================================================================
3    * Licensed to the Apache Software Foundation (ASF) under one
4    * or more contributor license agreements.  See the NOTICE file
5    * distributed with this work for additional information
6    * regarding copyright ownership.  The ASF licenses this file
7    * to you under the Apache License, Version 2.0 (the
8    * "License"); you may not use this file except in compliance
9    * with the License.  You may obtain a copy of the License at
10   *
11   *   http://www.apache.org/licenses/LICENSE-2.0
12   *
13   * Unless required by applicable law or agreed to in writing,
14   * software distributed under the License is distributed on an
15   * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
16   * KIND, either express or implied.  See the License for the
17   * specific language governing permissions and limitations
18   * under the License.
19   * ====================================================================
20   *
21   * This software consists of voluntary contributions made by many
22   * individuals on behalf of the Apache Software Foundation.  For more
23   * information on the Apache Software Foundation, please see
24   * <http://www.apache.org/>.
25   *
26   */
27  
28  package org.apache.http.localserver;
29  
30  import java.io.IOException;
31  import java.util.Locale;
32  
33  import org.apache.http.HttpEntity;
34  import org.apache.http.HttpEntityEnclosingRequest;
35  import org.apache.http.HttpException;
36  import org.apache.http.HttpRequest;
37  import org.apache.http.HttpResponse;
38  import org.apache.http.HttpStatus;
39  import org.apache.http.MethodNotSupportedException;
40  import org.apache.http.entity.ByteArrayEntity;
41  import org.apache.http.protocol.HttpContext;
42  import org.apache.http.protocol.HttpRequestHandler;
43  import org.apache.http.util.EntityUtils;
44  
45  
46  
47  /**
48   * A handler that echos the incoming request entity.
49   *
50   *
51   *
52   * <!-- empty lines to avoid 'svn diff' problems -->
53   */
54  public class EchoHandler
55      implements HttpRequestHandler {
56  
57      // public default constructor
58  
59      /**
60       * Handles a request by echoing the incoming request entity.
61       * If there is no request entity, an empty document is returned.
62       *
63       * @param request   the request
64       * @param response  the response
65       * @param context   the context
66       *
67       * @throws HttpException    in case of a problem
68       * @throws IOException      in case of an IO problem
69       */
70      @Override
71      public void handle(final HttpRequest request,
72                         final HttpResponse response,
73                         final HttpContext context)
74          throws HttpException, IOException {
75  
76          final String method = request.getRequestLine().getMethod().toUpperCase(Locale.ROOT);
77          if (!"GET".equals(method) &&
78              !"POST".equals(method) &&
79              !"PUT".equals(method)
80              ) {
81              throw new MethodNotSupportedException
82                  (method + " not supported by " + getClass().getName());
83          }
84  
85          HttpEntity entity = null;
86          if (request instanceof HttpEntityEnclosingRequest) {
87              entity = ((HttpEntityEnclosingRequest)request).getEntity();
88          }
89  
90          // For some reason, just putting the incoming entity into
91          // the response will not work. We have to buffer the message.
92          final byte[] data;
93          if (entity == null) {
94              data = new byte [0];
95          } else {
96              data = EntityUtils.toByteArray(entity);
97          }
98  
99          final ByteArrayEntity bae = new ByteArrayEntity(data);
100         if (entity != null) {
101             bae.setContentType(entity.getContentType());
102         }
103         entity = bae;
104 
105         response.setStatusCode(HttpStatus.SC_OK);
106         response.setEntity(entity);
107 
108     } // handle
109 
110 
111 } // class EchoHandler