55 lines
1.6 KiB
Go
55 lines
1.6 KiB
Go
package server
|
||
|
||
import (
|
||
"fmt"
|
||
"game/common/proto/pb"
|
||
"strings"
|
||
"sync"
|
||
)
|
||
|
||
var userServiceMgr userServiceManager
|
||
|
||
/*
|
||
记录玩家访问过的节点信息，玩家下线后清除相关信息，比如玩家在某个玩法服里掉线，重连后要回到该房间，应由对应的服务节点主动将玩家拉回房间
|
||
*/
|
||
type userServiceManager struct {
|
||
inService sync.Map // key:userId+service_type_id value:service_name
|
||
}
|
||
|
||
func (m *userServiceManager) makeKey(userId int64, typeId pb.ServiceTypeId) string {
|
||
return fmt.Sprintf("%s_%d", userId, typeId)
|
||
}
|
||
|
||
func (m *userServiceManager) Add(userId int64, typeId pb.ServiceTypeId, serviceName string) {
|
||
k := m.makeKey(userId, typeId)
|
||
m.inService.Store(k, serviceName)
|
||
}
|
||
|
||
// 玩家下线，清除他呆过的所有服务节点信息
|
||
func (m *userServiceManager) CleanUser(userId int64) {
|
||
var del []string
|
||
m.inService.Range(func(k, v interface{}) bool {
|
||
userServiceType := k.(string)
|
||
if strings.Contains(userServiceType, fmt.Sprintf("%d", userId)) {
|
||
del = append(del, userServiceType)
|
||
}
|
||
return true
|
||
})
|
||
for _, k := range del {
|
||
m.inService.Delete(k)
|
||
}
|
||
}
|
||
|
||
// todo:要考虑到旧服务可能已关闭，这里要访问etcd中是否有该服务节点，最好采用订阅机制，让etcd变动时清除对应服务节点
|
||
func (m *userServiceManager) FindServiceName(userId int64, typeId pb.ServiceTypeId) (serviceName string, ok bool) {
|
||
var v any
|
||
if v, ok = m.inService.Load(m.makeKey(userId, typeId)); ok {
|
||
serviceName = v.(string)
|
||
}
|
||
return
|
||
}
|
||
|
||
func (m *userServiceManager) Del(userId int64, typeId pb.ServiceTypeId) {
|
||
m.inService.Delete(m.makeKey(userId, typeId))
|
||
}
|