基于Java的Socket多客戶端Client-Server聊天程序的實(shí)現(xiàn)

更新時(shí)間：2020年03月12日 10:49:51 作者：進(jìn)階的JFarmer

這篇文章主要介紹了基于Java的Socket多客戶端Client-Server聊天程序的實(shí)現(xiàn)，文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧

任務(wù)要求

編寫一個(gè)簡(jiǎn)單的Socket多客戶端聊天程序：

客戶端程序，從控制臺(tái)輸入字符串，發(fā)送到服務(wù)器端，并將服務(wù)器返回的信息顯示出來(lái)
服務(wù)器端程序，從客戶機(jī)接收數(shù)據(jù)并打印，同時(shí)將從標(biāo)準(zhǔn)輸入獲取的信息發(fā)送給客戶機(jī)
滿足一個(gè)服務(wù)器可以服務(wù)多個(gè)客戶

低配版本鏈接

實(shí)現(xiàn)代碼

工具類

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);
	}

}

服務(wù)器端線程

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("歡迎來(lái)到聊天室！");
			login();
			System.out.println(currentUserNickName + "用戶登錄成功");
			write(currentUserNickName + "， 您已登錄。\n輸入【list users】可以查看當(dāng)前登錄用戶列表\n輸入【to all 消息內(nèi)容】可以群發(fā)消息\n輸入【to 某個(gè)用戶 消息內(nèi)容】可以給指定用戶發(fā)送消息\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("您輸入的命令不合法，請(qǐng)重新輸入！");
				}
				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("請(qǐng)輸入你的昵稱：");
		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("您輸入的昵稱已存在，請(qǐng)重新輸入：");
				}
			}
		}
	}

	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("您要單獨(dú)聊天的用戶【" + receiver + "】不存在或者已經(jīng)下線");
		}
	}

	private String formatMessage(String receiver, String originalMessage) {
		StringBuilder messageBuilder = new StringBuilder();
		messageBuilder.append(currentUserNickName).append(" 對(duì) ").append(receiver).append(" 說(shuō)：\n")
				.append(originalMessage).append("\n發(fā)送時(shí)間：")
				.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("當(dāng)前在線的用戶有：\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("與服務(wù)器的連接已中斷！");
					break;
				}
				if (!timeToStop) {
					e.printStackTrace();					
				}
			}
		}
	}
	
	public void stop() {
		timeToStop = true;
	}

}

服務(wù)器端程序

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("聊天室服務(wù)器端已啟動(dòng)，正在監(jiān)聽" + PORT + "端口");
			while (true) {
				try {
					Socket socket = ss.accept();
					System.out.println("有新用戶連接到服務(wù)器端，信息為：" + 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();
	}

}

運(yùn)行說(shuō)明

啟動(dòng)服務(wù)器：

啟動(dòng)第一個(gè)客戶端，輸入客戶昵稱：

服務(wù)器監(jiān)聽到了這個(gè)事件：

獲取所有用戶列表，發(fā)送給所有用戶“hhh”的信息：

服務(wù)器端接收到了這個(gè)事件：

新的客戶端登錄，注冊(cè)用戶昵稱：

服務(wù)器接收到這個(gè)事件：

用戶1向用戶2發(fā)送私聊消息：

用戶2收到用戶1的消息：

客戶2向所有用戶發(fā)送消息：

客戶1收到客戶2的群發(fā)消息：

服務(wù)器監(jiān)聽到了這些事件：

客戶2退出：

客戶1顯示的在線列表只有1人了：

客戶1也退出：

客戶端用戶退出的時(shí)候，該線程終止。

沒客戶端用戶，服務(wù)器也正常跑自己的事。

到此這篇關(guān)于基于Java的Socket多客戶端Client-Server聊天程序的實(shí)現(xiàn)的文章就介紹到這了,更多相關(guān)Java Socket多客戶端Client-Server聊天內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: