[开发测试]多用户即时通信系统

涉及到的内容

1、项目框架设计
2、java面向对象编程
3、网络编程
4、多线程
5、IO流
6、Mysql / 使用集合充当内存数据库

项目开发流程简介

1、需求分析

• 需求分析师：懂技术+行业
• 出一个需求分析报告（白皮书），该项目功能，客户具体要求

2、设计阶段

• 架构师 / 项目经理
• 设计工作（UML类图，流程图，模块设计，数据库，架构）
• 原型开发
• 组建团队

3、编码阶段

• 程序员 / 码农
• 完成架构师的模块功能
• 测试自己的模块

4、测试阶段

• 测试工程师
• 单元测试，测试用例，白盒测试，黑盒测试，集成测试

5、实施阶段

• 实施工程师（开发能力 / 环境配置部署能力）
• 项目正确的部署到客户的平台，并保证运行正常
• 身体好 ^_

6、维护阶段

• 发现bug解决 / 项目升级

需求分析

1、用户登录
2、拉取在线用户
3、无异常退出（客户端、服务端）
4、私聊
5、群发
6、发文件
7、服务器推送新闻

客户端界面设计

程序框架图

代码实现

服务端

Message.java

package com.leaveriver.common;

import java.io.Serializable;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 12:03
 * @Version 1.0
 * 说明：
 */
public class Message implements Serializable {
    private static final long serialVersionUID = 1L;   //为了保证序列化的兼容性。
    //消息发送者、和目标
    private String sender;
    private String getter;
    //消息内容
    private String content;
    //发送时间
    private String senderTime;
    //消息类型， 这个很重要
    private String mesType;

    //进行扩展，和文件相关的成员
    private byte[] fileBytes;
    private int fileLen = 0;
    private String srcFilePath;
    private String destFilePath;

    public byte[] getFileBytes() {
        return fileBytes;
    }

    public void setFileBytes(byte[] fileBytes) {
        this.fileBytes = fileBytes;
    }

    public int getFileLen() {
        return fileLen;
    }

    public void setFileLen(int fileLen) {
        this.fileLen = fileLen;
    }

    public String getSrcFilePath() {
        return srcFilePath;
    }

    public void setSrcFilePath(String srcFilePath) {
        this.srcFilePath = srcFilePath;
    }

    public String getDestFilePath() {
        return destFilePath;
    }

    public void setDestFilePath(String destFilePath) {
        this.destFilePath = destFilePath;
    }

    public String getSender() {
        return sender;
    }

    public void setSender(String sender) {
        this.sender = sender;
    }

    public String getGetter() {
        return getter;
    }

    public void setGetter(String getter) {
        this.getter = getter;
    }

    public String getContent() {
        return content;
    }

    public void setContent(String content) {
        this.content = content;
    }

    public String getSenderTime() {
        return senderTime;
    }

    public void setSenderTime(String senderTime) {
        this.senderTime = senderTime;
    }

    public String getMesType() {
        return mesType;
    }

    public void setMesType(String mesType) {
        this.mesType = mesType;
    }
}

MessageType.java

package com.leaveriver.common;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 12:09
 * @Version 1.0
 * 说明：
 */
public interface MessageType {

    String MESSAGE_LOGIN_SUCCEED = "1" ;//表示登录成功
    String MESSAGE_LOGIN_FAILED = "2" ;//表示登录失败
    String MESSAGE_COMM_MES = "3" ; //普通信息包
    String MESSAGE_GET_ONLINE_FRIEND = "4" ; //要求返回在线用户列表
    String MESSAGE_RET_ONLINE_FRIEND = "5" ; //返回在线用户列表
    String MESSAGE_CLIENT_EXIT = "6" ; //客户端请求退出
    String MESSAGE_TO_ALL_MES = "7"; //群发消息包

    String MESSAGE_FILE_OUTPUT = "8"; //传输文件

    String MESSAGE_SERVER_MES = "9";  //服务端推送消息

}

User.java

package com.leaveriver.common;

import java.io.Serializable;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 12:01
 * @Version 1.0
 * 说明：
 */
public class User implements Serializable {

    private static final long serialVersionUID = 1L;   //为了保证序列化的兼容性。
    private String userID ;     //  用户id/用户名
    private String password;    //  密码

    public User() {
    }

    public User(String userID, String password) {
        this.userID = userID;
        this.password = password;
    }

    public String getUserID() {
        return userID;
    }

    public void setUserID(String userID) {
        this.userID = userID;
    }

    public String getPassword() {
        return password;
    }

    public void setPassword(String password) {
        this.password = password;
    }
}

ManageServerConnectClientThread .java

package com.leaveriver.service;

import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 14:39
 * @Version 1.0
 * 说明：管理线程类
 */
public class ManageServerConnectClientThread {
    private static HashMap<String, ServerConnectClientThread> hashMap = new HashMap<>();

    public static boolean getOnline(String userId) {
        return hashMap.containsKey(userId);
    }

    public static HashMap<String, ServerConnectClientThread> getHashMap() {
        return hashMap;
    }

    public static void addServerConnectClientThread(
            String userId, ServerConnectClientThread serverConnectClientThread) {
        hashMap.put(userId, serverConnectClientThread);
    }
    
    public static ServerConnectClientThread getServerConnectClientThread(String userId){
        return hashMap.get(userId);
    }


