使用Golang net.Conn.Read读取整个数据

Question

所以我在Go中构建一个网络应用程序,我已经看到它Conn.Read读入了一个有限的字节数组,我创建了它make([]byte, 2048),现在的问题是我不知道内容的确切长度,所以它可能是太多或不够.
我的问题是如何才能读取确切的数据量.我想我必须使用bufio,但我不确定.

Answer 1

你可以使用这个ioutil.ReadAll功能:

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

func whois(domain, server string) ([]byte, error) {
    conn, err := net.Dial("tcp", server+":43")
    if err != nil {
        return nil, err
    }
    defer conn.Close()

    fmt.Fprintf(conn, "%s\r\n", domain)
    return ioutil.ReadAll(conn)
}

Answer 2

它在很大程度上取决于你想要做什么,以及你期望什么样的数据,例如,如果你只想阅读直到EOF,你可以使用这样的东西:

func main() {
    conn, err := net.Dial("tcp", "google.com:80")
    if err != nil {
        fmt.Println("dial error:", err)
        return
    }
    defer conn.Close()
    fmt.Fprintf(conn, "GET / HTTP/1.0\r\n\r\n")

    buf := make([]byte, 0, 4096) // big buffer
    tmp := make([]byte, 256)     // using small tmo buffer for demonstrating
    for {
        n, err := conn.Read(tmp)
        if err != nil {
            if err != io.EOF {
                fmt.Println("read error:", err)
            }
            break
        }
        //fmt.Println("got", n, "bytes.")
        buf = append(buf, tmp[:n]...)

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

//编辑:为了完整起见和@ fabrizioM的伟大建议,这完全忽略了我的想法:

func main() {
    conn, err := net.Dial("tcp", "google.com:80")
    if err != nil {
        fmt.Println("dial error:", err)
        return
    }
    defer conn.Close()
    fmt.Fprintf(conn, "GET / HTTP/1.0\r\n\r\n")
    var buf bytes.Buffer
    io.Copy(&buf, conn)
    fmt.Println("total size:", buf.Len())
}

Answer 3

您可以像这样读取数据：

// import net/textproto
import ("net/textproto", ...)

....

reader := bufio.NewReader(Conn)
tp := textproto.NewReader(reader)

defer Conn.Close()

for {
    // read one line (ended with \n or \r\n)
    line, _ := tp.ReadLine()
    // do something with data here, concat, handle and etc... 
}
....