Examples with PollItem org.zeromq.ZMQ.PollItem used on opensource projects

Example 1 with PollItem

use of org.zeromq.ZMQ.PollItem in project jeromq by zeromq.

the class TestZLoop method testZLoopEndReactorFromTimer.

@Test(timeout = 1000)
public void testZLoopEndReactorFromTimer() {
    int rc = 0;
    ZLoop loop = new ZLoop(ctx);
    assert (loop != null);
    ZLoop.IZLoopHandler timerEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            ((Socket) arg).send("PING", 0);
            return 0;
        }
    };
    ZLoop.IZLoopHandler socketEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            // After 10 msecs, fire an event that ends the reactor
            ZLoop.IZLoopHandler shutdownEvent = new ZLoop.IZLoopHandler() {

                @Override
                public int handle(ZLoop loop, PollItem item, Object arg) {
                    received = ((Socket) arg).recvStr(0);
                    // Just end the reactor
                    return -1;
                }
            };
            loop.addTimer(10, 1, shutdownEvent, arg);
            return 0;
        }
    };
    // Fire event that sends a ping message to output
    loop.addTimer(0, 1, timerEvent, input);
    // When we get the ping message, end the reactor
    PollItem pollInput = new PollItem(output, Poller.POLLIN);
    rc = loop.addPoller(pollInput, socketEvent, output);
    Assert.assertEquals(0, rc);
    loop.start();
    loop.removePoller(pollInput);
    Assert.assertEquals("PING", received);
}

Example 2 with PollItem

use of org.zeromq.ZMQ.PollItem in project jeromq by zeromq.

the class TestZLoop method testZLoopWithUDP.

@Test
public void testZLoopWithUDP() throws IOException {
    final int port = Utils.findOpenPort();
    final InetSocketAddress addr = new InetSocketAddress("127.0.0.1", port);
    ZLoop loop = new ZLoop(ctx);
    DatagramChannel udpIn = DatagramChannel.open();
    assertThat(udpIn, notNullValue());
    udpIn.configureBlocking(false);
    udpIn.socket().bind(new InetSocketAddress(port));
    DatagramChannel udpOut = DatagramChannel.open();
    assertThat(udpOut, notNullValue());
    udpOut.configureBlocking(false);
    udpOut.socket().connect(addr);
    final AtomicInteger counter = new AtomicInteger();
    final AtomicBoolean done = new AtomicBoolean();
    loop.addPoller(new PollItem(udpIn, ZMQ.Poller.POLLIN), new IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            DatagramChannel udpIn = (DatagramChannel) arg;
            ByteBuffer bb = ByteBuffer.allocate(3);
            try {
                udpIn.receive(bb);
                String read = new String(bb.array(), 0, bb.limit(), ZMQ.CHARSET);
                assertThat(read, is("udp"));
                done.set(true);
                counter.incrementAndGet();
            } catch (IOException e) {
                e.printStackTrace();
                fail();
            }
            return -1;
        }
    }, udpIn);
    loop.addPoller(new PollItem(udpOut, ZMQ.Poller.POLLOUT), new IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            DatagramChannel udpOut = (DatagramChannel) arg;
            try {
                ByteBuffer bb = ByteBuffer.allocate(3);
                bb.put("udp".getBytes(ZMQ.CHARSET));
                bb.flip();
                int written = udpOut.send(bb, addr);
                assertThat(written, is(3));
                counter.incrementAndGet();
            } catch (IOException e) {
                e.printStackTrace();
                fail();
            }
            return 0;
        }
    }, udpOut);
    loop.start();
    assertThat(done.get(), is(true));
    assertThat(counter.get(), is(2));
    udpIn.close();
    udpOut.close();
}

Example 3 with PollItem

use of org.zeromq.ZMQ.PollItem in project jeromq by zeromq.

the class TestZLoop method testZLoopAddTimerFromSocketHandler.

