如何从 TCP 套接字服务器读取所有数据并随后执行操作

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《如何从 TCP 套接字服务器读取所有数据并随后执行操作》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

问题内容

花了很多时间在这上面。我找不到一种方法来读取来自 tcp 套接字服务器的所有数据，然后进行操作，因为我找不到打破循环的方法。

套接字服务器发送包含许多以“\n”结尾的行的文本。客户端应该能够读取所有这些行并使用所有数据发出 post 请求，但循环总是挂起并且没有办法打破它。然后，它将继续等待更多数据，因此停止条件可能是三秒超时。

我尝试了不同的解决方案（scanner、readstring、readline、readall），但它总是挂起并且循环永远不会完成。

代码中的最后一行永远不会被打印。

conn, err := net.Dial("tcp", "127.0.0.1:15000")
 reader := bufio.NewReader(conn)
 message := ""
 for {
     line, err := reader.ReadString('\n')
     if err == io.EOF {
         break
     }
     message += line
 }
 log.Println(message)

解决方案

如果您唯一的选择是读取行直到超时，您可以在第一次读取完成后在连接上设置读取截止时间。然后，您可以拦截超时错误，并将其转换为 eof，以便缓冲读取器正确解释您的意图。

type timeoutReader struct {
    net.Conn
    once sync.Once
}

func (r *timeoutReader) Read(b []byte) (int, error) {
    n, err := r.Conn.Read(b)

    // Set a read deadline only after the first Read completes
    r.once.Do(func() {
        r.Conn.SetReadDeadline(time.Now().Add(3 * time.Second))
    })

    // If we got a timeout, treat it as an io.EOF so the bufio.Scanner handles
    // the error as if it was the normal end of the stream.
    var netErr net.Error
    if errors.As(err, &netErr) && netErr.Timeout() {
        return n, io.EOF
    }
    return n, err
}

func main() {
    conn, err := net.Dial("tcp", "127.0.0.1:15000")
    if err != nil {
        log.Fatal(err)
    }
    scanner := bufio.NewScanner(&timeoutReader{Conn: conn})
    message := ""
    for scanner.Scan() {
        message += scanner.Text()
    }
    if err := scanner.Err(); err != nil {
        log.Fatal(err)
    }
    log.Println(message)
}

好了，本文到此结束，带大家了解了《如何从 TCP 套接字服务器读取所有数据并随后执行操作》，希望本文对你有所帮助！关注编程网公众号，给大家分享更多Golang知识！