RecordingDataListener.java

package org.webpieces.nio.api.handlers;

import java.io.FileNotFoundException;
import java.io.FileOutputStream;
import java.nio.ByteBuffer;
import java.util.Hashtable;
import java.util.Map;

import org.webpieces.nio.api.channels.Channel;
import org.webpieces.recording.api.Recorder;
import org.webpieces.recording.api.RecordingPlaybackFactory;

public class RecordingDataListener implements DataListener {

	private DataListener realListener;
	private Map<Channel, Recorder> channelToRecorder = new Hashtable<>();
	private String fileSuffix;
	
	public RecordingDataListener(String fileSuffix, DataListener realListener) {
		this.fileSuffix = fileSuffix;
		this.realListener = realListener;
	}

	public void incomingData(Channel channel, ByteBuffer b) {
		Recorder recorder = getOrCreate(channel);
		recorder.record(b);
		realListener.incomingData(channel, b);
	}

	private Recorder getOrCreate(Channel channel) {
		Recorder recorder = channelToRecorder.get(channel);
		if(recorder == null) {
			recorder = createRecorder(channel);
			channelToRecorder.put(channel, recorder);
		}
		return recorder;
	}

	private Recorder createRecorder(Channel channel) {
		try {
			FileOutputStream str = new FileOutputStream(fileSuffix+channel.getChannelId()+".recording");
			return RecordingPlaybackFactory.createRecorder(str, 1);
		} catch (FileNotFoundException e) {
			throw new RuntimeException(e);
		}
	}

	public void farEndClosed(Channel channel) {
		realListener.farEndClosed(channel);
	}

	public void failure(Channel channel, ByteBuffer data, Exception e) {
		realListener.failure(channel, data, e);
	}

	public void applyBackPressure(Channel channel) {
		realListener.applyBackPressure(channel);
	}

	public void releaseBackPressure(Channel channel) {
		realListener.releaseBackPressure(channel);
	}
	
}