import java.io.IOException; import java.io.ObjectInputStream; import java.io.ObjectOutputStream; import java.net.Socket; /** * class for an individual connection to a client. allows to send messages to * this client and handles incoming messages. */ public class Connection extends Thread { protected Socket socket; protected ObjectInputStream inputStream; protected ObjectOutputStream outputStream; protected Server server; protected boolean connectionOpen = true; protected boolean activated = false; public Connection(Socket s, Server server) { this.socket = s; try { inputStream = new ObjectInputStream((s.getInputStream())); outputStream = new ObjectOutputStream((s.getOutputStream())); } catch (IOException e) { e.printStackTrace(); } this.server = server; } /** * waits for incoming messages from the socket */ public void run() { String clientName = socket.getInetAddress().toString(); try { server.broadcast(new TextMessage(clientName + " has joined.")); while (connectionOpen) { try { Object msg = inputStream.readObject(); handleIncomingMessage(clientName, msg); } catch (ClassNotFoundException e) { e.printStackTrace(); } } } catch (IOException ex) { if (!ex.getMessage().equals("Connection reset")) ex.printStackTrace(); } finally { server.removeConnection(this); server.broadcast(new TextMessage(clientName + " has left.")); try { socket.close(); } catch (IOException ex) { ex.printStackTrace(); } } } /** * decides what to do with incoming messages * * @param name * name of the client * @param msg * received message */ protected void handleIncomingMessage(String name, Object msg) { if (msg instanceof TextMessage) server.broadcast(new TextMessage(name + " " + ((TextMessage) msg).getContent())); } public void send(TextMessage msg) { try { synchronized (outputStream) { outputStream.writeObject(msg); } outputStream.flush(); } catch (IOException ex) { connectionOpen = false; } } }