修改计划任务的持久化，act与kv的缓存逻辑，预防多服务器重复持久化

cmd/update.go

@@ -119,8 +119,6 @@ var (
 						return
 					}
 				}
-				continue
-
 				g.Log().Debugf(ctx, "准备同步服务器:%v,url=%v", v.Name, address+"/callback/update")
 				get, err := client.Post(ctx, address+"/callback/update", &UpdateReq{
 					FileUrl: url[v.S3],

go.mod

@@ -12,12 +12,10 @@ require (
 	github.com/goccy/go-json v0.10.4
 	github.com/gogf/gf/contrib/config/apollo/v2 v2.9.0
 	github.com/gogf/gf/v2 v2.9.0
-	github.com/google/uuid v1.6.0
 	github.com/gorilla/websocket v1.5.3
 	github.com/lionsoul2014/ip2region/binding/golang v0.0.0-20241220152942-06eb5c6e8230
 	github.com/minio/minio-go/v7 v7.0.85
 	github.com/prometheus/client_golang v1.21.1
-	github.com/stretchr/testify v1.10.0
 	github.com/xuri/excelize/v2 v2.9.0
 	golang.org/x/oauth2 v0.24.0
 	google.golang.org/api v0.44.0
@@ -32,7 +30,6 @@ require (
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/clbanning/mxj/v2 v2.7.0 // indirect
-	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/dustin/go-humanize v1.0.1 // indirect
 	github.com/elastic/elastic-transport-go/v8 v8.6.1 // indirect
 	github.com/emirpasic/gods v1.18.1 // indirect
@@ -47,6 +44,7 @@ require (
 	github.com/go-pay/xtime v0.0.2 // indirect
 	github.com/golang/groupcache v0.0.0-20200121045136-8c9f03a8e57e // indirect
 	github.com/golang/protobuf v1.5.2 // indirect
+	github.com/google/uuid v1.6.0 // indirect
 	github.com/googleapis/gax-go/v2 v2.0.5 // indirect
 	github.com/grokify/html-strip-tags-go v0.1.0 // indirect
 	github.com/hashicorp/hcl v1.0.0 // indirect
@@ -62,7 +60,6 @@ require (
 	github.com/munnerz/goautoneg v0.0.0-20191010083416-a7dc8b61c822 // indirect
 	github.com/olekukonko/tablewriter v0.0.5 // indirect
 	github.com/pelletier/go-toml v1.9.3 // indirect
-	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/prometheus/client_model v0.6.1 // indirect
 	github.com/prometheus/common v0.62.0 // indirect
 	github.com/prometheus/procfs v0.15.1 // indirect

internal/boot/boot.go

@@ -1,6 +1,7 @@
 package boot
 
 import (
+	"context"
 	v1 "github.com/ayflying/utility_go/api/system/v1"
 	"github.com/ayflying/utility_go/service"
 	"github.com/gogf/gf/v2/os/gctx"
@@ -15,15 +16,16 @@ func Boot() (err error) {
 	err = service.SystemCron().StartCron()
 
 	//用户活动持久化
-	service.SystemCron().AddCron(v1.CronType_DAILY, func() error {
+	service.SystemCron().AddCronV2(v1.CronType_DAILY, func(ctx context.Context) error {
-		return service.GameAct().Saves()
+		err = service.GameKv().SavesV1()
+		err = service.GameAct().Saves(ctx)
+		return err
 	})
 
 	//初始化自启动方法
 	for _, v := range _func {
 		v()
 	}
-
 	return nil
 }
 

internal/logic/gameAct/gameAct.go

@@ -1,6 +1,7 @@
 package gameAct
 
 import (
+	"context"
 	"fmt"
 	"github.com/ayflying/utility_go/internal/model/do"
 	"github.com/ayflying/utility_go/internal/model/entity"
@@ -105,16 +106,24 @@ func (s *sGameAct) Set(uid int64, actId int, data interface{}) (err error) {
 	return
 }
 
-func (s *sGameAct) Saves() (err error) {
+func (s *sGameAct) Saves(ctx context.Context) (err error) {
+	getCache, _ := pkg.Cache("redis").Get(nil, "cron:game_act")
+	//如果没有执行过，设置时间戳
+	if getCache.Int64() > 0 {
+		return
+	} else {
+		pkg.Cache("redis").Set(nil, "cron:game_act", gtime.Now().Unix(), time.Hour)
+	}
+
 	//遍历执行
 	ActList.Iterator(func(i interface{}) bool {
-		err = s.Save(i.(int))
+		err = s.Save(ctx, i.(int))
 		return true
 	})
 	return
 }
 
-func (s *sGameAct) Save(actId int) (err error) {
+func (s *sGameAct) Save(ctx context.Context, actId int) (err error) {
 
 	cacheKey := fmt.Sprintf("act:%v:*", actId)
 	//获取当前用户的key值
@@ -149,8 +158,8 @@ func (s *sGameAct) Save(actId int) (err error) {
 			}
 
 			//如果有活跃，跳过持久化
-			if getBool, _ := pkg.Cache("redis").Contains(ctx,
+			if getBool, _ := pkg.Cache("redis").
-				fmt.Sprintf("act:update:%d", uid)); getBool {
+				Contains(ctx, fmt.Sprintf("act:update:%d", uid)); getBool {
 				continue
 			}
 

internal/logic/gameKv/gameKv.go

@@ -1,13 +1,17 @@
 package gameKv
 
 import (
+	"fmt"
+	"github.com/ayflying/utility_go/pkg"
 	"github.com/ayflying/utility_go/service"
 	"github.com/ayflying/utility_go/tools"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/os/gctx"
+	"github.com/gogf/gf/v2/os/gtime"
 	"strconv"
 	"strings"
 	"sync"
+	"time"
 )
 
 var (
@@ -25,10 +29,6 @@ func New() *sGameKv {
 
 func init() {
 	service.RegisterGameKv(New())
-
-	//支付钩子
-	//task.Task.Trigger(tasks.TaskType_PAY, service.GameKv().HookPay)
-	//task.Task.Trigger(tasks.TaskType_WARDROBE_LEVEL, service.GameKv().HookLevelRwd)
 }
 
 // SavesV1 方法
@@ -37,6 +37,14 @@ func init() {
 // @receiver s: sGameKv的实例。
 // @return err: 错误信息，如果操作成功，则为nil。
 func (s *sGameKv) SavesV1() (err error) {
+	getCache, err := pkg.Cache("redis").Get(nil, "cron:game_kv")
+	//如果没有执行过，设置时间戳
+	if getCache.Int64() > 0 {
+		return
+	} else {
+		pkg.Cache("redis").Set(nil, "cron:game_kv", gtime.Now().Unix(), time.Hour)
+	}
+
 	// 从Redis列表中获取所有用户KV索引的键
 	//keys, err := utils.RedisScan("user:kv:*")
 	err = tools.Redis.RedisScanV2("user:kv:*", func(keys []string) (err error) {
@@ -69,12 +77,17 @@ func (s *sGameKv) SavesV1() (err error) {
 			//if user.UpdatedAt.Seconds < gtime.Now().Add(consts.ActSaveTime).Unix() {
 			//	continue
 			//}
+			//如果有活跃，跳过持久化
+			if getBool, _ := pkg.Cache("redis").
+				Contains(ctx, fmt.Sprintf("act:update:%d", uid)); getBool {
+				continue
+			}
 
 			get, _ := g.Redis().Get(ctx, cacheKey)
 			var data interface{}
 			get.Scan(&data)
 			list = append(list, &ListData{
-				Uid: int64(uid),
+				Uid: uid,
 				Kv:  data,
 			})
 

internal/logic/ip2region/ip2region.go

@@ -1,7 +1,6 @@
 package ip2region
 
 import (
-	"github.com/ayflying/utility_go/internal/boot"
 	"github.com/ayflying/utility_go/service"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/os/gctx"
@@ -13,6 +12,7 @@ import (
 
 var (
 	ctx  = gctx.New()
+	wait = false
 )
 
 type sIp2region struct {
@@ -27,9 +27,9 @@ func New() *sIp2region {
 func init() {
 	service.RegisterIp2Region(New())
 
-	boot.AddFunc(func() {
+	//boot.AddFunc(func() {
-		service.Ip2Region().Load()
+	//	service.Ip2Region().Load()
-	})
+	//})
 }
 
 // Load 加载到内存中
@@ -39,10 +39,16 @@ func init() {
 func (s *sIp2region) Load() {
 	var err error
 
-	var url = "https://github.com/ayflying/resource/raw/refs/heads/main/attachment/ip2region.xdb"
 	var dbPath = "runtime/library/ip2region.xdb"
-
+	var url = "https://github.com/ayflying/resource/raw/refs/heads/main/attachment/ip2region.xdb"
+	if wait {
+		return
+	}
 	if gfile.IsEmpty(dbPath) {
+		wait = true
+		defer func() {
+			wait = false
+		}()
 		g.Log().Debug(ctx, "等待下载ip库文件")
 		//下载文件
 		putData, err2 := g.Client().Discovery(nil).Get(ctx, url)
@@ -71,6 +77,9 @@ func (s *sIp2region) Load() {
 }
 
 func (s *sIp2region) GetIp(ip string) (res []string) {
+	//初始化加载
+	s.Load()
+
 	res = make([]string, 5)
 	if s.searcher == nil {
 		return

internal/logic/systemCron/listener.go

@@ -7,6 +7,7 @@ import (
 	"github.com/ayflying/utility_go/pkg/notice"
 	"github.com/gogf/gf/v2/frame/g"
 	"github.com/gogf/gf/v2/net/gclient"
+	"github.com/gogf/gf/v2/os/gctx"
 )
 
 type Status struct {
@@ -22,10 +23,10 @@ func (s *sSystemCron) Guardian(DingTalkWebHook string) {
 		Name    string
 		Address string
 	}
-	cfg, _ := g.Cfg().Get(ctx, "serverList")
+	cfg, _ := g.Cfg().Get(gctx.New(), "serverList")
 	cfg.Scan(&list)
 	for _, v := range list {
-		get, err := g.Client().Discovery(nil).Get(ctx, v.Address+"/callback/status")
+		get, err := g.Client().Discovery(nil).Get(gctx.New(), v.Address+"/callback/status")
 
 		defer get.Close()
 		if err != nil {

internal/logic/systemCron/systemCron.go

@@ -14,7 +14,7 @@ import (
 )
 
 var (
-	ctx       = gctx.New()
+	//ctx       = gctx.New()
 	startTime *gtime.Time
 )
 
@@ -23,39 +23,44 @@ var (
 type sSystemCron struct {
 	//互斥锁
 	Lock        sync.Mutex
+	taskChan    chan func(context.Context) error
+	TaskTimeout time.Duration
 
 	// 每秒执行的任务
-	SecondlyTask []func() error
+	SecondlyTask []func(context.Context) error
 	// 每分钟执行的任务
-	MinutelyTask []func() error
+	MinutelyTask []func(context.Context) error
 	// 每小时执行的任务
-	HourlyTask []func() error
+	HourlyTask []func(context.Context) error
 	// 每天执行的任务
-	DailyTask []func() error
+	DailyTask []func(context.Context) error
 	// 每周执行的任务
-	WeeklyTask []func() error
+	WeeklyTask []func(context.Context) error
 	// 每月执行的任务
-	MonthlyTask []func() error
+	MonthlyTask []func(context.Context) error
 	// 每年执行的任务
-	YearlyTask []func() error
+	YearlyTask []func(context.Context) error
 	// 每周一执行的任务
-	MondayTask []func() error
+	MondayTask []func(context.Context) error
 	// 每周二执行的任务
-	TuesdayTask []func() error
+	TuesdayTask []func(context.Context) error
 	// 每周三执行的任务
-	WednesdayTask []func() error
+	WednesdayTask []func(context.Context) error
 	// 每周四执行的任务
-	ThursdayTask []func() error
+	ThursdayTask []func(context.Context) error
 	// 每周五执行的任务
-	FridayTask []func() error
+	FridayTask []func(context.Context) error
 	// 每周六执行的任务
-	SaturdayTask []func() error
+	SaturdayTask []func(context.Context) error
 	// 每周日执行的任务
-	SundayTask []func() error
+	SundayTask []func(context.Context) error
 }
 
 func New() *sSystemCron {
-	return &sSystemCron{}
+	return &sSystemCron{
+		taskChan:    make(chan func(context.Context) error, 2),
+		TaskTimeout: time.Minute * 30,
+	}
 }
 
 func init() {
@@ -65,13 +70,32 @@ func init() {
 // AddCron 添加一个定时任务到相应的调度列表中。
 //
 // @Description: 根据指定的类型将函数添加到不同的任务列表中，以供后续执行。
+// 确保自定义任务正确处理上下文取消信号，即可充分发挥超时打断功能。
 // @receiver s: sSystemCron的实例，代表一个调度系统。
 // @param typ: 任务的类型，决定该任务将被添加到哪个列表中。对应不同的时间间隔。
 // @param _func: 要添加的任务函数，该函数执行时应该返回一个error。
+// deprecated: 弃用，请使用 AddCronV2
 func (s *sSystemCron) AddCron(typ v1.CronType, _func func() error) {
+	//转换为带上下文的，提供打断
+	var _func2 = func(ctx context.Context) error {
+		return _func()
+	}
+	s.AddCronV2(typ, _func2)
+}
+
+// AddCronV2  添加一个定时任务到相应的调度列表中。
+//
+// @Description: 根据指定的类型将函数添加到不同的任务列表中，以供后续执行。
+// @receiver s: sSystemCron的实例，代表一个调度系统。
+// @param typ: 任务的类型，决定该任务将被添加到哪个列表中。对应不同的时间间隔。
+// @param _func: 要添加的任务函数，该函数执行时应该返回一个error。
+func (s *sSystemCron) AddCronV2(typ v1.CronType, _func func(context.Context) error) {
 	//加锁
 	s.Lock.Lock()
 	defer s.Lock.Unlock()
+	//
+	//ctx := gctx.New()
+	//newFunc := func()
 
 	switch typ {
 	case v1.CronType_SECOND:
@@ -118,140 +142,123 @@ func (s *sSystemCron) StartCron() (err error) {
 	}
 	startTime = gtime.Now()
 
-	g.Log().Debug(ctx, "启动计划任务定时器详情")
+	g.Log().Debug(gctx.New(), "启动计划任务定时器详情")
 	//每秒任务
-	gtimer.SetInterval(ctx, time.Second, func(ctx context.Context) {
+	gtimer.SetInterval(gctx.New(), time.Second, func(ctx context.Context) {
 		//g.Log().Debug(ctx, "每秒定时器")
-		err = s.secondlyTask()
+		s.secondlyTask()
 	})
 
 	//每分钟任务
-	_, err = gcron.AddSingleton(ctx, "0 * * * * *", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 * * * * *", func(ctx context.Context) {
 		//g.Log().Debug(ctx, "每分钟定时器")
-		err = s.minutelyTask()
+		s.minutelyTask()
 	})
 
 	//每小时任务
-	_, err = gcron.AddSingleton(ctx, "0 0 * * * *", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 * * * *", func(ctx context.Context) {
 		g.Log().Debug(ctx, "每小时定时器")
-		err = s.hourlyTask()
+		s.hourlyTask()
 	})
 
 	//每天任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * *", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * *", func(ctx context.Context) {
 		g.Log().Debug(ctx, "每日定时器")
-		err = s.dailyTask()
+		s.dailyTask()
 	})
 
 	//每周任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 1", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 1", func(ctx context.Context) {
 		g.Log().Debug(ctx, "每周一定时器")
-		err = s.weeklyTask(1)
+		s.weeklyTask(1)
 	})
-	//每周二任务
+	//每周二的任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 2", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 2", func(ctx context.Context) {
 		g.Log().Debug(ctx, "每周二定时器")
-		err = s.weeklyTask(2)
+		s.weeklyTask(2)
 	})
 	//周三任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 3", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 3", func(ctx context.Context) {
 		g.Log().Debug(ctx, "周三定时器")
-		err = s.weeklyTask(3)
+		s.weeklyTask(3)
 	})
 	//周四任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 4", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 4", func(ctx context.Context) {
 		g.Log().Debug(ctx, "周四定时器")
-		err = s.weeklyTask(4)
+		s.weeklyTask(4)
 	})
 	//周五任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 5", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 5", func(ctx context.Context) {
 		g.Log().Debug(ctx, "周五定时器")
-		err = s.fridayTask()
+		s.weeklyTask(5)
 	})
 	//周六任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 6", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 6", func(ctx context.Context) {
 		g.Log().Debug(ctx, "周六定时器")
-		err = s.weeklyTask(6)
+		s.weeklyTask(6)
 	})
 	//周日任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 * * 0", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 * * 0", func(ctx context.Context) {
 		g.Log().Debug(ctx, "周日定时器")
-		err = s.weeklyTask(7)
+		s.weeklyTask(7)
 	})
 
 	//每月任务
-	_, err = gcron.AddSingleton(ctx, "0 0 0 1 * *", func(ctx context.Context) {
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 1 * *", func(ctx context.Context) {
 		g.Log().Debug(ctx, "每月定时器")
-		err = s.monthlyTask()
+		s.monthlyTask()
 	})
 
-	_, err = gcron.AddSingleton(ctx, "0 0 0 1 1 *", func(ctx context.Context) {
+	//每年任务
+	_, err = gcron.AddSingleton(gctx.New(), "0 0 0 1 1 *", func(ctx context.Context) {
 		g.Log().Debug(ctx, "每年定时器")
-		err = s.monthlyTask()
+		s.yearlyTask()
 	})
 
+	//统一执行方法
+	s.RunFuncChan()
 	return
 }
 
 // 每妙任务
-func (s *sSystemCron) secondlyTask() (err error) {
+func (s *sSystemCron) secondlyTask() {
 	if len(s.SecondlyTask) == 0 {
 		return
 	}
-	for _, _func := range s.SecondlyTask {
+	s.AddFuncChan(s.SecondlyTask)
-		err = _func()
-		if err != nil {
-			g.Log().Error(ctx, err)
-		}
-	}
 	return
 }
 
 // 每分钟任务
-func (s *sSystemCron) minutelyTask() (err error) {
+func (s *sSystemCron) minutelyTask() {
 	if len(s.MinutelyTask) == 0 {
 		return
 	}
-	for _, _func := range s.MinutelyTask {
+	s.AddFuncChan(s.MinutelyTask)
-		err = _func()
-		if err != nil {
-			g.Log().Error(ctx, err)
-		}
-	}
 	return
 }
 
 // 每小时任务
-func (s *sSystemCron) hourlyTask() (err error) {
+func (s *sSystemCron) hourlyTask() {
 	if len(s.HourlyTask) == 0 {
 		return
 	}
-	for _, _func := range s.HourlyTask {
+	s.AddFuncChan(s.HourlyTask)
-		err = _func()
-		if err != nil {
-			g.Log().Error(ctx, err)
-		}
-	}
 	return
 }
 
 // 每天任务
-func (s *sSystemCron) dailyTask() (err error) {
+func (s *sSystemCron) dailyTask() {
 	if len(s.DailyTask) == 0 {
 		return
 	}
-	for _, _func := range s.DailyTask {
+	s.AddFuncChan(s.DailyTask)
-		err = _func()
-		if err != nil {
-			g.Log().Error(ctx, err)
-		}
-	}
 	return
 }
 
 // 每周任务
-func (s *sSystemCron) weeklyTask(day int) (err error) {
+func (s *sSystemCron) weeklyTask(day int) {
-	var arr []func() error
+	var arr []func(context.Context) error
 	switch day {
 	case 1:
 		arr = s.MondayTask
@@ -275,39 +282,91 @@ func (s *sSystemCron) weeklyTask(day int) (err error) {
 	if len(arr) == 0 {
 		return
 	}
-	for _, _func := range arr {
+	s.AddFuncChan(arr)
-		err = _func()
-		if err != nil {
-			g.Log().Error(ctx, err)
-		}
-	}
-	return
-}
-
-// 周五任务
-func (s *sSystemCron) fridayTask() (err error) {
-	if len(s.FridayTask) == 0 {
-		return
-	}
-	for _, _func := range s.FridayTask {
-		err = _func()
-		if err != nil {
-			g.Log().Error(ctx, err)
-		}
-	}
 	return
 }
 
 // 每月任务
-func (s *sSystemCron) monthlyTask() (err error) {
+func (s *sSystemCron) monthlyTask() {
 	if len(s.MonthlyTask) == 0 {
 		return
 	}
-	for _, _func := range s.MonthlyTask {
+	s.AddFuncChan(s.MonthlyTask)
-		err = _func()
+	return
+}
+
+//每年任务
+func (s *sSystemCron) yearlyTask() {
+	if len(s.YearlyTask) == 0 {
+		return
+	}
+	s.AddFuncChan(s.YearlyTask)
+
+}
+
+// AddFuncChan 添加方法到通道
+func (s *sSystemCron) AddFuncChan(list []func(context.Context) error) {
+	for _, v := range list {
+		s.taskChan <- v
+	}
+}
+
+// RunFuncChan 统一执行方法
+func (s *sSystemCron) RunFuncChan() {
+	go func() {
+		for task := range s.taskChan {
+			//ctx := gctx.New()
+			func() {
+				//超时释放资源
+				ctx, cancel := context.WithTimeout(context.Background(), s.TaskTimeout)
+				defer cancel()
+
+				// 使用匿名函数包裹来捕获 panic
+				defer func() {
+					if r := recover(); r != nil {
+						g.Log().Errorf(gctx.New(), "执行函数时发生 panic: %v", r)
+					}
+				}()
+
+				done := make(chan error)
+				go func() {
+					done <- task(ctx)
+				}()
+				//err := task()
+				//if err != nil {
+				//	g.Log().Error(ctx, err)
+				//}
+				select {
+				case taskErr := <-done:
+					if taskErr != nil {
+						// 使用新上下文记录错误
+						g.Log().Error(gctx.New(), taskErr)
+					}
+				case <-ctx.Done(): // 监听上下文取消（包括超时）
+					g.Log().Errorf(gctx.New(), "task timeout:%v", ctx.Err())
+				}
+			}()
+		}
+	}()
+}
+
+// RunFunc 统一执行方法
+// deprecated: 弃用，会造成周期任务并发执行，to service.SystemCron().AddFuncChan
+func (s *sSystemCron) RunFunc(list []func() error) {
+	for _, _func := range list {
+		ctx := gctx.New()
+		func() {
+			// 使用匿名函数包裹来捕获 panic
+			defer func() {
+				if r := recover(); r != nil {
+					g.Log().Errorf(ctx, "执行函数时发生 panic: %v", r)
+				}
+			}()
+			err := _func()
 			if err != nil {
 				g.Log().Error(ctx, err)
 			}
+		}()
 	}
 	return
 }

pkg/aycache/cache.go

@@ -12,34 +12,37 @@ import (
 	"math"
 )
 
+// Mod 定义缓存模块结构体，包含一个 gcache.Cache 客户端实例
 type Mod struct {
 	client *gcache.Cache
 }
 
-var (
+// QPSCount 记录缓存的 QPS 计数
-	QPSCount int
+var QPSCount int
-	QPS      = promauto.NewGauge(
+// QPS 是一个 Prometheus 指标，用于记录当前缓存的 QPS 数量
+var QPS = promauto.NewGauge(
 	prometheus.GaugeOpts{
 		Name: "Cache_QPS",
 		Help: "当前缓存QPS数量",
 	},
 )
-)
 
+// init 函数在包被导入时执行，用于初始化定时任务以更新 QPS 指标
 func init() {
 	boot.AddFunc(func() {
-		//初始化指标
+		// 初始化指标，每分钟计算一次平均 QPS 并重置计数器
 		service.SystemCron().AddCron(v1.CronType_MINUTE, func() error {
 			QPS.Set(math.Round(float64(QPSCount) / 60))
 			QPSCount = 0
 			return nil
 		})
 	})
-
 }
 
+// New 根据传入的名称创建不同类型的缓存适配器
+// 如果未传入名称，默认使用 "cache" 类型
+// 支持的类型包括 "cache"（内存缓存）、"redis"（Redis 缓存）、"file"（文件缓存）和 "es"（Elasticsearch 缓存）
 func New(_name ...string) gcache.Adapter {
-
 	var cacheAdapterObj gcache.Adapter
 	var name = "cache"
 	if len(_name) > 0 {
@@ -47,18 +50,23 @@ func New(_name ...string) gcache.Adapter {
 	}
 	switch name {
 	case "cache":
+		// 创建内存缓存适配器
 		cacheAdapterObj = drive2.NewAdapterMemory()
 	case "redis":
+		// 创建 Redis 缓存适配器
 		cacheAdapterObj = drive2.NewAdapterRedis()
 	case "file":
+		// 创建文件缓存适配器，指定缓存目录为 "runtime/cache"
 		cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
 	case "es":
+		// 创建 Elasticsearch 缓存适配器，需要传入额外参数
 		cacheAdapterObj = drive.NewAdapterElasticsearch(_name[1])
 	}
 
 	//var client = gcache.New()
 	//client.SetAdapter(cacheAdapterObj)
 
+	// 每次创建适配器时，QPS 计数加 1
 	QPSCount++
 	return cacheAdapterObj
 }

pkg/config/config.go

@@ -18,26 +18,40 @@ import (
 var (
 	//ApolloCfg      *apolloConfig.AppConfig
 	ApolloCfg      *apollo.Config
+	// ApolloListener 存储需要监听的 Apollo 配置项名称
 	ApolloListener []string
+	// Item2Obj 存储配置项名称和对应的加载器对象的映射
 	Item2Obj       = map[string]Load{}
 )
 
-// load接口定义了Load方法，用于加载数据
+// Load 接口定义了 Load 方法，用于加载数据
 type Load interface {
+	// Load 方法用于加载配置数据，支持传入可选的配置参数
 	Load(cfg ...string)
 }
 
+// NewV1 创建一个新的 Cfg 实例
 func NewV1() *Cfg {
 	return &Cfg{}
 }
 
+// Cfg 结构体包含配置操作的相关方法
 type Cfg struct {
+	// Lock 用于保证并发安全的互斥锁
 	Lock sync.Mutex
 }
 
+// GetDbFile 从数据库中获取指定名称的配置文件
+// 参数:
+//   name - 配置文件的名称
+// 返回值:
+//   *g.Var - 存储配置数据的变量
+//   error  - 操作过程中遇到的错误
 func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
+	// 从数据库的 game_config 表中查询指定名称的配置数据
 	get2, err := g.Model("game_config").
 		Where("name", name).Master().Value("data")
+	// 将查询结果扫描到 res 变量中
 	err = get2.Scan(&res)
 	if res == nil {
 		res = &gvar.Var{}
@@ -45,107 +59,81 @@ func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
 	return
 }
 
-func (c *Cfg) GetFile(filename string, obj ...Load) (jsonObj *gjson.Json, err error) {
+// GetFile 从文件系统或资源文件中加载 JSON 配置
+// 参数:
+//   filename   - 需要加载的配置文件名（不带扩展名）
+//   _pathStr   - 可选参数，指定配置文件目录路径，默认"manifest/game/"
+// 返回值:
+//   *gjson.Json - 解析后的 JSON 对象
+//   error       - 文件加载或解析过程中遇到的错误
+func (c *Cfg) GetFile(filename string, _pathStr ...string) (jsonObj *gjson.Json, err error) {
+	// 处理路径参数，使用默认路径或传入参数
 	pathStr := "manifest/game/"
+	if len(_pathStr) > 0 {
+		pathStr = _pathStr[0]
+	}
+	// 拼接完整的文件路径
 	filePath := pathStr + filename + ".json"
-	//err := gres.Load(pathStr + filename)
 
 	// 载入静态资源到文件对象
 	err = gres.Load(filePath)
 	var bytes []byte
 
+	// 优先从文件系统读取，不存在时从资源文件读取
 	if gfile.IsFile(filePath) {
-		bytes = gfile.GetBytes(filePath)
+		bytes = gfile.GetBytes(filePath) // 读取物理文件内容
 	} else {
-		bytes = gres.GetContent(filePath)
+		bytes = gres.GetContent(filePath) // 从打包资源中获取内容
 	}
 
+	// 解析 JSON 内容并返回结果
 	jsonObj, err = gjson.DecodeToJson(bytes)
-	//g.Dump(filePath, jsonObj)
 	return
 }
 
-// getUrlFile 获取远程配置
+// GetUrlFile 获取远程配置
+// 参数:
+//   name - 配置文件的名称
+// 返回值:
+//   *gjson.Json - 解析后的 JSON 对象
+//   error       - 请求或解析过程中遇到的错误
 func (c *Cfg) GetUrlFile(name string) (jsonObj *gjson.Json, err error) {
+	// 拼接远程配置文件的 URL
 	urlStr := fmt.Sprintf("http://sdf.sdfs.sdf/%s.json", name)
+	// 发送 HTTP 请求获取远程配置数据
 	getUrl, err := g.Client().Discovery(nil).Get(nil, urlStr)
+	// 读取响应内容
 	bytes := getUrl.ReadAll()
+	// 解析 JSON 内容并返回结果
 	jsonObj, err = gjson.DecodeToJson(bytes)
 	return
 }
 
-// 获取阿波罗
+// GetApollo 从 Apollo 配置中心获取指定名称的配置
-//func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
+// 参数:
-//	jsonObj, err = c.GetApolloV2(name, obj)
+//   name - 配置文件的名称
-//	return
+//   obj  - 实现了 Load 接口的加载器对象
-//
+// 返回值:
-//	//c.Lock.Lock()
+//   *gjson.Json - 解析后的 JSON 对象
-//	//defer c.Lock.Unlock()
+//   error       - 操作过程中遇到的错误
-//	//
-//	//Item2Obj[name+".json"] = obj
-//	//var cfg = apolloConfig.AppConfig{
-//	//	AppID:             ApolloCfg.AppID,
-//	//	Cluster:           ApolloCfg.Cluster,
-//	//	IP:                ApolloCfg.IP,
-//	//	NamespaceName:     name + ".json",
-//	//	Secret:            ApolloCfg.Secret,
-//	//	IsBackupConfig:    ApolloCfg.IsBackupConfig,
-//	//	BackupConfigPath:  ApolloCfg.BackupConfigPath,
-//	//	SyncServerTimeout: 60,
-//	//	MustStart:         true,
-//	//}
-//	////cfg.NamespaceName = name + ".json"
-//	//
-//	//client, err := agollo.StartWithConfig(func() (*apolloConfig.AppConfig, error) {
-//	//	return ApolloCfg, nil
-//	//})
-//	//if client == nil {
-//	//	return
-//	//}
-//	//var getStr string
-//	//var getApollo *storage.Config
-//	//for range 5 {
-//	//	getApollo = client.GetConfig(cfg.NamespaceName)
-//	//	if getApollo != nil {
-//	//		break
-//	//	}
-//	//	time.Sleep(time.Second * 5)
-//	//}
-//	//
-//	//if getApollo != nil {
-//	//	getStr = getApollo.GetValue("content")
-//	//	if getStr != "" {
-//	//		//写入配置
-//	//		gfile.PutContents(path.Join("manifest", "game", name+".json"), getStr)
-//	//	}
-//	//} else {
-//	//	jsonObj, err = c.GetFile(name)
-//	//}
-//	//jsonObj, err = gjson.DecodeToJson(getStr)
-//	////首次运行加入监听器
-//	//if !gstr.InArray(ApolloListener, name) {
-//	//	c2 := &CustomChangeListener{}
-//	//	client.AddChangeListener(c2)
-//	//	ApolloListener = append(ApolloListener, name)
-//	//}
-//	//return
-//}
-
 func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
+	// 将配置项名称和对应的加载器对象存入映射
 	Item2Obj[name+".json"] = obj
 
-	// 接入阿波罗配置
+	// 接入 Apollo 配置
 	ApolloCfg.NamespaceName = name + ".json"
+	// 创建 Apollo 配置适配器
 	adapter, err := apollo.New(nil, *ApolloCfg)
 	if err != nil {
+		// 配置适配器创建失败，记录致命错误日志
 		g.Log().Fatalf(nil, `%+v`, err)
 	}
-	// Change the adapter of default configuration instance.
+	// 更改默认配置实例的适配器
 	g.Cfg(name).SetAdapter(adapter)
 
 	// 首次运行加入监听器
 	if !gstr.InArray(ApolloListener, name+".json") {
-		//放置监听器
+		// 启动 Apollo 客户端
 		client, _ := agollo.StartWithConfig(func() (*apolloConfig.AppConfig, error) {
 			return &apolloConfig.AppConfig{
 				AppID:             ApolloCfg.AppID,
@@ -159,23 +147,32 @@ func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error)
 				MustStart:         ApolloCfg.MustStart,
 			}, nil
 		})
+		// 创建自定义监听器实例
 		c2 := &CustomChangeListener{}
+		// 为 Apollo 客户端添加监听器
 		client.AddChangeListener(c2)
+		// 将配置项名称添加到监听器列表
 		ApolloListener = append(ApolloListener, name+".json")
 	}
 
+	// 从配置中心获取指定配置项的值
 	cfg, err := g.Cfg(name).Get(nil, "content")
+	// 将配置值扫描到 jsonObj 中
 	cfg.Scan(&jsonObj)
 	return
 }
 
-// 阿波罗监听器
+// CustomChangeListener 是 Apollo 配置变化的自定义监听器
 type CustomChangeListener struct {
+	// wg 用于等待所有处理任务完成
 	wg sync.WaitGroup
 }
 
+// OnChange 当 Apollo 配置发生变化时触发
 func (c *CustomChangeListener) OnChange(changeEvent *storage.ChangeEvent) {
+	// 记录配置变化的日志
 	g.Log().Debugf(nil, "当前Namespace变化了：%v", changeEvent.Namespace)
+	// 获取变化的配置项名称
 	filename := changeEvent.Namespace
 	if obj, ok := Item2Obj[filename]; ok {
 		// 重载配置文件
@@ -183,7 +180,7 @@ func (c *CustomChangeListener) OnChange(changeEvent *storage.ChangeEvent) {
 	}
 }
 
+// OnNewestChange 当获取到最新配置时触发，当前为空实现
 func (c *CustomChangeListener) OnNewestChange(event *storage.FullChangeEvent) {
 	//write your code here
-
 }

pkg/s3/s3.go

@@ -16,26 +16,32 @@ import (
 	"github.com/minio/minio-go/v7/pkg/credentials"
 )
 
+// ctx 全局上下文，用于在整个包中传递请求范围的数据
 var (
 	//client *minio.Client
 	ctx = gctx.New()
 )
 
+// DataType 定义了 S3 配置的数据结构，用于存储访问 S3 所需的各种信息
 type DataType struct {
-	AccessKey     string `json:"access_key"`
+	AccessKey     string `json:"access_key"` // 访问 S3 的密钥 ID
-	SecretKey     string `json:"secret_key"`
+	SecretKey     string `json:"secret_key"` // 访问 S3 的密钥
-	Address       string `json:"address"`
+	Address       string `json:"address"`    // S3 服务的地址
-	Ssl           bool   `json:"ssl"`
+	Ssl           bool   `json:"ssl"`        // 是否使用 SSL 加密连接
-	Url           string `json:"url"`
+	Url           string `json:"url"`        // S3 服务的访问 URL
-	BucketName    string `json:"bucket_name"`
+	BucketName    string `json:"bucket_name"` // 默认存储桶名称
-	BucketNameCdn string `json:"bucket_name_cdn"`
+	BucketNameCdn string `json:"bucket_name_cdn"` // CDN 存储桶名称
 }
 
+// Mod 定义了 S3 模块的结构体，包含一个 S3 客户端实例和配置信息
 type Mod struct {
-	client *minio.Client
+	client *minio.Client // Minio S3 客户端实例
-	cfg    DataType
+	cfg    DataType      // S3 配置信息
 }
 
+// New 根据配置创建一个新的 S3 模块实例
+// 如果未提供名称，则从配置中获取默认的 S3 类型
+// 配置错误时会触发 panic
 func New(_name ...string) *Mod {
 	var name string
 	if len(_name) > 0 {
@@ -72,183 +78,111 @@ func New(_name ...string) *Mod {
 	return mod
 }
 
-//func (s *Mod) Load() {
+// GetCfg 获取当前 S3 模块的配置信息
-//	//导入配置
-//	get, err := g.Cfg().Get(ctx, "s3.type")
-//	cfgType := get.String()
-//	if cfgType == "" {
-//		cfgType = "default"
-//	}
-//
-//	cfgData, err := g.Cfg().Get(ctx, "s3."+cfgType)
-//	if cfgData.IsEmpty() {
-//		panic("当前配置中未配置s3：" + cfgType)
-//	}
-//
-//	get, err = g.Cfg().Get(ctx, "s3."+cfgType)
-//	err = get.Scan(&Cfg)
-//
-//	// 使用minio-go创建S3客户端
-//	obj, err := minio.New(
-//		Cfg.Address,
-//		&minio.Options{
-//			Creds:  credentials.NewStaticV4(Cfg.AccessKey, Cfg.SecretKey, ""),
-//			Secure: Cfg.Ssl,
-//		},
-//	)
-//	if err != nil {
-//		log.Fatalln(err)
-//	}
-//
-//	client = obj
-//}
-//
-//func (s *Mod) S3(name string) {
-//	get, err := g.Cfg().Get(ctx, "s3."+name)
-//	if err != nil {
-//		panic(err)
-//	}
-//	get.Scan(&Cfg)
-//
-//	// 使用minio-go创建S3客户端
-//	obj, err := minio.New(
-//		Cfg.Address,
-//		&minio.Options{
-//			Creds:  credentials.NewStaticV4(Cfg.AccessKey, Cfg.SecretKey, ""),
-//			Secure: Cfg.Ssl,
-//		},
-//	)
-//	if err != nil {
-//		log.Fatalln(err)
-//	}
-//
-//	client = obj
-//
-//}
-
-// GetCfg 获取配置
 func (s *Mod) GetCfg() *DataType {
 	return &s.cfg
 }
 
 // GetFileUrl 生成指向 S3 存储桶中指定文件的预签名 URL
-//
+// 预签名 URL 可用于在有限时间内访问 S3 存储桶中的文件
-// @Description: 生成一个具有有限有效期的预签名URL，可用于访问S3存储桶中的文件。
+// 支持从缓存中获取预签名 URL，以减少重复请求
-// @receiver s: S3的实例，用于执行S3操作。
-// @param name: 要获取预签名URL的文件名。
-// @param bucketName: 文件所在的存储桶名称。
-// @return presignedURL: 生成的预签名URL，可用于访问文件。
-// @return err: 在获取预签名URL过程中遇到的任何错误。
 func (s *Mod) GetFileUrl(name string, bucketName string, _expires ...time.Duration) (presignedURL *url.URL, err error) {
-	// 设置预签名URL的有效期为1小时
+	// 设置预签名 URL 的有效期为 1 小时，可通过参数覆盖
 	expires := time.Hour * 1
 	if len(_expires) > 0 {
 		expires = _expires[0]
 	}
+	// 生成缓存键
 	cacheKey := fmt.Sprintf("s3:%v:%v", name, bucketName)
+	// 尝试从缓存中获取预签名 URL
 	get, _ := gcache.Get(ctx, cacheKey)
-	//g.Dump(get.Vars())
 	if !get.IsEmpty() {
+		// 将缓存中的值转换为 *url.URL 类型
 		err = gconv.Struct(get.Val(), &presignedURL)
-		//presignedURL =
 		return
 	}
-	//expires := time.Duration(604800)
+	// 调用 S3 客户端生成预签名 URL
-	// 调用s3().PresignedGetObject方法生成预签名URL
 	presignedURL, err = s.client.PresignedGetObject(ctx, bucketName, name, expires, nil)
+	// 将生成的预签名 URL 存入缓存
 	err = gcache.Set(ctx, cacheKey, presignedURL, expires)
 	return
 }
 
-// PutFileUrl 生成一个用于上传文件到指定bucket的预签名URL
+// PutFileUrl 生成一个用于上传文件到指定存储桶的预签名 URL
-//
+// 预签名 URL 的有效期默认为 10 分钟
-//	@Description:
-//	@receiver s
-//	@param name 文件名
-//	@param bucketName 存储桶名称
-//	@return presignedURL 预签名的URL，用于上传文件
-//	@return err 错误信息，如果在生成预签名URL时发生错误
 func (s *Mod) PutFileUrl(name string, bucketName string) (presignedURL *url.URL, err error) {
-	// 设置预签名URL的有效期
+	// 设置预签名 URL 的有效期为 10 分钟
-	//expires := time.Now().Add(time.Minute * 30).Unix() // 例如：有效期30分钟
-	//expires2 := time.Duration(expires)
 	expires := time.Minute * 10
-	// 生成预签名URL
+	// 调用 S3 客户端生成预签名 URL
 	presignedURL, err = s.client.PresignedPutObject(ctx, bucketName, name, expires)
-
 	return
 }
 
-// 获取储存桶列表
+// ListBuckets 获取当前 S3 客户端可访问的所有存储桶列表
+// 出错时返回 nil
 func (s *Mod) ListBuckets() []minio.BucketInfo {
 	buckets, err := s.client.ListBuckets(ctx)
-	//g.Dump(buckets)
 	if err != nil {
-		//fmt.Println(err)
 		return nil
 	}
 	return buckets
 }
 
-// PutObject 上传文件到指定的存储桶中。
+// PutObject 上传文件到指定的存储桶中
-//
+// 支持指定文件大小，未指定时将读取文件直到结束
-// @Description: 上传一个文件到指定的存储桶。
-// @receiver s *Mod: 表示调用此方法的Mod实例。
-// @param f io.Reader: 文件的读取器，用于读取待上传的文件内容。
-// @param name string: 待上传文件的名称。
-// @param bucketName string: 存储桶的名称。
-// @param _size ...int64: 可选参数，指定上传文件的大小。如果未提供，则默认为-1，表示读取文件直到EOF。
-// @return res minio.UploadInfo: 上传成功后返回的上传信息。
-// @return err error: 如果上传过程中出现错误，则返回错误信息。
 func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...int64) (res minio.UploadInfo, err error) {
-	// 初始化文件大小为-1，表示将读取文件至结束。
+	// 初始化文件大小为 -1，表示将读取文件至结束
 	var size = int64(-1)
-	// 如果提供了文件大小，则使用提供的大小值。
 	if len(_size) > 0 {
 		size = _size[0]
 	}
-
+	// 调用 S3 客户端上传文件，设置内容类型为 "application/octet-stream"
-	// 调用client的PutObject方法上传文件，并设置内容类型为"application/octet-stream"。
 	res, err = s.client.PutObject(ctx, bucketName, name, f, size, minio.PutObjectOptions{ContentType: "application/octet-stream"})
 	if err != nil {
+		// 记录上传错误日志
 		g.Log().Error(ctx, err)
 	}
 	return
 }
 
-// RemoveObject 删除文件
+// RemoveObject 从指定存储桶中删除指定名称的文件
 func (s *Mod) RemoveObject(name string, bucketName string) (err error) {
 	opts := minio.RemoveObjectOptions{
 		//GovernanceBypass: true,
 		//VersionID:        "myversionid",
 	}
+	// 调用 S3 客户端删除文件
 	err = s.client.RemoveObject(ctx, bucketName, name, opts)
 	return
 }
 
-// ListObjects 文件列表
+// ListObjects 获取指定存储桶中指定前缀的文件列表
+// 返回一个包含文件信息的通道
 func (s *Mod) ListObjects(bucketName string, prefix string) (res <-chan minio.ObjectInfo, err error) {
+	// 调用 S3 客户端获取文件列表
 	res = s.client.ListObjects(ctx, bucketName, minio.ListObjectsOptions{
 		Prefix: prefix,
 	})
 	return
 }
 
-// SetBucketPolicy 设置bucket或对象前缀的访问权限
+// SetBucketPolicy 设置指定存储桶或对象前缀的访问策略
+// 目前使用固定的策略，可根据需求修改
 func (s *Mod) SetBucketPolicy(bucketName string, prefix string) (err error) {
-
+	// 定义访问策略
 	policy := `{"Version": "2012-10-17","Statement": [{"Action": ["s3:GetObject"],"Effect": "Allow","Principal": {"AWS": ["*"]},"Resource": ["arn:aws:s3:::my-bucketname/*"],"Sid": ""}]}`
-
+	// 调用 S3 客户端设置存储桶策略
 	err = s.client.SetBucketPolicy(ctx, bucketName, policy)
 	return
 }
 
-// GetUrl 获取文件访问地址
+// GetUrl 获取文件的访问地址
+// 支持返回默认文件地址，根据 SSL 配置生成不同格式的 URL
 func (s *Mod) GetUrl(filePath string, defaultFile ...string) (url string) {
 	bucketName := s.cfg.BucketNameCdn
 	get := s.cfg.Url
 
-	//如果没有图片，返回默认的图片地址
+	// 如果没有指定文件路径，且提供了默认文件路径，则使用默认路径
 	if filePath == "" && len(defaultFile) > 0 {
 		filePath = defaultFile[0]
 	}
@@ -262,6 +196,7 @@ func (s *Mod) GetUrl(filePath string, defaultFile ...string) (url string) {
 	return
 }
 
+// GetPath 从文件访问 URL 中提取文件路径
 func (s *Mod) GetPath(url string) (filePath string) {
 	bucketName := s.cfg.BucketNameCdn
 	get := s.cfg.Url
@@ -269,21 +204,22 @@ func (s *Mod) GetPath(url string) (filePath string) {
 	return url[len(get+bucketName)+1:]
 }
 
-// 复制文件
+// CopyObject 在指定存储桶内复制文件
+// 支持指定源文件和目标文件路径
 func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
-
+	// 定义目标文件选项
-	// 原始文件
 	var dst = minio.CopyDestOptions{
 		Bucket: bucketName,
 		Object: dstStr,
 	}
 
-	// 新文件
+	// 定义源文件选项
 	var src = minio.CopySrcOptions{
 		Bucket: bucketName,
 		Object: srcStr,
 	}
 
+	// 调用 S3 客户端复制文件
 	_, err = s.client.CopyObject(ctx, dst, src)
 	return
 }

pkg/websocket/socket_new.go

@@ -14,6 +14,7 @@ import (
 	"github.com/gogf/gf/v2/util/guid"
 	"github.com/gorilla/websocket"
 	"google.golang.org/protobuf/proto"
+	"net/http"
 	"sync"
 	"time"
 )
@@ -36,7 +37,7 @@ type WebsocketData struct {
 	Uid    int64           `json:"uid" dc:"用户编号"`
 	Groups []string        `json:"groups" dc:"群组"`
 	Ctx    context.Context `json:"ctx" dc:""`
-	RoomId int             `json:"roomId" dc:"房间编号"`
+	RoomId string          `json:"roomId" dc:"房间编号"`
 }
 
 func NewV1() *SocketV1 {
@@ -57,14 +58,21 @@ type SocketInterface interface {
 func (s *SocketV1) Load(serv *ghttp.Server, prefix string) {
 	//websocket服务启动
 	serv.Group(prefix, func(group *ghttp.RouterGroup) {
-
 		var websocketCfg = websocket.Upgrader{
 			ReadBufferSize:  1024,
 			WriteBufferSize: 1024,
+			CheckOrigin: func(r *http.Request) bool {
+				// In production, you should implement proper origin checking
+				return true
+			},
+			Error: func(w http.ResponseWriter, r *http.Request, status int, reason error) {
+				g.Log().Errorf(r.Context(), "WebSocket error: %v", reason)
+			},
 		}
 		group.Bind(
 			func(r *ghttp.Request) {
 				ctx := r.Context()
+
 				ws, err := websocketCfg.Upgrade(r.Response.Writer, r.Request, nil)
 				if err != nil {
 					glog.Error(ctx, err)
@@ -84,19 +92,17 @@ func (s *SocketV1) Load(serv *ghttp.Server, prefix string) {
 //	@Description:
 //	@receiver s
 //	@param conn
-func (s *SocketV1) OnConnect(ctx context.Context, conn *websocket.Conn) {
+func (s *SocketV1) OnConnect(ctx context.Context, ws *websocket.Conn) {
-
-	defer conn.Close()
 	id := guid.S()
-	ip := conn.RemoteAddr().String()
+	ip := ws.RemoteAddr().String()
-	data := &WebsocketData{
+	conn := &WebsocketData{
 		Uuid:   id,
-		Ws:     conn,
+		Ws:     ws,
 		Ctx:    ctx,
 		Groups: make([]string, 0),
-		RoomId: -1,
+		RoomId: "",
 	}
-	m.Set(id, data)
+	m.Set(id, conn)
 
 	//defer delete(Conn, id)
 
@@ -106,22 +112,23 @@ func (s *SocketV1) OnConnect(ctx context.Context, conn *websocket.Conn) {
 
 	//用户登录钩子执行
 	for _, connect := range OnConnectHandlers {
-		connect(data)
+		connect(conn)
 	}
 
 	for {
 		//进入当前连接线程拥堵
-		msgType, msg, err := conn.ReadMessage()
+		msgType, msg, err := ws.ReadMessage()
 		s.Type = msgType
 		if err != nil {
 			//客户端断开返回错误，断开当前连接
 			//g.Log().Error(ctx, err)
 			break
 		}
-		s.OnMessage(m.Get(id).(*WebsocketData), msg, msgType)
+		s.OnMessage(conn, msg, msgType)
 	}
 	//关闭连接触发
-	s.OnClose(data)
+
+	s.OnClose(conn)
 	g.Log().Debugf(ctx, "断开连接:uuid=%v,ip=%v", id, ip)
 }
 
@@ -338,8 +345,8 @@ func (s *SocketV1) OnClose(conn *WebsocketData) {
 	// 1. 更新连接状态或从连接池移除
 	// 2. 发送通知或清理关联资源
 	// 3. 执行特定于业务的断开处理
-	m.Remove(conn.Uuid)
 	conn.Ws.Close()
+	m.Remove(conn.Uuid)
 }
 
 // 是否在线

service/game_act.go

@@ -6,6 +6,8 @@
 package service
 
 import (
+	"context"
+
 	"github.com/gogf/gf/v2/frame/g"
 )
 
@@ -29,8 +31,8 @@ type (
 		// @param data interface{}: 要存储的活动信息数据。
 		// @return err error: 返回错误信息，如果操作成功，则返回nil。
 		Set(uid int64, actId int, data interface{}) (err error)
-		Saves() (err error)
+		Saves(ctx context.Context) (err error)
-		Save(actId int) (err error)
+		Save(ctx context.Context, actId int) (err error)
 		// 清空GetRedDot缓存
 		RefreshGetRedDotCache(uid int64)
 		Del(uid int64, actId int)

service/system_cron.go

@@ -6,6 +6,8 @@
 package service
 
 import (
+	"context"
+
 	v1 "github.com/ayflying/utility_go/api/system/v1"
 	"github.com/gogf/gf/v2/net/gclient"
 )
@@ -24,16 +26,32 @@ type (
 		// AddCron 添加一个定时任务到相应的调度列表中。
 		//
 		// @Description: 根据指定的类型将函数添加到不同的任务列表中，以供后续执行。
+		// 确保自定义任务正确处理上下文取消信号，即可充分发挥超时打断功能。
 		// @receiver s: sSystemCron的实例，代表一个调度系统。
 		// @param typ: 任务的类型，决定该任务将被添加到哪个列表中。对应不同的时间间隔。
 		// @param _func: 要添加的任务函数，该函数执行时应该返回一个error。
+		// deprecated: 弃用，请使用 AddCronV2
 		AddCron(typ v1.CronType, _func func() error)
+		// AddCronV2  添加一个定时任务到相应的调度列表中。
+		//
+		// @Description: 根据指定的类型将函数添加到不同的任务列表中，以供后续执行。
+		// @receiver s: sSystemCron的实例，代表一个调度系统。
+		// @param typ: 任务的类型，决定该任务将被添加到哪个列表中。对应不同的时间间隔。
+		// @param _func: 要添加的任务函数，该函数执行时应该返回一个error。
+		AddCronV2(typ v1.CronType, _func func(context.Context) error)
 		// StartCron 开始计划任务执行
 		//
 		//	@Description:
 		//	@receiver s
 		//	@return err
 		StartCron() (err error)
+		// AddFuncChan 添加方法到通道
+		AddFuncChan(list []func(context.Context) error)
+		// RunFuncChan 统一执行方法
+		RunFuncChan()
+		// RunFunc 统一执行方法
+		// deprecated: 弃用，会造成周期任务并发执行，to service.SystemCron().AddFuncChan
+		RunFunc(list []func() error)
 	}
 )
 

utility.go

@@ -17,6 +17,7 @@ func init() {
 	g.Log().Debug(ctx, "utility_go init启动完成")
 	// 初始化配置
 	var err = boot.Boot()
+
 	if err != nil {
 		panic(err)
 	}