MPSTEndpoint.java

package org.scribble.net.session;

import java.io.IOException;
import java.net.UnknownHostException;
import java.util.concurrent.Callable;

import org.scribble.main.ScribbleRuntimeException;
import org.scribble.net.ScribMessageFormatter;
import org.scribble.net.scribsock.ScribServerSocket;
import org.scribble.sesstype.name.Role;

// FIXME: factor out between role-endpoint based socket and channel-endpoint sockets
//.. initiator and joiner endpoints
public class MPSTEndpoint<S extends Session, R extends Role> extends SessionEndpoint<S, R>
{
	public MPSTEndpoint(S sess, R self, ScribMessageFormatter smf) throws IOException, ScribbleRuntimeException
	{
		super(sess, self, smf);
	}

	public void connect(Role role, Callable<? extends BinaryChannelEndpoint> cons, String host, int port) throws ScribbleRuntimeException, UnknownHostException, IOException
	{
		MPSTEndpoint.connect(this, role, cons, host, port);
	}

	// HACK FIXME: refactor (location, protected, etc)
	public static void connect(SessionEndpoint<?, ?> se, Role role, Callable<? extends BinaryChannelEndpoint> cons, String host, int port) throws ScribbleRuntimeException, UnknownHostException, IOException
	{
		// Can connect unlimited, as long as not already used via init
		//if (this.init)
		if (se.init)
		{
			//throw new ScribbleRuntimeException("Socket already initialised: " + this.getClass());
			throw new ScribbleRuntimeException("Socket already initialised: " + se.getClass());
		}
		//if (this.chans.containsKey(role))
		if (se.chans.containsKey(role))
		{
			throw new ScribbleRuntimeException("Already connected to: " + role);
		}
		try
		{
			BinaryChannelEndpoint c = cons.call();
			/*c.initClient(this, host, port);
			register(role, c);*/
			c.initClient(se, host, port);
			se.register(role, c);
		}
		catch (Exception e)
		{
			if (e instanceof IOException)
			{
				throw (IOException) e;
			}
			throw new IOException(e);
		}
	}

	public void accept(ScribServerSocket ss, Role role) throws IOException, ScribbleRuntimeException
	{
		accept(this, ss, role);
	}

	//public static void accept(SessionEndpoint<?, ?> se, ScribServerSocket ss, Role role) throws IOException, ScribbleRuntimeException
	public static void accept(SessionEndpoint<?, ?> se, ScribServerSocket ss, Role role) throws IOException, ScribbleRuntimeException
	{
		//if (this.init)
		if (se.init)
		{
			//throw new ScribbleRuntimeException("Socket already initialised: " + this.getClass());
			throw new ScribbleRuntimeException("Socket already initialised: " + se.getClass());
		}
		//if (this.chans.containsKey(role))
		if (se.chans.containsKey(role))
		{
			throw new ScribbleRuntimeException("Already connected to: " + role);
		}
		//register(role, ss.accept(this));  // FIXME: serv map in SessionEndpoint not currently used
		se.register(role, ss.accept(se));  // FIXME: serv map in SessionEndpoint not currently used
	}
	
	public void disconnect(Role role) throws IOException, ScribbleRuntimeException
	{
		MPSTEndpoint.disconnect(this, role);
	}

	public static void disconnect(SessionEndpoint<?, ?> se, Role role) throws IOException, ScribbleRuntimeException
	{
		//if (!this.chans.containsKey(role))
		if (!se.chans.containsKey(role))
		{
			throw new ScribbleRuntimeException("Not connected to: " + role);
		}
		//deregister(role);
		se.deregister(role);
	}
}