Golang使用conn.Write时如何检测远端服务器是否可达

TCP是面向流的可靠传输协议，调用conn.Write()时，操作系统仅将数据拷贝到内核TCP发送缓冲区就会返回成功，不会等待服务端返回ACK确认。当服务器故障、链路中断时，内核会按照TCP协议规则自动重试重传，该阶段上层的Write调用无法立刻感知到错误，只有当重传次数达到系统阈值后，后续的Write/Read操作才会返回错误。

你可以结合以下几种方案修改写入逻辑，快速感知连接异常：

• 启用TCP KeepAlive存活检测
  建立连接后开启TCP层面的存活探测，内核会定期向对端发送探测包，长时间未收到响应就会标记连接失效，后续操作会直接返回错误。

  conn, err := d.DialContext(ctx, string(transport), addr)
  if err != nil {
      // 处理建连错误
  }
  // 断言为TCP连接设置KeepAlive参数
  if tcpConn, ok := conn.(*net.TCPConn); ok {
      _ = tcpConn.SetKeepAlive(true)
      // 自定义探测周期，根据业务场景调整，示例为30秒
      _ = tcpConn.SetKeepAlivePeriod(30 * time.Second)
  }
  

• 给写入操作增加超时控制
  每次调用Write前设置写入超时，避免无限制阻塞，超时后直接判定当前连接不可用。

  // 设置写入超时，示例为5秒，可根据业务调整
  _ = conn.SetWriteDeadline(time.Now().Add(5 * time.Second))
  _, err := client.conn.Write([]byte(msg))
  if err != nil {
      // 出现错误直接关闭无效连接，触发重连逻辑
      _ = client.conn.Close()
      // 此处补充你的重连逻辑
      return err
  }
  

• 新增应用层心跳机制
  自定义业务心跳包，定期向服务端发送心跳请求，若规定时间内未收到服务端响应，直接标记连接失效，关闭旧连接重新建连。该方案灵活度最高，不受操作系统默认TCP参数限制。
• 配套独立读协程感知连接状态
  单独启动一个协程持续阻塞读取连接，一旦Read返回错误（包括EOF、连接重置等），立刻标记连接不可用，通知写入协程停止复用当前连接。

感知到连接错误后，必须销毁旧连接，不要继续复用，重新走DialContext建连逻辑再写入数据，避免数据丢失。

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