操作系統： CentOS 6.9_x64
go語言版本： 1.8.3
問題描述
現有一個tcp客戶端程序，需定期從服務器取數據，但由于種種原因（網絡不穩定等）需要自動重連。
測試服務器示例代碼：
/*tcp server for test*/package mainimport ( "fmt" "net" "os" "strings" "time")func checkError(err error) { if err != nil { fmt.Println(err) os.Exit(1) }}func handleClient(conn net.Conn) { conn.SetReadDeadline(time.Now().Add(3 * time.Minute)) request := make([]byte,1024) defer conn.Close() for { recv_len,err := conn.Read(request) if err != nil { fmt.Println(err) break } if recv_len == 0 { break } recvData := strings.TrimSpace(string(request[:recv_len])) fmt.Println("recv_len : ",recv_len) fmt.Println("recv_data : " + recvData) daytime := time.Now().String() conn.Write([]byte(daytime + "/n")) request = make([]byte,1024) }}func main() { bindInfo := ":12345" tcpAddr,err := net.ResolveTCPAddr("tcp4",bindInfo) checkError(err) listener,err := net.ListenTCP("tcp",tcpAddr) checkError(err) for { cc,err := listener.Accept() if err != nil { continue } go handleClient(cc) }}解決方案
/*tcp client with reconnect*/package mainimport ( "net" "fmt" "bufio" "time")func doTask(conn net.Conn) { for { fmt.Fprintf(conn,"test msg/n") msg,err := bufio.NewReader(conn).ReadString('/n') if err != nil { fmt.Println("recv data error") break }else{ fmt.Println("recv msg : ",msg) } time.Sleep(1 * time.Second) }}func main() { hostInfo := "127.0.0.1:12345" for { conn,err := net.Dial("tcp",hostInfo) fmt.Print("connect (",hostInfo) if err != nil { fmt.Println(") fail") }else{ fmt.Println(") ok") defer conn.Close() doTask(conn) } time.Sleep(3 * time.Second) }}運行效果：
[root@local t1]# ./tcpClient1connect (127.0.0.1:12345) okrecv msg : 2017-06-12 21:10:32.110977137 +0800 CSTrecv msg : 2017-06-12 21:10:33.111868746 +0800 CSTrecv data errorconnect (127.0.0.1:12345) failconnect (127.0.0.1:12345) failconnect (127.0.0.1:12345) okrecv msg : 2017-06-12 21:10:43.117203432 +0800 CSTrecv msg : 2017-06-12 21:10:44.11853427 +0800 CST
討論
這里只是個簡單的示例代碼，實現了tcp自動重連。
好，就這些了，希望對你有幫助。
以上這篇golang之tcp自動重連實現方法就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持VEVB武林網。
新聞熱點
疑難解答
