Examples with PooledByteBufAllocator io.netty.buffer.PooledByteBufAllocator used on opensource projects

Example 31 with PooledByteBufAllocator

use of io.netty.buffer.PooledByteBufAllocator in project jocean-http by isdom.

the class DefaultHttpClientTestCase method testInitiatorInteractionSuccessAsHttps.

@Test(timeout = 5000)
public void testInitiatorInteractionSuccessAsHttps() throws Exception {
    // 配置 池化分配器 为 取消缓存，使用 Heap
    configDefaultAllocator();
    final PooledByteBufAllocator allocator = defaultAllocator();
    assertEquals(0, allActiveAllocationsCount(allocator));
    final BlockingQueue<HttpTrade> trades = new ArrayBlockingQueue<>(1);
    final String addr = UUID.randomUUID().toString();
    final Subscription server = TestHttpUtil.createTestServerWith(addr, trades, enableSSL4ServerWithSelfSigned(), Feature.ENABLE_LOGGING_OVER_SSL);
    final DefaultHttpClient client = new DefaultHttpClient(new TestChannelCreator(), enableSSL4Client(), Feature.ENABLE_LOGGING_OVER_SSL);
    try {
        startInteraction(client.initiator().remoteAddress(new LocalAddress(addr)), Observable.just(fullHttpRequest()), new Interaction() {

            @Override
            public void interact(final HttpInitiator initiator, final Observable<DisposableWrapper<FullHttpResponse>> getresp) throws Exception {
                final Observable<DisposableWrapper<FullHttpResponse>> cached = getresp.cache();
                // initiator 开始发送 请求
                cached.subscribe();
                LOG.debug("before get tarde");
                // server side recv req
                final HttpTrade trade = trades.take();
                LOG.debug("after get tarde");
                // recv all request
                trade.inbound().toCompletable().await();
                final ByteBuf svrRespContent = allocator.buffer(CONTENT.length).writeBytes(CONTENT);
                // send back resp
                trade.outbound(TestHttpUtil.buildByteBufResponse("text/plain", svrRespContent));
                // wait for recv all resp at client side
                cached.toCompletable().await();
                svrRespContent.release();
                assertTrue(Arrays.equals(dumpResponseContentAsBytes(cached), CONTENT));
            }
        });
    } finally {
        assertEquals(0, allActiveAllocationsCount(allocator));
        client.close();
        server.unsubscribe();
    }
}

Example 32 with PooledByteBufAllocator

use of io.netty.buffer.PooledByteBufAllocator in project jocean-http by isdom.

the class DefaultHttpClientTestCase method testInitiatorInteractionFailedAsHttp10ConnectionCloseMissingPartContent.

