UserData for a agent

landspace · Jul 6, 2015 · d88e05e · d88e05e
1 parent b418216
commit d88e05e
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/gate/agent.go b/gate/agent.go
@@ -3,4 +3,6 @@ package gate
 type Agent interface {
 	WriteMsg(msg interface{})
 	Close()
+	UserData() interface{}
+	SetUserData(data interface{})
 }
diff --git a/gate/tcp_gate.go b/gate/tcp_gate.go
@@ -52,8 +52,9 @@ func (gate *TCPGate) Run(closeSig chan bool) {
 func (gate *TCPGate) OnDestroy() {}
 
 type TCPAgent struct {
-	conn *network.TCPConn
-	gate *TCPGate
+	conn     *network.TCPConn
+	gate     *TCPGate
+	userData interface{}
 }
 
 func (a *TCPAgent) Run() {
@@ -124,3 +125,11 @@ func (a *TCPAgent) WriteMsg(msg interface{}) {
 func (a *TCPAgent) Close() {
 	a.conn.Close()
 }
+
+func (a *TCPAgent) UserData() interface{} {
+	return a.userData
+}
+
+func (a *TCPAgent) SetUserData(data interface{}) {
+	a.userData = data
+}