Java团队课程设计-socket聊天室（Day3）

写了两天的项目，越写到后面越觉得这个结构太烂了，所以迫不得已只能重新搭建框架

此前代码的冗余度太高，复用率低，于是今天重构项目

重构后：

对Message类进行整理，分出HistoryMessage(历史消息),OfflineMessage(离线消息),OnlineMessage(在线消息)三个消息子类:

解决传输问题

问题1：

原来在ServiceController中将objectInputStream和objectOutputStream设置为static静态变量，这样做会导致多个线程连接服务器时，后一个线程的objectInputStream和objectOutputStream覆盖前一个线程，导致传输出错

解决：

创建每一个连接线程构造函数中传入各个用户对应的输入输出流

问题2:

原来的UserController(客户端)在与服务端交互时使用了多种对象流，导致接受端的判断变得困难且不利于数据的读取

解决:

统一使用ObjectInputStream对象流，在传输过程中，客户端先发送一个对象给服务端(用于判断发送的对象是什么)，再发送需要传输的Object。同时，接收方也需要进行两次相应读取操作

下面仅展示服务端的转发和客户端的接收代码：

1public void run() {
2            while (true) {
3                try {
4                    Object o = ois.readObject();
5                    if (o.getClass().toString().equals("class message.OnlineMessage")) {
6                        OnlineMessage onlineMessage = (OnlineMessage) o;
7                        if (sendMessage(onlineMessage)) {
8              //正确发送，Type为2，数据库我自己的表Flag为2，别人的表Flag为1
9                            InsertIntoMysql(onlineMessage, 1);
10                        } else {
11              //存入数据库，自己的表Type为2，对方的表Type为1，数据库我自己的表Flag为2，别人的表Flag为1
12                            InsertIntoMysql(onlineMessage, 2);
13                        }
14                    } else if (o.getClass().toString().equals("class message.HistoryMessage")) {
15                        getHistoryMessage();//发送历史消息
16                    }
17                    else if(o.getClass().toString().equals("class user.normal.impl.UserDocument"))
18                    {
19                        UserDocument userDocument=(UserDocument) ois.readObject();
20                        sendDocument(userDocument);
21                    }
22                }  catch (IOException |ClassNotFoundException e) {
23                    System.out.println(user.getUserName()+"线程意外退出！");
24                    return;
25                } catch (SQLException e){
26                    System.out.println(user.getUserName()+"此线程数据库异常！");
27                    return;
28                }
29            }
30        }
31
32
33public void run() {
34        while(true){
35            try {
36                Object object = objectInputStream.readObject();
37                if(object.equals("Find error in user regist!")){
38                    //注册失败
39                    RegisteJFrame.registrationReturnInformation("Find error in user regist!");
40                }else if(object.equals("user created successfully!")){
41                    //注册成功
42                    RegisteJFrame.registrationReturnInformation("user created successfully!");
43                }else if(object.equals("Login successful！")){
44                    Login.changePermission(2);
45                    //登陆成功
46                }else if(object.equals("Login error！")){
47                    Login.changePermission(1);
48                }else if(object.getClass().toString().equals("class user.normal.impl.User")){
49                    //返回数据库中列表信息cxxx
50                    User tempUser;
51                    ArrayList<User> personList = new ArrayList<>();
52                    while(true){
53                        tempUser = (User) objectInputStream.readObject();
54                        if(tempUser.getUserName()==null) break;
55                        personList.add(tempUser);
56                    }
57                    ContactJFrame.returnListInformation(personList);
58                    ContactJFrame.setButton();
59
60                }else if(object.getClass().toString().equals("class message.OfflineMessage")){
61
62                    Message tempMessage;
63                    ArrayList<OfflineMessage> offlineMessageArrayList = new ArrayList<>();
64                    while(true){
65                        tempMessage = (Message)objectInputStream.readObject();
66                        if(tempMessage.getSender()==null) break;
67                        OfflineMessage tempOfflineMessage = (OfflineMessage)tempMessage;
68                        offlineMessageArrayList.add(tempOfflineMessage);
69                        ContactJFrame.changeButton(tempMessage);
70                    }
71
72                    ContactJFrame.returnOfflineMessage(offlineMessageArrayList);
73
74                }else if(object.getClass().toString().equals("class message.OnlineMessage")){
75                    OnlineMessage onlineMessage = (OnlineMessage)objectInputStream.readObject();
76
77
78
79                    ContactJFrame.returnOnlineMessage(onlineMessage);
80
81                }else if(object.getClass().toString().equals("class message.HistoryMessage")){
82                    Object o;
83                    ArrayList<HistoryMessage> historyMessageArrayList = new ArrayList<>();
84                    while(true){
85                        o = objectInputStream.readObject();
86                        HistoryMessage t = (HistoryMessage)o;
87                        if(t.getSender()==null) break;
88                        historyMessageArrayList.add(t);
89                    }
90                    ContactJFrame.returnHistoryList(historyMessageArrayList);
91
92                }
93                else if(object.getClass().toString().equals("class user.normal.impl.UserDocument"))
94                {
95                    Object o=objectInputStream.readObject();
96                    UserDocument userDocument=(UserDocument) o;
97
98                    System.out.println(userDocument.toString());
99                    ContactJFrame.receiveUserDocument(userDocument);
100
101                }
102            } catch (IOException | ClassNotFoundException e) {
103                e.printStackTrace();
104                return;
105            }
106            try {
107                Thread.sleep(10);
108            } catch (InterruptedException e) {
109                e.printStackTrace();
110            }
111
112        }
113    }

Java团队课程设计-socket聊天室（Day1）

Java团队课程设计-socket聊天室（Day2）

Java团队课程设计-socket聊天室（Day4总结篇）