Search in sources :

Example 1 with SIZE_OF_LONG

use of org.agrona.BitUtil.SIZE_OF_LONG in project Aeron by real-logic.

the class HackSelectReceiveSendUdpPong method run.

private void run() throws IOException {
    final InetSocketAddress sendAddress = new InetSocketAddress("localhost", Common.PONG_PORT);
    final ByteBuffer buffer = ByteBuffer.allocateDirect(Configuration.MTU_LENGTH_DEFAULT);
    final DatagramChannel receiveChannel = DatagramChannel.open();
    Common.init(receiveChannel);
    receiveChannel.bind(new InetSocketAddress("localhost", Common.PING_PORT));
    final DatagramChannel sendChannel = DatagramChannel.open();
    Common.init(sendChannel);
    final Selector selector = Selector.open();
    final NioSelectedKeySet keySet = Common.keySet(selector);
    final ToIntFunction<SelectionKey> handler = (key) -> {
        try {
            buffer.clear();
            receiveChannel.receive(buffer);
            final long receivedSequenceNumber = buffer.getLong(0);
            final long receivedTimestamp = buffer.getLong(SIZE_OF_LONG);
            buffer.clear();
            buffer.putLong(receivedSequenceNumber);
            buffer.putLong(receivedTimestamp);
            buffer.flip();
            sendChannel.send(buffer, sendAddress);
        } catch (final IOException ex) {
            ex.printStackTrace();
        }
        return 1;
    };
    receiveChannel.register(selector, OP_READ, null);
    final AtomicBoolean running = new AtomicBoolean(true);
    SigInt.register(() -> running.set(false));
    while (true) {
        while (selector.selectNow() == 0) {
            if (!running.get()) {
                return;
            }
        }
        keySet.forEach(handler);
    }
}
Also used : SelectionKey(java.nio.channels.SelectionKey) DatagramChannel(java.nio.channels.DatagramChannel) Selector(java.nio.channels.Selector) ToIntFunction(java.util.function.ToIntFunction) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) IOException(java.io.IOException) InetSocketAddress(java.net.InetSocketAddress) SigInt(org.agrona.concurrent.SigInt) ByteBuffer(java.nio.ByteBuffer) SIZE_OF_LONG(org.agrona.BitUtil.SIZE_OF_LONG) Configuration(io.aeron.driver.Configuration) NioSelectedKeySet(org.agrona.nio.NioSelectedKeySet) OP_READ(java.nio.channels.SelectionKey.OP_READ) SelectionKey(java.nio.channels.SelectionKey) AtomicBoolean(java.util.concurrent.atomic.AtomicBoolean) NioSelectedKeySet(org.agrona.nio.NioSelectedKeySet) InetSocketAddress(java.net.InetSocketAddress) DatagramChannel(java.nio.channels.DatagramChannel) IOException(java.io.IOException) ByteBuffer(java.nio.ByteBuffer) Selector(java.nio.channels.Selector)

Aggregations

Configuration (io.aeron.driver.Configuration)1 IOException (java.io.IOException)1 InetSocketAddress (java.net.InetSocketAddress)1 ByteBuffer (java.nio.ByteBuffer)1 DatagramChannel (java.nio.channels.DatagramChannel)1 SelectionKey (java.nio.channels.SelectionKey)1 OP_READ (java.nio.channels.SelectionKey.OP_READ)1 Selector (java.nio.channels.Selector)1 AtomicBoolean (java.util.concurrent.atomic.AtomicBoolean)1 ToIntFunction (java.util.function.ToIntFunction)1 SIZE_OF_LONG (org.agrona.BitUtil.SIZE_OF_LONG)1 SigInt (org.agrona.concurrent.SigInt)1 NioSelectedKeySet (org.agrona.nio.NioSelectedKeySet)1