微信协议长连接保活机制在 Kubernetes 环境下的适配策略

微信长连接特性与 Kubernetes 冲突

部分微信私有协议（如微信 PC 客户端、企业微信内部通道）依赖TCP 长连接维持会话，通过周期性心跳包（如每 30 秒）防止 NAT 超时或中间设备断连。然而，在 Kubernetes 中，Pod 可能因滚动更新、节点驱逐或 HPA 缩容被无预警终止，导致连接中断且无法优雅迁移。此外，Service 的 iptables 或 IPVS 模式默认不感知应用层心跳，仅依赖 TCP keepalive（通常 >2 小时），远超微信协议容忍阈值。

客户端侧：主动心跳与重连逻辑

Java 客户端需实现自适应心跳，并监听连接状态：

package wlkankan.cn.wx.conn;

import io.netty.bootstrap.Bootstrap;
import io.netty.channel.*;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.timeout.IdleStateHandler;

public class WeChatLongConnection {

    private final String host;
    private final int port;
    private Channel channel;
    private EventLoopGroup group = new NioEventLoopGroup();

    public void connect() {
        Bootstrap b = new Bootstrap();
        b.group(group)
         .channel(NioSocketChannel.class)
         .option(ChannelOption.SO_KEEPALIVE, true)
         .option(ChannelOption.TCP_NODELAY, true)
         .handler(new ChannelInitializer<>() {
             @Override
             protected void initChannel(Channel ch) {
                 // 微信协议编解码器（略）
                 ch.pipeline().addLast(new WeChatMessageCodec());
                 // 心跳：15秒写空闲则发PING
                 ch.pipeline().addLast(new IdleStateHandler(0, 15, 0));
                 ch.pipeline().addLast(new HeartbeatHandler());
                 ch.pipeline().addLast(new ConnectionEventHandler());
             }
         });

        ChannelFuture f = b.connect(host, port).syncUninterruptibly();
        this.channel = f.channel();
    }

    static class HeartbeatHandler extends ChannelInboundHandlerAdapter {
        @Override
        public void userEventTriggered(ChannelHandlerContext ctx, Object evt) {
            if (evt instanceof IdleStateEvent) {
                ctx.writeAndFlush(new PingMessage()); // 微信PING包
            }
        }
    }

    static class ConnectionEventHandler extends ChannelInboundHandlerAdapter {
        @Override
        public void channelInactive(ChannelHandlerContext ctx) {
            // 触发重连
            wlkankan.cn.task.ReconnectScheduler.scheduleReconnect(ctx.channel());
        }
    }
}

Kubernetes 侧：Pod 生命周期协同

1. PreStop Hook 延迟终止

在 Pod 终止前，先通知微信服务端“下线”，再等待连接自然关闭：

apiVersion: v1
kind: Pod
spec:
  containers:
  - name: wechat-agent
    image: wechat-agent:latest
    lifecycle:
      preStop:
        exec:
          command: ["/bin/sh", "-c", "curl -sf http://localhost:8080/graceful-shutdown && sleep 30"]

Java 服务端点：

@RestController
public class ShutdownController {

    @PostMapping("/graceful-shutdown")
    public void gracefulShutdown() {
        // 1. 向微信服务器发送 LOGOUT 包（若协议支持）
        WeChatSessionManager.getInstance().broadcastLogout();
        // 2. 停止接受新连接
        serverChannel.close();
        // 3. 等待现有连接处理完或超时
        CompletableFuture.allOf(
            WeChatSessionManager.getActiveSessions()
                .stream()
                .map(session -> session.closeGracefully())
                .toArray(CompletableFuture[]::new)
        ).orTimeout(25, TimeUnit.SECONDS).join();
    }
}

2. Readiness Probe 动态就绪检测

当连接异常时，自动将 Pod 标记为未就绪，避免流量导入：

livenessProbe:
  exec:
    command: ["cat", "/tmp/wechat_conn_ok"]
  initialDelaySeconds: 30
  periodSeconds: 10
readinessProbe:
  exec:
    command: ["cat", "/tmp/wechat_conn_ok"]
  initialDelaySeconds: 10
  periodSeconds: 5

Java 连接状态维护：

public class WeChatSessionManager {
    private static volatile boolean connected = false;

    public static void markConnected() {
        connected = true;
        try (var writer = new FileWriter("/tmp/wechat_conn_ok")) {
            writer.write("ok");
        } catch (IOException e) { /* ignore */ }
    }

    public static void markDisconnected() {
        connected = false;
        new File("/tmp/wechat_conn_ok").delete();
    }
}

网络侧：调整内核参数与 Service 配置

1. 缩短 TCP keepalive

在容器启动脚本中设置：

sysctl -w net.ipv4.tcp_keepalive_time=60
sysctl -w net.ipv4.tcp_keepalive_intvl=10
sysctl -w net.ipv4.tcp_keepalive_probes=3

或通过 InitContainer 注入：

initContainers:
- name: sysctl-tuner
  image: busybox
  command:
  - sysctl
  - -w
  - net.ipv4.tcp_keepalive_time=60
  securityContext:
    privileged: true

2. 使用 Headless Service + StatefulSet（可选）

若需固定 Pod IP 或 DNS 名称以配合微信白名单：

apiVersion: apps/v1
kind: StatefulSet
metadata:
  name: wechat-agent
spec:
  serviceName: wechat-agent-headless
  replicas: 3
---
apiVersion: v1
kind: Service
metadata:
  name: wechat-agent-headless
spec:
  clusterIP: None
  selector:
    app: wechat-agent

监控与告警

暴露连接状态指标：

// Micrometer 示例
Gauge.builder("wechat.connection.active", WeChatSessionManager::getActiveCount)
     .register(Metrics.globalRegistry);

配置 Prometheus 告警规则：

- alert: WeChatConnectionDown
  expr: wechat_connection_active == 0
  for: 1m
  labels:
    severity: critical

通过客户端主动心跳 + PreStop 协同 + 动态就绪探针 + 内核参数调优，可在 Kubernetes 环境下有效维持微信长连接的稳定性，避免因平台调度导致的服务中断。