testCqe() — netty Function Reference

Source Code

transport-native-io_uring/src/test/java/io/netty/channel/uring/SubmissionQueueTest.java lines 203–251

    @ParameterizedTest
    @EnumSource(CqeSize.class)
    public void testCqe(CqeSize cqeSize) {
        assumeTrue(Native.ioUringSetupSupportsFlags(cqeSize.setupFlag()));

        RingBuffer ringBuffer = Native.createRingBuffer(8, cqeSize.setupFlag());
        ringBuffer.enable();
        try {
            SubmissionQueue submissionQueue = ringBuffer.ioUringSubmissionQueue();
            final CompletionQueue completionQueue = ringBuffer.ioUringCompletionQueue();

            assertNotNull(ringBuffer);
            assertNotNull(submissionQueue);
            assertNotNull(completionQueue);

            long result;
            if (cqeSize == CqeSize.MIXED) {
                result = submissionQueue.addNop((byte) 0, Native.IORING_NOP_CQE32, 1);
            } else {
                result = submissionQueue.addNop((byte) 0, 1);
            }
            assertThat(result).isNotZero();
            assertThat(submissionQueue.addNop((byte) 0, 1)).isNotZero();

            assertEquals(2, submissionQueue.submitAndGet());
            assertEquals(cqeSize == CqeSize.MIXED ? 3 : 2, completionQueue.count());
            int processed = completionQueue.process(new CompletionCallback() {
                private boolean first = true;
                @Override
                public void handle(int res, int flags, long udata, ByteBuffer extraCqeData) {
                    assertEquals(0, res);
                    assertEquals(cqeSize == CqeSize.MIXED && first ? Native.IORING_CQE_F_32 : 0, flags);
                    assertEquals(1, udata);
                    if (cqeSize == CqeSize.LARGE || (cqeSize == CqeSize.MIXED && first)) {
                        assertNotNull(extraCqeData);
                        assertEquals(0, extraCqeData.position());
                        assertEquals(16, extraCqeData.limit());
                    } else {
                        assertNull(extraCqeData);
                    }
                    first = false;
                }
            });
            assertEquals(2, processed);
            assertFalse(completionQueue.hasCompletions());
        } finally {
            ringBuffer.close();
        }
    }