客户端：
/** * 将聊天室客户端内容独立完成一次. * 下面内容可以选做: * 修改代码，使聊天室可以实现用户自定义昵称， * 以及私聊功能。 * * 参考思路: * 客户端连接服务端后，要求用户输入一个昵称， * 然后将改昵称发送给服务端，服务端那边读取到 * 客户端发送的第一个字符串认为是昵称。 * * 私聊功能可以定义格式，例如: * @张三:你好 * 服务端在读取客户端发送过来的昵称时，需要进行 * 解析，分解出昵称与聊天内容，然后将该聊天内容 * 单独发送给指定昵称的用户。 * 服务端的输出流需要使用Map进行维护，而不能再 * 使用List，Map的key可以是该用户昵称，value * 为该用户输出流。 * @author Xiloer * */ public class Client { private Socket socket; public Client() throws IOException{ socket=new Socket("176.195.107.88",8088); System.out.println("已连接服务器！"); } private void inputNickName(Scanner scan) throws Exception { String nickName = null; //创建输出流 PrintWriter pw = new PrintWriter( new OutputStreamWriter(socket.getOutputStream(), "UTF-8"),true); //创建输入流 BufferedReader br = new BufferedReader( new InputStreamReader( socket.getInputStream(),"UTF-8")); while(true) { System.out.println("请创建您的昵称："); nickName = scan.nextLine(); if (nickName.trim().equals("")) { System.out.println("昵称不得为空"); } else { pw.println(nickName); String pass = br.readLine(); if(pass!=null&&!pass.equals("OK")) { System.out.println("昵称已经被占用，请更换！"); break; } else { System.out.println("你好！"+nickName+"可以开始聊天了"); break; } } } } public void start(){ try { System.out.println("请输入："); Scanner scanner=new Scanner(System.in); inputNickName(scanner); ServerHandler server=new ServerHandler(); Thread t=new Thread(server); t.start(); OutputStream os = socket.getOutputStream(); OutputStreamWriter osw = new OutputStreamWriter(os,"UTF-8"); PrintWriter pw = new PrintWriter(osw,true); while(true) { pw.println(scanner.nextLine()); } } catch (Exception e) { e.printStackTrace(); }finally{ if(socket !=null) { try { socket.close(); } catch(IOException e) { e.printStackTrace(); } } } } public static void main(String[] args) throws IOException { Client client=new Client(); client.start(); } private class ServerHandler implements Runnable{ public void run() { try { InputStream is = socket.getInputStream(); InputStreamReader isr = new InputStreamReader(is,"UTF-8"); BufferedReader br = new BufferedReader(isr); String msgString = null; while((msgString = br.readLine())!= null) { System.out.println(msgString); } } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } } } 服务端： /** * 将聊天室服务端内容独立完成一次 * 下面内容可以选做: * 配合客户端实现支持昵称与私聊功能 * @author Xiloer * */ public class Server { private ServerSocket server; // private List<PrintWriter> allOut; private Map<String,PrintWriter> allOut; public Server(){ try { server=new ServerSocket(8088); allOut = new HashMap<String, PrintWriter>(); } catch (IOException e) { // TODO Auto-generated catch block e.printStackTrace(); } } private void addOut(String key,PrintWriter value) { synchronized(this) { allOut.put(key, value); } } private synchronized void removeOut(String key) { allOut.remove(key); System.out.println("当前在线人数为："+ allOut.size()); } private synchronized void sendMsgToAll(String message) { for(PrintWriter out: allOut.values()) { out.println(message); System.out.println("当前在线人数为："+ allOut.size()); } } private synchronized void sendMsgToPrivate(String nickname,String message) { PrintWriter pw = allOut.get(nickname); //将对应客户端的聊天信息取出作为私聊内容发送出去 if(pw!=null) { pw.println(message); System.out.println("1111111111111111111111111111111111111"); System.out.println("当前在线私聊人数为："+ allOut.size()); } } public void start(){ try{ while(true){ System.out.println("等待客户端连接......."); Socket socket=server.accept(); System.out.println("客户端已连接！"); ClientHandler client=new ClientHandler(socket); Thread t=new Thread(client); t.start(); } }catch(Exception e){ e.printStackTrace(); } } public static void main(String[] args) { Server server=new Server(); server.start(); } private class ClientHandler implements Runnable{ private Socket socket; private String host; private String nickName; public ClientHandler(Socket socket){ this.socket=socket; InetAddress adress=socket.getInetAddress(); host=adress.getHostAddress(); } private String getNickName() throws Exception { try { //服务端的输入流读取客户端发送来的昵称输出流 InputStream in = socket.getInputStream(); InputStreamReader isr = new InputStreamReader(in,"UTF-8"); BufferedReader bReader = new BufferedReader(isr); //服务端将昵称验证结果通过自身的输出流发送给客户端 OutputStream out = socket.getOutputStream(); OutputStreamWriter iosw = new OutputStreamWriter(out,"UTF-8"); PrintWriter ipw = new PrintWriter(iosw,true); //读取客户端发来的昵称 String nameString = bReader.readLine(); while(true) { if(nameString.trim().length()==0) { ipw.println("FAIL"); } if(allOut.containsKey(nameString)) { ipw.println("FAIL"); } else { ipw.println("OK"); return nameString; } } } catch(Exception e) { throw e; } } public void run() { PrintWriter pw = null; try { /* * 通过客户端的Socket获取客户端的输出流 * 用来将消息发送给客户端 */ OutputStream os = socket.getOutputStream(); OutputStreamWriter osw = new OutputStreamWriter(os,"UTF-8"); pw = new PrintWriter(osw,true); /* * 将客户昵称和其所说的话作为元素存入共享集合HashMap中 */ nickName = getNickName(); addOut(nickName, pw); Thread.sleep(100); /* * 服务端通知所有客户端，某用户登录 */ sendMsgToAll("[系统通知]：欢迎**"+nickName+"**登陆聊天室!"); /* * 通过客户端的Socket获取输入流 * 读取客户端发送来的信息 */ InputStream is = socket.getInputStream(); InputStreamReader isr = new InputStreamReader(is,"UTF-8"); BufferedReader br = new BufferedReader(isr); String msgString = null; while((msgString = br.readLine())!=null) { //验证是否是私聊 if(msgString.startsWith("@")) { /* * 私聊格式：@昵称：内容 */ int index = msgString.indexOf(":"); if(index >=0) { //获取昵称 String name = msgString.substring(1,index); String info = msgString.substring(index+1,msgString.length()); info = nickName + "对你说："+ info; //将私聊信息发送出去 sendMsgToPrivate(name, info); //服务端不在广播私聊的信息 continue; } System.out.println("已输出"); } /* * 遍历所有输出流，将该客户端发送的信息转发给所有客户端 */ System.out.println(nickName+"说："+ msgString); sendMsgToAll(nickName+"说："+ msgString); } } catch (Exception e) { /* * 因为Win系统用户的客户端断开连接后，br.readLine()方法读取 * 不到信息就会抛出异常，而Linux系统会持续发送null； * 因此这里就不在将捕获的异常抛出了。 */ } finally { /* * 当执行到此处时，说明客户端已经与服务端断开连接 * 则将该客户端存在共享集合中的输出流删除 */ removeOut(nickName); /* * 通知所有客户端，某某客户已经下线 */ sendMsgToAll("[系统通知]："+nickName + "已经下线了。"); /* * 关闭socket，则通过Socket获取的输入输出流也一同关闭了 */ if(socket!=null) { try { socket.close(); } catch(IOException e) { e.printStackTrace(); } } } } } }