昔念
2598dee881
refactor(common): 使用 sync.Map 优化全局客户端映射 将 common/cool/global.go 中的 Clientmap 从普通 map 替换为 sync.Map， 以提高并发安全性。同时迁移相关操作函数至 cool 包中统一管理。 更新了 rpc 和 service 层代码，确保正确调用新的客户端管理方法。 在 InfoService 中新增 Kick 方法用于处理用户踢出逻辑。 ```
89 lines
2.1 KiB
Go
89 lines
2.1 KiB
Go
package cool
|
||
|
||
import (
|
||
"blazing/common/utils"
|
||
"context"
|
||
"reflect"
|
||
"sync"
|
||
"time"
|
||
|
||
"github.com/antlabs/cronex"
|
||
"github.com/gogf/gf/v2/os/glog"
|
||
sensitive "github.com/zmexing/go-sensitive-word"
|
||
)
|
||
|
||
// 定义改为sync.Map
|
||
var Clientmap sync.Map
|
||
|
||
// ar Limiter = ratelimit.New(100) // per second
|
||
// var Limiter = limit.New()
|
||
var ctx = context.TODO()
|
||
|
||
type Cmd struct {
|
||
Func reflect.Value //方法函数
|
||
Req reflect.Type //请求体
|
||
//Res reflect.Value //返回体
|
||
}
|
||
|
||
var (
|
||
CmdCache = &utils.SyncMap[uint32, Cmd]{} //命令缓存
|
||
|
||
Loger = glog.New()
|
||
Cron = cronex.New() //时间轮
|
||
|
||
)
|
||
var Filter *sensitive.Manager
|
||
var DefaultGenerator = utils.NewGen(time.Date(2025, 1, 1, 0, 0, 0, 0, time.UTC), uint8(Config.PortBL))
|
||
|
||
func init() {
|
||
// 创建 IdGeneratorOptions 对象，可在构造函数中输入 WorkerId：
|
||
Loger.SetFlags(glog.F_TIME_STD | glog.F_FILE_LONG) //设置flag
|
||
Loger.Print(ctx, "初始化日志")
|
||
// for i := 0; i < 600; i++ {
|
||
// glog.Debug(context.Background(), i, "初始化雪花算法", DefaultGenerator.Get())
|
||
// }
|
||
|
||
Filter, _ = sensitive.NewFilter(
|
||
sensitive.StoreOption{Type: sensitive.StoreMemory}, // 基于内存
|
||
sensitive.FilterOption{Type: sensitive.FilterDfa}, // 基于DFA算法
|
||
)
|
||
go Cron.Run()
|
||
|
||
// if err != nil {
|
||
// log.Fatalf("敏感词服务启动失败, err:%v", err)
|
||
// return
|
||
// }
|
||
|
||
// 加载敏感词库
|
||
Filter.LoadDictEmbed(
|
||
sensitive.DictCovid19,
|
||
sensitive.DictOther,
|
||
sensitive.DictReactionary,
|
||
sensitive.DictViolence,
|
||
sensitive.DictPeopleLife,
|
||
sensitive.DictPornography,
|
||
sensitive.DictAdditional,
|
||
sensitive.DictCorruption,
|
||
sensitive.DictTemporaryTencent,
|
||
)
|
||
// if err != nil {
|
||
// log.Fatalf("加载词库发生了错误, err:%v", err)
|
||
// return
|
||
// }
|
||
|
||
// // 动态添加自定义敏感词
|
||
// err = Filter.Store.AddWord("李世民", "秦始皇")
|
||
// if err != nil {
|
||
// log.Fatalf("添加敏感词发生了错误, err:%v", err)
|
||
// return
|
||
// }
|
||
|
||
// // 动态自定义敏感词白名单
|
||
// err = Filter.Store.DelWord("武汉海鲜市场", "武汉")
|
||
// if err != nil {
|
||
// log.Fatalf("删除敏感词发生了错误, err:%v", err)
|
||
// return
|
||
// }
|
||
|
||
}
|