Bagaimana untuk Membaca Data Panjang Berubah daripada Sambungan Rangkaian dalam Go?

Membaca Data Panjang Pembolehubah dalam Go dengan net.Conn.Read

Apabila bekerja dengan aplikasi rangkaian dalam Go, anda sering menghadapi situasi di mana data yang diterima melalui sambungan adalah panjang berubah-ubah. Net.Conn perpustakaan standard menyediakan kaedah yang dipanggil Baca, yang mengisi tatasusunan bait dengan data yang diterima. Walau bagaimanapun, pendekatan ini boleh menimbulkan masalah jika anda tidak mengetahui panjang sebenar kandungan terlebih dahulu dan boleh mengakibatkan pembacaan data terlalu banyak atau tidak mencukupi.

Untuk menangani cabaran ini, satu pendekatan ialah menggunakan bufio pakej. Walau bagaimanapun, penyelesaian yang lebih cekap ialah menggunakan penimbal yang semakin meningkat dan membaca data sehingga fail akhir (EOF) ditemui:

package main

import (
    "fmt"
    "io"
    "net"
)

func main() {
    // Establish a connection with a remote host.
    conn, err := net.Dial("tcp", "google.com:80")
    if err != nil {
        fmt.Println("dial error:", err)
        return
    }
    defer conn.Close()

    // Write a request to the connection.
    fmt.Fprintf(conn, "GET / HTTP/1.0\r\n\r\n")

    // Create a buffer to store the received data.
    buf := make([]byte, 0, 4096)

    // Read from the connection in a loop until EOF.
    for {
        // Read into a temporary buffer to avoid overwriting existing data in `buf`.
        tmp := make([]byte, 256)
        n, err := conn.Read(tmp)
        if err != nil {
            if err != io.EOF {
                fmt.Println("read error:", err)
            }
            break
        }

        // Append the new data to the buffer.
        buf = append(buf, tmp[:n]...)
    }

    fmt.Println("total size:", len(buf))
}

Penyelesaian ini memperuntukkan penimbal awal yang besar dan mengembangkannya secara beransur-ansur sebagai lebih data diterima. Ia meneruskan bacaan sehingga EOF dicapai, memastikan penimbal mengandungi respons yang lengkap. Selain itu, ia menjejaki jumlah saiz data yang diterima.

Sebagai alternatif, anda boleh menggunakan bait.Jenis penimbal bersama-sama dengan io.Copy untuk mencapai hasil yang serupa:

package main

import (
    "bytes"
    "fmt"
    "io"
    "net"
)

func main() {
    // Establish a connection with a remote host.
    conn, err := net.Dial("tcp", "google.com:80")
    if err != nil {
        fmt.Println("dial error:", err)
        return
    }
    defer conn.Close()

    // Write a request to the connection.
    fmt.Fprintf(conn, "GET / HTTP/1.0\r\n\r\n")

    // Create a buffer to store the received data.
    var buf bytes.Buffer

    // Copy the data from the connection into the buffer.
    io.Copy(&buf, conn)

    fmt.Println("total size:", buf.Len())
}

Pendekatan ini menggunakan penimbal yang mengembang secara automatik mengikut keperluan dan menyediakan penyelesaian yang lebih bersih dan ringkas.

Atas ialah kandungan terperinci Bagaimana untuk Membaca Data Panjang Berubah daripada Sambungan Rangkaian dalam Go?. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!