Bugfix Handling if the TCP connection was closed.

Author: leviathan0992

client/util/client.go

@@ -28,7 +28,7 @@ func (c *client) Run() {
 		if len(connectionPool) < 10 {
 			srvConn, err := c.DialSrv()
 			if err != nil {
-				log.Println(err)
+				log.Printf("Connect to the server %s:%d failed: %s. \n", c.srvAddr.IP.String(), c.srvAddr.Port, err)
 				continue
 			}
 			log.Printf("Connect to the server %s:%d successful.\n", c.srvAddr.IP.String(), c.srvAddr.Port)

service/service.go

@@ -1,7 +1,6 @@
 package service
 
 import (
-	"io"
 	"net"
 )
 
@@ -35,11 +34,7 @@ func (s *Service) TransferToTCP(cliConn net.Conn, dstConn *net.TCPConn) error {
 		if nRead > 0 {
 			errWrite := s.TCPWrite(dstConn, buf[0:nRead])
 			if err != nil {
-				if errWrite == io.EOF {
-					return nil
-				} else {
-					return errWrite
-				}
+				return errWrite
 			}
 		}
 	}