client auto reconnect

wsboyxb · May 23, 2016 · a3f82e4 · a3f82e4
1 parent 088c6d4
commit a3f82e4
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/network/tcp_client.go b/network/tcp_client.go
@@ -13,6 +13,7 @@ type TCPClient struct {
 	ConnNum         int
 	ConnectInterval time.Duration
 	PendingWriteNum int
+	AutoReconnect   bool
 	NewAgent        func(*TCPConn) Agent
 	conns           ConnSet
 	wg              sync.WaitGroup
@@ -84,6 +85,7 @@ func (client *TCPClient) dial() net.Conn {
 func (client *TCPClient) connect() {
 	defer client.wg.Done()
 
+reconnect:
 	conn := client.dial()
 	if conn == nil {
 		return
@@ -108,6 +110,10 @@ func (client *TCPClient) connect() {
 	delete(client.conns, conn)
 	client.Unlock()
 	agent.OnClose()
+
+	if client.AutoReconnect {
+		goto reconnect
+	}
 }
 
 func (client *TCPClient) Close() {

diff --git a/network/ws_client.go b/network/ws_client.go
@@ -15,6 +15,7 @@ type WSClient struct {
 	PendingWriteNum  int
 	MaxMsgLen        uint32
 	HandshakeTimeout time.Duration
+	AutoReconnect    bool
 	NewAgent         func(*WSConn) Agent
 	dialer           websocket.Dialer
 	conns            WebsocketConnSet
@@ -85,6 +86,7 @@ func (client *WSClient) dial() *websocket.Conn {
 func (client *WSClient) connect() {
 	defer client.wg.Done()
 
+reconnect:
 	conn := client.dial()
 	if conn == nil {
 		return
@@ -110,6 +112,10 @@ func (client *WSClient) connect() {
 	delete(client.conns, conn)
 	client.Unlock()
 	agent.OnClose()
+
+	if client.AutoReconnect {
+		goto reconnect
+	}
 }
 
 func (client *WSClient) Close() {