在Go 的net/http 回應中取得底層Socket
Go 的net/http 套件為建置一個方便的框架。但是,在某些情況下,您可能需要存取回應流的底層套接字。本文透過探索檢索套接字連接的各種方法來解決這一需求。
Go 1.13 之前
Go 1.13 之前,取得底層套接字需要更複雜的技術。這裡有兩種方法:
基於事件的追蹤(僅限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>
監聽器覆蓋(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 的發布,一個更直接的解決方案是可用:
基於上下文的儲存:
從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中文網其他相關文章!