コード例 #1
0
ファイル: ProxyHandler.java プロジェクト: n1hility/undertow
 void cancel(final HttpServerExchange exchange) {
   final ProxyConnection connectionAttachment = exchange.getAttachment(CONNECTION);
   if (connectionAttachment != null) {
     ClientConnection clientConnection = connectionAttachment.getConnection();
     UndertowLogger.PROXY_REQUEST_LOGGER.timingOutRequest(
         clientConnection.getPeerAddress() + "" + exchange.getRequestURI());
     IoUtils.safeClose(clientConnection);
   } else {
     UndertowLogger.PROXY_REQUEST_LOGGER.timingOutRequest(exchange.getRequestURI());
   }
   if (exchange.isResponseStarted()) {
     IoUtils.safeClose(exchange.getConnection());
   } else {
     exchange.setStatusCode(StatusCodes.SERVICE_UNAVAILABLE);
     exchange.endExchange();
   }
 }
コード例 #2
0
  protected ResponseEntity<String> executeRequest(
      URI url, HttpString method, HttpHeaders headers, String body) {
    CountDownLatch latch = new CountDownLatch(1);
    List<ClientResponse> responses = new CopyOnWriteArrayList<ClientResponse>();

    try {
      ClientConnection connection =
          this.undertowBufferSupport
              .httpClientConnect(this.httpClient, url, this.worker, this.optionMap)
              .get();
      try {
        ClientRequest request = new ClientRequest().setMethod(method).setPath(url.getPath());
        request.getRequestHeaders().add(HttpString.tryFromString(HttpHeaders.HOST), url.getHost());
        if (body != null && !body.isEmpty()) {
          HttpString headerName = HttpString.tryFromString(HttpHeaders.CONTENT_LENGTH);
          request.getRequestHeaders().add(headerName, body.length());
        }
        addHttpHeaders(request, headers);
        connection.sendRequest(request, createRequestCallback(body, responses, latch));

        latch.await();
        ClientResponse response = responses.iterator().next();
        HttpStatus status = HttpStatus.valueOf(response.getResponseCode());
        HttpHeaders responseHeaders = toHttpHeaders(response.getResponseHeaders());
        String responseBody = response.getAttachment(RESPONSE_BODY);
        return (responseBody != null
            ? new ResponseEntity<String>(responseBody, responseHeaders, status)
            : new ResponseEntity<String>(responseHeaders, status));
      } finally {
        IoUtils.safeClose(connection);
      }
    } catch (IOException ex) {
      throw new SockJsTransportFailureException("Failed to execute request to " + url, ex);
    } catch (InterruptedException ex) {
      throw new SockJsTransportFailureException(
          "Interrupted while processing request to " + url, ex);
    }
  }