Examples with Callback org.eclipse.jetty.util.Callback used on opensource projects

Example 66 with Callback

use of org.eclipse.jetty.util.Callback in project jetty.project by eclipse.

the class HttpClientTest method testCompleteNotInvokedUntilContentConsumed.

@Test
public void testCompleteNotInvokedUntilContentConsumed() throws Exception {
    start(new AbstractHandler() {

        @Override
        public void handle(String target, org.eclipse.jetty.server.Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
            baseRequest.setHandled(true);
            ServletOutputStream output = response.getOutputStream();
            output.write(new byte[1024]);
        }
    });
    final AtomicReference<Callback> callbackRef = new AtomicReference<>();
    final CountDownLatch contentLatch = new CountDownLatch(1);
    final CountDownLatch completeLatch = new CountDownLatch(1);
    client.newRequest("localhost", connector.getLocalPort()).scheme(scheme).send(new Response.Listener.Adapter() {

        @Override
        public void onContent(Response response, ByteBuffer content, Callback callback) {
            // Do not notify the callback yet.
            callbackRef.set(callback);
            contentLatch.countDown();
        }

        @Override
        public void onComplete(Result result) {
            if (result.isSucceeded())
                completeLatch.countDown();
        }
    });
    Assert.assertTrue(contentLatch.await(5, TimeUnit.SECONDS));
    // Make sure the complete event is not emitted.
    Assert.assertFalse(completeLatch.await(1, TimeUnit.SECONDS));
    // Consume the content.
    callbackRef.get().succeeded();
    // Now the complete event is emitted.
    Assert.assertTrue(completeLatch.await(5, TimeUnit.SECONDS));
}

Example 67 with Callback

use of org.eclipse.jetty.util.Callback in project jetty.project by eclipse.

the class HttpClientStreamTest method testUploadWithDeferredContentProviderFailsMultipleOffers.

@Test
public void testUploadWithDeferredContentProviderFailsMultipleOffers() throws Exception {
    start(new EmptyServerHandler());
    final CountDownLatch failLatch = new CountDownLatch(2);
    final Callback callback = new Callback() {

        @Override
        public void failed(Throwable x) {
            failLatch.countDown();
        }
    };
    final CountDownLatch completeLatch = new CountDownLatch(1);
    final DeferredContentProvider content = new DeferredContentProvider();
    client.newRequest("localhost", connector.getLocalPort()).scheme(getScheme()).content(content).onRequestBegin(request -> {
        content.offer(ByteBuffer.wrap(new byte[256]), callback);
        content.offer(ByteBuffer.wrap(new byte[256]), callback);
        request.abort(new Exception("explicitly_thrown_by_test"));
    }).send(result -> {
        if (result.isFailed())
            completeLatch.countDown();
    });
    Assert.assertTrue(completeLatch.await(5, TimeUnit.SECONDS));
    Assert.assertTrue(failLatch.await(5, TimeUnit.SECONDS));
    // Make sure that adding more content results in the callback to be failed.
    final CountDownLatch latch = new CountDownLatch(1);
    content.offer(ByteBuffer.wrap(new byte[128]), new Callback() {

        @Override
        public void failed(Throwable x) {
            latch.countDown();
        }
    });
    Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
}

Example 68 with Callback

use of org.eclipse.jetty.util.Callback in project jetty.project by eclipse.

the class HttpClientStreamTest method testInputStreamResponseListenerFailedWhileWaiting.

