add read timeout when first reading data

xindong · Apr 5, 2016 · f877a7d · f877a7d
1 parent 227c325
commit f877a7d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/main.go b/main.go
@@ -141,6 +141,8 @@ func handleConn(c net.Conn) {
 		}
 	}()
 
+	c.SetReadDeadline(time.Now().Add(_ConnReadTimeout))
+
 	rdr := bufio.NewReader(c)
 
 	addr, err := handleBinaryHdr(rdr, c)