再谈 Golang 的 TCP 客户端重新连接问题：是否明智使用递归函数？

对于 TCP 客户端在连接失败后是否应该使用递归函数重新连接的问题，存在争议。递归函数可能会导致内存耗尽，尤其是在连接尝试在长时间内持续进行的情况下。因此，建议避免使用递归函数，而采用更稳健的方法，例如使用循环或定时器来定期尝试重新连接。

我有这个有效的 tcp 客户端代码。当它在 tcp 连接上写入或读取失败时，它会使用递归函数 tcpreconnect() 创建一个新连接。

这安全吗还是会填满 ram？它可能会在几天（周末或节假日）内尝试重新连接。此代码是监视工业机器状态的驱动程序的一部分。

也许这个问题有更好的解决方案。我没能找到。

ps：我不喜欢投票

package main

import (
    "fmt"
    "net"
    "time"
)

var pollTime = 1000 //ms
var host = "127.0.0.1"
var port = "11000"

func main() {
    finished := make(chan bool)
    go Driver()
    <-finished
}

func tcpReconnect() net.Conn {
    newConn, err := net.Dial("tcp", host+":"+port)
    if err != nil {
        fmt.Println("Failed to reconnect:", err.Error())
        time.Sleep(time.Millisecond * time.Duration(2000))
        newConn = tcpReconnect()
    }
    return newConn
}

func Driver() {
    var conn net.Conn
    conn, err := net.Dial("tcp", host+":"+port)
    if err != nil {
        fmt.Println("Failed to initialize Connection, trying to reconnect:", err.Error())
        conn = tcpReconnect()
    }

    for {
        _, err = conn.Write([]byte("11|5546|STATUS" + "\r\n"))
        if err != nil {
            println("Write to server failed:", err.Error())
            println("Trying reset the connection...")
            conn = tcpReconnect()
        }

        var replyBuffer = make([]byte, 256)
        _, err = conn.Read(replyBuffer)
        if err != nil {
            println("Read from server failed:", err.Error())
            println("Trying reset the connection...")
            conn = tcpReConnect()
        }

        var reply string
        for i, val := range replyBuffer {
            if val == 13 { //13 is CR and marks the end of the message
                reply = string(replyBuffer[:i])
                break
            }
        }

        fmt.Printf("reply from server=%s\n", reply)
        time.Sleep(time.Millisecond * time.Duration(pollTime))
    }
}

解决方案

这就是我想出来的。致谢@tkausl 和@thundercat

func Driver() {
    for {
        conn, err := net.Dial("tcp", host+":"+port)
        if err != nil {
            fmt.Println("Failed to connect:", err.Error())
            fmt.Println("Trying reset the connection...")
            time.Sleep(time.Millisecond * time.Duration(2000))
        } else {
            for {

                _, err = conn.Write([]byte("11|5546|STATUS" + "\r\n"))
                if err != nil {
                    fmt.Println("Write to server failed:", err.Error())
                    fmt.Println("Trying reset the connection...")
                    break
                }

                var replyBuffer = make([]byte, 256)
                _, err = conn.Read(replyBuffer)
                if err != nil {
                    fmt.Println("Read from server failed:", err.Error())
                    fmt.Println("Trying reset the connection...")
                    break
                }

                var reply string
                for i, val := range replyBuffer {
                    if val == 13 { //13 is CR and marks the end of the message
                        reply = string(replyBuffer[:i])
                        break
                    }
                }

                fmt.Printf("reply from server=_%s_\n", reply)
                time.Sleep(time.Millisecond * time.Duration(pollTime))
            }
        }
    }
}

本篇关于《再谈 Golang 的 TCP 客户端重新连接问题：是否明智使用递归函数？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！