Java用TCP手写聊天室 可以 私聊版加群聊版

一：引言

想要私聊必须有规定的格式：@名字：要说的话

二：上码

1.服务端

package com.wyj.talkhome;
/**
* 实现一个用户可以接发多条消息
* 
* 
*/
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.concurrent.CopyOnWriteArrayList;

public class Demo5_服务端私聊版 {
	
	//在遍历一个list的时候，还向list中添加元素
	private static CopyOnWriteArrayList<pipe> all =new CopyOnWriteArrayList<pipe>();
	public static void main(String[] args) throws IOException {
		
		System.out.println("---server---");
		
		//1.使用ServerSocket创建一个服务端 并指定一个端口号
		ServerSocket server = new ServerSocket(7777); 
		//阻塞式等待链接
		boolean aa = true;
		
		while( aa ) {
			Socket client = server.accept();
			System.out.println("一个客户端链接成功");
			pipe p = new pipe(client);
			all.add(p);
			
			new Thread(p).start();
		}
		
		
	}
	
	static class pipe implements Runnable{

		private Socket client;
		private DataInputStream dis;
		private DataOutputStream dos;
		private boolean runing = true;
		private String name;
		
		public pipe( Socket client ) {
			this.client = client;
			
		    try {
				dis = new DataInputStream(client.getInputStream());
				dos = new DataOutputStream(client.getOutputStream());
				
				this.name = receive();
				
				this.send("欢迎你的到来");
				sendothers(this.name+"来到了聊天室",true);//系统消息
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
				release();
			}
		}
		//收到
		private String receive() {
			
			String msg = "";
			try {
				if (msg != null )
				 msg = dis.readUTF();
			} catch (IOException e) {
				// TODO Auto-generated catch block
			//	e.printStackTrace();
				release();
			}
			
			return msg;
			
		}
		
		//发送
		private void send ( String msg ) {
			try {
				dos.writeUTF(msg);
				dos.flush();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				//e.printStackTrace();
				this.runing = false;
				release();
			}
		}
		
		//群聊：获取自己的消息 发送信息给其他人
		//私聊：约定数据格式：@xxx:msg;
		private void sendothers(String msg,boolean isSys) {
			
			boolean isPrivate = msg.startsWith("@");
			
			if(isPrivate) {
				int idx = msg.indexOf(":");
				String tagetName = msg.substring(1,idx);
				msg = msg.substring(idx+1);
				
				for( pipe others :all) {
					
					if( others.name.equals(tagetName)) {
						others.send(this.name+"悄悄对你说："+msg);
					}
				}
				
				
			}else {
				for( pipe others:all) {
					if( others == this) {
						continue;
					}
					if(!isSys)//群消息
						others.send(this.name+"对所有人说："+msg);
					else//系统消息
						others.send(msg);
				}		
			}
		}
		
		
		//释放资源
		private void release() {
			try {
				if( dos != null )
				dos.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
	        try {
	        	if( dis != null )
				dis.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
	        
	        try {
	        	if( client != null )
				client.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}
		
		@Override
		public void run() {
			// TODO Auto-generated method stub

			//Io流 
			//输入流 接受客户端的请求
			
			while( runing ) {
				//收到请求
				String msg  = receive();
			
				//反馈结果
				if( !msg.equals(""))
				    sendothers(msg,false);
				else
					runing = false;
			}
			//释放资源
			release();
	    
		}
		
	}
}

2.客户端

package com.wyj.talkhome;
/**
 * 实现多用户可以接发多条消息
 * 
 */


import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;
import java.net.UnknownHostException;

public class Demo5_客户端私聊版 {

	public static void main(String[] args) throws UnknownHostException, IOException  {
		// TODO Auto-generated method stub
		
		System.out.println("---client---");
		
		BufferedReader brr = new BufferedReader( new InputStreamReader(System.in));
		System.out.print("请输入你在群聊中的名称：");
		String name = brr.readLine();
		
		
		//使用Socket创建客户端并指定IP和端口号
		Socket client = new Socket("localhost",7777);
		
		new Thread(new Demo4_客户端send(client,name)).start();
		new Thread(new Demo4_客户端receive(client)).start();
		
		
		
	}

}

3：客户端实现发送消息给服务端

package com.wyj.talkhome;

import java.io.BufferedReader;
import java.io.DataInputStream;
import java.io.DataOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Socket;

public class Demo4_客户端send implements Runnable{

	private Socket client;
	private DataOutputStream dos; 
	private DataInputStream dis;
	private BufferedReader br;
	private boolean runing = true;
	private String name;
	
	public Demo4_客户端send( Socket client ,String name) {
		
		this.client = client;
		this.name = name;
	    br = new BufferedReader(new InputStreamReader(System.in));
	    try {
			dos = new DataOutputStream(client.getOutputStream());
			send(this.name);//建立好通道  直接发送名字
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			this.release();
		}
	}
	
	//释放资源
	private void release() {
		try {
			if( dos != null )
			dos.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
        try {
        	if( dis != null )
			dis.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
        
        try {
        	if( client != null )
			client.close();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
		}
	}
	
	
	//从控制台获取信息
	private String fromconsole() {
		try {
			return  br.readLine();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			this.release();
		}
		return "";
	} 
	//向服务端发送信息
	private void send( String msg) {
		try {
			dos.writeUTF(msg);
			dos.flush();
		} catch (IOException e) {
			// TODO Auto-generated catch block
			//e.printStackTrace();
			this.runing = false;
			this.release();
		}
	}
	
		
	public void run() {
		// TODO Auto-generated method stub
		
		while(runing) {
			String msg = fromconsole();
			if( !msg.equals("")) {
				send(msg);
			}
		}	
	}
	
}

4.客户端接受消息从服务端

package com.wyj.talkhome;

import java.io.DataInputStream;
import java.io.IOException;
import java.net.Socket;

public class Demo4_客户端receive implements Runnable{

	private DataInputStream dis; 
	private Socket client;
	boolean runing = true;
	
	public Demo4_客户端receive(Socket client) {
		try {
			dis = new DataInputStream(client.getInputStream());
		} catch (IOException e) {
			// TODO Auto-generated catch block
			e.printStackTrace();
			this.release();
		}
	}
	private String receive() {
		
		String msg = "";
		try {
			if (msg != null )
			 msg = dis.readUTF();
		} catch (IOException e) {
			// TODO Auto-generated catch block
		//	e.printStackTrace();
			release();
		}
		
		return msg;
		
	}
	
	//释放资源
		private void release() {
	        try {
	        	if( dis != null )
				dis.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
	        
	        try {
	        	if( client != null )
				client.close();
			} catch (IOException e) {
				// TODO Auto-generated catch block
				e.printStackTrace();
			}
		}
	
	@Override
	public void run() {
		// TODO Auto-generated method stub
		while( runing ) {
		
			//收到反馈结果
			String msg1 = receive();
			if( !msg1.equals(""));
			System.out.println(msg1);
		}
	}
		
}

演示结果：