    //拉取在线用户
    public static String getOnlineUser(){
        //集合遍历，遍历hashmap的key
        Iterator<String> iterator = hashMap.keySet().iterator();
        String onlineUserList = "";

        while (iterator.hasNext()) {
            onlineUserList += iterator.next().toString() + " " ;   //规定用空格拼接所有用户id

        }
        return onlineUserList;
    }

    //移除一个线程
    public static void removeServerConnectClientThread(String userId){
        hashMap.remove(userId);
    }
}

ManageUsers .java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.User;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Author Administrator
 * @Date 2021/11/4 0004 12:37
 * @Version 1.0
 * 说明：
 */
public class ManageUsers {

    private static ConcurrentHashMap<String, ArrayList<Message>> offLineDb = new ConcurrentHashMap<>();


    //创建一个集合，存放多个用户，如果是这些用户登录，就认为是合法的
    private static HashMap<String, User> validUsers = new HashMap<>();


    static {//在静态代码块，初始化validUsers
        validUsers.put("100", new User("100", "123456"));
        validUsers.put("200", new User("200", "123456"));
        validUsers.put("300", new User("300", "123456"));
        validUsers.put("至尊宝", new User("至尊宝", "123456"));
        validUsers.put("紫霞仙子", new User("紫霞仙子", "123456"));
    }

    public static User get(String userId){
        return validUsers.get(userId);
    }
    public static void addOffLineDb(String getter, ArrayList<Message> message) {
        offLineDb.put(getter, message);
    }

    public static ArrayList<Message> getOffLineDb(String getter){
        return offLineDb.get(getter);
    }
    public static boolean containsOffLineDb(String getter) {
        return offLineDb.containsKey(getter);
    }

    public static void removeOffLineDb(String getter) {
        offLineDb.remove(getter);
    }
}

PushMesServerService .java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;
import com.leaveriver.utils.Utility;

import java.io.IOException;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.ServerSocket;
import java.net.Socket;
import java.text.SimpleDateFormat;
import java.util.HashMap;
import java.util.Iterator;

/**
 * @Author Administrator
 * @Date 2021/11/3 0003 15:36
 * @Version 1.0
 * 说明：
 */
public class PushMesServerService extends Thread {

    @Override
    public void run() {

        while (true) {
            System.out.println("请输入你要推送的消息：");
            String newsMes = Utility.readString(100);
            //创建message对象
            Message message = new Message();
            message.setMesType(MessageType.MESSAGE_TO_ALL_MES);
            message.setSender("服务器");
            message.setContent(newsMes);
            message.setSenderTime(new SimpleDateFormat("yyyy-MM-dd HH:mm:ss").format(System.currentTimeMillis()));

            //创建完之后发送。要拿到目标用户的socket
            HashMap<String, ServerConnectClientThread> hashMap = ManageServerConnectClientThread.getHashMap();
            Iterator<String> iterator = hashMap.keySet().iterator();

            while (iterator.hasNext()) {
                String next = iterator.next();
                try {
                    OutputStream os = hashMap.get(next).getSocket().getOutputStream();
                    ObjectOutputStream oos = new ObjectOutputStream(os);
                    oos.writeObject(message);
                } catch (IOException e) {
                    e.printStackTrace();
                }
            }
            System.out.println("[服务器推送记录][" + message.getSenderTime() + "]" +
                    message.getSender() + "推送信息：" + newsMes);
        }
    }
}

ServerConnectClientThread.java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;

import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.io.OutputStream;
import java.net.Socket;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.Iterator;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 14:28
 * @Version 1.0
 * 说明：与客户端保持通信
 */
public class ServerConnectClientThread extends Thread {

    private Socket socket;
    private String userId;  //连接到服务器的用户id


    //
    public ServerConnectClientThread(Socket socket, String userId) {
        this.socket = socket;
        this.userId = userId;
    }


    public Socket getSocket() {
        return socket;
    }

