在Golang中实现TCP客户端之间的数据中继

偷偷努力，悄无声息地变强，然后惊艳所有人！哈哈，小伙伴们又来学习啦~今天我将给大家介绍《在Golang中实现TCP客户端之间的数据中继》，这篇文章主要会讲到等等知识点，不知道大家对其都有多少了解，下面我们就一起来看一吧！当然，非常希望大家能多多评论，给出合理的建议，我们一起学习，一起进步！

我正在编写一个 tcp 服务器，它同时接受来自移动设备和一些 wifi 设备 (iot) 的多个连接。连接一旦建立就需要维持，如果没有收到心跳，则超时 30 秒。所以它类似于以下内容：

// clientsMap map[string] conn

func someFunction() {
    conn, err := s.listener.Accept()
    // I store the conn in clientsMap
    // so I can access it, for brevity not 
    // shown here, then:
    go serve(connn)
}

func serve(conn net.Conn) {
    timeoutDuration := 30 * time.Second
    conn.SetReadDeadline(time.Now().Add(timeoutDuration))
    for {
        msgBuffer := make([]byte, 2048)
        msgBufferLen, err := conn.Read(msgBuffer)
        // do something with the stuff
    }
}

因此，每个客户端都有一个 goroutine。每个客户端一旦连接到服务器，就会等待读取。然后服务器处理读取的内容。

问题是我有时需要从一个客户端读取数据，然后将数据传递到另一个客户端（在移动设备和 wifi 设备之间）。我已将连接存储在 clientsmap 中。所以我可以随时访问它。但是由于每个客户端都由一个 goroutine 处理，我是否应该使用通道将数据从一个客户端传递到另一个客户端？但是如果 goroutine 被阻塞等待挂起的读取，我如何让它也等待来自通道的数据？或者我应该从clientsmap中获取对方的连接并写入它？

解决方案

The documentation for net.Conn明确指出：

多个 goroutine 可以同时调用 conn 上的方法。

所以是的，只需写入连接就可以了。您应该注意对每条要发送的消息发出一次 write 调用。如果您多次调用 write，则可能会面临来自不同移动设备的消息交错的风险。这意味着直接调用 write，而不是通过其他 api（换句话说，不包装连接）。例如，以下内容不是安全的：

json.NewEncoder(conn).Encode(myValue) // use json.Marshal(myValue) instead
io.Copy(conn, src)                    // use io.ReadAll(src) instead

今天关于《在Golang中实现TCP客户端之间的数据中继》的内容就介绍到这里了，是不是学起来一目了然！想要了解更多关于的内容请关注golang学习网公众号！