Bagaimana Menghuraikan Permintaan dan Respons HTTP daripada Fail Teks dalam Go?

Menghuraikan Permintaan dan Respons HTTP daripada Fail Teks dalam Go

Apabila bekerja dengan aliran saluran paip HTTP yang disimpan dalam fail teks, menghuraikan permintaan dan respons adalah penting untuk pemprosesan selanjutnya. Dalam artikel ini, kami akan meneroka cara untuk mencapai ini dengan berkesan dalam Go menggunakan fungsi http.ReadRequest dan http.ReadResponse.

Masalah

Diberikan fail teks yang mengandungi urutan permintaan dan respons HTTP, matlamatnya adalah untuk menghuraikan komponen ini ke dalam struktur data, khususnya sepotong Sambungan contoh.

Penyelesaian

Memanfaatkan fungsi yang disebutkan di atas menyediakan pendekatan yang mudah untuk menghuraikan kedua-dua permintaan dan respons daripada fail teks:

import (
    "bufio"
    "bytes"
    "fmt"
    "io"
    "io/ioutil"
    "log"
    "net/http"
    "net/http/httputil"
    "os"
)

type Connection struct {
    Request  *http.Request
    Response *http.Response
}

func ReadHTTPFromFile(r io.Reader) ([]Connection, error) {
    buf := bufio.NewReader(r)
    stream := make([]Connection, 0)

    for {
        req, err := http.ReadRequest(buf)
        if err == io.EOF {
            break
        }
        if err != nil {
            return stream, err
        }

        resp, err := http.ReadResponse(buf, req)
        if err != nil {
            return stream, err
        }

        // Save response body
        b := new(bytes.Buffer)
        io.Copy(b, resp.Body)
        resp.Body.Close()
        resp.Body = ioutil.NopCloser(b)

        stream = append(stream, Connection{Request: req, Response: resp})
    }
    return stream, nil
}

Penggunaan

Untuk menggunakan fungsi ini, cuma baca input fail teks ke dalam pembaca dan panggil ReadHTTPFromFile. Fungsi ini akan mengembalikan sekeping contoh Sambungan, di mana setiap sambungan mewakili permintaan dan respons yang berkaitan.

f, err := os.Open("/tmp/test.http")
if err != nil {
    log.Fatal(err)
}
defer f.Close()
stream, err := ReadHTTPFromFile(f)
if err != nil {
    log.Fatalln(err)
}

Nota Tambahan

• http.ReadRequest dan http.ReadResponse boleh dipanggil berulang kali pada pembaca buffer yang sama sehingga ia mencapai akhir fail.
• Untuk memastikan keserasian dengan fail input anda, sahkan bahawa ia termasuk pengepala Panjang Kandungan dengan nilai yang tepat.
• Jika fail input anda mengandungi jawapan terpenggal, seperti dalam contoh yang diberikan, anda mungkin mempertimbangkan untuk menggunakan httputil.DumpResponse untuk memastikan perwakilan kandungan respons yang lebih komprehensif.

Atas ialah kandungan terperinci Bagaimana Menghuraikan Permintaan dan Respons HTTP daripada Fail Teks dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!