@Test(timeout = 5000)
public void testInitiatorInteractionFailedAsHttp10ConnectionCloseMissingPartContent() throws Exception {
    // 配置 池化分配器 为 取消缓存，使用 Heap
    configDefaultAllocator();
    final PooledByteBufAllocator allocator = defaultAllocator();
    final BlockingQueue<HttpTrade> trades = new ArrayBlockingQueue<>(1);
    final String addr = UUID.randomUUID().toString();
    final Subscription server = TestHttpUtil.createTestServerWith(addr, trades, Feature.ENABLE_LOGGING);
    final DefaultHttpClient client = new DefaultHttpClient(new TestChannelCreator(), Feature.ENABLE_LOGGING);
    try {
        final HttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_0, HttpMethod.GET, "/");
        request.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.CLOSE);
        startInteraction(client.initiator().remoteAddress(new LocalAddress(addr)), Observable.just(request), new Interaction() {

            @Override
            public void interact(final HttpInitiator initiator, final Observable<DisposableWrapper<FullHttpResponse>> getresp) throws Exception {
                final CountDownLatch cdl4initiator = new CountDownLatch(1);
                initiator.doOnTerminate(new Action0() {

                    @Override
                    public void call() {
                        cdl4initiator.countDown();
                    }
                });
                assertEquals(0, allActiveAllocationsCount(allocator));
                final TestSubscriber<DisposableWrapper<FullHttpResponse>> subscriber = new TestSubscriber<>();
                getresp.subscribe(subscriber);
                // server side recv req
                final HttpTrade trade = trades.take();
                // recv all request
                trade.inbound().toCompletable().await();
                final ByteBuf svrRespContent = allocator.buffer(CONTENT.length).writeBytes(CONTENT);
                // for HTTP 1.0 Connection: Close response behavior
                final FullHttpResponse fullrespfromsvr = new DefaultFullHttpResponse(HttpVersion.HTTP_1_0, HttpResponseStatus.OK, svrRespContent);
                fullrespfromsvr.headers().set(HttpHeaderNames.CONTENT_TYPE, "text/plain");
                // missing Content-Length
                fullrespfromsvr.headers().set(HttpHeaderNames.CONTENT_LENGTH, fullrespfromsvr.content().readableBytes() + 1);
                fullrespfromsvr.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.CLOSE);
                trade.outbound(Observable.just(fullrespfromsvr));
                TerminateAware.Util.awaitTerminated(trade);
                subscriber.awaitTerminalEvent();
                assertTrue(svrRespContent.release());
                subscriber.assertError(TransportException.class);
                // assertTrue(subscriber.getValueCount() > 0);
                cdl4initiator.await();
                assertEquals(0, allActiveAllocationsCount(allocator));
            }
        });
    } finally {
        client.close();
        server.unsubscribe();
    }
}

Example 33 with PooledByteBufAllocator

use of io.netty.buffer.PooledByteBufAllocator in project jocean-http by isdom.

the class DefaultHttpClientTestCase method testInitiatorInteractionDisconnectFromServerAfterConnectedAsHttps.

@Test(timeout = 5000)
public void testInitiatorInteractionDisconnectFromServerAfterConnectedAsHttps() throws Exception {
    // 配置 池化分配器 为 取消缓存，使用 Heap
    configDefaultAllocator();
    final PooledByteBufAllocator allocator = defaultAllocator();
    final BlockingQueue<HttpTrade> trades = new ArrayBlockingQueue<>(1);
    final String addr = UUID.randomUUID().toString();
    final Subscription server = TestHttpUtil.createTestServerWith(addr, trades, enableSSL4ServerWithSelfSigned(), Feature.ENABLE_LOGGING_OVER_SSL);
    final DefaultHttpClient client = new DefaultHttpClient(new TestChannelCreator(), enableSSL4Client(), Feature.ENABLE_LOGGING_OVER_SSL);
    assertEquals(0, allActiveAllocationsCount(allocator));
    try {
        startInteraction(client.initiator().remoteAddress(new LocalAddress(addr)), Observable.just(fullHttpRequest()), new Interaction() {

            @Override
            public void interact(final HttpInitiator initiator, final Observable<DisposableWrapper<FullHttpResponse>> getresp) throws Exception {
                final TestSubscriber<DisposableWrapper<FullHttpResponse>> subscriber = new TestSubscriber<>();
                getresp.subscribe(subscriber);
                // server side recv req
                final HttpTrade trade = trades.take();
                // recv request from client side
                trade.inbound().toCompletable().await();
                // disconnect
                trade.close();
                subscriber.awaitTerminalEvent();
                subscriber.assertError(TransportException.class);
                subscriber.assertNotCompleted();
                subscriber.assertNoValues();
            }
        });
        assertEquals(0, allActiveAllocationsCount(allocator));
    } finally {
        client.close();
        server.unsubscribe();
    }
}

Example 34 with PooledByteBufAllocator

use of io.netty.buffer.PooledByteBufAllocator in project jocean-http by isdom.

the class DefaultHttpClientTestCase method testInitiatorMultiCalldefineInteraction.

