first commit

net/connMgr.go

@@ -0,0 +1,202 @@
+package net
+
+import (
+	"github.com/gorilla/websocket"
+	"log/slog"
+	"slgserver/server/slgserver/conn"
+	"slgserver/server/slgserver/pos"
+	"sync"
+	"sync/atomic"
+)
+
+var ConnMgr = Mgr{}
+var cid int64 = 0
+
+type Mgr struct {
+	cm sync.RWMutex
+	um sync.RWMutex
+	rm sync.RWMutex
+
+	connCache map[int64]WSConn
+	userCache map[int]WSConn
+	roleCache map[int]WSConn
+}
+
+func (this *Mgr) NewConn(wsSocket *websocket.Conn, needSecret bool) *ServerConn {
+	this.cm.Lock()
+	defer this.cm.Unlock()
+
+	id := atomic.AddInt64(&cid, 1)
+	if this.connCache == nil {
+		this.connCache = make(map[int64]WSConn)
+	}
+
+	if this.userCache == nil {
+		this.userCache = make(map[int]WSConn)
+	}
+
+	if this.roleCache == nil {
+		this.roleCache = make(map[int]WSConn)
+	}
+
+	c := NewServerConn(wsSocket, needSecret)
+	c.SetProperty("cid", id)
+	this.connCache[id] = c
+
+	return c
+}
+
+func (this *Mgr) UserLogin(conn WSConn, session string, uid int) {
+	this.um.Lock()
+	defer this.um.Unlock()
+
+	oldConn, ok := this.userCache[uid]
+	if ok {
+		if conn != oldConn {
+			slog.Warn("rob login",
+				"uid", uid,
+				"oldAddr", oldConn.Addr(),
+				"newAddr", conn.Addr())
+
+			//这里需要通知旧端被抢登录
+			oldConn.Push("robLogin", nil)
+		}
+	}
+	this.userCache[uid] = conn
+	conn.SetProperty("session", session)
+	conn.SetProperty("uid", uid)
+}
+
+func (this *Mgr) UserLogout(conn WSConn) {
+	this.removeUser(conn)
+}
+
+func (this *Mgr) removeUser(conn WSConn) {
+	this.um.Lock()
+	uid, err := conn.GetProperty("uid")
+	if err == nil {
+		//只删除自己的conn
+		id := uid.(int)
+		c, ok := this.userCache[id]
+		if ok && c == conn {
+			delete(this.userCache, id)
+		}
+	}
+	this.um.Unlock()
+
+	this.rm.Lock()
+	rid, err := conn.GetProperty("rid")
+	if err == nil {
+		//只删除自己的conn
+		id := rid.(int)
+		c, ok := this.roleCache[id]
+		if ok && c == conn {
+			delete(this.roleCache, id)
+		}
+	}
+	this.rm.Unlock()
+
+	conn.RemoveProperty("session")
+	conn.RemoveProperty("uid")
+	conn.RemoveProperty("role")
+	conn.RemoveProperty("rid")
+}
+
+func (this *Mgr) RoleEnter(conn WSConn, rid int) {
+	this.rm.Lock()
+	defer this.rm.Unlock()
+	conn.SetProperty("rid", rid)
+	this.roleCache[rid] = conn
+}
+
+func (this *Mgr) RemoveConn(conn WSConn) {
+	this.cm.Lock()
+	cid, err := conn.GetProperty("cid")
+	if err == nil {
+		delete(this.connCache, cid.(int64))
+		conn.RemoveProperty("cid")
+	}
+	this.cm.Unlock()
+
+	this.removeUser(conn)
+}
+
+func (this *Mgr) PushByRoleId(rid int, msgName string, data interface{}) bool {
+	if rid <= 0 {
+		return false
+	}
+	this.rm.Lock()
+	defer this.rm.Unlock()
+	conn, ok := this.roleCache[rid]
+	if ok {
+		conn.Push(msgName, data)
+		return true
+	} else {
+		return false
+	}
+}
+
+func (this *Mgr) Count() int {
+	this.cm.RLock()
+	defer this.cm.RUnlock()
+
+	return len(this.connCache)
+}
+
+func (this *Mgr) Push(pushSync conn.PushSync) {
+
+	proto := pushSync.ToProto()
+	belongRIds := pushSync.BelongToRId()
+	isCellView := pushSync.IsCellView()
+	x, y := pushSync.Position()
+	cells := make(map[int]int)
+
+	//推送给开始位置
+	if isCellView {
+		cellRIds := pos.RPMgr.GetCellRoleIds(x, y, 8, 6)
+		for _, rid := range cellRIds {
+			//是否能出现在视野
+			if can := pushSync.IsCanView(rid, x, y); can {
+				this.PushByRoleId(rid, pushSync.PushMsgName(), proto)
+				cells[rid] = rid
+			}
+		}
+	}
+
+	//推送给目标位置
+	tx, ty := pushSync.TPosition()
+	if tx >= 0 && ty >= 0 {
+		var cellRIds []int
+		if isCellView {
+			cellRIds = pos.RPMgr.GetCellRoleIds(tx, ty, 8, 6)
+		} else {
+			cellRIds = pos.RPMgr.GetCellRoleIds(tx, ty, 0, 0)
+		}
+
+		for _, rid := range cellRIds {
+			if _, ok := cells[rid]; !ok {
+				if can := pushSync.IsCanView(rid, tx, ty); can {
+					this.PushByRoleId(rid, pushSync.PushMsgName(), proto)
+					cells[rid] = rid
+				}
+			}
+		}
+	}
+
+	//推送给自己
+	for _, rid := range belongRIds {
+		if _, ok := cells[rid]; !ok {
+			this.PushByRoleId(rid, pushSync.PushMsgName(), proto)
+		}
+	}
+
+}
+
+func (this *Mgr) pushAll(msgName string, data interface{}) {
+
+	this.rm.Lock()
+	defer this.rm.Unlock()
+	for _, conn := range this.roleCache {
+		conn.Push(msgName, data)
+	}
+}