com.vmlens.trace.agent.bootstrap.event.StreamWrapperWithSlidingWindow Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of trace-agent-bootstrap Show documentation
There is a newer version: 1.1.5
Show newest version
package com.vmlens.trace.agent.bootstrap.event;



import java.io.*;

import java.nio.*;
import java.nio.channels.Channel;
import java.nio.channels.FileChannel;

import gnu.trove.list.TLinkable;
import gnu.trove.list.linked.TLinkedList;

public class StreamWrapperWithSlidingWindow extends AbstractStreamWrapper  {
	
	

	private DataOutputStream streamStatistic;
	private int lastWrittenSlidingWindow = -1; 
	
	private long start;

	
	
	protected final String eventDir;
	protected final String name;
	protected final int writerNumber;
	
	
	
    private ByteBuffer mappedByteBuffer;
    private FileChannel fileChannel;
  


	public StreamWrapperWithSlidingWindow(
			String eventDir, String name, int writerNumber,TLinkedList list) {
		super(list);
		
		this.eventDir = eventDir;
		this.name = name;
		this.writerNumber = writerNumber;
	}
	
	
	public StreamWrapperWithSlidingWindow(
			String eventDir, String name, int writerNumber) {
		super();
		
		this.eventDir = eventDir;
		this.name = name;
		this.writerNumber = writerNumber;
	}


//	protected  void openAdditionalStreams()  throws Exception
//	{
//		streamStatistic  =  new DataOutputStream(new  BufferedOutputStream(new FileOutputStream(eventDir + "/" +  name + "Statistic_" +writerNumber + ".vmlens" )));
//	}
//	
//	protected  void closeAdditionalStreams()  throws Exception
//	{
//		 streamStatistic.close();
//	}
	
	
	public void flush() throws IOException
	{
		if(streamStatistic != null )
		{
			
			 streamStatistic.writeInt(lastWrittenSlidingWindow);
			 streamStatistic.writeLong( start + mappedByteBuffer.position() );
			
			 streamStatistic.flush();
			 fileChannel.force(false);;
		}
	}
	
	
	
	public void close() throws Exception
	{
		
		if(streamStatistic != null )
		{
			 streamStatistic.writeInt(lastWrittenSlidingWindow);
			 streamStatistic.writeLong( start + mappedByteBuffer.position() );
			
			
			 streamStatistic.close();
			 fileChannel.close();
		}
		
	
	}


  
	
	

	public ByteBuffer getByteBuffer(int slidingWindowId,int arraySize , int blocksize)  throws Exception {
		
		
		if(lastWrittenSlidingWindow != slidingWindowId )
		{
		
			
			
			if(  mappedByteBuffer == null )
			{
				fileChannel = (new RandomAccessFile( new File(eventDir + "/" +  name + "_" + +writerNumber + ".vmlens" ) , "rw")).getChannel();
				mappedByteBuffer = fileChannel.map(FileChannel.MapMode.READ_WRITE, start, blocksize );
				streamStatistic  =  new DataOutputStream((new FileOutputStream(eventDir + "/" +  name + "Statistic_" +writerNumber + ".vmlens" )));
			}
			
			
			 
			 
			 streamStatistic.writeInt(lastWrittenSlidingWindow);
			 streamStatistic.writeLong( start + mappedByteBuffer.position() );
			 
			 
			 lastWrittenSlidingWindow = slidingWindowId;
			 
			 
			
		}
		
		
		if( (mappedByteBuffer.position()  + arraySize ) >=  blocksize)
		{
			start = start + mappedByteBuffer.position();
			mappedByteBuffer = fileChannel.map(FileChannel.MapMode.READ_WRITE, start,  blocksize);
		}
		
		
	
		
		return mappedByteBuffer;
		
	}
	
	
	
	
	
	

}