/** * Delegates to {@link BodyDeferringAsyncHandler#getResponse()}. Will * blocks as long as headers arrives only. Might return * <code>null</code>. See * {@link BodyDeferringAsyncHandler#getResponse()} method for details. * * @return a {@link Response} * @throws InterruptedException if the latch is interrupted * @throws IOException if the handler completed with an exception */ public Response getAsapResponse() throws InterruptedException, IOException { return bdah.getResponse(); }
@Test public void testPipedStreams() throws Exception { try (AsyncHttpClient client = asyncHttpClient(getAsyncHttpClientConfig())) { PipedOutputStream pout = new PipedOutputStream(); try (PipedInputStream pin = new PipedInputStream(pout)) { BodyDeferringAsyncHandler handler = new BodyDeferringAsyncHandler(pout); ListenableFuture<Response> respFut = client.prepareGet(getTargetUrl()).execute(handler); Response resp = handler.getResponse(); if (resp.getStatusCode() == 200) { try (BodyDeferringInputStream is = new BodyDeferringInputStream(respFut, handler, pin)) { String body = IOUtils.toString(is, StandardCharsets.UTF_8); assertTrue(body.contains("ABCDEF")); } } else { throw new IOException("HTTP error " + resp.getStatusCode()); } } } }
@Test public void deferredSimple() throws IOException, ExecutionException, InterruptedException { try (AsyncHttpClient client = asyncHttpClient(getAsyncHttpClientConfig())) { BoundRequestBuilder r = client.prepareGet(getTargetUrl()); CountingOutputStream cos = new CountingOutputStream(); BodyDeferringAsyncHandler bdah = new BodyDeferringAsyncHandler(cos); Future<Response> f = r.execute(bdah); Response resp = bdah.getResponse(); assertNotNull(resp); assertEquals(resp.getStatusCode(), HttpServletResponse.SC_OK); assertEquals(resp.getHeader(CONTENT_LENGTH), String.valueOf(CONTENT_LENGTH_VALUE)); // we got headers only, it's probably not all yet here (we have BIG file // downloading) assertTrue(cos.getByteCount() <= CONTENT_LENGTH_VALUE); // now be polite and wait for body arrival too (otherwise we would be // dropping the "line" on server) f.get(); // it all should be here now assertEquals(cos.getByteCount(), CONTENT_LENGTH_VALUE); } }
@Test(expectedExceptions = RemotelyClosedException.class, enabled = false) public void deferredSimpleWithFailure() throws Throwable { try (AsyncHttpClient client = asyncHttpClient(getAsyncHttpClientConfig())) { BoundRequestBuilder r = client.prepareGet(getTargetUrl()).addHeader("X-FAIL-TRANSFER", Boolean.TRUE.toString()); CountingOutputStream cos = new CountingOutputStream(); BodyDeferringAsyncHandler bdah = new BodyDeferringAsyncHandler(cos); Future<Response> f = r.execute(bdah); Response resp = bdah.getResponse(); assertNotNull(resp); assertEquals(resp.getStatusCode(), HttpServletResponse.SC_OK); assertEquals(resp.getHeader(CONTENT_LENGTH), String.valueOf(CONTENT_LENGTH_VALUE)); // we got headers only, it's probably not all yet here (we have BIG file // downloading) assertTrue(cos.getByteCount() <= CONTENT_LENGTH_VALUE); // now be polite and wait for body arrival too (otherwise we would be // dropping the "line" on server) try { f.get(); } catch (ExecutionException e) { // good // it's incomplete, there was an error assertNotEquals(cos.getByteCount(), CONTENT_LENGTH_VALUE); throw e.getCause(); } } }
@Test(expectedExceptions = IOException.class) public void testConnectionRefused() throws IOException, InterruptedException { int newPortWithoutAnyoneListening = findFreePort(); try (AsyncHttpClient client = asyncHttpClient(getAsyncHttpClientConfig())) { BoundRequestBuilder r = client.prepareGet("http://localhost:" + newPortWithoutAnyoneListening + "/testConnectionRefused"); CountingOutputStream cos = new CountingOutputStream(); BodyDeferringAsyncHandler bdah = new BodyDeferringAsyncHandler(cos); r.execute(bdah); bdah.getResponse(); } }
/** * Delegates to {@link BodyDeferringAsyncHandler#getResponse()}. Will * blocks as long as headers arrives only. Might return * <code>null</code>. See * {@link BodyDeferringAsyncHandler#getResponse()} method for details. * * @return a {@link Response} * @throws InterruptedException */ public Response getAsapResponse() throws InterruptedException, IOException { return bdah.getResponse(); }