wip

input/init.go

@@ -1,8 +1,6 @@
 package input
 
-import (
-	"io"
-)
+import "io"
 
 type Plugin interface {
 	Name() string

input/listen.go

@@ -141,7 +141,7 @@ func (l *Listener) acceptTcpConn(c net.Conn) {
 		}
 	}()
 
-	l.handler.Handle(c)
+	l.handler.Handle(NewTimeoutConn(c, 2*time.Minute))
 	c.Close()
 }
 

input/timeout_conn.go

@@ -0,0 +1,24 @@
+package input
+
+import (
+	"net"
+	"time"
+)
+
+// TimeoutConn automatically applies a read deadline on a conn upon every read
+type TimeoutConn struct {
+	conn    net.Conn
+	timeout time.Duration
+}
+
+func NewTimeoutConn(conn net.Conn, timeout time.Duration) TimeoutConn {
+	return TimeoutConn{
+		conn:    conn,
+		timeout: timeout,
+	}
+}
+
+func (t TimeoutConn) Read(p []byte) (n int, err error) {
+	t.conn.SetReadDeadline(time.Now().Add(2 * time.Minute))
+	return t.conn.Read(p)
+}