Imanol-Mikel Barba Sabariego / PAD · Files

package pad.prac2;

import java.io.IOException;
import java.util.Collection;
import java.util.Iterator;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.locks.Condition;
import java.util.concurrent.locks.Lock;
import java.util.concurrent.locks.ReentrantLock;

public class Server
{
	private MyServerSocket ss;
	private int roomSize;
	private boolean kill;
	private Connection[] workerPool;
	private int freeWorkers;
	private Lock lock;
	private Condition availableWorkers;
	private ConcurrentHashMap<Connection,String> activeConnections;
	
	public Server(String ip, int port, int rS)
	{
		try
		{
			ss = new MyServerSocket();
			lock = new ReentrantLock();
			activeConnections = new ConcurrentHashMap<Connection,String>();
			roomSize = rS;
			freeWorkers = roomSize;
			kill = false;
			availableWorkers = lock.newCondition();
			workerPool = new Connection[roomSize];
			for(int i = 0; i < roomSize; i++)
			{
				workerPool[i] = new Connection(this);
				workerPool[i].start();
			}
			ss.bind(ip, port);
		}
		catch(IOException ioExc)
		{
			System.out.println("TCP: Error initializating server socket");
		}
	}
	
	public void startServer()
	{
		while(!kill)
		{
			try
			{
				MySocket incoming = ss.accept();
				startWorker(incoming);
			}
			catch(IOException ioExc)
			{
				if(!kill)
				{
					System.out.println("TCP: Error accepting connection");
					break;
				}
			}
		}
		
	}
	
	public void startWorker(MySocket s)
	{
		lock.lock();
		while(freeWorkers == 0)
		{
			try
			{
				s.sendMsg("CHATFULL");
				s.close();
				lock.unlock();
				return;
			}
			catch(IOException ioExc)
			{
				System.out.println("TCP: Error while sending message to client");
			}
		}
		workerPool[roomSize - freeWorkers].setSock(s);
		workerPool[roomSize - freeWorkers].awake();
		workerPool[roomSize - freeWorkers] = null;
		freeWorkers--;
		lock.unlock();
	}
	
	public void finishWorker(Connection c)
	{
		lock.lock();
		removeFromChatroom(c);
		freeWorkers++;
		workerPool[roomSize - freeWorkers] = c;
		availableWorkers.signal();
		lock.unlock();
	}
	
	public void sendTo(Connection orig, String nick, String msg) throws IOException, ChatException
	{
		lock.lock();
		System.out.println("FROM " + getNickname(orig) + " TO " + nick + ": " + msg);
		msg = getNickname(orig) + ": " + msg;
		Set<Connection> conns = activeConnections.keySet();
		Iterator<Connection> it = conns.iterator();
		Connection dest;
		while(it.hasNext())
		{
			dest = it.next();
			if(getNickname(dest).equals(nick))
			{
				dest.sendMessage(msg);
				lock.unlock();
				return;
			}
		}
		lock.unlock();
		throw new ChatException("No such nickname");
	}
	
	public void sendToChat(Connection origin, String message) throws IOException
	{
		lock.lock();
		String nickname = activeConnections.get(origin);
		Set<Connection> connections = activeConnections.keySet();
		Iterator<Connection> it = connections.iterator();
		Connection conn;
		while(it.hasNext())
		{
			conn = it.next();
			if(conn != origin)
			{
				conn.sendMessage("0"+nickname + ": " + message);
			}
		}
		lock.unlock();
	}
	
	public String getNickname(Connection c)
	{
		return activeConnections.get(c);
	}
	
	public void notifyAllConnections()
	{
		lock.lock();
		Set<Connection> connections = activeConnections.keySet();
		Iterator<Connection> it = connections.iterator();
		Connection conn;
		while(it.hasNext())
		{
			conn = it.next();
			try
			{
				conn.sendMessage("USERLIST:"+nickList());
			}
			catch(IOException ioExc)
			{
				System.out.println("TCP: Error writing to socket");
			}
		}
		lock.unlock();
	}
	
	public void addToChatroom(Connection c, String nickName)
	{
		activeConnections.put(c, nickName);
		System.out.println(nickName + " has entered the room");
		notifyAllConnections();
	}
	
	private void removeFromChatroom(Connection c)
	{
		activeConnections.remove(c);
		notifyAllConnections();
	}
	
	public boolean isOnline(String nick)
	{
		return activeConnections.contains(nick);
	}
	
	public String listOnline()
	{
		lock.lock();
		String ret = new Integer(activeConnections.size()).toString();
		ret += " people currently online:\n";
		ret += nickList();
		lock.unlock();
		return ret;
	}
	
	public String nickList()
	{
		String ret = "";
		Collection<String> nickNames = activeConnections.values();
		Iterator<String> it = nickNames.iterator();
		while(it.hasNext())
		{
			ret += it.next();
			ret += "\n";
		}
		return ret;
	}
	
	public void killServer()
	{
		kill = true;
		ss.close();
		finishConnections();
	}
	
	public void finishConnections()
	{
		Set<Connection> conns = activeConnections.keySet();
		Iterator<Connection> it = conns.iterator();
		while(it.hasNext())
		{
			it.next().finish();
		}
	}
}