这篇文章主要为大家详细介绍了Java Socket实现聊天室功能,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
本文实例为大家分享了Java Socket实现聊天室的具体代码,供大家参考,具体内容如下
1 创建登录判断类UserLogin
import java.util.HashSet;
import java.util.Set;
public class UserLogin {
public static boolean login(String username) {
Set<String> set = initUser();
// set中含有该username则登录成功
return set.contains(username);
}
public static Set<String> initUser() {
Set<String> set = new HashSet<>();
set.add("111");
set.add("222");
set.add("333");
set.add("444");
set.add("555");
return set;
}
}
2 创建登录服务器LoginServer
import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
public class LoginServer {
public static void main(String[] args) {
int port = 6666;
try (ServerSocket serverSocket = new ServerSocket(port)) {
System.out.println("登录服务器已经启动");
while (true) {
Socket socket = serverSocket.accept();
// 接收Client输出的信息(username)
InputStream inputStream = socket.getInputStream();
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
String username = bufferedReader.readLine();
// 调用登录方法,判断是否成功登录
OutputStream outputStream = socket.getOutputStream();
PrintStream printStream = new PrintStream(outputStream);
if (UserLogin.login(username)) {
System.out.println(username + "登录成功");
// 登录成功输出"true"到Client
printStream.println("true");
} else {
System.out.println(username + "登录失败");
// 登录失败则输出"false"到Client
printStream.println("false");
}
printStream.flush();
}
} catch (IOException e) {
e.printStackTrace();
}
}
}
3 创建聊天服务器ChatServer
import com.socket.socketChatroom.bio.thread.ChatThread;
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.HashMap;
import java.util.Map;
public class ChatServer {
// 保存当前聊天中的所有Map<username,Socket>
public static Map<String, Socket> socketMap = new HashMap<>();
public static void main(String[] args) {
int port = 9999;
try (ServerSocket serverSocket = new ServerSocket(port)) {
System.out.println("聊天服务器已经启动");
while (true) {
Socket socket = serverSocket.accept();
// 接收Client输出的username
InputStream inputStream = socket.getInputStream();
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
String username = bufferedReader.readLine();
System.out.println(username + "连接到聊天服务器");
// 连接聊天服务器成功则将它的<username,Socket>放入socketMap中
socketMap.put(username, socket);
System.out.println(socketMap);
// 因为可能有多个Client与ChatServer进行交互,所以一旦有Client连接成功就新创建一个线程与之交互
new ChatThread(username,socket).start();
}
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
4 创建客户端Client
import com.socket.socketChatroom.bio.thread.ReadThread;
import com.socket.socketChatroom.bio.thread.WriteThread;
import java.io.*;
import java.net.Socket;
import java.util.Scanner;
public class Client {
public static void main(String[] args) {
// 服务端地址,端口号
String ip = "127.0.0.1";// 本机IP地址
int port = 6666; // 必须和LoginServer的端口一样
try {
Socket client = new Socket(ip, port);
Scanner scanner = new Scanner(System.in);
System.out.println("请输入用户名");
String username = scanner.nextLine();
// 将用户名输出到LoginServer
OutputStream outputStream = client.getOutputStream();
PrintStream printStream = new PrintStream(outputStream);
printStream.println(username);
printStream.flush();
// 接收LoginServer输出的是否登录成功标志("true"或"false")
InputStream inputStream = client.getInputStream();
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
String state = bufferedReader.readLine();
// 如果为"true",则为登录成功,可以开始聊天
if (Boolean.parseBoolean(state)) {
chat(username);
}
} catch (IOException e) {
e.printStackTrace();
}
}
public static void chat(String username) {
try {
String ip = "127.0.0.1";// 本机IP地址
int port = 9999;// 必须和ChatServer的端口一样
// 连接聊天服务器ChatServer
Socket socket = new Socket(ip, port);
// 将username输出到ChatServer
OutputStream outputStream = socket.getOutputStream();
PrintStream printStream = new PrintStream(outputStream);
printStream.println(username);
printStream.flush();
// 因为聊天的收发消息不一定是与同一个人,所以创建两个线程分别用于收和发
new ReadThread(socket).start();
new WriteThread(socket).start();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
5 创建服务器用于处理聊天的线程类ChatThread
import com.socket.socketChatroom.bio.server.ChatServer;
import java.io.*;
import java.net.Socket;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
public class ChatThread extends Thread {
private final String username;
private final Socket socket;
public ChatThread(String username, Socket socket) {
this.username = username;
this.socket = socket;
}
@Override
public void run() {
while (true) {
try {
// 接收Client的WriteThread线程输出的消息
InputStream inputStream = socket.getInputStream();
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
String message = bufferedReader.readLine();
System.out.println(username + "说: " + message);
// 将当前所有在线的Client的username存入usernameSet中
Map<String, Socket> socketMap = ChatServer.socketMap;
Set<String> usernameSet = new HashSet<>();
for (Map.Entry<String, Socket> map : socketMap.entrySet()) {
usernameSet.add(map.getKey());
}
// 以"-"作为标识符,判断是群聊还是单聊
String[] values = message.split("-");
if (values.length == 2 && usernameSet.contains(values[0])) { //单聊
// values[0]为要指定发送的Client的username
OutputStream outputStream = socketMap.get(values[0]).getOutputStream();
PrintStream printStream = new PrintStream(outputStream);
// values[1]为消息内容
printStream.println(username + "说: " + values[1]);
printStream.flush();
} else {//群聊
// 将消息发送到除自己之外的所有Client
for (String username : usernameSet) {
if (socket != socketMap.get(username)) {
OutputStream outputStream = socketMap.get(username).getOutputStream();
PrintStream printStream = new PrintStream(outputStream);
printStream.println(this.username + "说: " + message);
printStream.flush();
}
}
}
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}
6 创建客户端Client用于发送消息的线程类WriteThread
import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;
import java.net.Socket;
import java.util.Scanner;
public class WriteThread extends Thread {
private final Socket socket;
public WriteThread(Socket socket) {
this.socket = socket;
}
@Override
public void run() {
while (true) {
try {
// 获取输出流
OutputStream outputStream = socket.getOutputStream();
PrintStream printStream = new PrintStream(outputStream);
// 键盘输入要发送的消息,若为单聊,则以要发送的Client的username+"-",开头,比如"111-hello",表示向111这个Client发送"hello"
Scanner scanner = new Scanner(System.in);
System.out.println("请输入聊天内容: ");
String message = scanner.nextLine();
// 输出消息到ChatServer
printStream.println(message);
printStream.flush();
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}
7 创建客户端Client用于接收消息的线程类ReadThread
import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.net.Socket;
public class ReadThread extends Thread {
private final Socket socket;
public ReadThread(Socket socket) {
this.socket = socket;
}
@Override
public void run() {
while (true) {
try {
// 从ChatServer获取消息
InputStream inputStream = socket.getInputStream();
BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(inputStream));
String message = bufferedReader.readLine();
System.out.println(message);
} catch (IOException e) {
throw new RuntimeException(e);
}
}
}
}
项目结构如下图
8 测试
先运行LoginServer和ChatServer,再运行多个Client
设置IDEA允许运行多个实例
8.1 登录
8.2 群聊
8.3 单聊
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持编程学习网。
沃梦达教程
本文标题为:Java Socket实现聊天室功能
基础教程推荐
猜你喜欢
- ConditionalOnProperty配置swagger不生效问题及解决 2023-01-02
- java实现多人聊天系统 2023-05-19
- java基础知识之FileInputStream流的使用 2023-08-11
- Java并发编程进阶之线程控制篇 2023-03-07
- Java文件管理操作的知识点整理 2023-05-19
- JDK数组阻塞队列源码深入分析总结 2023-04-18
- springboot自定义starter方法及注解实例 2023-03-31
- Java实现线程插队的示例代码 2022-09-03
- Java实现查找文件和替换文件内容 2023-04-06
- Java数据结构之对象比较详解 2023-03-07