@Test(timeout = 5000)
public void testInitiatorMultiCalldefineInteraction() throws Exception {
    // 配置 池化分配器 为 取消缓存，使用 Heap
    configDefaultAllocator();
    final PooledByteBufAllocator allocator = defaultAllocator();
    assertEquals(0, allActiveAllocationsCount(allocator));
    final BlockingQueue<HttpTrade> trades = new ArrayBlockingQueue<>(1);
    final String addr = UUID.randomUUID().toString();
    final Subscription server = TestHttpUtil.createTestServerWith(addr, trades, Feature.ENABLE_LOGGING);
    final DefaultHttpClient client = new DefaultHttpClient(new TestChannelCreator(), Feature.ENABLE_LOGGING);
    try (final HttpInitiator initiator = client.initiator().remoteAddress(new LocalAddress(addr)).build().toBlocking().single()) {
        initiator.defineInteraction(Observable.just(fullHttpRequest()));
        initiator.defineInteraction(Observable.just(fullHttpRequest()));
        assertEquals(0, allActiveAllocationsCount(allocator));
    } finally {
        client.close();
        server.unsubscribe();
    }
}

Example 35 with PooledByteBufAllocator

use of io.netty.buffer.PooledByteBufAllocator in project jocean-http by isdom.

the class DefaultHttpClientTestCase method testInitiatorInteractionFailedAsHttps10ConnectionCloseMissingPartContent.

@Test(timeout = 5000)
public void testInitiatorInteractionFailedAsHttps10ConnectionCloseMissingPartContent() throws Exception {
    // 配置 池化分配器 为 取消缓存，使用 Heap
    configDefaultAllocator();
    final PooledByteBufAllocator allocator = defaultAllocator();
    final BlockingQueue<HttpTrade> trades = new ArrayBlockingQueue<>(1);
    final String addr = UUID.randomUUID().toString();
    final Subscription server = TestHttpUtil.createTestServerWith(addr, trades, enableSSL4ServerWithSelfSigned(), Feature.ENABLE_LOGGING_OVER_SSL);
    final DefaultHttpClient client = new DefaultHttpClient(new TestChannelCreator(), enableSSL4Client(), Feature.ENABLE_LOGGING_OVER_SSL);
    try {
        final HttpRequest request = new DefaultFullHttpRequest(HttpVersion.HTTP_1_0, HttpMethod.GET, "/");
        request.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.CLOSE);
        startInteraction(client.initiator().remoteAddress(new LocalAddress(addr)), Observable.just(request), new Interaction() {

            @Override
            public void interact(final HttpInitiator initiator, final Observable<DisposableWrapper<FullHttpResponse>> getresp) throws Exception {
                final CountDownLatch cdl4initiator = new CountDownLatch(1);
                initiator.doOnTerminate(new Action0() {

                    @Override
                    public void call() {
                        cdl4initiator.countDown();
                    }
                });
                final TestSubscriber<DisposableWrapper<FullHttpResponse>> subscriber = new TestSubscriber<>();
                getresp.subscribe(subscriber);
                // server side recv req
                final HttpTrade trade = trades.take();
                // recv all request
                trade.inbound().toCompletable().await();
                final ByteBuf svrRespContent = allocator.buffer(CONTENT.length).writeBytes(CONTENT);
                // for HTTP 1.0 Connection: Close response behavior
                final FullHttpResponse fullrespfromsvr = new DefaultFullHttpResponse(HttpVersion.HTTP_1_0, HttpResponseStatus.OK, svrRespContent);
                fullrespfromsvr.headers().set(HttpHeaderNames.CONTENT_TYPE, "text/plain");
                // missing Content-Length
                fullrespfromsvr.headers().set(HttpHeaderNames.CONTENT_LENGTH, fullrespfromsvr.content().readableBytes() + 1);
                fullrespfromsvr.headers().set(HttpHeaderNames.CONNECTION, HttpHeaderValues.CLOSE);
                trade.outbound(Observable.just(fullrespfromsvr));
                TerminateAware.Util.awaitTerminated(trade);
                subscriber.awaitTerminalEvent();
                assertTrue(svrRespContent.release());
                subscriber.assertError(TransportException.class);
                // assertTrue(subscriber.getValueCount() > 0);
                cdl4initiator.await();
                assertEquals(0, allActiveAllocationsCount(allocator));
            }
        });
    } finally {
        client.close();
        server.unsubscribe();
    }
}