解决客户端内存泄漏问题

client.go

@@ -76,13 +76,23 @@ func (c *TRPClient) process(conn net.Conn) error {
 		case "vkey":
 			log.Fatal("vkey不正确,请检查配置文件")
 		case "sign":
+			c.deal(conn)
+		case "msg0":
+			log.Println("服务端返回错误。")
+		default:
+			log.Println("无法解析该错误。")
+		}
+	}
+	return nil
+}
+func (c *TRPClient) deal(conn net.Conn) error {
+	val := make([]byte, 4)
 	_, err := conn.Read(val)
 	nlen := binary.LittleEndian.Uint32(val)
 	log.Println("收到服务端数据，长度：", nlen)
 	if nlen <= 0 {
 		log.Println("数据长度错误。")
 		c.werror(conn)
-				continue
 	}
 	raw := make([]byte, nlen)
 	n, err := conn.Read(raw)
@@ -92,13 +102,11 @@ func (c *TRPClient) process(conn net.Conn) error {
 	if n != int(nlen) {
 		log.Printf("读取服务端数据长度错误，已经读取%dbyte，总长度%d字节\n", n, nlen)
 		c.werror(conn)
-				continue
 	}
 	req, err := DecodeRequest(raw)
 	if err != nil {
 		log.Println("DecodeRequest错误：", err)
 		c.werror(conn)
-				continue
 	}
 	rawQuery := ""
 	if req.URL.RawQuery != "" {
@@ -114,7 +122,6 @@ func (c *TRPClient) process(conn net.Conn) error {
 	if err != nil && !disRedirect {
 		log.Println("请求本地客户端错误：", err)
 		c.werror(conn)
-				continue
 	}
 	if !disRedirect {
 		defer resp.Body.Close()
@@ -126,7 +133,6 @@ func (c *TRPClient) process(conn net.Conn) error {
 	if err != nil {
 		log.Println("EncodeResponse错误：", err)
 		c.werror(conn)
-				continue
 	}
 	n, err = conn.Write(respBytes)
 	if err != nil {
@@ -137,11 +143,5 @@ func (c *TRPClient) process(conn net.Conn) error {
 	} else {
 		log.Printf("本次请求成功完成，共发送：%dbyte\n", n)
 	}
-		case "msg0":
-			log.Println("服务端返回错误。")
-		default:
-			log.Println("无法解析该错误。")
-		}
-	}
 	return nil
 }