Wie kann ich in Go auf den zugrunde liegenden Socket einer Netz-/http-Antwort zugreifen?

Abrufen des zugrunde liegenden Sockets einer Net/http-Antwort

Im Bereich der Netzwerkprogrammierung mit dem Net/http-Paket von Go müssen wir manchmal direkt auf den zugrunde liegenden Socket zugreifen Netzwerkverbindung (net.Conn).

Problembeschreibung

Ein Entwickler versucht, Dateien über net/http bereitzustellen, stößt jedoch auf ein Problem: Er benötigt Zugriff auf den zugrunde liegenden Socket von http.ResponseWriter in der Handlerfunktion (net.Conn). um plattformspezifische Systemaufrufe durchzuführen.

Problemumgehung

In Go 1.13 und höher kann net.Conn im Anforderungskontext gespeichert werden, indem Sie die folgenden Schritte ausführen:

• Verwenden Sie die SaveConnInContext-Funktion, um net zu speichern .Conn wird im Anforderungskontext gespeichert.
• Rufen Sie net.Conn aus der Anfrage mithilfe der GetConn-Funktion ab.

Vor dieser Version gab es zwei Alternativen:

Verwenden Sie die Remote-Adresszeichenfolge

Zum Abhören eines TCP-Ports für Server , net.Conn.RemoteAddr().String() ist für jede Verbindung eindeutig und kann als Schlüssel für die globale Verbindungszuordnung verwendet werden.

net.Listener.Accept() überschreiben

Für Server, die auf UNIX-Sockets lauschen, können wir net.Listener.Accept() überschreiben, um den Dateideskriptor für die Rückgabe zu verwenden ein einzigartigerer Wert.

Codebeispiel

Go 1.13 und höher

<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>

Für TCP-Verbindung

<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>

Für UNIX-Verbindungen

<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>

Durch diese Methoden können Entwickler einfach auf den zugrunde liegenden Socket von http.ResponseWriter zugreifen, was ihnen eine Reihe benutzerdefinierter Netzwerkverarbeitungsmöglichkeiten eröffnet.

Das obige ist der detaillierte Inhalt vonWie kann ich in Go auf den zugrunde liegenden Socket einer Netz-/http-Antwort zugreifen?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!