    @Override
    public void run() {
        //这里线程处于运行状态，可以发送/接受消息
        while (true) {
            try {
                System.out.println("服务器和客户端[" + userId + "]保持通信，读取数据。。。。");
                ObjectInputStream ois = new ObjectInputStream(socket.getInputStream());
                Message message = (Message) ois.readObject();

                /*
                 *这块加入一个验证对方用户是否在线的检测
                 *1、看管理线程中，是否有对方用户的线程，没有说明他不在线
                 *  */

                if (!ManageServerConnectClientThread.getOnline(message.getGetter()) && (ManageUsers.get(message.getGetter()) != null)) {
                    //如果不在线,将信息存在 ConcurrentHashMap中
                    if (!ManageUsers.containsOffLineDb(message.getGetter())) {
                        ArrayList<Message> arrayList = new ArrayList<>();
                        arrayList.add(message);
                        ManageUsers.addOffLineDb(message.getGetter(), arrayList);
                    } else {
                        ArrayList<Message> offLineDb = ManageUsers.getOffLineDb(message.getGetter());
                        offLineDb.add(message);
                        ManageUsers.addOffLineDb(message.getGetter(), offLineDb);
                    }
                    //区分文件和普通消息，服务端控制台输出不同的信息
                    if (message.getMesType().equals(MessageType.MESSAGE_FILE_OUTPUT)) {
                        System.out.println("[离线文件记录][" + message.getSenderTime() + "]" +
                                message.getSender() + "发送文件给" + message.getGetter() + "\n" +
                                message.getSrcFilePath() + " 到 " + message.getDestFilePath());

                    } else {
                        System.out.println("[离线消息记录][" + message.getSenderTime() + "]" +
                                message.getSender() + " 对 " + message.getGetter() + "说： " + message.getContent());
                    }

                } else {


                    //后面会使用message
                    if (message.getMesType().equals(MessageType.MESSAGE_GET_ONLINE_FRIEND)) {
                        System.out.println(message.getSender() + "要获取在线用户");
                        String onlineUser = ManageServerConnectClientThread.getOnlineUser();
                        //返回给客户端
                        //构建一个message对象，
                        Message mes = new Message();
                        mes.setMesType(MessageType.MESSAGE_RET_ONLINE_FRIEND);
                        mes.setContent(onlineUser);
                        mes.setGetter(mes.getSender());
                        //写入
                        ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
                        oos.writeObject(mes);

                    } else if (message.getMesType().equals(MessageType.MESSAGE_CLIENT_EXIT)) {
                        System.out.println(message.getSender() + "退出");
                        //
                        ManageServerConnectClientThread.removeServerConnectClientThread(message.getSender());
                        socket.close();
                        break; //退出循环
                    } else if (message.getMesType().equals(MessageType.MESSAGE_COMM_MES)) {

                        //在服务端输出一句话。记录
                        System.out.println("[私聊记录][" + message.getSenderTime() + "]" +
                                message.getSender() + " 对 " + message.getGetter() + "说： " + message.getContent());
                        //转发给目标用户
                        //获取对应用户的socket的输出流，将信息转发出去
                        ServerConnectClientThread scct = ManageServerConnectClientThread.getServerConnectClientThread(message.getGetter());
                        ObjectOutputStream oos = new ObjectOutputStream(scct.getSocket().getOutputStream());
                        oos.writeObject(message);

                    } else if (message.getMesType().equals(MessageType.MESSAGE_TO_ALL_MES)) {


                        System.out.println("[群聊消息记录][" + message.getSenderTime() + "]" +
                                message.getSender() + " 对大家说： " + message.getContent());
                        HashMap<String, ServerConnectClientThread> hashMap = ManageServerConnectClientThread.getHashMap();
                        Iterator<String> iterator = hashMap.keySet().iterator();
                        while (iterator.hasNext()) {
                            String next = iterator.next();
                            if (!next.equals(message.getSender())) {
                                OutputStream os = hashMap.get(next).getSocket().getOutputStream();
                                ObjectOutputStream oos = new ObjectOutputStream(os);
                                oos.writeObject(message);
                            }
                        }
                    } else if (message.getMesType().equals(MessageType.MESSAGE_FILE_OUTPUT)) {


                            System.out.println("[文件传输记录][" + message.getSenderTime() + "]" +
                                    message.getSender() + "发送文件给" + message.getGetter() + "\n" +
                                    message.getSrcFilePath() + " 到 " + message.getDestFilePath());
                            ServerConnectClientThread scct = ManageServerConnectClientThread.getServerConnectClientThread(message.getGetter());
                            OutputStream os = scct.getSocket().getOutputStream();
                            ObjectOutputStream oos = new ObjectOutputStream(os);
                            oos.writeObject(message);

                    } else {
                        System.out.println("其他类型暂时不处理");
                    }
                }
            } catch (Exception e) {
                e.printStackTrace();
            }

        }
    }
}

ServerService.java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;
import com.leaveriver.common.User;

import java.io.*;
import java.net.ServerSocket;
import java.net.Socket;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.concurrent.ConcurrentHashMap;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 14:12
 * @Version 1.0
 * 说明：这是服务器，在监听9999，等待客户端的连接，并保持通信。
 */
public class ServerService {

    private ServerSocket serverSocket;//因为ServerSocket可能在其他地方要使用，所以也作为成员变量。


    //验证用户是否有效的方法
    private boolean checkUser(String userId, String password) {
        User user = ManageUsers.get(userId);
        if (user == null)
            return false;
        if (!user.getPassword().equals(password))
            return false;
        return true;
    }


