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.nio.integration;
29  
30  import java.io.IOException;
31  
32  import org.apache.http.HttpEntity;
33  import org.apache.http.HttpEntityEnclosingRequest;
34  import org.apache.http.HttpException;
35  import org.apache.http.HttpRequest;
36  import org.apache.http.HttpResponse;
37  import org.apache.http.HttpStatus;
38  import org.apache.http.entity.ContentType;
39  import org.apache.http.nio.entity.NStringEntity;
40  import org.apache.http.protocol.HttpContext;
41  import org.apache.http.protocol.HttpRequestHandler;
42  import org.apache.http.util.EntityUtils;
43  
44  final class SimpleRequestHandler implements HttpRequestHandler {
45  
46      private final boolean chunking;
47  
48      SimpleRequestHandler() {
49          this(false);
50      }
51  
52      SimpleRequestHandler(boolean chunking) {
53          super();
54          this.chunking = chunking;
55      }
56  
57      public void handle(
58              final HttpRequest request,
59              final HttpResponse response,
60              final HttpContext context) throws HttpException, IOException {
61  
62          String content = null;
63          if (request instanceof HttpEntityEnclosingRequest) {
64              HttpEntity entity = ((HttpEntityEnclosingRequest) request).getEntity();
65              if (entity != null) {
66                  content = EntityUtils.toString(entity);
67              } else {
68                  response.setStatusCode(HttpStatus.SC_BAD_REQUEST);
69                  content = "Request entity not avaialble";
70              }
71          } else {
72              String s = request.getRequestLine().getUri();
73              int idx = s.indexOf('x');
74              if (idx == -1) {
75                  throw new HttpException("Unexpected request-URI format");
76              }
77              String pattern = s.substring(0, idx);
78              int count = Integer.parseInt(s.substring(idx + 1, s.length()));
79  
80              StringBuilder buffer = new StringBuilder();
81              for (int i = 0; i < count; i++) {
82                  buffer.append(pattern);
83              }
84              content = buffer.toString();
85          }
86          NStringEntity entity = new NStringEntity(content, ContentType.DEFAULT_TEXT);
87          entity.setChunked(this.chunking);
88          response.setEntity(entity);
89      }
90  
91  }