操作系統(tǒng)： CentOS 6.9_x64
go語(yǔ)言版本： 1.8.3
問(wèn)題描述
現(xiàn)有一個(gè)tcp客戶端程序，需定期從服務(wù)器取數(shù)據(jù)，但由于種種原因（網(wǎng)絡(luò)不穩(wěn)定等）需要自動(dòng)重連。
測(cè)試服務(wù)器示例代碼：
/*
tcp server for test
*/
package main
import (
"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 main
import (
"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)
}
}
運(yùn)行效果：
[root@local t1]# ./tcpClient1
connect (127.0.0.1:12345) ok
recv msg : 2017-06-12 21:10:32.110977137 +0800 CST
recv msg : 2017-06-12 21:10:33.111868746 +0800 CST
recv data error
connect (127.0.0.1:12345) fail
connect (127.0.0.1:12345) fail
connect (127.0.0.1:12345) ok
recv msg : 2017-06-12 21:10:43.117203432 +0800 CST
recv msg : 2017-06-12 21:10:44.11853427 +0800 CST
討論
這里只是個(gè)簡(jiǎn)單的示例代碼，實(shí)現(xiàn)了tcp自動(dòng)重連。
好，就這些了，希望對(duì)你有幫助。
以上這篇golang之tcp自動(dòng)重連實(shí)現(xiàn)方法就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
您可能感興趣的文章:- 基于go手動(dòng)寫個(gè)轉(zhuǎn)發(fā)代理服務(wù)的代碼實(shí)現(xiàn)
- 利用Golang實(shí)現(xiàn)TCP連接的雙向拷貝詳解
- golang 實(shí)現(xiàn)tcp轉(zhuǎn)發(fā)代理的方法