    public ServerService() {
        //注意：端口可以写在配置文件中。
        try {
            System.out.println("服务端在9999端口监听。");

            new PushMesServerService().start();

            serverSocket = new ServerSocket(9999);
            while (true) {      //当和某个客户端连接后，会继续监听，因此while
                Socket socket = serverSocket.accept();   //如果没有客户端连接，就会阻塞
                //得到socket关联的对象输入流
                InputStream is = socket.getInputStream();
                ObjectInputStream ois = new ObjectInputStream(is);

                //得到socket关联的对象输出流，用于回复message给客户端
                ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());


                //读取从客户端发来的user对象。
                User user = (User) ois.readObject();

                //创建一个message对象，准备回复客户端
                Message message = new Message();
                if (checkUser(user.getUserID(), user.getPassword())) {
                    //账号密码正确,登录成功
                    message.setMesType(MessageType.MESSAGE_LOGIN_SUCCEED);
                    //回复客户端
                    oos.writeObject(message);
                    //创建一个线程与客户端保持通信。该线程需要持有socket对象。
                    //
                    ServerConnectClientThread scct = new ServerConnectClientThread(socket, user.getUserID());
                    scct.start();
                    ManageServerConnectClientThread.addServerConnectClientThread(user.getUserID(), scct);

                    //登录成功后，获取离线消息。
                    //在map中找 key为 user.getUserId()的,如果有，将value发送给该用户。
                    ArrayList<Message> offLineDb = ManageUsers.getOffLineDb(user.getUserID());
                    if (offLineDb != null)
                        for (int i = 0; i < offLineDb.size() ; i++) {
                            Message mes = offLineDb.get(i);
                            OutputStream outputStream = ManageServerConnectClientThread.getServerConnectClientThread(mes.getGetter()).getSocket().getOutputStream();
                            ObjectOutputStream objectOutputStream = new ObjectOutputStream(outputStream);
                            objectOutputStream.writeObject(mes);
                            objectOutputStream.flush();

                            //获取发送完之后，就移除，不然下次登录还会发送。（服务器没重启的话。）
                            ManageUsers.removeOffLineDb(mes.getGetter());
                        }

                } else {
                    //如果验证失败，返回消息类型为2
                    message.setMesType(MessageType.MESSAGE_LOGIN_FAILED);
                    oos.writeObject(message);

                    //关闭socket
                    socket.close();
                }

            }

        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            //如果服务器退出了while。说明服务器端不再监听，因此关闭serverSocket
            try {
                serverSocket.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

    }
}

Utility.java

package com.leaveriver.utils;


/**
 * 工具类的作用:
 * 处理各种情况的用户输入,并且能按照程序员对的需求,得到用户的控制台输入
 */

import java.util.Scanner;


public class Utility {
    //静态属性
    private static Scanner scanner = new Scanner(System.in);


    /**
     * 功能:读取键盘输入的一个菜单项,值:1-5范围
     */
    public static char readMenuSelection() {
        char c;
        for (; ; ) {
            String str = readKeyBoard(1, false);
            c = str.charAt(0);//将字符串转换为字符
            if (c != '1' && c != '2' && c != '3' && c != '4' && c != '5') {
                System.out.println("选择错误，请重新输入：");
            } else break;
        }
        return c;
    }

    /**
     * 功能:读取键盘输入的一个字符
     */
    public static char readChar() {
        String str = readKeyBoard(1, false);//就是一个字符
        return str.charAt(0);
    }

    /**
     * 功能;读取键盘输入的一个字符,如果直接回车,则返回指定的默认值
     * defaultValue 指定的默认值
     * return 默认值或者输入的字符
     */
    public static char readChar(char defaultValue) {
        String str = readKeyBoard(1, true);//要么是空字符,要么??
        return str.length() == 0 ? defaultValue : str.charAt(0);
    }


    /**
     * 读取键盘输入的整型,长度少于两位
     * return 整数
     */
    public static int readInt() {
        int n;
        for (; ; ) {
            String str = readKeyBoard(2, false);
            try {
                n = Integer.parseInt(str);
                break;
            } catch (NumberFormatException e) {
                System.out.println("选择错误，请重新输入：");
            }
        }
        return n;
    }

    /**
     * 功能:读取键盘输入的整数或者默认值,如果直接回车,则返回默认值
     * defaultValue 指定的默认值
     * return 整数或者默认值
     */
    public static int readInt(int defaultValue) {
        int n;
        for (; ; ) {
            String str = readKeyBoard(2, false);
            if (str.equals("")) {
                return defaultValue;
            }

            try {
                n = Integer.parseInt(str);
                break;
            } catch (NumberFormatException e) {
                System.out.println("选择错误，请重新输入：");
            }
        }
        return n;
    }

    /**
     * 功能:读取键盘输入的指定长度的字符串
     * limit:限制的长度
     * return:指定长度的字符串
     */
    public static String readString(int limit) {
        return readKeyBoard(limit, false);
    }

    /**
     * 功能:读取键盘输入的指定长度的字符串或者默认值,如果直接回车,返回默认值
     * limit:限制的长度
     * defaultValue 指定的默认值
     * return 指定长度的字符串
     */
    public static String readString(int limit, String defaultValue) {
        String str = readKeyBoard(limit, true);
        return str.equals("") ? defaultValue : str;
    }

    /**
     * 功能:读取键盘输入的确认选项,Y或N
     * 将小的功能,封装到一个方法中
     * return Y/N
     */
    public static char readConfirmSelection() {
        System.out.println("请输入你的选择(Y/N)");
        char c;
        for (; ; ) {//无限循环
            //在这里,将接收到字符,转换程大写字母
            //y=>Y,n=>N
            String str = readKeyBoard(1, false).toUpperCase();
            c = str.charAt(0);
            if (c == 'Y' || c == 'N') {
                break;
            } else {
                System.out.print("选择错误，请重新输入：");
            }
        }
        return c;
    }

