Close connection more aggressively.

common/net/timed_io.go

@@ -0,0 +1,30 @@
+package net
+
+import (
+	"net"
+	"time"
+)
+
+var (
+	emptyTime time.Time
+)
+
+type TimeOutReader struct {
+	timeout    int
+	connection net.Conn
+}
+
+func NewTimeOutReader(timeout int, connection net.Conn) *TimeOutReader {
+	return &TimeOutReader{
+		timeout:    timeout,
+		connection: connection,
+	}
+}
+
+func (reader *TimeOutReader) Read(p []byte) (n int, err error) {
+	deadline := time.Duration(reader.timeout) * time.Second
+	reader.connection.SetReadDeadline(time.Now().Add(deadline))
+	n, err = reader.connection.Read(p)
+	reader.connection.SetReadDeadline(emptyTime)
+	return
+}