WebSocket服务实现类net/http优雅关机的相关技术问题咨询

• 加10%随机抖动是为了消除固定间隔的同步效应：如果用固定间隔轮询，很可能出现大量连接刚好卡在轮询间隙同时从活跃转为空闲的情况，导致单次轮询要关闭大量连接，产生CPU、IO尖峰；同时也能避免固定间隔导致的空闲连接判定对齐，拉长整体关机耗时。
• 搭配指数退避（间隔每次翻倍直到上限）是为了平衡关机速度和轮询开销：刚触发关机时活跃连接多，用短间隔可以快速回收空闲连接；随着关机推进，剩余活跃连接越来越少，拉长间隔可以避免无意义的频繁遍历，降低不必要的性能损耗。

net/http的检测逻辑是基于连接状态+状态更新时间戳实现的，具体规则如下：

• 每个连接会维护当前状态（StateNew/StateActive/StateIdle等）和最近一次状态变更的Unix时间戳
• 常规情况下，只有状态为StateIdle（当前没有处理任何请求，处于等待请求状态）且时间戳有效的连接才会被判定为空闲
• 特殊兼容逻辑：如果是刚建立的StateNew状态连接，超过5秒还没读到第一个请求头，也会被判定为空闲，避免僵死的新连接一直阻塞关机流程
• 刚创建、还没设置过状态的连接（时间戳为0）不会被判定为空闲，避免误关

你可以复用net/http的设计思路，结合WebSocket长连接的特性做适配，步骤如下：

1. 先停止接收新连接：触发关机后第一时间关闭端口监听，不再处理新的WebSocket握手请求，对应http Shutdown里numListeners() == 0的判断。
2. 维护连接元数据：给每个WebSocket连接维护两个核心字段：最近一次收发消息的时间戳、当前是否正处于业务逻辑处理中（比如正执行消息推送、业务逻辑计算）；每次收到客户端消息、向客户端推送消息时都更新活跃时间戳。
3. 实现轮询关闭逻辑：
   • 复用你现有的sync.WaitGroup：每个连接建立时wg.Add(1)，连接正常/异常关闭时wg.Done()，轮询时如果wg计数为0可以直接结束关机。
   • 照搬指数退避+随机抖动的轮询间隔逻辑，每次轮询遍历所有活跃连接：
     • 符合两个条件之一判定为空闲：一是当前没有执行业务逻辑，且超过你设定的空闲阈值（比如30s）没有任何消息收发；二是连接已经超过你设置的最长存活时间，没有任何读写操作。
     • 对空闲连接先发送WebSocket CLOSE帧（状态码1001表示服务端关机），等待短时间握手后关闭连接，从活跃连接列表移除。
   • 每次轮询后如果没有剩余活跃连接，直接返回关机成功。
4. 添加上下文超时兜底：给整个关机流程加超时控制，如果超过用户设定的最大等待时间还有活跃连接，直接强制关闭所有连接，返回超时错误，避免关机流程无限阻塞。

内容的提问来源于stack exchange，提问作者Nakem1