Search in sources :

Example 6 with DataByteBuffer

use of alluxio.network.protocol.databuffer.DataByteBuffer in project alluxio by Alluxio.

the class RPCBlockReadResponseTest method getPayloadDataBuffer.

/**
   * Tests the {@link RPCBlockReadResponse#getPayloadDataBuffer()} method.
   */
@Test
public void getPayloadDataBuffer() {
    int length = 10;
    DataByteBuffer payload = new DataByteBuffer(ByteBuffer.allocate(length), length);
    RPCBlockReadResponse resp = new RPCBlockReadResponse(BLOCK_ID, OFFSET, LENGTH, payload, STATUS);
    assertValid(resp);
    Assert.assertEquals(payload, resp.getPayloadDataBuffer());
}
Also used : DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) Test(org.junit.Test)

Example 7 with DataByteBuffer

use of alluxio.network.protocol.databuffer.DataByteBuffer in project alluxio by Alluxio.

the class RPCFileReadResponseTest method getPayloadDataBuffer.

/**
   * Tests the {@link RPCBlockReadResponse#getPayloadDataBuffer()} method.
   */
@Test
public void getPayloadDataBuffer() {
    int length = 10;
    DataByteBuffer payload = new DataByteBuffer(ByteBuffer.allocate(length), length);
    RPCFileReadResponse resp = new RPCFileReadResponse(TEMP_UFS_FILE_ID, OFFSET, LENGTH, payload, STATUS);
    assertValid(resp);
    Assert.assertEquals(payload, resp.getPayloadDataBuffer());
}
Also used : DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) Test(org.junit.Test)

Example 8 with DataByteBuffer

use of alluxio.network.protocol.databuffer.DataByteBuffer in project alluxio by Alluxio.

the class LocalFilePacketReader method readPacket.

@Override
public DataBuffer readPacket() throws IOException {
    if (mPos >= mEnd) {
        return null;
    }
    ByteBuffer buffer = mReader.read(mPos, Math.min(LOCAL_READ_PACKET_SIZE, mEnd - mPos));
    DataBuffer dataBuffer = new DataByteBuffer(buffer, buffer.remaining());
    mPos += dataBuffer.getLength();
    return dataBuffer;
}
Also used : DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) ByteBuffer(java.nio.ByteBuffer) DataBuffer(alluxio.network.protocol.databuffer.DataBuffer)

Example 9 with DataByteBuffer

use of alluxio.network.protocol.databuffer.DataByteBuffer in project alluxio by Alluxio.

the class RPCMessageIntegrationTest method RPCBlockReadResponse.

@Test
public void RPCBlockReadResponse() {
    ByteBuffer payload = BufferUtils.getIncreasingByteBuffer((int) OFFSET, (int) LENGTH);
    RPCBlockReadResponse msg = new RPCBlockReadResponse(BLOCK_ID, OFFSET, LENGTH, new DataByteBuffer(payload, LENGTH), RPCResponse.Status.SUCCESS);
    RPCBlockReadResponse decoded = (RPCBlockReadResponse) encodeThenDecode(msg);
    assertValid(msg, decoded);
}
Also used : DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) ByteBuffer(java.nio.ByteBuffer) DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) Test(org.junit.Test)

Example 10 with DataByteBuffer

use of alluxio.network.protocol.databuffer.DataByteBuffer in project alluxio by Alluxio.

the class NettyRemoteBlockReaderTest method createRPCBlockReadResponse.

private RPCBlockReadResponse createRPCBlockReadResponse(RPCResponse.Status status) {
    ByteBuffer byteBuffer = ByteBuffer.allocateDirect((int) LENGTH);
    byteBuffer.put("alluxio".getBytes());
    DataBuffer dataBuffer = new DataByteBuffer(byteBuffer, LENGTH);
    return new RPCBlockReadResponse(BLOCK_ID, OFFSET, LENGTH, dataBuffer, status);
}
Also used : DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) RPCBlockReadResponse(alluxio.network.protocol.RPCBlockReadResponse) ByteBuffer(java.nio.ByteBuffer) DataByteBuffer(alluxio.network.protocol.databuffer.DataByteBuffer) DataBuffer(alluxio.network.protocol.databuffer.DataBuffer)

Aggregations

DataByteBuffer (alluxio.network.protocol.databuffer.DataByteBuffer)12 ByteBuffer (java.nio.ByteBuffer)7 Test (org.junit.Test)6 DataBuffer (alluxio.network.protocol.databuffer.DataBuffer)5 RPCBlockReadResponse (alluxio.network.protocol.RPCBlockReadResponse)2 ChannelFuture (io.netty.channel.ChannelFuture)2 IOException (java.io.IOException)2 BlockDoesNotExistException (alluxio.exception.BlockDoesNotExistException)1 RPCFileReadResponse (alluxio.network.protocol.RPCFileReadResponse)1 BlockReader (alluxio.worker.block.io.BlockReader)1 InputStream (java.io.InputStream)1