Examples with Page org.opensearch.nio.Page used on opensource projects

Example 1 with Page

use of org.opensearch.nio.Page in project OpenSearch by opensearch-project.

the class PagedByteBufTests method testBytesAreUsed.

public void testBytesAreUsed() {
    byte[] bytes1 = new byte[10];
    byte[] bytes2 = new byte[10];
    for (int i = 0; i < 10; ++i) {
        bytes1[i] = (byte) i;
    }
    for (int i = 10; i < 20; ++i) {
        bytes2[i - 10] = (byte) i;
    }
    Page[] pages = new Page[2];
    pages[0] = new Page(ByteBuffer.wrap(bytes1), () -> {
    });
    pages[1] = new Page(ByteBuffer.wrap(bytes2), () -> {
    });
    ByteBuf byteBuf = PagedByteBuf.byteBufFromPages(pages);
    assertEquals(20, byteBuf.readableBytes());
    for (int i = 0; i < 20; ++i) {
        assertEquals((byte) i, byteBuf.getByte(i));
    }
    Page[] pages2 = new Page[2];
    ByteBuffer firstBuffer = ByteBuffer.wrap(bytes1);
    firstBuffer.position(2);
    ByteBuffer secondBuffer = ByteBuffer.wrap(bytes2);
    secondBuffer.limit(8);
    pages2[0] = new Page(firstBuffer, () -> {
    });
    pages2[1] = new Page(secondBuffer, () -> {
    });
    ByteBuf byteBuf2 = PagedByteBuf.byteBufFromPages(pages2);
    assertEquals(16, byteBuf2.readableBytes());
    for (int i = 2; i < 18; ++i) {
        assertEquals((byte) i, byteBuf2.getByte(i - 2));
    }
}

Example 2 with Page

use of org.opensearch.nio.Page in project OpenSearch by opensearch-project.

the class PagedByteBufTests method testReleasingPage.

public void testReleasingPage() {
    AtomicInteger integer = new AtomicInteger(0);
    int pageCount = randomInt(10) + 1;
    ArrayList<Page> pages = new ArrayList<>();
    for (int i = 0; i < pageCount; ++i) {
        pages.add(new Page(ByteBuffer.allocate(10), integer::incrementAndGet));
    }
    ByteBuf byteBuf = PagedByteBuf.byteBufFromPages(pages.toArray(new Page[0]));
    assertEquals(0, integer.get());
    byteBuf.retain();
    byteBuf.release();
    assertEquals(0, integer.get());
    ByteBuf secondBuf = byteBuf.retainedSlice();
    byteBuf.release();
    assertEquals(0, integer.get());
    secondBuf.release();
    assertEquals(pageCount, integer.get());
}

Example 3 with Page

use of org.opensearch.nio.Page in project OpenSearch by opensearch-project.

the class TcpReadWriteHandler method consumeReads.

@Override
public int consumeReads(InboundChannelBuffer channelBuffer) throws IOException {
    Page[] pages = channelBuffer.sliceAndRetainPagesTo(channelBuffer.getIndex());
    BytesReference[] references = new BytesReference[pages.length];
    for (int i = 0; i < pages.length; ++i) {
        references[i] = BytesReference.fromByteBuffer(pages[i].byteBuffer());
    }
    Releasable releasable = () -> IOUtils.closeWhileHandlingException(pages);
    try (ReleasableBytesReference reference = new ReleasableBytesReference(CompositeBytesReference.of(references), releasable)) {
        pipeline.handleBytes(channel, reference);
        return reference.length();
    }
}