如何在 Go 中访问 net/http 响应的底层套接字？

Retrieving the Underlying Socket of a net/http Response

In the realm of network programming with Go's net/http package,有时我们可能需要直接访问底层的网络连接(net.Conn)。

问题陈述

一位开发者正在尝试使用net/http为文件提供服务，却遇到了一个问题：需要访问处理程序函数中http.ResponseWriter的底层套接字(net.Conn)，以便进行特定于平台的系统调用。

解决方法

在Go 1.13及更高版本中，net.Conn可以通过以下步骤存储在请求上下文中：

• 使用SaveConnInContext函数将net.Conn保存到请求上下文中。
• 使用GetConn函数从请求中检索net.Conn。

在此版本之前，有两种替代方法：

使用远程地址字符串

对于在TCP端口上侦听的服务器，net.Conn.RemoteAddr().String()对于每个连接都是唯一的，可以用作全局连接映射的键。

覆盖net.Listener.Accept ()

对于在UNIX套接字上侦听的服务器，我们可以重写net.Listener.Accept()以使用文件描述符来返回一个更独特的值。

代码示例

Go 1.13及更高版本

<code class="go">// SaveConnInContext stores the net.Conn in the request context.
func SaveConnInContext(ctx context.Context, c net.Conn) context.Context {
  return context.WithValue(ctx, ConnContextKey, c)
}

// GetConn retrieves the net.Conn from the request context.
func GetConn(r *http.Request) net.Conn {
  return r.Context().Value(ConnContextKey).(net.Conn)
}</code>

对于TCP连接

<code class="go">// ConnStateEvent handles connection state events.
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())
  }
}

// GetConn retrieves the net.Conn from a map using the remote address as a key.
func GetConn(r *http.Request) net.Conn {
  return conns[r.RemoteAddr]
}</code>

对于UNIX连接

<code class="go">// NewUnixListener creates a new UNIX listener with a modified Accept() method.
func NewUnixListener(path string) (net.Listener, error) {
  // ... (setup code)

  l, err := net.Listen("unix", path)
  if err != nil {
    return nil, err
  }

  return NewConnSaveListener(l), nil
}

// NewConnSaveListener wraps a listener and overrides the Accept() method.
func NewConnSaveListener(wrap net.Listener) net.Listener {
  return connSaveListener{wrap}
}

// remoteAddrPtrConn overrides the RemoteAddr() method to return a unique value.
type remoteAddrPtrConn struct {
  net.Conn
  ptrStr string
}

func (self remoteAddrPtrConn) RemoteAddr() net.Addr {
  return remoteAddrPtr{self.ptrStr}
}

// remoteAddrPtr implements the net.Addr interface.
type remoteAddrPtr struct {
  ptrStr string
}

func (remoteAddrPtr) Network() string {
  return ""
}

func (self remoteAddrPtr) String() string {
  return self.ptrStr
}

// Accept overrides the default Accept() method to store the net.Conn in a map.
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
}</code>

通过这些方法，开发人员可以轻松访问http.ResponseWriter的底层套接字，从而为其自定义网络处理打开了一系列可能性。

以上是如何在 Go 中访问 net/http 响应的底层套接字？的详细内容。更多信息请关注PHP中文网其他相关文章！