EchoServer.java

package org.webpieces.nio.test;

import java.io.IOException;
import java.net.InetAddress;
import java.net.InetSocketAddress;
import java.nio.ByteBuffer;
import java.util.LinkedList;
import java.util.List;
import java.util.logging.Level;
import org.webpieces.util.logging.Logger;

import org.webpieces.nio.api.channels.Channel;
import org.webpieces.nio.api.channels.RegisterableChannel;
import org.webpieces.nio.api.channels.TCPChannel;
import org.webpieces.nio.api.channels.TCPServerChannel;
import org.webpieces.nio.api.deprecated.ChannelService;
import org.webpieces.nio.api.deprecated.Settings;
import org.webpieces.nio.api.handlers.ConnectionListener;
import org.webpieces.nio.api.handlers.DataListener;
import org.webpieces.nio.api.handlers.NullWriteCallback;
import org.webpieces.nio.api.testutil.MockNIOServer;


public class EchoServer implements DataListener, ConnectionListener {


	public static final String CONNECTED = "connected";
	public static final String CONN_FAILED = "failed";
	
	private static final Logger log = LoggerFactory.getLogger(MockNIOServer.class);
	private ChannelService chanMgr;
	private TCPServerChannel srvrChannel;
	private List<TCPChannel> sockets = new LinkedList<TCPChannel>();
	private Settings factoryHolder;
	
	public EchoServer(ChannelService svr, Settings h) {
		this.chanMgr = svr;
		this.factoryHolder = h;
	}
	
	public InetSocketAddress start() throws IOException, InterruptedException {
		int port = 0;
	
		chanMgr.start();
		
		InetAddress loopBack = InetAddress.getByName("127.0.0.1");
		InetSocketAddress svrAddr = new InetSocketAddress(loopBack, port);		
		srvrChannel = chanMgr.createTCPServerChannel("TCPServerChannel", factoryHolder);
		srvrChannel.setReuseAddress(true);
		srvrChannel.bind(svrAddr);	
		srvrChannel.registerServerSocketChannel(this);
		
		return srvrChannel.getLocalAddress();
	}
	
	public void stop() throws IOException, InterruptedException {		
		srvrChannel.closeServerChannel();
		for(int i = 0; i < sockets.size(); i++) {
			Channel channel = sockets.get(i);
			channel.oldClose();
		}
		chanMgr.stop();		
	}

	public void connected(Channel channel) throws IOException {
		log.trace(channel+"mockserver accepted connection");
		sockets.add((TCPChannel) channel);
		channel.registerForReads(this);
	}

	public void failed(RegisterableChannel channel, Throwable e) {
		log.error("exception", e);
	}
	
	public String toString() {
		return chanMgr.toString();
	}

	private int id = 0;
	public void incomingData(Channel channel, ByteBuffer chunk) throws IOException {		
		channel.oldWrite(chunk, NullWriteCallback.singleton());
	}

	public void farEndClosed(Channel channel) {
        channel.oldClose(null);
	}

	public void failure(Channel channel, ByteBuffer data, Exception e) {
		log.erroring(channel+"Data not received");
	}

}