com.barchart.udt.nio.SelectionKeyUDT Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2009-2012 Barchart, Inc. 
 *
 * All rights reserved. Licensed under the OSI BSD License.
 *
 * http://www.opensource.org/licenses/bsd-license.php
 */
package com.barchart.udt.nio;

import java.net.InetSocketAddress;
import java.nio.channels.CancelledKeyException;
import java.nio.channels.SelectableChannel;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.spi.AbstractSelectionKey;

import com.barchart.udt.ExceptionUDT;
import com.barchart.udt.SocketUDT;

/**
 * NOTE: 1<->1 mapping between: SelectionKeyUDT -> ChannelUDT -> SocketUDT
 */
public class SelectionKeyUDT extends AbstractSelectionKey {

	final SelectorUDT selectorUDT;
	final ChannelUDT channelUDT;

	/**
	 * used by JNI
	 */
	final int socketID;
	final SocketUDT socketUDT;

	SelectionKeyUDT(SelectorUDT selector, ChannelUDT channel,
			Object attachment, int ops) {

		this.channelUDT = channel;
		this.selectorUDT = selector;
		super.attach(attachment);
		this.interestOps = ops;

		this.socketUDT = channel.getSocketUDT();
		this.socketID = socketUDT.getSocketId();

		if (channel.getChannelKind() == KindUDT.CONNECTOR) {
			// special relationship for connectors
			ChannelSocketUDT socketChannel = (ChannelSocketUDT) channel;
			socketChannel.setRegistredKey(this);
		}

	}

	volatile int interestOps;
	volatile int readyOps;

	//

	@Override
	public SelectableChannel channel() {
		return (SelectableChannel) channelUDT;
	}

	@Override
	public int interestOps() {
		if (!isValid()) {
			throw new CancelledKeyException();
		}
		return interestOps;
	}

	@Override
	public SelectionKey interestOps(int ops) {
		if (!isValid()) {
			throw new CancelledKeyException();
		}
		if ((ops & ~(channel().validOps())) != 0) {
			throw new IllegalArgumentException("invalid ops");
		}
		if (((ops & OP_CONNECT) != 0) && socketUDT.isConnected()) {
			throw new IllegalArgumentException("already connected");
		}
		interestOps = ops;
		return this;
	}

	@Override
	public int readyOps() {
		if (!isValid()) {
			throw new CancelledKeyException();
		}
		return readyOps;
	}

	@Override
	public Selector selector() {
		return selectorUDT;
	}

	//

	@Override
	public int hashCode() {
		return socketID;
	}

	@Override
	public boolean equals(Object otherKey) {
		if (otherKey instanceof SelectionKeyUDT) {
			SelectionKeyUDT other = (SelectionKeyUDT) otherKey;
			return other.socketID == this.socketID;
		}
		return false;
	}

	//

	@Override
	public String toString() {
		InetSocketAddress local;
		try {
			local = socketUDT.getLocalSocketAddress();
		} catch (ExceptionUDT e) {
			local = null;
		}
		InetSocketAddress remote;
		try {
			remote = socketUDT.getRemoteSocketAddress();
		} catch (ExceptionUDT e) {
			remote = null;
		}

		return " socketID=" + socketID
				+ //
				" type=" + channelUDT.getChannelKind()
				+ //
				" readyOps=" + toStringOps(readyOps) + "(" + readyOps + ")"
				+ //
				" ntrstOps=" + toStringOps(interestOps) + "(" + interestOps
				+ ")" + //
				" local=" + local + //
				" remote=" + remote + //
				"";
	}

	public static final String toStringOps(int keyOps) {
		char R = (OP_READ & keyOps) != 0 ? 'R' : '-';
		char W = (OP_WRITE & keyOps) != 0 ? 'W' : '-';
		char C = (OP_CONNECT & keyOps) != 0 ? 'C' : '-';
		char A = (OP_ACCEPT & keyOps) != 0 ? 'A' : '-';
		return String.format("%c%c%c%c", A, C, W, R);
	}

}