实现简单多人聊天并@功能

实现多人的聊天功能并能@的功能

    分析: 实现多人的群聊功能，服务器端必须能够记录所有人的socket套接字，本次代码操作中以map来存储用户的名字和套接字，然后对于每次接受的数据都群发一遍，对于@功能的实现，提炼出名字，找到对应的套接字然后回送数据。

服务端的实现:

package com.mixm0924;
 
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.ServerSocket;
import java.net.Socket;
import java.text.SimpleDateFormat;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
 
/**
 * @ClassName: Client01
 * @Description: 实现群聊并能解析@的功能
 * @author: Mixm
 * @date: 2015年9月28日 上午12:03:53
 */
public class Server01 {
 
	public static void main(String[] args) {
 
		// 用来存放socket和名字的对应关系 默认名字不能重复
		Map<String, Socket> map = null;
 
		ServerSocket ss = null;
 
		Socket s = null;
 
		try {
			map = new HashMap<String, Socket>();
			ss = new ServerSocket(10001);
 
			while (true) {
				s = ss.accept();
				new ServerThread(s, map).start();
			}
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				ss.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
 
	}
 
}
 
/**
 * @ClassName: ClientThread
 * @Description: 读取中默认第一次读取的是个文字，就要加入到map中，用名字作为键，对于有@的拆分，并且找到人物然后进行转发
 * @author: Mixm
 * @date: 2015年9月28日 上午12:25:10
 */
class ServerThread extends Thread {
 
	private Socket s = null;
	private Map<String, Socket> map = null;
	private String name = null;
 
	public ServerThread(Socket s, Map<String, Socket> map) {
		this.s = s;
		this.map = map;
	}
 
	@Override
	public void run() {
		BufferedReader br = null;
 
		try {
 
			// 判断是否为第一次，第一次的数据默认为名字,flag 为true就代表要接受的是名字
			boolean flag = true;
 
			// 关联流
			br = new BufferedReader(new InputStreamReader(s.getInputStream()));
			while (true) {
				if (flag) {
					name = readLine(br);
					if (name == null) {
						break;
					}
					System.out.println(name + "加入聊天室");
					map.put(name, s);
					flag = false;
				} else {
					// 得到信息
					String msg_receive = readLine(br);
					
					if (name == null) {
						break;
					}
 
					// 判断是否为@
					if (msg_receive.startsWith("@")) {
						int index = msg_receive.indexOf(":");
						String temp_name = msg_receive.substring(1, index);
						String temp_msg = msg_receive.substring(index + 1);
						sendMsg(temp_msg, map.get(temp_name));
					} else {
						for (Socket s : map.values()) {
							sendMsg(msg_receive, s);
						}
					}
				}
			}
 
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				br.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
	}
 
	//为什么这个流不能关   找时间会解答一下
	private void sendMsg(String msg, Socket s) {
		SimpleDateFormat sdf = new SimpleDateFormat("HH:mm:ss");
		BufferedWriter bw = null;
 
		try {
			bw = new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
			bw.write(name + "  " + sdf.format(new Date()) + "  " + msg);
			bw.newLine();
			bw.flush();
		} catch (Exception e) {
			e.printStackTrace();
		} 
//		finally {
//			try {
//				///bw.close();//这个流 不能关 我很忧伤  不知道原因
//			} catch (IOException e) {
//				e.printStackTrace();
//			}
//		}
 
	}
	
	//这个函数是解决用户强行退出 会抛异常的问题
	public String readLine(BufferedReader br) {
		String msg = null;
		try {
			msg = br.readLine();
		} catch (Exception e) {
			// 说明有客户端退出
			System.out.println(name + "退出聊天室");
			// 从map集合里面移除name对应的Socket
			map.remove(name);
		}
		return msg;
	}
}

客户端代码:

package com.mixm0924;
 
import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.net.Socket;
 
public class Client01 {
 
	public static void main(String[] args) {
 
		Socket s = null;
		// 用来读键盘
		BufferedReader br_input = null;
		BufferedWriter bw = null;
 
		try {
			s = new Socket("127.0.0.1", 10001);
			new ReadThread(s).start();
			br_input = new BufferedReader(new InputStreamReader(System.in));
 
			bw = new BufferedWriter(new OutputStreamWriter(s.getOutputStream()));
 
			String str = null;
			// 第一次提示输入名字
			System.out.println("请输入你的名字");
 
			while (true) {
				str = br_input.readLine();
				bw.write(str);
				bw.newLine();
				bw.flush();
			}
 
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				br_input.close();
				bw.close();
				s.close();
			} catch (IOException e) {
				e.printStackTrace();
			}
		}
 
	}
 
}
 
/**
 * @ClassName: ReadThread
 * @Description: 由于s.read是阻塞式方法 所以为了读的时候能写 就把读用线程在后台实现
 * @author: Mixm
 * @date: 2015年9月28日 上午12:54:23
 */
class ReadThread extends Thread {
	private Socket s = null;
 
	public ReadThread(Socket s) {
		this.s = s;
	}
 
	@Override
	public void run() {
 
		BufferedReader br = null;
		try {
			br = new BufferedReader(new InputStreamReader(s.getInputStream()));
 
			String msg = null;
			while ((msg = br.readLine()) != null) {
				System.out.println(msg);
			}
 
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			try {
				br.close();
			} catch (Exception e2) {
				e2.printStackTrace();
			}
		}
	}
 
}