多用户及时通信系统之无异常退出功能实现(4)

问题

目前退出会出现的问题，输入9后退出，没有真正的退出，一直处于待输入的状态。

是因为现在退出时，程序将循环标志loop设为false，退出了内层循环，而外层循环因为也用了loop来作为循环条件，外层循环也同样退出。此时在客户端 类QQView中的主线程已经结束，但是在循环过程中，因为与服务端连接而产生的线程并没有结束，整个进程也就没有结束，因此程序仍在运行中。

解决方法：

客户端：在main线程中调用方法，给服务端发送一个退出系统的message对象，然后调用System.exit(0)指令，正常退出。这样整个进程就可以关闭。

服务器端：在服务器这边，接收到一个退出系统的message对象后，把这个客户端对应的线程所持有的socket关闭，然后退出该线程。

功能实现

客户端:

1. 在UserClientService中添加登出方法，结束进程
2. 在QQView中调用登出方法

UserClientService.java : 添加logout()方法

package com.ming.qqclient.service;

import com.ming.qqcommon.Message;
import com.ming.qqcommon.MessageType;
import com.ming.qqcommon.User;

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.InetAddress;
import java.net.Socket;

/**
 * @Author: mei_ming
 * @DateTime: 2022/10/2 20:13
 * @Description: 用户登录、注册等功能的实现类
 */
public class UserClientService {

    //因为我们可能在其他地方使用user信息，因此作为成员属性
    private User u = new User();
    //因为Socket在其他地方也有可能使用
    private Socket socket;

    //根据userId 和 pwd 到服务器验证该用户是否合法
    public boolean checkUser(String userId,String pwd){
        boolean b = false;
        //创建User对象
        u.setUserId(userId);
        u.setPasswd(pwd);

        try {
            //连接到服务端，发送User对象 u
            socket = new Socket(InetAddress.getByName("127.0.0.1"), 9999);
            //得到ObjectOutputStream对象
            ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
            oos.writeObject(u);

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

            if (message.getMesType().equals(MessageType.MESSAGE_LOGIN_SUCCEED)){  //登录成功
                //创建一个和服务器端保持通信的线程 ClientConnectServerThread 类
                //将socket放入线程中
                ClientConnectServerThread connectServerThread = new ClientConnectServerThread(socket);
                connectServerThread.start();
                //将线程放在集合中
                ManageClientConnectServerThread.addClientConnectServerThread(userId,connectServerThread);
                b=true;
            }else{  //登录失败
                //关闭socket
                socket.close();
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return b;
    }

    //向服务端请求在线用户列表
    public void onlineFriendList(){
        //发送一个Message，类型是
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_GET_ONLINE_FRIEND);
        message.setSender(u.getUserId());

        //发送给服务端
        try {
            //1. 从管理线程的集合中，通过userId，得到这个 线程 对象
            ClientConnectServerThread clientConnectServerThread = ManageClientConnectServerThread.getClientConnectServerThread(u.getUserId());
            //2. 得到当前线程的 Socket
            Socket socket = clientConnectServerThread.getSocket();
            //3. 得到当前Socket相关联的 ObjectOutPutStream
            ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
            //向服务端 发送 Message
            oos.writeObject(message);

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

    //退出客户端，发送Message断开连接通道
    public void logout(){
        Message message = new Message();
        message.setMesType(MessageType.MESSAGE_CLIENT_EXIT);
        message.setSender(u.getUserId());  //指定关闭哪一个客户端

        //发送message
        try {
            //ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
            Socket socket = ManageClientConnectServerThread.getClientConnectServerThread(u.getUserId()).getSocket();
            ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
            oos.writeObject(message);
            System.out.println(u.getUserId()+" 退出系统");
            System.exit(0);  //结束进程
        } catch (IOException e) {
            e.printStackTrace();
        }

    }
}

QQView.java : 调用logout()方法

switch (key){
     case "1":
         //调用UserClientService中的方法：
         userClientService.onlineFriendList();
         break;
     case "2":
         System.out.println("群发消息");
         break;
     case "3":
         System.out.println("私聊消息");
         break;
     case "4":
         System.out.println("发送文件");
         break;
     case "9":
         //调用方法 退出
         userClientService.logout();
         loop=false;
         break;
}

---

服务端:

1. 在ServerConnectClientThread中判断messageType的类型
2. 在ManageClientThreads中添加删除线程的方法

ServerConnectClientThread.java : 添加判断messageType的类型，如果等于MESSAGE_CLIENT_EXIT 则进行删除线程操作

package com.ming.qqserver.service;

import com.ming.qqcommon.Message;
import com.ming.qqcommon.MessageType;
import com.sun.deploy.net.proxy.ProxyUnavailableException;

import java.io.IOException;
import java.io.ObjectInputStream;
import java.io.ObjectOutputStream;
import java.net.Socket;

/**
 * @Author: mei_ming
 * @DateTime: 2022/10/2 21:09
 * @Description: 该类的一个对象和某个客户端保持通信
 */
public class ServerConnectClientThread extends Thread {

    private Socket socket;
    private String userId;  //区分是哪一个客户端

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

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

                //根据message类型，做相应的业务处理
                if (message.getMesType().equals(MessageType.MESSAGE_GET_ONLINE_FRIEND)){
                    //客户端要在线用户列表
                    // 格式 "100 200 孙悟空"
                    System.out.println(message.getSender() + " 要在线用户列表");
                    String onlineUser = ManageClientThreads.getOnlineUser();
                    //返回message
                    Message message2 = new Message();
                    message2.setMesType(MessageType.MESSAGE_RET_ONLINE_FRIEND);
                    message2.setContent(onlineUser);
                    message2.setGetter(message.getSender());

                    //返回客户端
                    ObjectOutputStream oos = new ObjectOutputStream(socket.getOutputStream());
                    oos.writeObject(message2);
                }else if(message.getMesType().equals(MessageType.MESSAGE_CLIENT_EXIT)){
                    //客户端退出
                    System.out.println(message.getSender()+" 退出");
                    //将这个客户端对应的线程从集合中删除
                    ManageClientThreads.removeServerConnectClientThread(message.getSender());
                    socket.close();  //关闭连接
                    break;  //退出线程
                }else{
                    System.out.println("是其他类型的message,暂不处理");
                }
            } catch (IOException e) {
                e.printStackTrace();
            } catch (ClassNotFoundException e) {
                e.printStackTrace();
            }
        }
    }
}

ManageClientThreads.java : 添加 删除线程 方法

package com.ming.qqserver.service;

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

/**
 * @Author: mei_ming
 * @DateTime: 2022/10/2 21:20
 * @Description: 该类用于管理和客户端通信的线程
 */
public class ManageClientThreads {

    private static HashMap<String,ServerConnectClientThread> hm=new HashMap<>();

    //添加线程到hm中
    public static void addServerConnectClientThread(String userId,ServerConnectClientThread serverConnectClientThread){
        hm.put(userId,serverConnectClientThread);
    }

    //删除线程
    public static void removeServerConnectClientThread(String userId){
        hm.remove(userId);
    }

    //根据userId返回ServerConnectClientThread线程
    public static ServerConnectClientThread getServerConnectClientThread(String userId){
        return hm.get(userId);
    }

    //返回在线列表
    public static String getOnlineUser(){
        //集合遍历
        Iterator<String> iterator = hm.keySet().iterator();
        String onlineUserList = "";
        while(iterator.hasNext()){
            onlineUserList += iterator.next()+" ";
        }
        return onlineUserList;
    }
}

运行结果

 下一篇： 用户私聊/群发功能