Java Code Examples for org.apache.flink.runtime.io.network.buffer.NetworkBuffer#release()

private void readAndVerify(int bufferSize, InputChannelInfo channelInfo, byte[] data, ChannelStateReader reader) throws IOException {
	int dataSize = data.length;
	int iterations = dataSize / bufferSize + (-(dataSize % bufferSize) >>> 31);
	NetworkBuffer buffer = getBuffer(bufferSize);
	try {
		for (int i = 0; i < iterations; i++) {
			String hint = String.format("dataSize=%d, bufferSize=%d, iteration=%d/%d", dataSize, bufferSize, i + 1, iterations);
			boolean isLast = i == iterations - 1;
			assertEquals(hint, isLast ? NO_MORE_DATA : HAS_MORE_DATA, reader.readInputData(channelInfo, buffer));
			assertEquals(hint, isLast ? dataSize - bufferSize * i : bufferSize, buffer.readableBytes());
			assertArrayEquals(hint, Arrays.copyOfRange(data, i * bufferSize, Math.min(dataSize, (i + 1) * bufferSize)), toBytes(buffer));
			buffer.resetReaderIndex();
			buffer.resetWriterIndex();
		}
	} finally {
		buffer.release();
	}
}
 

private void write(byte[] data, ChannelStateSerializerImpl serializer, OutputStream baos) throws IOException {
	DataOutputStream out = new DataOutputStream(baos);
	serializer.writeHeader(out);
	NetworkBuffer buffer = new NetworkBuffer(MemorySegmentFactory.allocateUnpooledSegment(data.length), FreeingBufferRecycler.INSTANCE);
	try {
		buffer.writeBytes(data);
		serializer.writeData(out, buffer);
		out.flush();
	} finally {
		buffer.release();
	}
}
 

private void readAndCheck(byte[] data, ChannelStateSerializerImpl serializer, ByteArrayInputStream is) throws IOException {
	serializer.readHeader(is);
	int size = serializer.readLength(is);
	assertEquals(data.length, size);
	NetworkBuffer buffer = new NetworkBuffer(MemorySegmentFactory.allocateUnpooledSegment(data.length), FreeingBufferRecycler.INSTANCE);
	try {
		int read = serializer.readData(is, wrap(buffer), size);
		assertEquals(size, read);
		assertArrayEquals(data, readBytes(buffer));
	} finally {
		buffer.release();
	}
}
 

private byte[] writeSomeBytes(int bytesCount, DataOutputStream out, ChannelStateSerializer serializer) throws IOException {
	byte[] bytes = generateData(bytesCount);
	NetworkBuffer buf = getBuffer(bytesCount);
	try {
		buf.writeBytes(bytes);
		serializer.writeData(out, buf);
		return bytes;
	} finally {
		buf.release();
	}
}