SocketChannelController.java

package com.google.code.hs4j.network.nio.impl;

import java.io.IOException;
import java.nio.channels.SelectionKey;
import java.nio.channels.SocketChannel;
import java.util.Queue;

import com.google.code.hs4j.network.config.Configuration;
import com.google.code.hs4j.network.core.CodecFactory;
import com.google.code.hs4j.network.core.EventType;
import com.google.code.hs4j.network.core.Handler;
import com.google.code.hs4j.network.core.Session;
import com.google.code.hs4j.network.core.WriteMessage;
import com.google.code.hs4j.network.core.impl.StandardSocketOption;
import com.google.code.hs4j.network.nio.NioSession;
import com.google.code.hs4j.network.nio.NioSessionConfig;

/**
 * Nio tcp socket controller
 * 
 * @author dennis
 * 
 */
public abstract class SocketChannelController extends NioController {

	protected boolean soLingerOn = false;

	public void setSoLinger(boolean on, int value) {
		this.soLingerOn = on;
		this.socketOptions.put(StandardSocketOption.SO_LINGER, value);
	}

	public SocketChannelController() {
		super();
	}

	public SocketChannelController(Configuration configuration) {
		super(configuration, null, null);

	}

	public SocketChannelController(Configuration configuration,
			CodecFactory codecFactory) {
		super(configuration, null, codecFactory);
	}

	public SocketChannelController(Configuration configuration,
			Handler handler, CodecFactory codecFactory) {
		super(configuration, handler, codecFactory);
	}

	@Override
	protected final void dispatchReadEvent(SelectionKey key) {
		Session session = (Session) key.attachment();
		if (session != null) {
			((NioSession) session).onEvent(EventType.READABLE, key.selector());
		} else {
			log
					.warn("Could not find session for readable event,maybe it is closed");
		}
	}

	@Override
	protected final void dispatchWriteEvent(SelectionKey key) {
		Session session = (Session) key.attachment();
		if (session != null) {
			((NioSession) session).onEvent(EventType.WRITEABLE, key.selector());
		} else {
			log
					.warn("Could not find session for writable event,maybe it is closed");
		}

	}

	protected NioSession buildSession(SocketChannel sc) {
		Queue<WriteMessage> queue = buildQueue();
		NioSessionConfig sessionConfig = buildSessionConfig(sc, queue);
		NioSession session = new NioTCPSession(sessionConfig,
				this.configuration.getSessionReadBufferSize());
		return session;
	}

	/**
	 * Confiure socket channel
	 * 
	 * @param sc
	 * @throws IOException
	 */
	protected final void configureSocketChannel(SocketChannel sc)
			throws IOException {
		sc.socket().setSoTimeout(this.soTimeout);
		sc.configureBlocking(false);
		if (this.socketOptions.get(StandardSocketOption.SO_REUSEADDR) != null) {
			sc.socket().setReuseAddress(
					StandardSocketOption.SO_REUSEADDR.type().cast(
							this.socketOptions
									.get(StandardSocketOption.SO_REUSEADDR)));
		}
		if (this.socketOptions.get(StandardSocketOption.SO_SNDBUF) != null) {
			sc.socket().setSendBufferSize(
					StandardSocketOption.SO_SNDBUF.type().cast(
							this.socketOptions
									.get(StandardSocketOption.SO_SNDBUF)));
		}
		if (this.socketOptions.get(StandardSocketOption.SO_KEEPALIVE) != null) {
			sc.socket().setKeepAlive(
					StandardSocketOption.SO_KEEPALIVE.type().cast(
							this.socketOptions
									.get(StandardSocketOption.SO_KEEPALIVE)));
		}
		if (this.socketOptions.get(StandardSocketOption.SO_LINGER) != null) {
			sc.socket().setSoLinger(
					this.soLingerOn,
					StandardSocketOption.SO_LINGER.type().cast(
							this.socketOptions
									.get(StandardSocketOption.SO_LINGER)));
		}
		if (this.socketOptions.get(StandardSocketOption.SO_RCVBUF) != null) {
			sc.socket().setReceiveBufferSize(
					StandardSocketOption.SO_RCVBUF.type().cast(
							this.socketOptions
									.get(StandardSocketOption.SO_RCVBUF)));

		}
		if (this.socketOptions.get(StandardSocketOption.TCP_NODELAY) != null) {
			sc.socket().setTcpNoDelay(
					StandardSocketOption.TCP_NODELAY.type().cast(
							this.socketOptions
									.get(StandardSocketOption.TCP_NODELAY)));
		}
	}

}