    /**
     * 功能:读取键盘输入,如果输入为空,或者输入大于limit的长度,就会提示重新输入
     */
    private static String readKeyBoard(int limit, boolean blankReturn){
        //定义了字符串
        String line="";

        //scanner.hasNextLine()判断有没有下一行
        while (scanner.hasNextLine()){
            line=scanner.nextLine();//读取下一行

            //如果Line.length=0,即用户没有输入任何内容,直接回车
            if(line.length()==0){
                if(blankReturn) return line;
                else continue;//如果blackRuturn=false,不接受空串,必须输入内容
            }

            //如果用户输入的内容大于了limit,就提示重写输入
            //如果用户输入的内容哦个>0<=limit,我就接受
            if(line.length()<1 || line.length()>limit){
                System.out.println("输入长度（不大于 "+ limit + "）错误，请重新输入：");
                continue;
            }
            break;
        }
        return line;
    }
}

QQFrame.java

package com.leaveriver.qqframe;
import com.leaveriver.service.ServerService;
/**
 * @Author Administrator
 * @Date 2021/10/29 0029 15:00
 * @Version 1.0
 * 说明：该类创建一个QQServer对象。启动后台的服务
 */
public class QQFrame {
    public static void main(String[] args) {

              new ServerService();
    }
}

客户端

ClientConnectServerThread .java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;
import com.leaveriver.utils.Utility;

import java.io.BufferedOutputStream;
import java.io.FileOutputStream;
import java.io.IOException;
import java.io.ObjectInputStream;
import java.net.Socket;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 13:33
 * @Version 1.0
 * 说明：
 */
public class ClientConnectServerThread extends Thread {
    //该线程需要持有socket
    private Socket socket;

    //构造器可以接受一个Socket对象
    public ClientConnectServerThread(Socket socket) {
        this.socket = socket;
    }

    //
    @Override
    public void run() {
        //因为Thread需要在后台和服务器通信，因此我们做一个while循环
        while (true) {

            try {
                //System.out.println("客户端线程，等待读取从服务端发送的消息");
                ObjectInputStream ois = new ObjectInputStream(socket.getInputStream());
                Message message = (Message) ois.readObject();   //如果服务器没有发送Message对象，线程会在这里阻塞
                //该message为与服务器通信的信息，后续再完善。
                //判断message的类型。然后做相应的业务处理。
                //如果读到的是服务端返回的在线用户列表
                if (message.getMesType().equals(MessageType.MESSAGE_RET_ONLINE_FRIEND)) {
                    //取出在线列表信息，并显示
                    //规定服务端返回的是一串用户id.中间用空格隔开。 那么这里接收时，就用空格划分。
                    String[] s = message.getContent().split(" ");
                    System.out.println("\n============当前在线用户列表=================");
                    for (int i = 0; i < s.length; i++) {
                        System.out.println("用户： " + s[i]);
                    }
                } else if (message.getMesType().equals(MessageType.MESSAGE_COMM_MES)) {
                    System.out.println("\n [" + message.getSenderTime() + "] " + message.getSender() + "对你说： " + message.getContent());
                } else if (message.getMesType().equals(MessageType.MESSAGE_TO_ALL_MES)) {
                    System.out.println("\n [" + message.getSenderTime() + "] " + message.getSender() + "对大家说： " + message.getContent());
                } else if (message.getMesType().equals(MessageType.MESSAGE_SERVER_MES)) {
                    System.out.println("\n [" + message.getSenderTime() + "] " + message.getSender() + "对大家说： " + message.getContent());
                } else if (message.getMesType().equals(MessageType.MESSAGE_FILE_OUTPUT)) {
                    //接受并保存在指定位置。
                    FileOutputStream fos = new FileOutputStream(message.getDestFilePath());
                    fos.write(message.getFileBytes());
                    fos.flush();
                    System.out.println("\n" + message.getSender() + "给你发送文件，保存在：" + message.getDestFilePath() + "路径");
                    System.out.println("保存成功");
                    fos.close();
                } else {
                    System.out.println("其他类型的信息， 这里暂时不处理");
                }
            } catch (Exception e) {
                e.printStackTrace();
            }
        }
    }

    //为了更方便得到Socket, 增加一个get方法
    public Socket getSocket() {
        return socket;
    }
}

FileService .java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;
import com.leaveriver.utils.StreamUtils;

import java.io.*;
import java.text.SimpleDateFormat;

/**
 * @Author Administrator
 * @Date 2021/11/3 0003 12:23
 * @Version 1.0
 * 说明：文件传输服务
 */
public class FileService {

