Wie überprüfe und ändere ich den Antworttext eines Reverse-Proxys in Go?

So überprüfen Sie den Antworttext des Reverse Proxy

Im angegebenen Go-Code wird ein Reverse Proxy eingerichtet, um Anfragen an www.google.com umzuleiten. Während der Reverse-Proxy ordnungsgemäß funktioniert, stellt sich die Frage: Wie kann man auf den Antworttext der Proxy-Anfrage zugreifen?

Die Lösung hierfür liegt im httputil-Paket, das den ReverseProxy-Typ bereitstellt. In der neuesten Version des Pakets bietet ReverseProxy eine optionale ModifyResponse-Funktion, die die Anpassung und Überprüfung der vom Backend kommenden Antwort ermöglicht.

Konkret können Entwickler durch die Implementierung dieser Funktion die *http.Response vor der Übergabe ändern es an den Kunden weiter. Ein möglicher Anwendungsfall für ModifyResponse ist das Lesen und Bearbeiten des Antworttexts.

Um dies zu veranschaulichen, betrachten Sie den folgenden geänderten Code:

package main

import (
    "bytes"
    "fmt"
    "io/ioutil"
    "net/http"
    "net/http/httputil"
    "net/url"
)

func main() {
    target := &url.URL{Scheme: "http", Host: "www.google.com"}
    proxy := httputil.NewSingleHostReverseProxy(target)

    proxy.ModifyResponse = func(resp *http.Response) (err error) {
        b, err := ioutil.ReadAll(resp.Body) // Read the HTML response body
        if err != nil {
            return err
        }

        resp.Body.Close() // The body is closed by the ReadAll function, but it's good practice to close it again
        b = bytes.Replace(b, []byte("server"), []byte("schmerver"), -1) // Modify the HTML response by replacing "server" with "schmerver"

        newBody := ioutil.NopCloser(bytes.NewReader(b))
        resp.Body = newBody
        resp.ContentLength = int64(len(b))
        resp.Header.Set("Content-Length", strconv.Itoa(len(b)))
        fmt.Println(resp) // Print the modified response for debugging purposes

        return nil
    }

    http.Handle("/google", proxy)
    http.ListenAndServe(":8099", nil)
}

In diesem Code ist die ModifyResponse-Funktion zum Lesen definiert Ändern Sie den HTML-Antworttext, indem Sie eine bestimmte Zeichenfolge ersetzen, und legen Sie dann den geänderten Text als Antworttext fest. Durch die Verarbeitung des URL-Pfads „/google“ werden Anfragen an www.google.com weitergeleitet und die geänderte Antwort an den Client zurückgesendet.

Das obige ist der detaillierte Inhalt vonWie überprüfe und ändere ich den Antworttext eines Reverse-Proxys in Go?. Für weitere Informationen folgen Sie bitte anderen verwandten Artikeln auf der PHP chinesischen Website!