首頁  >  文章  >  後端開發  >  如何存取 Go 的 net/http 回應中的底層套接字?

如何存取 Go 的 net/http 回應中的底層套接字?

Barbara Streisand
Barbara Streisand原創
2024-11-04 12:42:29725瀏覽

How Do I Access the Underlying Socket in Go's net/http Response?

在Go 的net/http 回應中取得底層Socket

Go 的net/http 套件為建置一個方便的框架。但是,在某些情況下,您可能需要存取回應流的底層套接字。本文透過探索檢索套接字連接的各種方法來解決這一需求。

Go 1.13 之前

Go 1.13 之前,取得底層套接字需要更複雜的技術。這裡有兩種方法:

  1. 基於事件的追蹤(僅限TCP 連接):
    對於透過TCP 連接埠的連接,您可以利用net. Listener。 ConnState 事件處理程序用於追蹤活動連線及其關聯的套接字。此資訊可以儲存在全域映射中以供以後檢索。

    <code class="go">var conns = make(map[string]net.Conn)
    
    func ConnStateEvent(conn net.Conn, event http.ConnState) {
      if event == http.StateActive {
        conns[conn.RemoteAddr().String()] = conn
      } else if event == http.StateHijacked || event == http.StateClosed {
        delete(conns, conn.RemoteAddr().String())
      }
    }
    
    func GetConn(r *http.Request) (net.Conn) {
      return conns[r.RemoteAddr]
    }</code>
  2. 監聽器覆蓋(Unix 套接字):
    Unix 套接字需要更複雜的方法,因為它們缺乏每個連接方法的唯一識別碼。一種解決方案是重寫 net.Listener.Accept 方法,為每個連接分配唯一的字串,並將這些字串用作儲存套接字資訊的鍵。

    <code class="go">func NewConnSaveListener(wrap net.Listener) (net.Listener) {
      return connSaveListener{wrap}
    }
    
    func (self connSaveListener) Accept() (net.Conn, error) {
      conn, err := self.Listener.Accept()
      ptrStr := fmt.Sprintf("%d", &conn)
      conns[ptrStr] = conn
      return remoteAddrPtrConn{conn, ptrStr}, err
    }
    
    func GetConn(r *http.Request) (net.Conn) {
      return conns[r.RemoteAddr]
    }</code>

Go 1.13 和超越

隨著Go 1.13 的發布,一個更直接的解決方案是可用:

  1. 基於上下文的儲存:
    從Go 1.13 開始,net.Conn 可以儲存在HTTP 請求的上下文中。這提供了一種乾淨且有效率的方式來存取套接字連接。

    <code class="go">type contextKey struct {
      key string
    }
    var ConnContextKey = &contextKey{"http-conn"}
    
    func SaveConnInContext(ctx context.Context, c net.Conn) (context.Context) {
      return context.WithValue(ctx, ConnContextKey, c)
    }
    
    func GetConn(r *http.Request) (net.Conn) {
      return r.Context().Value(ConnContextKey).(net.Conn)
    }</code>

透過了解這些技術,您可以有效地擷取 net/http 回應處理程序中的底層套接字,從而啟用自訂操作和特定於平台的功能。

以上是如何存取 Go 的 net/http 回應中的底層套接字?的詳細內容。更多資訊請關注PHP中文網其他相關文章!

陳述:
本文內容由網友自願投稿,版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容,請聯絡admin@php.cn