设置 *tls.Conn 的 Keep-Alive 周期

亲爱的编程学习爱好者，如果你点开了这篇文章，说明你对《设置 *tls.Conn 的 Keep-Alive 周期》很感兴趣。本篇文章就来给大家详细解析一下，主要介绍一下，希望所有认真读完的童鞋们，都有实质性的提高。

我想增加 http 和 https 请求的 tcp 连接的保持活动时间。

对于 http 请求，可以这样完成：

package main

import (
    "fmt"
    "io"
    "log"
    "net"
    "net/http"
    "time"
)

func main() {
    server := &http.server{addr: ":8080", handler: http.handlerfunc(func(w http.responsewriter, r *http.request) {
        io.writestring(w, "hello, world!")
    })}

    server.connstate = func(conn net.conn, state http.connstate) {
        if state == http.statenew {
            if err := conn.(*net.tcpconn).setkeepaliveperiod(1000 * time.second); err != nil {
                fmt.println("could not set keep alive period", err)
            } else {
                fmt.println("update keep alive period")
            }
        }
    }

    log.fatal(server.listenandserve())
}

对于 https 请求，这不能通过 server.connstate 完成，因为将在函数内部传递的 net.conn 是 *tls.conn。此连接不会公开 setkeepaliveperiod 等函数，也不会提供对底层 *net.tcpconn 的访问。

func main() {
    server := &http.Server{Addr: ":8080", Handler: http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
        io.WriteString(w, "Hello, World!")
    })}

    server.ConnState = func(conn net.Conn, state http.ConnState) {
        if state == http.StateNew {
            tlsConn := conn.(*tls.Conn)
            // how to set SetKeepAlivePeriod
        }
    }

    log.Fatal(server.ListenAndServeTLS("../example.crt", "../example.key"))
}

如何设置 tls 连接的保持活动时间？

解决方案

（至少）有两种方法可以做到这一点：

使用 net.ListenConfig：

net.listenconfig 对象具有 keepalive time.duration 字段。当非零时，这将用于在接受的连接上设置保持活动状态（例如：for TCP on posix）。

您可以将侦听器传递给 servetls：

server := &http.server{...}

lc := net.listenconfig{keepalive: 1000 * time.second}
ln, err := lc.listen(context.background(), "tcp", ":8080")
if err != nil {
  panic(err)
}
defer ln.close()

log.fatal(server.servetls(ln, "../example.crt", "../example.key"))

如上所述，接受的 tcp 连接将自动启用保持活动状态，并将周期设置为指定值。

使用 tls.Config 回调：

您可以通过设置 tls.Config getconfigforclient 或 getcertificate 回调来访问 tls.conn 底层的 net.conn。

只要返回 nil 以使 tls 代码回退到默认行为，使用哪一个并不重要。重要的部分是访问 tls.ClientHelloInfo，它有一个指向底层连接的 .conn 字段。这将是 net.tcpconn。

setTCPKeepAlive := func(clientHello *tls.ClientHelloInfo) (*tls.Config, error) {
  // Check that the underlying connection really is TCP.
  if tcpConn, ok := clientHello.Conn.(*net.TCPConn); ok {
    if err := tcpConn.SetKeepAlivePeriod(1000 * time.Second); err != nil {
      fmt.Println("Could not set keep alive period", err)
    } else {
      fmt.Println("update keep alive period")
    }
  } else {
    fmt.Println("TLS over non-TCP connection")
  }

  // Make sure to return nil, nil to let the caller fall back on the default behavior.
  return nil, nil
}

tlsConfig := &tls.Config{
    ...
    GetConfigForClient: setTCPKeepAlive,
    ...
}

server := &http.Server{
    Addr:      ":8080",
    TLSConfig: tlsConfig,
}

server.ListenAndServeTLS("../example.crt", "../example.key")

今天带大家了解了的相关知识，希望对你有所帮助；关于Golang的技术知识我们会一点点深入介绍，欢迎大家关注golang学习网公众号，一起学习编程~