From a699ff4f66e264d88faa93c0aeee732ca3d9aa1f Mon Sep 17 00:00:00 2001
From: liuxiaolong <736321739@qq.com>
Date: 星期五, 11 十月 2019 10:01:30 +0800
Subject: [PATCH] add queryResponse timeout
---
transport.go | 15 ++++++++++-----
1 files changed, 10 insertions(+), 5 deletions(-)
diff --git a/transport.go b/transport.go
index 3b2c634..c7032ba 100644
--- a/transport.go
+++ b/transport.go
@@ -27,14 +27,17 @@
}
func RawReceiveTcpMsg() {
- var tcpAddr *net.TCPAddr
- tcpAddr,_ = net.ResolveTCPAddr("tcp","127.0.0.1:"+strconv.Itoa(TcpTransportPort))
- listener,_ := net.ListenTCP("tcp",tcpAddr)
+ tcpAddr := "0.0.0.0:"+strconv.Itoa(TcpTransportPort)
+ listener,err := net.Listen("tcp",tcpAddr)
+ if err != nil {
+ logger.Debug("RawReceive server listen err:",err)
+ return
+ }
defer listener.Close()
for{
- conn,err := listener.AcceptTCP()
+ conn,err := listener.Accept()
if err!=nil {
logger.Debug("listener.Accept err:", err)
continue
@@ -44,7 +47,9 @@
}
}
-func readStream(conn *net.TCPConn) {
+func readStream(conn net.Conn) {
+ defer conn.Close()
+
data := make([]byte,0)
buf := make([]byte, 4096)
for {
--
Gitblit v1.8.0