@Test(timeout = 1000)
public void testZLoopAddTimerFromSocketHandler() {
    int rc = 0;
    ZLoop loop = new ZLoop(ctx);
    assert (loop != null);
    ZLoop.IZLoopHandler timerEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            ((Socket) arg).send("PING", 0);
            return 0;
        }
    };
    ZLoop.IZLoopHandler socketEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            final long now = System.currentTimeMillis();
            ZLoop.IZLoopHandler timerEvent2 = new ZLoop.IZLoopHandler() {

                @Override
                public int handle(ZLoop loop, PollItem item, Object arg) {
                    final long now2 = System.currentTimeMillis();
                    Assert.assertTrue(now2 >= now + 10);
                    received = ((Socket) arg).recvStr(0);
                    // Just end the reactor
                    return -1;
                }
            };
            // After 10 msec fire a timer that ends the reactor
            loop.addTimer(10, 1, timerEvent2, arg);
            return 0;
        }
    };
    // Fire a timer that sends the ping message
    loop.addTimer(0, 1, timerEvent, input);
    // When we get the ping message, end the reactor
    PollItem pollInput = new PollItem(output, Poller.POLLIN);
    rc = loop.addPoller(pollInput, socketEvent, output);
    Assert.assertEquals(0, rc);
    loop.start();
    loop.removePoller(pollInput);
    Assert.assertEquals("PING", received);
}

Example 4 with PollItem

use of org.zeromq.ZMQ.PollItem in project jeromq by zeromq.

the class ZLoopTest method testZLoop.

@Test
public void testZLoop() {
    int rc = 0;
    ZLoop loop = new ZLoop(ctx);
    assertThat(loop, notNullValue());
    ZLoop.IZLoopHandler timerEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            ((Socket) arg).send("PING", 0);
            return 0;
        }
    };
    ZLoop.IZLoopHandler socketEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            received = ((Socket) arg).recvStr(0);
            // Just end the reactor
            return -1;
        }
    };
    // After 10 msecs, send a ping message to output
    loop.addTimer(10, 1, timerEvent, input);
    // When we get the ping message, end the reactor
    PollItem pollInput = new PollItem(output, Poller.POLLIN);
    rc = loop.addPoller(pollInput, socketEvent, output);
    assertThat(rc, is(0));
    loop.start();
    loop.removePoller(pollInput);
    assertThat(received, is("PING"));
}

Example 5 with PollItem

use of org.zeromq.ZMQ.PollItem in project jeromq by zeromq.

the class ZLoopTest method testZLoopEndReactorFromTimer.

@Test(timeout = 1000)
public void testZLoopEndReactorFromTimer() {
    int rc = 0;
    ZLoop loop = new ZLoop(ctx);
    assertThat(loop, notNullValue());
    ZLoop.IZLoopHandler timerEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            ((Socket) arg).send("PING", 0);
            return 0;
        }
    };
    ZLoop.IZLoopHandler socketEvent = new ZLoop.IZLoopHandler() {

        @Override
        public int handle(ZLoop loop, PollItem item, Object arg) {
            // After 10 msecs, fire an event that ends the reactor
            ZLoop.IZLoopHandler shutdownEvent = new ZLoop.IZLoopHandler() {

                @Override
                public int handle(ZLoop loop, PollItem item, Object arg) {
                    received = ((Socket) arg).recvStr(0);
                    // Just end the reactor
                    return -1;
                }
            };
            loop.addTimer(10, 1, shutdownEvent, arg);
            return 0;
        }
    };
    // Fire event that sends a ping message to output
    loop.addTimer(0, 1, timerEvent, input);
    // When we get the ping message, end the reactor
    PollItem pollInput = new PollItem(output, Poller.POLLIN);
    rc = loop.addPoller(pollInput, socketEvent, output);
    assertThat(rc, is(0));
    loop.start();
    loop.removePoller(pollInput);
    assertThat(received, is("PING"));
}