    //    Message类进行扩展，和文件相关的成员
    //    private byte[] fileBytes;
    //    private int fileLen = 0;
    //    private String srcFilePath;
    //    private String destFilePath;
    //发送文件
    public void sendFile(String sender, String fileToUser, String srcFilePath, String destFilePath){
        //创建message对象。
        Message message = new Message();
        message.setSender(sender);
        message.setGetter(fileToUser);
        message.setMesType(MessageType.MESSAGE_FILE_OUTPUT);
        message.setSrcFilePath(srcFilePath);
        message.setDestFilePath(destFilePath);

        message.setSenderTime(new SimpleDateFormat(
                "yyyy-MM-dd HH:mm:ss").format(System.currentTimeMillis()));



        //从磁盘上读取文件
        FileInputStream fis = null;
        byte[] fileBytes;
        try {
            fis = new FileInputStream(srcFilePath);
            fileBytes = new byte[(int) new File(srcFilePath).length()];
            fis.read(fileBytes);
            message.setFileBytes(fileBytes);
        } catch (IOException e) {
            e.printStackTrace();
        }finally {
            try {
                fis.close();
            } catch (IOException e) {
                e.printStackTrace();
            }
        }

        //发送给服务器
        try {
            ObjectOutputStream oos = new ObjectOutputStream(
                    ManageClientConnetServerThread.getClientConnectServerThread(sender).getSocket().getOutputStream());
            oos.writeObject(message);
            System.out.println("文件发送完成===");
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

ManageClientConnetServerThread .java

package com.leaveriver.service;

import jdk.nashorn.internal.ir.CallNode;

import java.util.HashMap;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 13:48
 * @Version 1.0
 * 说明：管理线程类
 */
public class ManageClientConnetServerThread {
    //放入hashmap集合中，key为用户id,value为线程
    private static HashMap<String, ClientConnectServerThread> hashMap = new HashMap<>();

    //将某个线程加入到集合中。
    public static void  addClientConnectServerThread(
            String userId, ClientConnectServerThread clientConnectServerThread){
        hashMap.put(userId,clientConnectServerThread);
    }

    //通过userId可以得到对应的线程
    public static ClientConnectServerThread getClientConnectServerThread(String userId) {
        return hashMap.get(userId);
    }

}

MessageService .java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;
import com.leaveriver.common.User;
import com.leaveriver.utils.StreamUtils;
import com.leaveriver.utils.Utility;

import java.io.*;
import java.net.Socket;
import java.text.SimpleDateFormat;

/**
 * @Author Administrator
 * @Date 2021/11/2 0002 16:30
 * @Version 1.0
 * 说明：消息服务
 */
public class MessageService {


    //用户私聊
    public void userOneChat(String senderId, String chatContent, String chatToUserId) {
        //创建message对象。类型为 MESSAGE_USER_CHAT_ONE
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_COMM_MES);
        message.setSender(senderId);
        //将目标用户和内容都加入
        message.setGetter(chatToUserId);
        message.setContent(chatContent);
        //将当前时间设置为发送时间
        SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        message.setSenderTime(df.format(System.currentTimeMillis()));

        //发送

        try {

            ClientConnectServerThread ccst = ManageClientConnetServerThread.getClientConnectServerThread(senderId);
            OutputStream os = ccst.getSocket().getOutputStream();
            /*OutputStream os = socket.getOutputStream();*/
            ObjectOutputStream oos = new ObjectOutputStream(os);
            oos.writeObject(message);
            System.out.println("私聊内容发送成功");

        } catch (IOException e) {
            e.printStackTrace();
        }


    }

    //群发消息
    public void chatAllUsers(String sender, String content) {
        //创建message对象
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_TO_ALL_MES);
        message.setSender(sender);


        //设置时间
        SimpleDateFormat df = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss");
        message.setSenderTime(df.format(System.currentTimeMillis()));

        message.setContent(content);

        //发送

        try {
            ObjectOutputStream oos = new ObjectOutputStream(
                    ManageClientConnetServerThread.getClientConnectServerThread(sender).getSocket().getOutputStream());
            oos.writeObject(message);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

UserClientService .java

package com.leaveriver.service;

import com.leaveriver.common.Message;
import com.leaveriver.common.MessageType;
import com.leaveriver.common.User;
import com.leaveriver.utils.Utility;

import java.io.*;
import java.net.InetAddress;
import java.net.Socket;
import java.net.UnknownHostException;
import java.text.SimpleDateFormat;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 12:52
 * @Version 1.0
 * 说明：该类完成用户登录验证和用户注册等功能
 */
public class UserClientService {

    private User user = new User();  //  User类作为成员属性。 因为我们可能在其他地方要使用user信息
    private Socket socket;//因为socket可能在其他地方要使用，所以也作为成员变量。

    //根据id和密码到服务器验证是否合法
    public boolean checkUser(String userId, String password) {

        boolean b = false;    //该方法的返回值。 true为成功。
        //初始化User对象的id和密码
        user.setUserID(userId);
        user.setPassword(password);
        //连接到服务端，发送user对象
        try {
            socket = new Socket(InetAddress.getLocalHost(), 9999);
            //得到ObjectOutputStream对象
            OutputStream os = socket.getOutputStream();
            ObjectOutputStream oos = new ObjectOutputStream(os);
            oos.writeObject(user);//发送user对象

            //发送user对象到服务器后，服务器会验证账号密码，返回相应的message信息。由该信息可确认是否验证成功

            //读取从服务端回送的Message对象
            ObjectInputStream ois = new ObjectInputStream(socket.getInputStream());
            Message message = (Message) ois.readObject();

            if (message.getMesType().equals(MessageType.MESSAGE_LOGIN_SUCCEED)) {

                //创建一个和服务器端保持通信的线程 ---->创建一个类  ClientConnectServerThread
                ClientConnectServerThread ccst = new ClientConnectServerThread(socket);
                //启动线程
                ccst.start();
                //为了后面客户端的拓展，我们将线程放在集合中。
                ManageClientConnetServerThread.addClientConnectServerThread(userId, ccst);


                //验证合法，返回true
                b = true;
            } else {
                //如果登录失败，要关闭socket
                socket.close();
            }

        } catch (Exception e) {
            e.printStackTrace();
        }


        return b;
    }


    //拉取在线人数服务
    public void onlineFriendList(){

        //发送一个message，类型为   MESSAGE_GET_ONLINE_FRIEND
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_GET_ONLINE_FRIEND);
        message.setSender(user.getUserID());
        try {
            //发送给服务器:
            //从线程集合中得到当前线程、
            //再拿到对应的socket,然后创建对应的输出流，直接发送message即可。
            /*ClientConnectServerThread ccst = ManageClientConnetServerThread.getClientConnectServerThread(user.getUserID());
            OutputStream os = ccst.getSocket().getOutputStream();*/
            OutputStream os = socket.getOutputStream();
            ObjectOutputStream oos = new ObjectOutputStream(os);
            oos.writeObject(message);

        } catch (Exception e) {
            e.printStackTrace();
        }

    }

    //编写方法， 退出客户端，并给服务器发送一个退出系统的message对象。
    public void systemExit(){

        //发送一个message，类型为   MESSAGE_CLIENT_EXIT
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_CLIENT_EXIT);
        message.setSender(user.getUserID());

        //发送
        try {
            ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
            oos.writeObject(message);
            System.out.println("用户 ["+user.getUserID()+"] 退出");
            System.exit(0);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

StreamUtils .java

package com.leaveriver.utils;

import java.io.*;

public class StreamUtils {

    /*
    *streamToByteArray(),  该方法功能：将输入流转化为byte[]，即可以把文件的内容读入到byte[]
    * */

    public static byte[] streamToByteArray(InputStream is) throws IOException {
        ByteArrayOutputStream bos = new ByteArrayOutputStream();  //创建输出流对象
        byte[] bytes = new byte[1024];  //字节数组
        int readLen;
        while ((readLen = is.read(bytes)) != -1){ //循环读取
            bos.write(bytes, 0, readLen);       //把读取到数据写入bos
        }
        byte[] array = bos.toByteArray();  //然后将bos转成字节数组
        bos.close();
        return array;
    }

    public static String streamToString(InputStream is) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(is));
        StringBuilder stringBuilder = new StringBuilder();
        String line;
        while ((line = bufferedReader.readLine()) != null){
            stringBuilder.append(line+"\n");
        }
        return stringBuilder.toString();
    }
}

Utility.java

package com.leaveriver.utils;


/**
 * 工具类的作用:
 * 处理各种情况的用户输入,并且能按照程序员对的需求,得到用户的控制台输入
 */

import java.util.Scanner;
import java.util.*;


public class Utility {
    //静态属性
    private static Scanner scanner = new Scanner(System.in);


    /**
     * 功能:读取键盘输入的一个菜单项,值:1-5范围
     */
    public static char readMenuSelection() {
        char c;
        for (; ; ) {
            String str = readKeyBoard(1, false);
            c = str.charAt(0);//将字符串转换为字符
            if (c != '1' && c != '2' && c != '3' && c != '4' && c != '5') {
                System.out.println("选择错误，请重新输入：");
            } else break;
        }
        return c;
    }

    /**
     * 功能:读取键盘输入的一个字符
     */
    public static char readChar() {
        String str = readKeyBoard(1, false);//就是一个字符
        return str.charAt(0);
    }

    /**
     * 功能;读取键盘输入的一个字符,如果直接回车,则返回指定的默认值
     * defaultValue 指定的默认值
     * return 默认值或者输入的字符
     */
    public static char readChar(char defaultValue) {
        String str = readKeyBoard(1, true);//要么是空字符,要么??
        return str.length() == 0 ? defaultValue : str.charAt(0);
    }


    /**
     * 读取键盘输入的整型,长度少于两位
     * return 整数
     */
    public static int readInt() {
        int n;
        for (; ; ) {
            String str = readKeyBoard(2, false);
            try {
                n = Integer.parseInt(str);
                break;
            } catch (NumberFormatException e) {
                System.out.println("选择错误，请重新输入：");
            }
        }
        return n;
    }

    /**
     * 功能:读取键盘输入的整数或者默认值,如果直接回车,则返回默认值
     * defaultValue 指定的默认值
     * return 整数或者默认值
     */
    public static int readInt(int defaultValue) {
        int n;
        for (; ; ) {
            String str = readKeyBoard(2, false);
            if (str.equals("")) {
                return defaultValue;
            }

            try {
                n = Integer.parseInt(str);
                break;
            } catch (NumberFormatException e) {
                System.out.println("选择错误，请重新输入：");
            }
        }
        return n;
    }

    /**
     * 功能:读取键盘输入的指定长度的字符串
     * limit:限制的长度
     * return:指定长度的字符串
     */
    public static String readString(int limit) {
        return readKeyBoard(limit, false);
    }

    /**
     * 功能:读取键盘输入的指定长度的字符串或者默认值,如果直接回车,返回默认值
     * limit:限制的长度
     * defaultValue 指定的默认值
     * return 指定长度的字符串
     */
    public static String readString(int limit, String defaultValue) {
        String str = readKeyBoard(limit, true);
        return str.equals("") ? defaultValue : str;
    }

    /**
     * 功能:读取键盘输入的确认选项,Y或N
     * 将小的功能,封装到一个方法中
     * return Y/N
     */
    public static char readConfirmSelection() {
        System.out.println("请输入你的选择(Y/N)");
        char c;
        for (; ; ) {//无限循环
            //在这里,将接收到字符,转换程大写字母
            //y=>Y,n=>N
            String str = readKeyBoard(1, false).toUpperCase();
            c = str.charAt(0);
            if (c == 'Y' || c == 'N') {
                break;
            } else {
                System.out.print("选择错误，请重新输入：");
            }
        }
        return c;
    }

    /**
     * 功能:读取键盘输入,如果输入为空,或者输入大于limit的长度,就会提示重新输入
     */
    private static String readKeyBoard(int limit, boolean blankReturn){
        //定义了字符串
        String line="";

        //scanner.hasNextLine()判断有没有下一行
        while (scanner.hasNextLine()){
            line=scanner.nextLine();//读取下一行

            //如果Line.length=0,即用户没有输入任何内容,直接回车
            if(line.length()==0){
                if(blankReturn) return line;
                else continue;//如果blackRuturn=false,不接受空串,必须输入内容
            }

            //如果用户输入的内容大于了limit,就提示重写输入
            //如果用户输入的内容哦个>0<=limit,我就接受
            if(line.length()<1 || line.length()>limit){
                System.out.println("输入长度（不大于 "+ limit + "）错误，请重新输入：");
                continue;
            }
            break;
        }
        return line;
    }

}

View.java

package com.leaveriver.view;

import com.leaveriver.service.FileService;
import com.leaveriver.service.MessageService;
import com.leaveriver.service.UserClientService;
import com.leaveriver.utils.Utility;

/**
 * @Author Administrator
 * @Date 2021/10/29 0029 12:16
 * @Version 1.0
 * 说明：
 */
public class View {
    private boolean loop = true; //控制是否显示主菜单
    private String key = ""; //接收用户的键盘输入

    private UserClientService userClientService = new UserClientService(); //用于登录服务器。
    private MessageService messageService = new MessageService();  //消息服务
    private FileService fileService = new FileService(); //文件服务
/*    public static void main(String[] args) {
        new View().mainMenu();
        System.out.println("退出系统。");
    }*/
    //显示主菜单
    public void mainMenu(){

        while(loop) {

            System.out.println("=============欢迎登录网络通讯系统=============");
            System.out.println("\t\t 1 登录系统");
            System.out.println("\t\t 9 退出系统");

            System.out.print("请输入你的选择：");
            key = Utility.readString(1);  //工具类。 读取字符串，控制在1个即可
            //根据用户的输入，来处理不同的逻辑
            switch (key){
                case "1" :
                    System.out.print("请输入用户号：");
                    String userID = Utility.readString(20);
                    System.out.print("请输入密  码：");
                    String password = Utility.readString(20);

                    //到这里，需要构建user对象，去服务端验证。
                    //这里编写一个类Service ，用于用户登录/注册服务
                    boolean b = userClientService.checkUser(userID, password);
                    if (b) {
                        System.out.println("=============欢迎 (用户" + userID +")=============");
                        //进入到二级菜单
                        while (loop) {
                            System.out.println("=============网络通讯系统二级菜单 (用户" + userID +")=============");
                            System.out.println("\t\t 1 显示在线用户列表");
                            System.out.println("\t\t 2 群发消息");
                            System.out.println("\t\t 3 私聊消息");
                            System.out.println("\t\t 4 发送文件");
                            System.out.println("\t\t 9 退出系统");

                            System.out.print("请输入你的选择：");
                            key = Utility.readString(1);  //工具类。 读取字符串，控制在1个即可

                            //具体实现后面完善
                            switch (key){
                                case "1":
                                    //这里在Service中编写一个方法实现。
                                    userClientService.onlineFriendList();
                                    break;
                                case "2":
                                    //System.out.println("群发消息");
                                    System.out.print("请输入群发消息：");
                                    String chatAll = Utility.readString(50);
                                    //编写群发方法, 规定接受方为all
                                    messageService.chatAllUsers(userID, chatAll);
                                    break;
                                case "3":
                                    //输入要私聊的对象用户名：
                                    System.out.print("请输入要私聊的对象用户名：");
                                    String chatToUserId = Utility.readString(20);
                                    //输入私聊内容：
                                    System.out.print("私聊内容：");
                                    String chatContent = Utility.readString(100);
                                    //这里在Service中编写一个方法实现。
                                    messageService.userOneChat(userID, chatContent, chatToUserId);

                                    break;
                                case "4":
                                    System.out.println("发送文件");
                                    System.out.print("你要发送文件给用户： ");
                                    String fileToUser = Utility.readString(20);
                                    System.out.print("你要发送的文件的完整路径: ");
                                    String fileStartPath = Utility.readString(50);
                                    System.out.print("保存下对方的路径: ");
                                    String fileTarPath = Utility.readString(50);


                                    //编写一个方法 实现。
                                    fileService.sendFile(userID, fileToUser, fileStartPath,fileTarPath);

                                    break;
                                case "9": //退出菜单
                                    //调用方法，给服务器发送一个退出系统的message
                                    userClientService.systemExit();
                                    loop = false;
                                    break;
                            }
                        }
                    }else {  //登录系统失败
                        System.out.println("============登录失败==================");

                    }
                    break;
                case "9" :
                    loop = false;
                    break;


            }
        }
    }
}

QQFrame.java

package com.leaveriver.qqframe;
import com.leaveriver.view.View;
/**
 * @Author Administrator
 * @Date 2021/10/29 0029 15:02
 * @Version 1.0
 * 说明：
 */
public class QQFrame {
    public static void main(String[] args) {
        View view = new View();
        view.mainMenu();
        System.out.println("退出系统。");
    }
}