Java Socket实现聊天室功能

更新时间：2022年09月02日 16:24:43 作者：dreaming9420

这篇文章主要为大家详细介绍了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实现并查集示例详解
这篇文章主要通过一个题目示例为大家详细介绍Java如何实现并查集，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-12-12
java并发编程之原子性、可见性、有序性
这篇文章主要给大家分享的是java并发编程的原子性、可见性和有序性，文章会具体举例说明，感兴趣的小伙伴可以参考一下文章的具体内容
2021-09-09
关于Java的Character类详解
这篇文章主要介绍了关于Java的Character类详解,Java中的Character类是一个包装类，用于封装一个基本数据类型char的值,它提供了一些静态方法来操作字符,需要的朋友可以参考下
2023-05-05
java服务器的简单实现过程记录
在线浏览网页离不开服务器,用户发出请求request,服务器做出响应response,提供给用户需要的页面,这篇文章主要给大家介绍了关于java服务器简单实现的相关资料,需要的朋友可以参考下
2021-11-11
深入Java分布式计算的使用分析
本篇文章对Java分布式计算的使用进行了详细的介绍。需要的朋友参考下
2013-05-05
Windows10安装IDEA 2020.1.2的方法步骤
这篇文章主要介绍了Windows10安装IDEA 2020.1.2的方法步骤，文中通过图文介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-08-08
Java实现的文本字符串操作工具类实例【数据替换，加密解密操作】
这篇文章主要介绍了Java实现的文本字符串操作工具类,可实现数据替换、加密解密等操作,涉及java字符串遍历、编码转换、替换等相关操作技巧,需要的朋友可以参考下
2017-10-10
java json不生成null或者空字符串属性(详解)
下面小编就为大家带来一篇java json不生成null或者空字符串属性(详解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2017-02-02
详解Mybatis 传递参数类型为List的取值问题
这篇文章主要介绍了详解Mybatis 传递参数类型为List的取值问题,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-10-10
springboot整合redis修改分区的操作流程
这篇文章主要介绍了springboot整合redis修改分区的操作流程，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教
2021-07-07

Java Socket实现聊天室功能

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具