java Socket实现多人群聊与私聊功能
作者:OneDay-X
这篇文章主要为大家详细介绍了java Socket实现多人群聊与私聊功能,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
本文实例为大家分享了java Socket实现多人群聊与私聊的具体代码,供大家参考,具体内容如下
关于Socket套接字的一些基本知识与认识可以参见上一篇或自行查阅。
ServerSocket和Socket实现群聊与私聊涉及到多线程编程,实现过程的重点是利用Socket通信的原理,即不断的在服务端和客户端创建输入输出流来相互传递、交换数据等以达到通信的目的。具体实现如下:
服务端:
import java.io.*; import java.net.*; import java.util.HashMap; import java.util.Map; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.ThreadPoolExecutor; public class TCPServer { private ServerSocket serverSocket; /** * 创建线程池来管理客户端的连接线程 * 避免系统资源过度浪费 */ private ExecutorService exec; // 存放客户端之间私聊的信息 private Map<String,PrintWriter> storeInfo; public TCPServer() { try { serverSocket = new ServerSocket(6789); storeInfo = new HashMap<String, PrintWriter>(); exec = Executors.newCachedThreadPool(); } catch (Exception e) { e.printStackTrace(); } } // 将客户端的信息以Map形式存入集合中 private void putIn(String key,PrintWriter value) { synchronized(this) { storeInfo.put(key, value); } } // 将给定的输出流从共享集合中删除 private synchronized void remove(String key) { storeInfo.remove(key); System.out.println("当前在线人数为:"+ storeInfo.size()); } // 将给定的消息转发给所有客户端 private synchronized void sendToAll(String message) { for(PrintWriter out: storeInfo.values()) { out.println(message); } } // 将给定的消息转发给私聊的客户端 private synchronized void sendToSomeone(String name,String message) { PrintWriter pw = storeInfo.get(name); //将对应客户端的聊天信息取出作为私聊内容发送出去 if(pw != null) pw.println(message); } public void start() { try { while(true) { System.out.println("等待客户端连接... ... "); Socket socket = serverSocket.accept(); // 获取客户端的ip地址 InetAddress address = socket.getInetAddress(); System.out.println("客户端:“" + address.getHostAddress() + "”连接成功! "); /** * 启动一个线程,由线程来处理客户端的请求,这样可以再次监听 * 下一个客户端的连接 */ exec.execute(new ListenrClient(socket)); //通过线程池来分配线程 } } catch(Exception e) { e.printStackTrace(); } } /** * 该线程体用来处理给定的某一个客户端的消息,循环接收客户端发送 * 的每一个字符串,并输出到控制台 */ class ListenrClient implements Runnable { private Socket socket; private String name; public ListenrClient(Socket socket) { this.socket = socket; } // 创建内部类来获取昵称 private String getName() throws Exception { try { //服务端的输入流读取客户端发送来的昵称输出流 BufferedReader bReader = new BufferedReader( new InputStreamReader(socket.getInputStream(), "UTF-8")); //服务端将昵称验证结果通过自身的输出流发送给客户端 PrintWriter ipw = new PrintWriter( new OutputStreamWriter(socket.getOutputStream(), "UTF-8"),true); //读取客户端发来的昵称 while(true) { String nameString = bReader.readLine(); if ((nameString.trim().length() == 0) || storeInfo.containsKey(nameString)) { ipw.println("FAIL"); } else { ipw.println("OK"); return nameString; } } } catch(Exception e) { throw e; } } @Override public void run() { try { /* * 通过客户端的Socket获取客户端的输出流 * 用来将消息发送给客户端 */ PrintWriter pw = new PrintWriter( new OutputStreamWriter(socket.getOutputStream(), "UTF-8"), true); /* * 将客户昵称和其所说的内容存入共享集合HashMap中 */ name = getName(); putIn(name, pw); Thread.sleep(100); // 服务端通知所有客户端,某用户上线 sendToAll("[系统通知] “" + name + "”已上线"); /* * 通过客户端的Socket获取输入流 * 读取客户端发送来的信息 */ BufferedReader bReader = new BufferedReader( new InputStreamReader(socket.getInputStream(), "UTF-8")); String msgString = null; while((msgString = bReader.readLine()) != null) { // 检验是否为私聊(格式:@昵称:内容) if(msgString.startsWith("@")) { int index = msgString.indexOf(":"); if(index >= 0) { //获取昵称 String theName = msgString.substring(1, index); String info = msgString.substring(index+1, msgString.length()); info = name + ":"+ info; //将私聊信息发送出去 sendToSomeone(theName, info); continue; } } // 遍历所有输出流,将该客户端发送的信息转发给所有客户端 System.out.println(name+":"+ msgString); sendToAll(name+":"+ msgString); } } catch (Exception e) { // e.printStackTrace(); } finally { remove(name); // 通知所有客户端,某某客户已经下线 sendToAll("[系统通知] "+name + "已经下线了。"); if(socket!=null) { try { socket.close(); } catch(IOException e) { e.printStackTrace(); } } } } } public static void main(String[] args) { TCPServer server = new TCPServer(); server.start(); } }
客户端:
import java.io.*; import java.net.*; import java.util.Scanner; import java.util.concurrent.ExecutorService; import java.util.concurrent.Executors; import java.util.concurrent.ThreadPoolExecutor; public class TCPClient { static private Socket clientSocket; public TCPClient() {} public static void main(String[] args) throws Exception { Scanner scanner = new Scanner(System.in); String serverIP; System.out.println("请设置服务器IP:"); serverIP = scanner.next(); clientSocket = new Socket(serverIP, 6789); TCPClient client = new TCPClient(); client.start(); } public void start() { try { Scanner scanner = new Scanner(System.in); setName(scanner); // 接收服务器端发送过来的信息的线程启动 ExecutorService exec = Executors.newCachedThreadPool(); exec.execute(new ListenrServser()); // 建立输出流,给服务端发信息 PrintWriter pw = new PrintWriter( new OutputStreamWriter(clientSocket.getOutputStream(), "UTF-8"), true); while(true) { pw.println(scanner.nextLine()); } } catch(Exception e) { e.printStackTrace(); } finally { if (clientSocket !=null) { try { clientSocket.close(); } catch(IOException e) { e.printStackTrace(); } } } } private void setName(Scanner scan) throws Exception { String name; //创建输出流 PrintWriter pw = new PrintWriter( new OutputStreamWriter(clientSocket.getOutputStream(), "UTF-8"),true); //创建输入流 BufferedReader br = new BufferedReader( new InputStreamReader(clientSocket.getInputStream(),"UTF-8")); while(true) { System.out.println("请创建您的昵称:"); name = scan.nextLine(); if (name.trim().equals("")) { System.out.println("昵称不得为空"); } else { pw.println(name); String pass = br.readLine(); if (pass != null && (!pass.equals("OK"))) { System.out.println("昵称已经被占用,请重新输入:"); } else { System.out.println("昵称“"+name+"”已设置成功,可以开始聊天了"); break; } } } } // 循环读取服务端发送过来的信息并输出到客户端的控制台 class ListenrServser implements Runnable { @Override public void run() { try { BufferedReader br = new BufferedReader( new InputStreamReader(clientSocket.getInputStream(), "UTF-8")); String msgString; while((msgString = br.readLine())!= null) { System.out.println(msgString); } } catch(Exception e) { e.printStackTrace(); } } } }
运行结果:
开始自己的实现也不是很完整,后来也是借鉴别人比较好的思想后完善的,权当分享。
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。