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

Question

在花了很多时间之后。我找不到读取来自 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)

Answer 1

如果您唯一的选择是读取行直到超时，您可以在第一次读取完成后在连接上设置读取截止日期。然后，您可以拦截超时错误，并将其转换为 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)

}

Answer 2

如果条件是收到第一行后3秒超时，解决办法是收到第一行后3秒关闭socket。

var firstLineReceived bool

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

    if !firstLineReceived {

        firstLineReceived = true

        go func(){

            time.Sleep(3*time.Second)

            conn.Close()

        }()

    }

}

log.Println(message)