ChatServerMain.java

package chat;

import java.io.IOException;
import java.net.InetSocketAddress;
import java.net.ServerSocket;
import java.net.SocketException;
import java.net.UnknownHostException;
import java.nio.channels.SelectionKey;
import java.nio.channels.Selector;
import java.nio.channels.ServerSocketChannel;
import java.nio.channels.SocketChannel;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Set;

import csc4509.ReadMessageStatus;
import csc4509.Message;

public class ChatServerMain {
	
	boolean debug = false;

	public static void main(String[] argv) {
		ReadMessageStatus rms = null;
		if (argv.length != 1) {
			System.out.println("usage: java MainTcpNioServer <port>");
			return;
		}
		try {
			ServerSocketChannel ssc = ServerSocketChannel.open();
			/* Socket non bloquant */
			ssc.configureBlocking(false);
	
			ServerSocket ss = ssc.socket();
			InetSocketAddress rcvAddress = new InetSocketAddress( Integer.parseInt(argv[0]) );
			ss.bind( rcvAddress );
			
			/* Initalisation du Selector */
			Selector selector = Selector.open();
			ssc.register(selector, SelectionKey.OP_ACCEPT);
			
			HashMap<SelectionKey,Message> hm = new HashMap<SelectionKey,Message>();
			
			/* Socket en écoute */
			while (true) {
				/* Init selector */
				selector.select();
				Set<SelectionKey> readyKeys = selector.selectedKeys();
				Iterator<SelectionKey> iterator = readyKeys.iterator();
				while (iterator.hasNext()) {
					SelectionKey key = (SelectionKey) iterator.next();
					
			        if (key.isValid() && key.isAcceptable()) {
						/* Accepter une nouvelle connexion */
						SocketChannel sc = ssc.accept();
						sc.configureBlocking(false);
						
						
						/* Nouveau message */
						Message mess = new Message(1);
						mess.setSc(sc);
						
						/* Ajout de cette connexion au selector */
				        SelectionKey newKey = sc.register( selector, SelectionKey.OP_READ);
				        
						/* Sauvegarde du mesasge dans hashmap */
						hm.put(newKey, mess);
				        
				        iterator.remove();
				        
				        System.out.println("Nouveau client:" + mess.getHost());
					}
			        else if (key.isValid() && key.isReadable()) {
						SocketChannel sc = (SocketChannel)key.channel();
						
						Message mess = hm.get(key);
						
						
						rms = mess.readMessage(sc);
						if (rms == ReadMessageStatus.ReadDataCompleted) {
							/* Le message est reçu complètement */
							String s = null;
							try {
								s = (String) mess.getData();
							} catch (IOException e) {
								e.printStackTrace();
								return;
							}
							System.out.println("Message reçu (" + mess.getHost() +") :" + s);
							/* Il faut l'envoyer a tous les autres*/
							System.out.println("Message reçu !");
							for (Iterator<Message> i = hm.values().iterator() ; i.hasNext() ;){
								Message m = i.next();
								if (!m.equals(mess)) {
									/* On envoie pas au client son propre message */
									m.setMessage(1, s);
									m.write(m.getSc());
								}
							}
							//System.out.println(s);
						}
						else if (rms == ReadMessageStatus.ChannelClosed) {
							System.out.println("Client supprimé:" + mess.getHost());
							hm.remove(key);
						}
						
						iterator.remove();
					}
					else if (!key.isValid()) {
						hm.remove(key);
						iterator.remove();
					}
				}
			}
			
		} catch (SocketException se) {
			se.printStackTrace();
			return;
		} catch (UnknownHostException ue) {
			ue.printStackTrace();
			return;
		} catch (IOException ie) {
			ie.printStackTrace();
			return;
		}
	}
}