Голанг: Рекурсивная функция для повторного подключения TCP-клиента ... Плохая идея? - PullRequest
0 голосов
/ 21 февраля 2019

У меня есть этот рабочий код клиента TCP.Когда не удается выполнить запись или чтение по TCP-соединению, создается новое соединение с рекурсивной функцией tcpReconnect ().

Безопасно ли это или заполнит ОЗУ? ЭтоВозможно, что он пытается восстановить соединение в течение нескольких дней (выходные или праздничные дни).Этот код является частью драйвера, который контролирует состояние промышленной машины.

Возможно, есть лучшее решение для этой проблемы.Я не смог его найти.

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))
    }
}

1 Ответ

0 голосов
/ 21 февраля 2019

Это то, что я придумал.Кредиты отправляются на @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))
            }
        }
    }
}
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...