Example usage for io.netty.channel WriteBufferWaterMark low

Introduction

In this page you can find the example usage for io.netty.channel WriteBufferWaterMark low.

Prototype

int low

To view the source code for io.netty.channel WriteBufferWaterMark low.

Click Source Link

Usage

From source file:io.grpc.netty.NettyServerTest.java

License:Apache License

@Test(timeout = 60000)
public void childChannelOptions() throws Exception {
    final int originalLowWaterMark = 2097169;
    final int originalHighWaterMark = 2097211;

    Map<ChannelOption<?>, Object> channelOptions = new HashMap<>();

    channelOptions.put(ChannelOption.WRITE_BUFFER_WATER_MARK,
            new WriteBufferWaterMark(originalLowWaterMark, originalHighWaterMark));

    final AtomicInteger lowWaterMark = new AtomicInteger(0);
    final AtomicInteger highWaterMark = new AtomicInteger(0);

    final CountDownLatch countDownLatch = new CountDownLatch(1);

    InetSocketAddress addr = new InetSocketAddress(0);
    NettyServer ns = new NettyServer(addr, Utils.DEFAULT_SERVER_CHANNEL_FACTORY, channelOptions,
            SharedResourcePool.forResource(Utils.DEFAULT_BOSS_EVENT_LOOP_GROUP),
            SharedResourcePool.forResource(Utils.DEFAULT_WORKER_EVENT_LOOP_GROUP),
            ProtocolNegotiators.plaintext(), Collections.<ServerStreamTracer.Factory>emptyList(),
            TransportTracer.getDefaultFactory(), 1, // ignore
            1, // ignore
            1, // ignore
            1, // ignore
            1, // ignore
            1, 1, // ignore
            1, 1, // ignore
            true, 0, // ignore
            channelz);//ww w . j a  v a 2  s . c o  m
    ns.start(new ServerListener() {
        @Override
        public ServerTransportListener transportCreated(ServerTransport transport) {
            Channel channel = ((NettyServerTransport) transport).channel();
            WriteBufferWaterMark writeBufferWaterMark = channel.config()
                    .getOption(ChannelOption.WRITE_BUFFER_WATER_MARK);
            lowWaterMark.set(writeBufferWaterMark.low());
            highWaterMark.set(writeBufferWaterMark.high());

            countDownLatch.countDown();

            return new NoopServerTransportListener();
        }

        @Override
        public void serverShutdown() {
        }
    });

    Socket socket = new Socket();
    socket.connect(ns.getListenSocketAddress(), /* timeout= */ 8000);
    countDownLatch.await();
    socket.close();

    assertThat(lowWaterMark.get()).isEqualTo(originalLowWaterMark);
    assertThat(highWaterMark.get()).isEqualTo(originalHighWaterMark);

    ns.shutdown();
}

---