基於Java的Socket多客戶端Client-Server聊天程式的實現
阿新 • • 發佈:2020-03-13
任務要求
編寫一個簡單的Socket多客戶端聊天程式:
- 客戶端程式,從控制檯輸入字串,傳送到伺服器端,並將伺服器返回的資訊顯示出來
- 伺服器端程式,從客戶機接收資料並列印,同時將從標準輸入獲取的資訊傳送給客戶機
- 滿足一個伺服器可以服務多個客戶
低配版本連結
實現程式碼
工具類
import java.io.DataOutputStream; import java.io.IOException; import java.io.OutputStream; import java.net.Socket; public class SocketUtils { public static void writeToSocket(Socket socket,String message) throws IOException { writeToOutputStream(socket.getOutputStream(),message); } public static void writeToDataOutputStream(DataOutputStream dos,String message) throws IOException { dos.writeUTF(message); dos.flush(); } public static void writeToOutputStream(OutputStream os,String message) throws IOException { writeToDataOutputStream(new DataOutputStream(os),message); } }
伺服器端執行緒
import java.io.DataInputStream; import java.io.DataOutputStream; import java.io.IOException; import java.net.Socket; import java.text.SimpleDateFormat; import java.util.Date; import java.util.Map; public class ChatServerRunnable implements Runnable { private Socket socket; private DataOutputStream dos; private DataInputStream dis; private String currentUserNickName; public ChatServerRunnable(Socket socket) throws IOException { this.socket = socket; this.dos = new DataOutputStream(socket.getOutputStream()); this.dis = new DataInputStream(socket.getInputStream()); } @Override public void run() { try { write("歡迎來到聊天室!"); login(); System.out.println(currentUserNickName + "使用者登入成功"); write(currentUserNickName + ", 您已登入。\n輸入【list users】可以檢視當前登入使用者列表\n輸入【to all 訊息內容】可以群發訊息\n輸入【to 某個使用者 訊息內容】可以給指定使用者傳送訊息\n輸入【exit】可以退出聊天"); String input = dis.readUTF(); while (!ChatServer.EXIT.equals(input)) { System.out.println(currentUserNickName + "輸入了" + input); if (input.startsWith("to ")) { sendMessage(input); } else if ("list users".equals(input)) { showOnlineUsers(); } else { write("您輸入的命令不合法,請重新輸入!"); } input = dis.readUTF(); } } catch (IOException e) { e.printStackTrace(); } finally { ChatServer.nickNameSocketMap.remove(currentUserNickName); try { dis.close(); } catch (IOException e) { e.printStackTrace(); } try { dos.close(); } catch (IOException e) { e.printStackTrace(); } try { socket.close(); } catch (IOException e) { e.printStackTrace(); } } } private void login() throws IOException { write("請輸入你的暱稱:"); while (true) { String nickName = dis.readUTF(); System.out.println("使用者輸入了暱稱:" + nickName); synchronized (ChatServerRunnable.class) { if (!ChatServer.nickNameSocketMap.containsKey(nickName)) { currentUserNickName = nickName; ChatServer.nickNameSocketMap.put(nickName,socket); break; } else { write("您輸入的暱稱已存在,請重新輸入:"); } } } } private void sendMessage(String input) throws IOException { int receiverEndIndex = input.indexOf(" ",3); String receiver = input.substring(3,receiverEndIndex); String message = input.substring(receiverEndIndex + 1); if ("all".equals(receiver)) { broadcast(message); } else { sendIndividualMessage(receiver,message); } } private void sendIndividualMessage(String receiver,String orignalMessage) throws IOException { Socket receiverSocket = ChatServer.nickNameSocketMap.get(receiver); if (receiverSocket != null) { SocketUtils.writeToSocket(receiverSocket,formatMessage("你",orignalMessage)); } else { write("您要單獨聊天的使用者【" + receiver + "】不存在或者已經下線"); } } private String formatMessage(String receiver,String originalMessage) { StringBuilder messageBuilder = new StringBuilder(); messageBuilder.append(currentUserNickName).append(" 對 ").append(receiver).append(" 說:\n") .append(originalMessage).append("\n傳送時間:") .append(new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(new Date())); return messageBuilder.toString(); } private void broadcast(String orignalMessage) throws IOException { for (Map.Entry<String,Socket> entry : ChatServer.nickNameSocketMap.entrySet()) { if (!currentUserNickName.equals(entry.getKey())) { SocketUtils.writeToSocket(entry.getValue(),formatMessage("所有人",orignalMessage)); } } } private void showOnlineUsers() throws IOException { StringBuilder users = new StringBuilder(); users.append("當前線上的使用者有:\n"); for (String nickName : ChatServer.nickNameSocketMap.keySet()) { users.append("【").append(nickName).append("】\n"); } write(users.toString()); } private void write(String message) throws IOException { SocketUtils.writeToDataOutputStream(dos,message); } }
客戶端執行緒
import java.io.DataInputStream; import java.io.IOException; public class ClientMessageReceiver implements Runnable { private DataInputStream dis; private boolean timeToStop = false; public ClientMessageReceiver(DataInputStream dis) { this.dis = dis; } @Override public void run() { while (!timeToStop) { try { System.out.println(dis.readUTF()); } catch (IOException e) { if ("Connection reset".equals(e.getMessage())) { System.out.println("與伺服器的連線已中斷!"); break; } if (!timeToStop) { e.printStackTrace(); } } } } public void stop() { timeToStop = true; } }
伺服器端程式
import java.io.IOException; import java.net.ServerSocket; import java.net.Socket; import java.util.HashMap; import java.util.Map; public class ChatServer { public static final String EXIT = "exit"; public static final int PORT = 8888; static Map<String,Socket> nickNameSocketMap = new HashMap<>(); public static void main(String[] args) { try (ServerSocket ss = new ServerSocket(PORT)) { System.out.println("聊天室伺服器端已啟動,正在監聽" + PORT + "埠"); while (true) { try { Socket socket = ss.accept(); System.out.println("有新使用者連線到伺服器端,資訊為:" + socket); new Thread(new ChatServerRunnable(socket)).start(); } catch (Exception e) { e.printStackTrace(); } } } catch (IOException e) { e.printStackTrace(); } } }
客戶端程式
import java.io.DataInputStream; import java.io.DataOutputStream; import java.io.IOException; import java.net.Socket; import java.util.Scanner; public class ChatClient { private static Scanner scanner = new Scanner(System.in); public static void main(String[] args) { try (Socket socket = new Socket("127.0.0.1",8888); DataInputStream dis = new DataInputStream(socket.getInputStream()); DataOutputStream dos = new DataOutputStream(socket.getOutputStream())) { ClientMessageReceiver messageReceiver = new ClientMessageReceiver(dis); new Thread(messageReceiver).start(); String input = null; do { input = scanner.nextLine(); write(dos,input); } while (!ChatServer.EXIT.equals(input)); messageReceiver.stop(); } catch (IOException e) { e.printStackTrace(); } } private static void write(DataOutputStream dos,String message) throws IOException { dos.writeUTF(message); dos.flush(); } }
執行說明
啟動伺服器:
啟動第一個客戶端,輸入客戶暱稱:
伺服器監聽到了這個事件:
獲取所有使用者列表,傳送給所有使用者“hhh”的資訊:
伺服器端接收到了這個事件:
新的客戶端登入,註冊使用者暱稱:
伺服器接收到這個事件:
使用者1向用戶2傳送私聊訊息:
使用者2收到使用者1的訊息:
客戶2向所有使用者傳送訊息:
客戶1收到客戶2的群發訊息:
伺服器監聽到了這些事件:
客戶2退出:
客戶1顯示的線上列表只有1人了:
客戶1也退出:
客戶端使用者退出的時候,該執行緒終止。
沒客戶端使用者,伺服器也正常跑自己的事。
到此這篇關於基於Java的Socket多客戶端Client-Server聊天程式的實現的文章就介紹到這了,更多相關Java Socket多客戶端Client-Server聊天內容請搜尋我們以前的文章或繼續瀏覽下面的相關文章希望大家以後多多支援我們!