timeout på readline hvis der ikke kommer noget input

package chat.simple.server;

import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.List;

public class Server {
    private List<ClientHandler> active;
    private ServerSocket ss;
    public Server(int port) throws IOException {
        active = new ArrayList<ClientHandler>();
        ss = new ServerSocket(12345);
    }
    public void run() throws IOException {
        while(true) {
            Socket s = ss.accept();
            ClientHandler cli = new ClientHandler(s, this);
            synchronized(active) {
                active.add(cli);
            }
            cli.start();
        }
    }
    public void sendAll(String msg) {
        synchronized(active) {
            for(ClientHandler ch : active) {
                ch.send(msg);
            }
        }
    }
    public void removeClient(ClientHandler ch) {
        synchronized(active) {
            active.remove(ch);
        }
    }
    public static void main(String[] args) throws IOException {
        Server srv = new Server(12345);
        srv.run();
    }
}

package chat.simple.server;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;

public class ClientHandler extends Thread {
    private PrintWriter pw;
    private BufferedReader br;
    private Server srv;
    private String usr;
    public ClientHandler(Socket s, Server srv) throws IOException {
        pw = new PrintWriter(s.getOutputStream());
        br = new BufferedReader(new InputStreamReader(s.getInputStream()));
        this.srv = srv;
        usr = "<no login>";
    }
    public void run() {
        String line;
        try {
            while((line = br.readLine()) != null) {
                if(line.startsWith("USR:")) {
                    usr = line.substring(4);
                } else if(line.startsWith("MSG:")) {
                    String msg = line.substring(4);
                    srv.sendAll(usr + ":" + msg);
                } else {
                    close();
                    return;
                }
            }
        } catch (IOException e) {
            close();
        }
    }
    public void send(String msg) {
        pw.println("MSG:" + msg);
        pw.flush();
    }
    private void close() {
        srv.removeClient(this);
        try {
            br.close();
            pw.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
        System.out.println(usr + ":logout");
    }
}

package chat.simple.client;

import java.awt.BorderLayout;
import java.awt.EventQueue;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.io.IOException;
import java.net.UnknownHostException;

import javax.swing.JButton;
import javax.swing.JFrame;
import javax.swing.JLabel;
import javax.swing.JPanel;
import javax.swing.JScrollPane;
import javax.swing.JTextArea;
import javax.swing.JTextField;
import javax.swing.SwingUtilities;

public class Client extends JFrame implements ServerHandler.Display {
    private JTextField un;
    private JButton login;
    private JTextArea msgs;
    private JTextField msg;
    private JButton send;
    private ServerHandler sh;
    public Client() throws UnknownHostException, IOException {
        setDefaultCloseOperation(JFrame.EXIT_ON_CLOSE);
        setTitle("Picture swap");
        getContentPane().setLayout(new BorderLayout());
        un = new JTextField();
        un.setColumns(40);
        login = new JButton("Login");
        JPanel n = new JPanel();
        n.setLayout(new BorderLayout());
        n.add(new JLabel("Username:"), BorderLayout.WEST);
        n.add(un, BorderLayout.CENTER);
        n.add(login, BorderLayout.EAST);
        getContentPane().add(n, BorderLayout.NORTH);
        msgs = new JTextArea();
        msgs.setRows(20);
        getContentPane().add(new JScrollPane(msgs), BorderLayout.CENTER);
        msg = new JTextField();
        msg.setColumns(40);
        send = new JButton("Send");
        JPanel s = new JPanel();
        s.setLayout(new BorderLayout());
        s.add(new JLabel("Message:"), BorderLayout.WEST);
        s.add(msg, BorderLayout.CENTER);
        s.add(send, BorderLayout.EAST);
        getContentPane().add(s, BorderLayout.SOUTH);
        sh = new ServerHandler("localhost", 12345, this);
        sh.start();
        login.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                sh.send("USR:" + un.getText());
            }
        });
        send.addActionListener(new ActionListener() {
            public void actionPerformed(ActionEvent e) {
                sh.send("MSG:" + msg.getText());
            }
        });
        pack();   
    }
    public void add(final String msg) {
        EventQueue.invokeLater(new Runnable() {
            public void run() {
                msgs.append(msg + "\r\n");
            }
        });
    }
    public static void main(String[] args) {
        SwingUtilities.invokeLater(new Runnable() {
            public void run() {
                try {
                    JFrame f = new Client();
                    f.setVisible(true);
                } catch (UnknownHostException e) {
                    e.printStackTrace();
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
        });
    }
}

package chat.simple.client;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.net.UnknownHostException;

public class ServerHandler extends Thread {
    public static interface Display {
        public void add(String msg);
    }
    private PrintWriter pw;
    private BufferedReader br;
    private Display disp;
    public ServerHandler(String host, int port, Display disp) throws UnknownHostException, IOException {
        Socket s = new Socket(host, port);
        pw = new PrintWriter(s.getOutputStream());
        br = new BufferedReader(new InputStreamReader(s.getInputStream()));
        this.disp = disp;
    }
    public void run() {
        String line;
        try {
            while((line = br.readLine()) != null) {
                if(line.startsWith("MSG:")) {
                    String msg = line.substring(4);
                    disp.add(msg);
                } else {
                    close();
                    return;
                }
            }
        } catch (IOException e) {
            close();
        }
    }
    public void send(String msg) {
        pw.println(msg);
        pw.flush();
    }
    private void close() {
        try {
            br.close();
            pw.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}