TCP协议实现的聊天室

分析功能：

1、客户端登录

（1）服务器
1.使用循环一直监听客户端的连接
2.服务器的收和发线程写到一起–ServerThread类
3.在服务器中使用字符流进行收发，先读客户端登录名数据，再转发出去
（2）客户端
1.发送数据到服务器-----SendThread
2.接收服务器中转发的数据----RecvThread

2、群发

（1）服务器
1.服务器接收到数据，并群发给除自身以外的其他用户
2.将服务器中连接上来的socket所在的Thread用List存储起来
3.ServerThread中接收到数据后，循环遍历集合的数据，并群发出去
（2）客户端
1.例如， 张三发送一条信息
2.客户端收到服务器转发的的群发信息

3、私聊

（1）规则： @用户名#内容----》 @zs#哈哈哈
（2）客户端
1.例如发送内容： @zs#哈哈哈
2.客户端收到服务器中转发的私发的数据
（3）服务器
1.接收到数据后，拆分数据，获取到用户名和内容
2.并将拆分的内容发送给获取到的用户
3.循环遍历集合，判断线程名与指定用户名一致，则发一份出去（登录时要设置线程名）

4、客户端退出

（1）在服务器中捕获住了客户端的退出
（2）并将当前用户退出消息群发给其他客户端
（3）并在集合中将当前线程对象移除

实现代码：

public class TcpClient {//客户端
	public static void main(String[] args) throws UnknownHostException, IOException {
		Socket socket = new Socket("127.0.0.1", 10086);
		
		new WriteThread(socket).start();   //客户端发送数据
		
		new ReadThread(socket).start();    //客户端接收数据
	}
}
public class TcpServer {//服务端
	//实例化集合，用于存储线程对象
	public static List<ServerThread> list = new ArrayList<>();
	
public static void main(String[] args) throws IOException {
	ServerSocket server = new ServerSocket(10086);
	System.out.println("服务器监听10086端口...");
	while(true){
		Socket socket = server.accept();
		System.out.println("有一个客户端进行了连接");
		ServerThread st = new ServerThread(socket);
		st.start();
		
		list.add(st);   //存线程对象
	}
}
}
public class ServerThread extends Thread {//服务端收发信息线程
	private BufferedReader br;  //读
	private BufferedWriter bw;  //写
	
private Socket socket;
public ServerThread(Socket socket) throws IOException {
	br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
	bw = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
	this.socket = socket;
}

@Override
public void run() {
	welcome();   //欢迎信息
	
	try {
		while(true){
			
			String msg = br.readLine();  //读取信息
			
			if(msg.startsWith("@")&&msg.contains("#")){
				//私聊
				sendToOne(msg);
			}else{
				//群发
				sendToOther(this.getName()+"说："+msg);
			}
			
		}
		
	} catch (IOException e) {
		sendToOther(this.getName()+"退出了聊天室");
		//在集合中移除该线程对象
		TcpServer.list.remove(this);
	}finally{
		CloseUtils.closeAll(br,bw,socket);
	}
}

private void sendToOne(String msg) {
	//私聊   substring:提取子串   split：按字符串拆分
	String[] a = msg.substring(1).split("#");
	String name = a[0];
	String content = a[1];
	for(ServerThread st : TcpServer.list){
		if(st.getName().equals(name)){
			st.sendMsg(this.getName()+"悄悄对你说:"+content);
		}
	}
}

private void sendToOther(String msg) {
	for(ServerThread st : TcpServer.list){
		if(st != this){ //除自身以外，都转发一份出去
			st.sendMsg(msg);
		}
	}
}

private void sendMsg(String msg) {
	try {
		bw.write(msg);
		bw.newLine();
		bw.flush();
	} catch (IOException e) {
		e.printStackTrace();
	}  
	}

	private void welcome() {
		try {
			String msg = br.readLine();
			this.setName(msg);  //设置线程名
			msg = "【温馨提示】:欢迎"+msg+"进入聊天室";
			sendMsg(msg);
			
		} catch (IOException e) {
			e.printStackTrace();
		}
	}
}

public class ReadThread extends Thread {//客户端读线程
	private BufferedReader br;
	private Socket socket;
	public ReadThread(Socket socket) throws IOException {
		br = new BufferedReader(new InputStreamReader(socket.getInputStream()));
		this.socket = socket;
	}
@Override
public void run() {
	String msg;
	try {
		while(true){
			msg = br.readLine();
			System.out.println(msg);
		}
	} catch (IOException e) {
		e.printStackTrace();
	}finally{
		CloseUtils.closeAll(br,socket);
	}
		
	}
}

public class WriteThread extends Thread {//客户端写线程
	private Socket socket;
	private BufferedWriter bw;   //写
	private BufferedReader br;   //控制台录入
	public WriteThread(Socket socket) throws IOException {
		this.socket = socket;
		bw = new BufferedWriter(new OutputStreamWriter(socket.getOutputStream()));
		br = new BufferedReader(new InputStreamReader(System.in));
	}
@Override
public void run() {
	System.out.println("请输入用户名：");
	
	try {
		while(true){
			String msg = br.readLine();  //接收控制台数据
			bw.write(msg);
			bw.newLine();  //换行
			bw.flush();    //刷新缓冲区
		}
	} catch (IOException e) {
		e.printStackTrace();
	}finally {
		CloseUtils.closeAll(bw,br,socket);
	} 
		
	
	}
}

关闭资源的工具类：

public class CloseUtils {
	public static void closeAll(Closeable...closeables){
		for(Closeable c : closeables){
			if(c!=null){
				try {
					c.close();
				} catch (IOException e) {
					e.printStackTrace();
				}
			}
		}
	}
}