Java心跳服务器如何判断客户端断开

在开发网络应用程序时,经常会遇到需要实现心跳功能来保持客户端与服务器之间的连接状态。通过定期发送心跳包,服务器可以判断客户端是否断开连接,从而及时处理断开连接的情况。

实现逻辑

  1. 客户端和服务器建立连接后,客户端定时发送心跳包给服务器。
  2. 服务器接收到心跳包后,更新客户端的最后活动时间。
  3. 服务器定时检查客户端的最后活动时间,如果超过一定时间没有收到心跳包,则判断客户端已断开连接。

代码示例

客户端代码

import java.io.*;
import java.net.*;

public class Client {
    private static final int HEARTBEAT_INTERVAL = 5000; // 心跳间隔时间,单位毫秒

    public static void main(String[] args) {
        try {
            Socket socket = new Socket("localhost", 8888);
            PrintWriter out = new PrintWriter(socket.getOutputStream(), true);

            while (true) {
                out.println("heartbeat"); // 发送心跳包
                Thread.sleep(HEARTBEAT_INTERVAL);
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
    }
}

服务器端代码

import java.io.*;
import java.net.*;
import java.util.*;

public class Server {
    private static final int TIMEOUT = 10000; // 客户端超时时间,单位毫秒
    private Map<Socket, Long> clients = new HashMap<>();

    public static void main(String[] args) {
        new Server().start();
    }

    public void start() {
        try {
            ServerSocket serverSocket = new ServerSocket(8888);

            while (true) {
                Socket clientSocket = serverSocket.accept();
                clients.put(clientSocket, System.currentTimeMillis());

                new Thread(() -> {
                    try {
                        BufferedReader in = new BufferedReader(new InputStreamReader(clientSocket.getInputStream()));

                        String message;
                        while ((message = in.readLine()) != null) {
                            if (message.equals("heartbeat")) {
                                clients.put(clientSocket, System.currentTimeMillis());
                            }
                        }
                    } catch (IOException e) {
                        e.printStackTrace();
                    }
                }).start();
            }

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

序列图

sequenceDiagram
    participant Client
    participant Server
    Client->>Server: 发送心跳包
    Server->>Server: 更新客户端活动时间
    Server->>Server: 定时检查客户端活动时间
    Server-->>Server: 判断客户端是否断开连接

结论

通过定时发送心跳包并在服务器端更新客户端的最后活动时间,可以实现心跳服务器判断客户端是否断开连接的功能。这种方法可以帮助及时发现客户端断开连接的情况,并进行处理。在实际开发中,可以根据需求调整心跳间隔时间和超时时间,以达到更好的效果。