Bagaimana untuk Mengakses Badan Respons dalam Proksi Go Reverse?

Mengakses Badan Tindak Balas dalam Proksi Terbalik

Apabila bekerja dengan Reverse Proxy, anda mungkin menghadapi keperluan untuk mengakses badan respons yang diterima dari bahagian belakang pelayan. Artikel ini menyelidiki cara untuk mendapatkan semula badan respons dalam Go menggunakan pakej httputil.

Dalam coretan kod yang disediakan, anda mempunyai proksi terbalik mudah yang memajukan permintaan kepada Google:

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

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

Kunci untuk mengakses badan tindak balas terletak pada medan ModifyResponse jenis ReverseProxy. Mengikut dokumentasi rasmi, ModifyResponse ialah fungsi pilihan yang membolehkan anda mengubah suai respons daripada bahagian belakang.

Berikut ialah contoh yang membaca dan mengubah suai badan respons sebelum menghantarnya kepada pelanggan:

import (
    "bytes"
    "fmt"
    "io/ioutil"
)

func rewriteBody(resp *http.Response) (err error) {
    b, err := ioutil.ReadAll(resp.Body) //Read html
    if err != nil {
        return err
    }
    err = resp.Body.Close()
    if err != nil {
        return err
    }
    b = bytes.Replace(b, []byte("server"), []byte("schmerver"), -1) // replace html
    body := ioutil.NopCloser(bytes.NewReader(b))
    resp.Body = body
    resp.ContentLength = int64(len(b))
    resp.Header.Set("Content-Length", strconv.Itoa(len(b)))
    return nil
}

func main() {
    target, _ := url.Parse("http://example.com")
    proxy := httputil.NewSingleHostReverseProxy(target)
    proxy.ModifyResponse = rewriteBody

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

Dalam kod ini, fungsi rewriteBody membaca badan tindak balas, mengubah suainya (dalam kes ini, menggantikan perkataan "pelayan" dengan "schmerver"), dan kemudian menetapkan badan yang diubah suai sebagai badan tindak balas baharu.

Atas ialah kandungan terperinci Bagaimana untuk Mengakses Badan Respons dalam Proksi Go Reverse?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!