增加链路属性

demo/Server.go

@@ -43,11 +43,27 @@ func DoConnectionBegin(conn iface.IConnection) {
 	if err := conn.SendMsg(202, []byte("DoConnection Begin")); err != nil {
 		fmt.Println(err)
 	}
+
+	fmt.Println("set conn name")
+
+	conn.SetProperty("Name", "LiuLiGames")
+	conn.SetProperty("GitHub", "https://github.com/liuligames")
+	conn.SetProperty("Blob", "https://liuligames.com")
 }
 
 func DoConnectionLost(conn iface.IConnection) {
 	fmt.Println("DoConnectionLast is Called ...")
 	fmt.Println("conn id = ", conn.GetConnID(), "is lost")
+
+	if name, err := conn.GetProperty("Name"); err == nil {
+		fmt.Println("Name = ", name)
+	}
+	if github, err := conn.GetProperty("GitHub"); err == nil {
+		fmt.Println("GitHub = ", github)
+	}
+	if blob, err := conn.GetProperty("Blob"); err == nil {
+		fmt.Println("Blob = ", blob)
+	}
 }
 
 func main() {

iface/IConnection.go

@@ -14,6 +14,12 @@ type IConnection interface {
 	GetRemoteAddr() net.Addr
 
 	SendMsg(msgId uint32, data []byte) error
+
+	SetProperty(key string, value interface{})
+
+	GetProperty(key string) (interface{}, error)
+
+	RemoveProperty(key string)
 }
 
 type HandleFunc func(*net.TCPConn, []byte, int) error

net/Connection.go

@@ -5,18 +5,21 @@ import (
 	"fmt"
 	"io"
 	"net"
+	"sync"
 	"wukong/iface"
 	"wukong/utils"
 )
 
 type Connection struct {
-	TcpServer  iface.IServer
-	Conn       *net.TCPConn
-	ConnID     uint32
-	isClosed   bool
-	ExitChan   chan bool
-	msgChan    chan []byte
-	MsgHandler iface.IMsgHandler
+	TcpServer    iface.IServer
+	Conn         *net.TCPConn
+	ConnID       uint32
+	isClosed     bool
+	ExitChan     chan bool
+	msgChan      chan []byte
+	MsgHandler   iface.IMsgHandler
+	property     map[string]interface{}
+	propertyLock sync.RWMutex
 }
 
 func NewConnection(tcpServer iface.IServer, conn *net.TCPConn, connID uint32, msgHandler iface.IMsgHandler) *Connection {
@@ -28,6 +31,7 @@ func NewConnection(tcpServer iface.IServer, conn *net.TCPConn, connID uint32, ms
 		isClosed:   false,
 		msgChan:    make(chan []byte),
 		ExitChan:   make(chan bool, 1),
+		property:   make(map[string]interface{}),
 	}
 	c.TcpServer.GetConnManager().Add(c)
 	return c
@@ -161,3 +165,27 @@ func (c *Connection) SendMsg(msgId uint32, data []byte) error {
 
 	return nil
 }
+
+func (c *Connection) SetProperty(key string, value interface{}) {
+	c.propertyLock.Lock()
+	defer c.propertyLock.Unlock()
+
+	c.property[key] = value
+}
+
+func (c *Connection) GetProperty(key string) (interface{}, error) {
+	c.propertyLock.RLock()
+	defer c.propertyLock.RUnlock()
+	if value, ok := c.property[key]; ok {
+		return value, nil
+	} else {
+		return nil, errors.New("no property found")
+	}
+}
+
+func (c *Connection) RemoveProperty(key string) {
+	c.propertyLock.Lock()
+	defer c.propertyLock.Unlock()
+
+	delete(c.property, key)
+}