Java BlockChannelReader.closeAndDelete方法代码示例

import org.apache.flink.runtime.io.disk.iomanager.BlockChannelReader; //导入方法依赖的package包/类
@Test
public void channelReadWriteOneSegment()
{
	final int NUM_IOS = 1111;
	
	try {
		final Channel.ID channelID = this.ioManager.createChannel();
		final BlockChannelWriter writer = this.ioManager.createBlockChannelWriter(channelID);
		
		MemorySegment memSeg = this.memoryManager.allocatePages(new DummyInvokable(), 1).get(0);
		
		for (int i = 0; i < NUM_IOS; i++) {
			for (int pos = 0; pos < memSeg.size(); pos += 4) {
				memSeg.putInt(pos, i);
			}
			
			writer.writeBlock(memSeg);
			memSeg = writer.getNextReturnedSegment();
		}
		
		writer.close();
		
		final BlockChannelReader reader = this.ioManager.createBlockChannelReader(channelID);
		for (int i = 0; i < NUM_IOS; i++) {
			reader.readBlock(memSeg);
			memSeg = reader.getNextReturnedSegment();
			
			for (int pos = 0; pos < memSeg.size(); pos += 4) {
				if (memSeg.getInt(pos) != i) {
					Assert.fail("Read memory segment contains invalid data.");
				}
			}
		}
		
		reader.closeAndDelete();
		
		this.memoryManager.release(memSeg);
		
	} catch (Exception ex) {
		ex.printStackTrace();
		Assert.fail("TEst encountered an exception: " + ex.getMessage());
	}
}
 

import org.apache.flink.runtime.io.disk.iomanager.BlockChannelReader; //导入方法依赖的package包/类
private final void testChannelWithSegments(int numSegments) throws Exception
	{
		final List<MemorySegment> memory = this.memManager.allocatePages(memoryOwner, numSegments);
		final Channel.ID channel = this.ioManager.createChannel();
		
		BlockChannelWriter writer = null;
		BlockChannelReader reader = null;
		
		try {	
			writer = this.ioManager.createBlockChannelWriter(channel);
			final ChannelWriterOutputView out = new ChannelWriterOutputView(writer, memory, this.memManager.getPageSize());
			
			long writeStart = System.currentTimeMillis();
			
			int valsLeft = NUM_INTS_WRITTEN;
			while (valsLeft-- > 0) {
				out.writeInt(valsLeft);
			}
			
			out.close();
			final int numBlocks = out.getBlockCount();
			writer.close();
			writer = null;
			
			long writeElapsed = System.currentTimeMillis() - writeStart;
			
			// ----------------------------------------------------------------
			
			reader = ioManager.createBlockChannelReader(channel);
			final ChannelReaderInputView in = new ChannelReaderInputView(reader, memory, numBlocks, false);
			
			long readStart = System.currentTimeMillis();
			
			valsLeft = NUM_INTS_WRITTEN;
			while (valsLeft-- > 0) {
				in.readInt();
//				Assert.assertTrue(rec.getValue() == valsLeft);
			}
			
			in.close();
			reader.close();
			
			long readElapsed = System.currentTimeMillis() - readStart;
			
			reader.deleteChannel();
			reader = null;
			
			LOG.info("IOManager with " + numSegments + " mem segments: write " + writeElapsed + " msecs, read " + readElapsed + " msecs.");
			
			memManager.release(memory);
		}
		finally {
			if (reader != null) {
				reader.closeAndDelete();
			}
			if (writer != null) {
				writer.closeAndDelete();
			}
		}
	}