添加链接属性配置功能

ziface/iconnection.go

@@ -8,16 +8,26 @@ type IConnection interface {
 	Start()
 	//停止连接，结束当前连接状态M
 	Stop()
+
 	//从当前连接获取原始的socket TCPConn
 	GetTCPConnection() *net.TCPConn
 	//获取当前连接ID
 	GetConnID() uint32
 	//获取远程客户端地址信息
 	RemoteAddr() net.Addr
+
 	//直接将Message数据发送数据给远程的TCP客户端(无缓冲)
 	SendMsg(msgId uint32, data []byte) error
 	//直接将Message数据发送给远程的TCP客户端(有缓冲)
 	SendBuffMsg(msgId uint32, data []byte) error
+
+	//设置链接属性
+	SetProperty(key string, value interface{})
+	//获取链接属性
+	GetProperty(key string)(interface{}, error)
+	//移除链接属性
+	RemoveProperty(key string)
+
 }
 
 

znet/connection.go

@@ -5,6 +5,7 @@ import (
 	"fmt"
 	"io"
 	"net"
+	"sync"
 	"zinx/utils"
 	"zinx/ziface"
 )
@@ -26,21 +27,26 @@ type Connection struct {
 	msgChan chan []byte
 	//有关冲管道，用于读、写两个goroutine之间的消息通信
 	msgBuffChan chan []byte
-}
 
+	//链接属性
+	property     map[string]interface{}
+	//保护链接属性修改的锁
+	propertyLock sync.RWMutex
+}
 
 //创建连接的方法
-func NewConntion(server ziface.IServer, conn *net.TCPConn, connID uint32, msgHandler ziface.IMsgHandle) *Connection{
+func NewConntion(server ziface.IServer, conn *net.TCPConn, connID uint32, msgHandler ziface.IMsgHandle) *Connection {
 	//初始化Conn属性
 	c := &Connection{
-		TcpServer:server,
+		TcpServer:    server,
 		Conn:         conn,
 		ConnID:       connID,
 		isClosed:     false,
 		MsgHandler:   msgHandler,
 		ExitBuffChan: make(chan bool, 1),
-		msgChan:make(chan []byte),
-		msgBuffChan:make(chan []byte, utils.GlobalObject.MaxMsgChanLen),
+		msgChan:      make(chan []byte),
+		msgBuffChan:  make(chan []byte, utils.GlobalObject.MaxMsgChanLen),
+		property:     make(map[string]interface{}),
 	}
 
 	//将新创建的Conn添加到链接管理中
@@ -50,8 +56,8 @@ func NewConntion(server ziface.IServer, conn *net.TCPConn, connID uint32, msgHan
 
 /*
 	写消息Goroutine， 用户将数据发送给客户端
- */
- func (c *Connection) StartWriter() {
+*/
+func (c *Connection) StartWriter() {
 	fmt.Println("[Writer Goroutine is running]")
 	defer fmt.Println(c.RemoteAddr().String(), "[conn Writer exit!]")
 
@@ -63,7 +69,7 @@ func NewConntion(server ziface.IServer, conn *net.TCPConn, connID uint32, msgHan
 				fmt.Println("Send Data error:, ", err, " Conn Writer exit")
 				return
 			}
- 			case data, ok:= <-c.msgBuffChan:
+		case data, ok := <-c.msgBuffChan:
 			if ok {
 				//有数据要写给客户端
 				if _, err := c.Conn.Write(data); err != nil {
@@ -78,12 +84,11 @@ func NewConntion(server ziface.IServer, conn *net.TCPConn, connID uint32, msgHan
 			return
 		}
 	}
- }
-
+}
 
 /*
 	读消息Goroutine，用于从客户端中读取数据
- */
+*/
 func (c *Connection) StartReader() {
 	fmt.Println("[Reader Goroutine is running]")
 	defer fmt.Println(c.RemoteAddr().String(), "[conn Reader exit!]")
@@ -101,7 +106,7 @@ func (c *Connection) StartReader() {
 		}
 
 		//拆包，得到msgid 和 datalen 放在msg中
-		msg , err := dp.Unpack(headData)
+		msg, err := dp.Unpack(headData)
 		if err != nil {
 			fmt.Println("unpack error ", err)
 			break
@@ -120,8 +125,8 @@ func (c *Connection) StartReader() {
 
 		//得到当前客户端请求的Request数据
 		req := Request{
-			conn:c,
-			msg:msg,
+			conn: c,
+			msg:  msg,
 		}
 
 		if utils.GlobalObject.WorkerPoolSize > 0 {
@@ -175,7 +180,7 @@ func (c *Connection) GetTCPConnection() *net.TCPConn {
 }
 
 //获取当前连接ID
-func (c *Connection) GetConnID() uint32{
+func (c *Connection) GetConnID() uint32 {
 	return c.ConnID
 }
 
@@ -220,3 +225,31 @@ func (c *Connection) SendBuffMsg(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)
+}