实现WebSocket心跳机制,以及超过30分钟无活动自动关闭会话的功能

实现WebSocket心跳机制,以及超过30分钟无活动自动关闭会话的功能,涉及到后端和前端的协作。下面分别介绍后端(使用Spring WebFlux)和前端如何实现这一机制。

后端实现(Spring WebFlux)

在Spring WebFlux中,你可以定期发送心跳消息(例如,使用ping消息)来维持连接。同时,你需要追踪最后一次活动时间,如果超过30分钟无活动,则关闭会话。

  1. 心跳消息发送:

在WebSocketHandler的实现中,你可以通过定时发送ping消息来实现心跳。使用Flux.interval创建一个定期执行的Flux,并发送ping消息。

java 复制代码
@Override
public Mono<Void> handle(WebSocketSession session) {
    Flux<WebSocketMessage> heartBeat = Flux.interval(Duration.ofSeconds(15))  // 每15秒发送一次心跳
            .map(tick -> session.pingMessage(DataBufferFactory -> DataBufferFactory.allocateBuffer(0)));

    Mono<Void> output = session.send(heartBeat);

    Mono<Void> input = session.receive()  // 处理接收到的消息
            .doOnNext(message -> {
                // 更新最后活动时间
            })
            .then();

    return Mono.zip(input, output).then();
}
  1. 活动追踪和超时处理:

你可以定义一个变量来追踪最后一次活动时间。每次接收到消息时更新这个时间。同时,你可以使用一个定时器来检查是否已经超过30分钟无活动。

java 复制代码
AtomicLong lastActivityTime = new AtomicLong(System.currentTimeMillis());

Flux<WebSocketMessage> timeoutChecker = Flux.interval(Duration.ofMinutes(1))
    .flatMap(tick -> {
        long inactiveDuration = System.currentTimeMillis() - lastActivityTime.get();
        if (inactiveDuration > Duration.ofMinutes(30).toMillis()) {
            // 超过30分钟无活动,可以关闭会话
            return Flux.just(session.close(CloseStatus.NORMAL.withReason("Inactive for more than 30 minutes")));
        } else {
            return Flux.empty();  // 无操作
        }
    });

Mono<Void> output = session.send(Flux.merge(heartBeat, timeoutChecker));

前端实现

前端需要实现两部分逻辑:一是响应心跳消息,二是处理连接超时关闭。

  1. WebSocket心跳响应:

前端WebSocket客户端需要能够处理后端发送的心跳ping消息,并据此保持连接。

javascript 复制代码
let socket = new WebSocket("ws://example.com/ws");
socket.onmessage = function(event) {
    // 处理服务器发送的消息
};

// 处理Ping消息,维持连接
socket.onping = function(event) {
    // 可以发送Pong消息回应,或不做任何操作
};
  1. 超时关闭处理:

前端还需要能够处理超过30分钟无数据交换时的情况。这通常通过设置一个定时器实现,每次收到消息时重置定时器。

javascript 复制代码
let timeoutInterval = 30 * 60 * 1000;  // 30分钟
let timeoutTimer;

function resetTimeout() {
    clearTimeout(timeoutTimer);
    timeoutTimer = setTimeout(() => {
        socket.close();
        alert("WebSocket closed due to inactivity.");
    }, timeoutInterval);
}

socket.onopen = resetTimeout;
socket.onmessage = resetTimeout;
socket.onping = resetTimeout;  // 如果前端能收到ping消息,也应重置定时器

通过上述方法,你可以在客户端和服务器之间实现一个心跳机制,以及在超过一定时间无活动时自动关闭WebSocket连接的功能。

相关推荐
wdeok15 小时前
Socket编程UDP-简单的聊天室
网络·网络协议·udp
严肃小白兔1 天前
TanStack Router 更新构建工具链:Node.js 兼容性、路由分割与前端工程实践
前端·node.js·状态模式
LK_072 天前
【FastAPI】基础篇 01|开始之前:API、HTTP 与 FastAPI 到底是什么
网络协议·http·fastapi
阿星_2 天前
pydio登陆不了分析问题
网络协议·http·https
Joy T2 天前
Gzip 压缩原理与实践:HTTP 响应压缩、性能收益与工程配置
网络·网络协议·http
博客zhu虎康4 天前
实现局域网P2P通信:内网发送消息
网络·网络协议·p2p
xixingzhe24 天前
AI 对话打字机效果:SSE 和 HTTP 的区别与适用场景
网络·网络协议·http
叼烟扛炮5 天前
Linux 学习笔记(五十一):数据链路层与 ARP 协议——局域网寻址与黑客骗局
linux·网络·网络协议·php·tcpdump
沫璃染墨5 天前
从零入门计算机网络系列(十四):TCP报文与字节流本质——从数据封装到报文边界
linux·服务器·网络·网络协议·tcp/ip·tcp
byte轻骑兵5 天前
【LMP】规范精讲[13]: 参数的户口本、公钥的集装箱与出厂的兜底值
人工智能·网络协议·人机交互·lmp·蓝牙链路管理