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

在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中文網其他相關文章！