addr interface (gate/agent)

tangeping · Feb 15, 2017 · 2c74d13 · 2c74d13
1 parent e216b1f
commit 2c74d13
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/gate/agent.go b/gate/agent.go
@@ -1,7 +1,13 @@
 package gate
 
+import (
+	"net"
+)
+
 type Agent interface {
 	WriteMsg(msg interface{})
+	LocalAddr() net.Addr
+	RemoteAddr() net.Addr
 	Close()
 	Destroy()
 	UserData() interface{}

diff --git a/gate/gate.go b/gate/gate.go
@@ -4,6 +4,7 @@ import (
 	"github.com/name5566/leaf/chanrpc"
 	"github.com/name5566/leaf/log"
 	"github.com/name5566/leaf/network"
+	"net"
 	"reflect"
 	"time"
 )
@@ -130,6 +131,14 @@ func (a *agent) WriteMsg(msg interface{}) {
 	}
 }
 
+func (a *agent) LocalAddr() net.Addr {
+	return a.conn.LocalAddr()
+}
+
+func (a *agent) RemoteAddr() net.Addr {
+	return a.conn.RemoteAddr()
+}
+
 func (a *agent) Close() {
 	a.conn.Close()
 }