Change rwc to rwc with addresses

wrappers/timeoutrwc.go

@@ -1,31 +1,34 @@
 package wrappers
 
 import (
-	"io"
 	"net"
 	"time"
 
 	"github.com/9seconds/mtg/config"
 )
 
-type TimeoutReadWriteCloser struct {
+type TimeoutReadWriteCloserWithAddr struct {
 	conn net.Conn
 }
 
-func (t *TimeoutReadWriteCloser) Read(p []byte) (int, error) {
+func (t *TimeoutReadWriteCloserWithAddr) Read(p []byte) (int, error) {
 	t.conn.SetReadDeadline(time.Now().Add(config.TimeoutRead))
 	return t.conn.Read(p)
 }
 
-func (t *TimeoutReadWriteCloser) Write(p []byte) (int, error) {
+func (t *TimeoutReadWriteCloserWithAddr) Write(p []byte) (int, error) {
 	t.conn.SetWriteDeadline(time.Now().Add(config.TimeoutWrite))
 	return t.conn.Write(p)
 }
 
-func (t *TimeoutReadWriteCloser) Close() error {
+func (t *TimeoutReadWriteCloserWithAddr) Close() error {
 	return t.conn.Close()
 }
 
-func NewTimeoutRWC(conn net.Conn) io.ReadWriteCloser {
-	return &TimeoutReadWriteCloser{conn}
+func (t *TimeoutReadWriteCloserWithAddr) Addr() *net.TCPAddr {
+	return t.conn.RemoteAddr().(*net.TCPAddr)
+}
+
+func NewTimeoutRWC(conn net.Conn) ReadWriteCloserWithAddr {
+	return &TimeoutReadWriteCloserWithAddr{conn}
 }