我确实有一个进程从第二个本地进程接收数据。 我需要测试连接错误是否得到妥善处理，以及连接是否自动重新连接并在断开连接后继续接收数据。

为此，我试图使其突然断开连接或将TCP连接置于单元测试的错误状态。
如在此问题中看到的，以检查连接是否关闭，我正在检查数据是否到来并测试它是否返回错误。
我不确定如何：

  • 不良地关闭连接
  • 使它处于错误状态

这是我的数据接收器的本质：

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
    import (
       "bufio"
       "encoding/json"
       "fmt"
       "io"
       "net"
    )

    type Message struct {
        ID string `json:"id"`
    }

    func ReceiveData(listener Listener) {
        var tcpConn net.Conn
        var addr string ="127.0.0.1:9999"
        tcpConn, err := net.Dial("tcp", addr)
        socketReader := bufio.NewReader(tcpConn)
        decoder := json.NewDecoder(socketReader)

        for {
            var msg Message
            if err := decoder.Decode(&msg); err == io.EOF {
                listener.ProcessUpdate(Message{}, fmt.Errorf("Received EOF"), nil)
                tcpConn = nil
                return
            } else if err != nil {
                listener.ProcessUpdate(Message{}, nil, fmt.Errorf("Error decoding message: %s", err.Error()))
                tcpConn = nil
                return
        }

        // process message
        _ = msg

        // Test disconnection
        // This does not disconnect:
        // tcpConn = nil
        // This does but gracefully:
        // tcpConn.Close()
    }

我没有嘲笑TCP连接，因为我想尝试使用真正的数据生成器。 如果需要的话，我会看的。

  • 确切地说，"不恰当地"关闭连接是什么意思？
  • 意外断开连接，人们拔下电缆，另一侧的电源损耗，tcp消息格式错误。 我怀疑其中大多数都会产生超时，我不知道tcpConn.Read()可能产生的所有错误。
  • 如果您希望远程端模拟断开连接，例如拉电缆或断电，请确保已禁用keepalive并停止响应。 如果要以其他各种方式断开连接，则需要在网络上的较低级别进行连接。

一种解决方案是为TCP连接本身设置截止日期：

1
tcpConn.SetDeadline(time.Now())

稍后，这将触发超时错误，该错误可能由以下原因捕获：

1
2
3
4
5
6
7
8
err := decoder.Decode(&msg);
if err != nil {
    if neterr, ok := err.(net.Error); ok && neterr.Timeout() {
        fmt.Errorf("TCP timeout : %s", err.Error())
    } else {
        fmt.Errorf("Received error decoding message: %s", err.Error())
    }
}