@Test
public void testInputStreamResponseListenerFailedWhileWaiting() throws Exception {
    start(new AbstractHandler() {

        @Override
        public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
            baseRequest.setHandled(true);
            byte[] data = new byte[1024];
            response.setContentLength(data.length);
            ServletOutputStream output = response.getOutputStream();
            output.write(data);
        }
    });
    CountDownLatch failedLatch = new CountDownLatch(1);
    CountDownLatch contentLatch = new CountDownLatch(1);
    InputStreamResponseListener listener = new InputStreamResponseListener() {

        @Override
        public void onContent(Response response, ByteBuffer content, Callback callback) {
            super.onContent(response, content, new Callback() {

                @Override
                public void failed(Throwable x) {
                    failedLatch.countDown();
                    callback.failed(x);
                }
            });
            contentLatch.countDown();
        }
    };
    client.newRequest("localhost", connector.getLocalPort()).scheme(getScheme()).send(listener);
    Response response = listener.get(5, TimeUnit.SECONDS);
    Assert.assertEquals(HttpStatus.OK_200, response.getStatus());
    // Wait until we get some content.
    Assert.assertTrue(contentLatch.await(5, TimeUnit.SECONDS));
    // Abort the response.
    response.abort(new Exception());
    // Make sure that the callback has been invoked.
    Assert.assertTrue(failedLatch.await(5, TimeUnit.SECONDS));
}

Example 69 with Callback

use of org.eclipse.jetty.util.Callback in project jetty.project by eclipse.

the class HttpClientStreamTest method testInputStreamResponseListenerClosedBeforeContent.

@Test(expected = AsynchronousCloseException.class)
public void testInputStreamResponseListenerClosedBeforeContent() throws Exception {
    AtomicReference<AsyncContext> contextRef = new AtomicReference<>();
    start(new AbstractHandler() {

        @Override
        public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
            baseRequest.setHandled(true);
            contextRef.set(request.startAsync());
            response.flushBuffer();
        }
    });
    CountDownLatch latch = new CountDownLatch(1);
    InputStreamResponseListener listener = new InputStreamResponseListener() {

        @Override
        public void onContent(Response response, ByteBuffer content, Callback callback) {
            super.onContent(response, content, new Callback() {

                @Override
                public void failed(Throwable x) {
                    latch.countDown();
                    callback.failed(x);
                }
            });
        }
    };
    client.newRequest("localhost", connector.getLocalPort()).scheme(getScheme()).send(listener);
    Response response = listener.get(5, TimeUnit.SECONDS);
    Assert.assertEquals(HttpStatus.OK_200, response.getStatus());
    InputStream input = listener.getInputStream();
    input.close();
    AsyncContext asyncContext = contextRef.get();
    asyncContext.getResponse().getOutputStream().write(new byte[1024]);
    asyncContext.complete();
    Assert.assertTrue(latch.await(5, TimeUnit.SECONDS));
    // Throws
    input.read();
}

Example 70 with Callback

use of org.eclipse.jetty.util.Callback in project jetty.project by eclipse.

the class HttpClientTest method testAsyncResponseContentBackPressure.

@Test
public void testAsyncResponseContentBackPressure() throws Exception {
    start(new AbstractHandler() {

        @Override
        public void handle(String target, Request baseRequest, HttpServletRequest request, HttpServletResponse response) throws IOException, ServletException {
            baseRequest.setHandled(true);
            // Large write to generate multiple DATA frames.
            response.getOutputStream().write(new byte[256 * 1024]);
        }
    });
    CountDownLatch completeLatch = new CountDownLatch(1);
    AtomicInteger counter = new AtomicInteger();
    AtomicReference<Callback> callbackRef = new AtomicReference<>();
    AtomicReference<CountDownLatch> latchRef = new AtomicReference<>(new CountDownLatch(1));
    client.newRequest("localhost", connector.getLocalPort()).scheme(getScheme()).onResponseContentAsync((response, content, callback) -> {
        if (counter.incrementAndGet() == 1) {
            callbackRef.set(callback);
            latchRef.get().countDown();
        } else {
            callback.succeeded();
        }
    }).send(result -> completeLatch.countDown());
    Assert.assertTrue(latchRef.get().await(5, TimeUnit.SECONDS));
    // Wait some time to verify that back pressure is applied correctly.
    Thread.sleep(1000);
    Assert.assertEquals(1, counter.get());
    callbackRef.get().succeeded();
    Assert.assertTrue(completeLatch.await(5, TimeUnit.SECONDS));
}