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());
}
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());
}
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;
}
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);
}
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);
}
Aggregations