Compare commits

..

16 Commits

Author SHA1 Message Date
ayflying
569937c67f 增加计划任务的打断机制 2025-04-22 12:15:53 +08:00
ayflying
d6bfe1c2fb 计划任务定时器修改,改为通道顺序执行 2025-04-22 11:15:51 +08:00
ayflying
16da554a60 完善weboscket的启动 2025-04-21 14:35:24 +08:00
ayflying
7f6635fb91 分离ip库的下载时机 2025-04-21 14:16:58 +08:00
ayflying
0605302db6 调整接口的逻辑 2025-04-18 18:36:50 +08:00
ayflying
5f2fe5dcb2 增加群组广播 2025-04-14 17:05:08 +08:00
ayflying
452aefe3d0 调整长连接库 2025-04-14 11:17:33 +08:00
ayflying
97bf7fc390 修改上传版本呢目录 2025-04-14 11:17:19 +08:00
ayflying
08f2b2d9bc 长连接接口增加玩家信息绑定 2025-04-10 15:42:08 +08:00
ayflying
366ddb45ea 修改导表工具,修改websocket协议的钩子 2025-04-09 16:50:54 +08:00
ayflying
09a9f14a29 修改ExcelTime2Time方法 2025-04-09 11:47:07 +08:00
ayflying
e0afb55bb2 随机算法小修改 2025-04-08 18:43:37 +08:00
ayflying
a469692358 修改随机方法 2025-04-02 16:20:03 +08:00
ayflying
bce7131d9f 开启qps 2025-04-02 15:55:43 +08:00
ayflying
a53f7b718d 修改目录 2025-04-02 15:53:05 +08:00
ayflying
75624ff0b7 使用boot统一注册执行代码 2025-04-02 15:49:42 +08:00
22 changed files with 693 additions and 236 deletions

View File

@@ -26,7 +26,8 @@ var (
}, },
Examples: "make -m act -i 1: 创建活动1的接口与服务文件 \n" + Examples: "make -m act -i 1: 创建活动1的接口与服务文件 \n" +
"make -m logic -n test: 创建test的服务文件 \n" + "make -m logic -n test: 创建test的服务文件 \n" +
"make -m config -n test: 创建配置文件", "make -m config -n test: 创建配置文件 \n" +
"make -m socket -n test: 创建socket文件 \n",
Func: func(ctx context.Context, parser *gcmd.Parser) (err error) { Func: func(ctx context.Context, parser *gcmd.Parser) (err error) {
//g.Dump(parser.GetOptAll(), parser.GetArgAll()) //g.Dump(parser.GetOptAll(), parser.GetArgAll())
@@ -53,6 +54,12 @@ var (
return return
} }
err = this.Config(name) err = this.Config(name)
case "socket":
var name = parser.GetOpt("name").String()
if name == "" {
return
}
err = this.Socket(name)
} }
return return
@@ -116,3 +123,25 @@ func (c *cMake) Config(name string) (err error) {
return return
} }
func (c *cMake) Socket(name string) (err error) {
var filePath = fmt.Sprintf("internal/socket/%s/%s_new.go", name, gstr.CaseSnake(name))
//生成文件不覆盖
if !gfile.Exists(filePath) {
// 生成目录文件
get, _ := fs.ReadFile(ConfigFiles, "make/socket")
fileStr := string(get)
fileStr = gstr.Replace(fileStr, "{name}", name)
err = gfile.PutContents(filePath, fileStr)
//生成方法文件
var filePath2 = fmt.Sprintf("internal/socket/%s/%s.go", name, gstr.CaseSnake(name))
get, _ = fs.ReadFile(ConfigFiles, "make/socket2")
fileStr = string(get)
fileStr = gstr.Replace(fileStr, "{name}", name)
fileStr = gstr.Replace(fileStr, "{func}", gstr.CaseCamel(name))
err = gfile.PutContents(filePath2, fileStr)
}
return
}

10
cmd/make/socket Normal file
View File

@@ -0,0 +1,10 @@
package {name}
type {name} struct {
}
func New() *{name} {
return &{name}{}
}
func init() {}

19
cmd/make/socket2 Normal file
View File

@@ -0,0 +1,19 @@
package {name}
import (
"github.com/ayflying/utility_go/pkg"
"github.com/ayflying/utility_go/pkg/websocket"
"google.golang.org/protobuf/proto"
)
func (s *{name}) {func}Handler(conn *websocket.WebsocketData, req any) (err error) {
var data = &v1.{func}2S{}
err = proto.Unmarshal(req.([]byte), data)
var res = &v1.{func}2C{}
resp, err := proto.Marshal(res)
pkg.Websocket().Send(000000, conn.Uid, resp)
return
}

View File

@@ -2,13 +2,14 @@ package cmd
import ( import (
"context" "context"
"github.com/ayflying/utility_go/package/s3" "github.com/ayflying/utility_go/pkg"
"github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/net/ghttp" "github.com/gogf/gf/v2/net/ghttp"
"github.com/gogf/gf/v2/os/gcfg" "github.com/gogf/gf/v2/os/gcfg"
"github.com/gogf/gf/v2/os/gcmd" "github.com/gogf/gf/v2/os/gcmd"
"github.com/gogf/gf/v2/os/gctx" "github.com/gogf/gf/v2/os/gctx"
"os" "os"
"path"
"time" "time"
) )
@@ -17,6 +18,8 @@ type serverCfg struct {
Address string `json:"address" dc:"服务地址"` Address string `json:"address" dc:"服务地址"`
Prod bool `json:"prod" dc:"是否生产环境"` Prod bool `json:"prod" dc:"是否生产环境"`
S3 string `json:"s3" dc:"使用哪个对象储存中转"` S3 string `json:"s3" dc:"使用哪个对象储存中转"`
Arch string `json:"arch" dc:"架构"`
System string `json:"system" dc:"系统"`
} }
type UpdateReq struct { type UpdateReq struct {
@@ -24,8 +27,6 @@ type UpdateReq struct {
FileUrl string `json:"file_url" dc:"文件地址"` FileUrl string `json:"file_url" dc:"文件地址"`
} }
var s3Mod *s3.Mod
var ( var (
Update = gcmd.Command{ Update = gcmd.Command{
Name: "update", Name: "update",
@@ -36,9 +37,30 @@ var (
g.Log().Info(ctx, "准备上传更新文件") g.Log().Info(ctx, "准备上传更新文件")
//加载编辑配置文件 //加载编辑配置文件
g.Cfg("hack").GetAdapter().(*gcfg.AdapterFile).SetFileName("hack/config.yaml") g.Cfg("hack").GetAdapter().(*gcfg.AdapterFile).SetFileName("hack/config.yaml")
//获取文件名
getFileName, err := g.Cfg("hack").Get(ctx, "gfcli.build.name") getFileName, err := g.Cfg("hack").Get(ctx, "gfcli.build.name")
Filename := getFileName.String() Filename := getFileName.String()
////获取架构
//getArch, err := g.Cfg("hack").Get(ctx, "gfcli.build.arch")
//Arch := getArch.String()
//if Arch == "" {
// Arch = "amd64"
//}
//
////获取操作系统
//getSystem, err := g.Cfg("hack").Get(ctx, "gfcli.build.system")
//System := getSystem.String()
//
//if System == "" {
// System = "linux"
//}
//var systemName = System + "_" + Arch
//获取版本号
getVersion, err := g.Cfg("hack").Get(ctx, "gfcli.build.version")
Version := getVersion.String()
var list []*serverCfg var list []*serverCfg
serverList := g.Cfg().MustGet(ctx, "server_list") serverList := g.Cfg().MustGet(ctx, "server_list")
serverList.Scan(&list) serverList.Scan(&list)
@@ -59,7 +81,9 @@ var (
g.Dump("需要更新的服务器", list) g.Dump("需要更新的服务器", list)
//获取上传链接 //获取上传链接
var url = make(map[string]string) var url = make(map[string]string)
filename := "linux_amd64/" + Filename var system = make(map[string]string)
//filename := "linux_amd64/" + Filename
//filename := path.Join(Version, "linux_amd64", Filename)
client := g.Client() client := g.Client()
client.SetTimeout(time.Minute) client.SetTimeout(time.Minute)
@@ -73,14 +97,28 @@ var (
} }
//查询当前上传地址是否存在 //查询当前上传地址是否存在
if _, ok := url[v.S3]; !ok { _, ok2 := system[v.System+v.Arch]
if _, ok := url[v.S3]; !ok || !ok2 {
var systemName = "linux_amd64"
if v.Arch != "" && v.System != "" {
systemName = v.System + "_" + v.Arch
}
var filename = path.Join(Version, systemName, Filename)
g.Log().Debugf(ctx, "当前上传文件:"+filename)
url[v.S3], err = UploadS3(v.S3, filename) url[v.S3], err = UploadS3(v.S3, filename)
if err != nil { if err != nil {
g.Log().Error(ctx, err) g.Log().Error(ctx, err)
return return
} }
} system[v.System+v.Arch] = filename
if err != nil {
g.Log().Error(ctx, err)
return
}
}
g.Log().Debugf(ctx, "准备同步服务器:%v,url=%v", v.Name, address+"/callback/update") g.Log().Debugf(ctx, "准备同步服务器:%v,url=%v", v.Name, address+"/callback/update")
get, err := client.Post(ctx, address+"/callback/update", &UpdateReq{ get, err := client.Post(ctx, address+"/callback/update", &UpdateReq{
FileUrl: url[v.S3], FileUrl: url[v.S3],
@@ -107,9 +145,13 @@ var (
func UploadS3(typ string, filename string) (res string, err error) { func UploadS3(typ string, filename string) (res string, err error) {
//updateServerS3Name, _ := g.Config().Get(ctx, "update_server_s3_name") //updateServerS3Name, _ := g.Config().Get(ctx, "update_server_s3_name")
s3Mod = s3.New(typ) var s3Mod = pkg.S3(typ)
bucketName := s3Mod.GetCfg().BucketName bucketName := s3Mod.GetCfg().BucketName
obj, err := os.Open(filename) obj, err := os.Open(filename)
if err != nil {
return
}
defer obj.Close()
ff, err := obj.Stat() ff, err := obj.Stat()
_, err = s3Mod.PutObject(obj, filename, bucketName, ff.Size()) _, err = s3Mod.PutObject(obj, filename, bucketName, ff.Size())
if err != nil { if err != nil {

View File

@@ -1,6 +1,7 @@
package elasticsearch package elasticsearch
import ( import (
"github.com/ayflying/utility_go/internal/boot"
"github.com/gogf/gf/v2/database/gdb" "github.com/gogf/gf/v2/database/gdb"
"github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/frame/g"
) )
@@ -14,17 +15,20 @@ const (
quoteChar = "`" quoteChar = "`"
) )
func Init() { func init() {
var ( boot.AddFunc(func() {
err error var (
driverObj = New() err error
driverNames = g.SliceStr{"es", "elasticsearch"} driverObj = New()
) driverNames = g.SliceStr{"es", "elasticsearch"}
for _, driverName := range driverNames { )
if err = gdb.Register(driverName, driverObj); err != nil { for _, driverName := range driverNames {
panic(err) if err = gdb.Register(driverName, driverObj); err != nil {
panic(err)
}
} }
} })
} }
// New create and returns a driver that implements gdb.Driver, which supports operations for MySQL. // New create and returns a driver that implements gdb.Driver, which supports operations for MySQL.

5
go.mod
View File

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

View File

@@ -1,12 +1,10 @@
package boot package boot
import ( import (
"context"
v1 "github.com/ayflying/utility_go/api/system/v1" v1 "github.com/ayflying/utility_go/api/system/v1"
"github.com/ayflying/utility_go/drivers/db/elasticsearch"
"github.com/ayflying/utility_go/pkg/aycache"
"github.com/ayflying/utility_go/service" "github.com/ayflying/utility_go/service"
"github.com/gogf/gf/v2/os/gctx" "github.com/gogf/gf/v2/os/gctx"
"math"
) )
var ( var (
@@ -18,25 +16,14 @@ func Boot() (err error) {
err = service.SystemCron().StartCron() 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() return service.GameAct().Saves(ctx)
})
//初始化ES
elasticsearch.Init()
//初始化指标
service.SystemCron().AddCron(v1.CronType_MINUTE, func() error {
aycache.QPS.Set(math.Round(float64(aycache.QPSCount) / 60))
aycache.QPSCount = 0
return nil
}) })
//初始化自启动方法 //初始化自启动方法
for _, v := range _func { for _, v := range _func {
v() v()
} }
return nil return nil
} }

View File

@@ -1,6 +1,7 @@
package gameAct package gameAct
import ( import (
"context"
"fmt" "fmt"
"github.com/ayflying/utility_go/internal/model/do" "github.com/ayflying/utility_go/internal/model/do"
"github.com/ayflying/utility_go/internal/model/entity" "github.com/ayflying/utility_go/internal/model/entity"
@@ -105,16 +106,16 @@ func (s *sGameAct) Set(uid int64, actId int, data interface{}) (err error) {
return return
} }
func (s *sGameAct) Saves() (err error) { func (s *sGameAct) Saves(ctx context.Context) (err error) {
//遍历执行 //遍历执行
ActList.Iterator(func(i interface{}) bool { ActList.Iterator(func(i interface{}) bool {
err = s.Save(i.(int)) err = s.Save(ctx, i.(int))
return true return true
}) })
return 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) cacheKey := fmt.Sprintf("act:%v:*", actId)
//获取当前用户的key值 //获取当前用户的key值
@@ -149,8 +150,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 continue
} }

View File

@@ -11,7 +11,8 @@ import (
) )
var ( var (
ctx = gctx.New() ctx = gctx.New()
wait = false
) )
type sIp2region struct { type sIp2region struct {
@@ -19,13 +20,16 @@ type sIp2region struct {
} }
func New() *sIp2region { func New() *sIp2region {
s := &sIp2region{}
s.Load() return &sIp2region{}
return s
} }
func init() { func init() {
service.RegisterIp2Region(New()) service.RegisterIp2Region(New())
//boot.AddFunc(func() {
// service.Ip2Region().Load()
//})
} }
// Load 加载到内存中 // Load 加载到内存中
@@ -34,12 +38,20 @@ func init() {
// @receiver s *sIp2region: sIp2region的实例。 // @receiver s *sIp2region: sIp2region的实例。
func (s *sIp2region) Load() { func (s *sIp2region) Load() {
var err error var err error
var dbPath = "/runtime/library/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) { if gfile.IsEmpty(dbPath) {
wait = true
defer func() {
wait = false
}()
g.Log().Debug(ctx, "等待下载ip库文件")
//下载文件 //下载文件
putData, err2 := g.Client().Discovery(nil). putData, err2 := g.Client().Discovery(nil).Get(ctx, url)
Get(ctx, "https://resource.luoe.cn/attachment/ip2region.xdb")
if err2 != nil { if err2 != nil {
return return
} }
@@ -65,6 +77,9 @@ func (s *sIp2region) Load() {
} }
func (s *sIp2region) GetIp(ip string) (res []string) { func (s *sIp2region) GetIp(ip string) (res []string) {
//初始化加载
s.Load()
res = make([]string, 5) res = make([]string, 5)
if s.searcher == nil { if s.searcher == nil {
return return

View File

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

View File

@@ -14,7 +14,7 @@ import (
) )
var ( var (
ctx = gctx.New() //ctx = gctx.New()
startTime *gtime.Time startTime *gtime.Time
) )
@@ -22,40 +22,45 @@ var (
// 它包含了不同时间周期的任务,如秒、分钟、小时、天、周、月、年以及特定的工作日任务。 // 它包含了不同时间周期的任务,如秒、分钟、小时、天、周、月、年以及特定的工作日任务。
type sSystemCron struct { type sSystemCron struct {
//互斥锁 //互斥锁
Lock sync.Mutex 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 { func New() *sSystemCron {
return &sSystemCron{} return &sSystemCron{
taskChan: make(chan func(context.Context) error, 2),
TaskTimeout: time.Minute * 30,
}
} }
func init() { func init() {
@@ -65,13 +70,32 @@ func init() {
// AddCron 添加一个定时任务到相应的调度列表中。 // AddCron 添加一个定时任务到相应的调度列表中。
// //
// @Description: 根据指定的类型将函数添加到不同的任务列表中,以供后续执行。 // @Description: 根据指定的类型将函数添加到不同的任务列表中,以供后续执行。
// 确保自定义任务正确处理上下文取消信号,即可充分发挥超时打断功能。
// @receiver s: sSystemCron的实例代表一个调度系统。 // @receiver s: sSystemCron的实例代表一个调度系统。
// @param typ: 任务的类型,决定该任务将被添加到哪个列表中。对应不同的时间间隔。 // @param typ: 任务的类型,决定该任务将被添加到哪个列表中。对应不同的时间间隔。
// @param _func: 要添加的任务函数该函数执行时应该返回一个error。 // @param _func: 要添加的任务函数该函数执行时应该返回一个error。
// deprecated: 弃用,请使用 AddCronV2
func (s *sSystemCron) AddCron(typ v1.CronType, _func func() error) { 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() s.Lock.Lock()
defer s.Lock.Unlock() defer s.Lock.Unlock()
//
//ctx := gctx.New()
//newFunc := func()
switch typ { switch typ {
case v1.CronType_SECOND: case v1.CronType_SECOND:
@@ -118,140 +142,123 @@ func (s *sSystemCron) StartCron() (err error) {
} }
startTime = gtime.Now() 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, "每秒定时器") //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, "每分钟定时器") //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, "每小时定时器") 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, "每日定时器") 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, "每周一定时器") 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, "每周二定时器") 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, "周三定时器") 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, "周四定时器") 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, "周五定时器") 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, "周六定时器") 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, "周日定时器") 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, "每月定时器") 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, "每年定时器") g.Log().Debug(ctx, "每年定时器")
err = s.monthlyTask() s.yearlyTask()
}) })
//统一执行方法
s.RunFuncChan()
return return
} }
// 每妙任务 // 每妙任务
func (s *sSystemCron) secondlyTask() (err error) { func (s *sSystemCron) secondlyTask() {
if len(s.SecondlyTask) == 0 { if len(s.SecondlyTask) == 0 {
return return
} }
for _, _func := range s.SecondlyTask { s.AddFuncChan(s.SecondlyTask)
err = _func()
if err != nil {
g.Log().Error(ctx, err)
}
}
return return
} }
// 每分钟任务 // 每分钟任务
func (s *sSystemCron) minutelyTask() (err error) { func (s *sSystemCron) minutelyTask() {
if len(s.MinutelyTask) == 0 { if len(s.MinutelyTask) == 0 {
return return
} }
for _, _func := range s.MinutelyTask { s.AddFuncChan(s.MinutelyTask)
err = _func()
if err != nil {
g.Log().Error(ctx, err)
}
}
return return
} }
// 每小时任务 // 每小时任务
func (s *sSystemCron) hourlyTask() (err error) { func (s *sSystemCron) hourlyTask() {
if len(s.HourlyTask) == 0 { if len(s.HourlyTask) == 0 {
return return
} }
for _, _func := range s.HourlyTask { s.AddFuncChan(s.HourlyTask)
err = _func()
if err != nil {
g.Log().Error(ctx, err)
}
}
return return
} }
// 每天任务 // 每天任务
func (s *sSystemCron) dailyTask() (err error) { func (s *sSystemCron) dailyTask() {
if len(s.DailyTask) == 0 { if len(s.DailyTask) == 0 {
return return
} }
for _, _func := range s.DailyTask { s.AddFuncChan(s.DailyTask)
err = _func()
if err != nil {
g.Log().Error(ctx, err)
}
}
return 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 { switch day {
case 1: case 1:
arr = s.MondayTask arr = s.MondayTask
@@ -275,39 +282,91 @@ func (s *sSystemCron) weeklyTask(day int) (err error) {
if len(arr) == 0 { if len(arr) == 0 {
return 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 return
} }
// 每月任务 // 每月任务
func (s *sSystemCron) monthlyTask() (err error) { func (s *sSystemCron) monthlyTask() {
if len(s.MonthlyTask) == 0 { if len(s.MonthlyTask) == 0 {
return return
} }
for _, _func := range s.MonthlyTask { s.AddFuncChan(s.MonthlyTask)
err = _func() return
if err != nil { }
g.Log().Error(ctx, err)
//每年任务
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 return
} }

View File

@@ -111,8 +111,12 @@ func (s *Excel) RemoveComments(list []interface{}, json []string) []interface{}
// 遍历当前元素的每个键值对 // 遍历当前元素的每个键值对
for _, v3 := range v2.(g.Map) { for _, v3 := range v2.(g.Map) {
// 如果字符串中存在//则跳过不写入temp // 如果字符串中存在//则跳过不写入temp
if gstr.Contains(gconv.String(v3), "//") { //if gstr.Contains(gconv.String(v3), "//") {
//delKey = append(delKey, k2) // //delKey = append(delKey, k2)
// add = false
// break
//}
if strings.HasPrefix(gconv.String(v3), "//") {
add = false add = false
break break
} }

View File

@@ -1,11 +1,15 @@
package aycache package aycache
import ( import (
v1 "github.com/ayflying/utility_go/api/system/v1"
"github.com/ayflying/utility_go/internal/boot"
"github.com/ayflying/utility_go/pkg/aycache/drive" "github.com/ayflying/utility_go/pkg/aycache/drive"
drive2 "github.com/ayflying/utility_go/pkg/aycache/drive" drive2 "github.com/ayflying/utility_go/pkg/aycache/drive"
"github.com/ayflying/utility_go/service"
"github.com/gogf/gf/v2/os/gcache" "github.com/gogf/gf/v2/os/gcache"
"github.com/prometheus/client_golang/prometheus" "github.com/prometheus/client_golang/prometheus"
"github.com/prometheus/client_golang/prometheus/promauto" "github.com/prometheus/client_golang/prometheus/promauto"
"math"
) )
type Mod struct { type Mod struct {
@@ -22,6 +26,18 @@ var (
) )
) )
func init() {
boot.AddFunc(func() {
//初始化指标
service.SystemCron().AddCron(v1.CronType_MINUTE, func() error {
QPS.Set(math.Round(float64(QPSCount) / 60))
QPSCount = 0
return nil
})
})
}
func New(_name ...string) gcache.Adapter { func New(_name ...string) gcache.Adapter {
var cacheAdapterObj gcache.Adapter var cacheAdapterObj gcache.Adapter

View File

@@ -1,16 +1,49 @@
package websocket package websocket
import "context" import "google.golang.org/protobuf/proto"
// 定义一个处理方法的类型 // 定义一个处理方法的类型
type Handler func(ctx context.Context, req any) (err error) type Handler func(conn *WebsocketData, req any) (err error)
type Handler2 func(conn *WebsocketData)
type HandlerMessage func(conn *WebsocketData, req any)
type PbType func(cmd int32, data []byte, code int32, msg string) proto.Message
type PbType2 func(data []byte) (int, []byte)
// 路由器的处理映射 // 路由器的处理映射
var ( var (
handlers = make(map[int]Handler) handlers = make(map[int]Handler)
OnConnectHandlers = make([]Handler2, 0)
OnCloseHandlers = make([]Handler2, 0)
onMessageHandlers = make([]HandlerMessage, 0)
Byte2Pb = make([]PbType, 0)
Pb2Bytes = make([]PbType2, 0)
) )
// 注册方法将某个消息路由器ID和对应的处理方法关联起来 // 注册方法将某个消息路由器ID和对应的处理方法关联起来
func (s *SocketV1) RegisterRouter(cmd int, handler Handler) { func (s *SocketV1) RegisterRouter(cmd int, handler Handler) {
handlers[cmd] = handler handlers[cmd] = handler
} }
//注册方法,讲长连接登陆方法进行注册
func (s *SocketV1) RegisterOnConnect(_func Handler2) {
OnConnectHandlers = append(OnConnectHandlers, _func)
}
func (s *SocketV1) RegisterOnClose(_func Handler2) {
OnCloseHandlers = append(OnCloseHandlers, _func)
}
//注册方法长连接消息体
func (s *SocketV1) RegisterMessage(_func HandlerMessage) {
onMessageHandlers = append(onMessageHandlers, _func)
}
func (s *SocketV1) RegisterByte2Pb(_func PbType) {
Byte2Pb = append(Byte2Pb, _func)
}
func (s *SocketV1) RegisterPb2Byte(_func PbType2) {
Pb2Bytes = append(Pb2Bytes, _func)
}

View File

@@ -2,57 +2,77 @@ package websocket
import ( import (
"context" "context"
"fmt"
"github.com/ayflying/utility_go/pkg/aycache"
"github.com/ayflying/utility_go/tools"
"github.com/gogf/gf/v2/container/gmap" "github.com/gogf/gf/v2/container/gmap"
"github.com/gogf/gf/v2/encoding/gjson"
"github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/net/ghttp"
"github.com/gogf/gf/v2/os/gctx" "github.com/gogf/gf/v2/os/gctx"
"github.com/gogf/gf/v2/os/glog" "github.com/gogf/gf/v2/os/glog"
"github.com/gogf/gf/v2/util/gconv" "github.com/gogf/gf/v2/util/guid"
"github.com/google/uuid"
"github.com/gorilla/websocket" "github.com/gorilla/websocket"
"google.golang.org/protobuf/proto"
"net/http"
"sync" "sync"
"github.com/gogf/gf/v2/net/ghttp" "time"
) )
type SocketV1 struct{} type SocketV1 struct {
Type int `json:"type"`
}
var ( var (
//ctx = gctx.New() //ctx = gctx.New()
//Conn map[uuid.UUID]*WebsocketData //Conn map[uuid.UUID]*WebsocketData
lock sync.Mutex lock sync.Mutex
cache = aycache.New("redis")
m = gmap.New(true) m = gmap.NewHashMap(true)
) )
type WebsocketData struct { type WebsocketData struct {
Ws *websocket.Conn Ws *websocket.Conn `json:"ws" dc:"websocket连接池"`
Uuid uuid.UUID Uuid string `json:"uuid" dc:"用户唯一标识"`
Uid int64 Uid int64 `json:"uid" dc:"用户编号"`
Ctx context.Context Groups []string `json:"groups" dc:"群组"`
Ctx context.Context `json:"ctx" dc:""`
RoomId int `json:"roomId" dc:"房间编号"`
} }
func NewV1() *SocketV1 { func NewV1() *SocketV1 {
return &SocketV1{} return &SocketV1{
Type: 2,
}
} }
type SocketInterface interface { type SocketInterface interface {
OnConnect(*websocket.Conn) Load(serv *ghttp.Server, prefix string)
OnMessage(*WebsocketData, []byte, int) OnConnect(ctx context.Context, conn *websocket.Conn)
Send(uuid.UUID, []byte) (err error) OnMessage(conn *WebsocketData, req []byte, msgType int)
SendAll(data []byte) Send(cmd int32, uid int64, req proto.Message)
OnClose(conn *websocket.Conn) SendAll(cmd int32, req proto.Message)
OnClose(conn *WebsocketData)
} }
func (s *SocketV1) Load(serv *ghttp.Server, prefix string) { func (s *SocketV1) Load(serv *ghttp.Server, prefix string) {
//websocket服务启动 //websocket服务启动
serv.Group(prefix, func(group *ghttp.RouterGroup) { serv.Group(prefix, func(group *ghttp.RouterGroup) {
var websocketCfg = websocket.Upgrader{ var websocketCfg = websocket.Upgrader{
ReadBufferSize: 1024, ReadBufferSize: 1024,
WriteBufferSize: 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( group.Bind(
func(r *ghttp.Request) { func(r *ghttp.Request) {
ctx := r.Context() ctx := r.Context()
ws, err := websocketCfg.Upgrade(r.Response.Writer, r.Request, nil) ws, err := websocketCfg.Upgrade(r.Response.Writer, r.Request, nil)
if err != nil { if err != nil {
glog.Error(ctx, err) glog.Error(ctx, err)
@@ -72,37 +92,43 @@ func (s *SocketV1) Load(serv *ghttp.Server, prefix string) {
// @Description: // @Description:
// @receiver s // @receiver s
// @param conn // @param conn
func (s *SocketV1) OnConnect(ctx context.Context, conn *websocket.Conn) { func (s *SocketV1) OnConnect(ctx context.Context, ws *websocket.Conn) {
//lock.Lock() id := guid.S()
//defer lock.Unlock() ip := ws.RemoteAddr().String()
conn := &WebsocketData{
defer conn.Close() Uuid: id,
id, _ := uuid.NewUUID() Ws: ws,
ip := conn.RemoteAddr().String() Ctx: ctx,
Groups: make([]string, 0),
data := &WebsocketData{ RoomId: -1,
Uuid: id,
Ws: conn,
Ctx: ctx,
} }
m.Set(id, data) m.Set(id, conn)
//defer delete(Conn, id) //defer delete(Conn, id)
//to := fmt.Sprintf("创建连接:%v,ip=%v", id, ip) to := fmt.Sprintf("创建连接:%v,ip=%v", id, ip)
g.Log().Debugf(ctx, to)
//s.Send(id, []byte(to)) //s.Send(id, []byte(to))
//用户登录钩子执行
for _, connect := range OnConnectHandlers {
connect(conn)
}
for { for {
//进入当前连接线程拥堵 //进入当前连接线程拥堵
msgType, msg, err := conn.ReadMessage() msgType, msg, err := ws.ReadMessage()
s.Type = msgType
if err != nil { if err != nil {
//客户端断开返回错误,断开当前连接 //客户端断开返回错误,断开当前连接
//g.Log().Error(ctx, err)
break break
} }
s.OnMessage(m.Get(id).(*WebsocketData), msg, msgType) s.OnMessage(conn, msg, msgType)
} }
//关闭连接触发 //关闭连接触发
s.OnClose(id, conn)
s.OnClose(conn)
g.Log().Debugf(ctx, "断开连接:uuid=%v,ip=%v", id, ip) g.Log().Debugf(ctx, "断开连接:uuid=%v,ip=%v", id, ip)
} }
@@ -113,67 +139,221 @@ func (s *SocketV1) OnConnect(ctx context.Context, conn *websocket.Conn) {
// @param msg // @param msg
// @param msgType // @param msgType
func (s *SocketV1) OnMessage(conn *WebsocketData, req []byte, msgType int) { func (s *SocketV1) OnMessage(conn *WebsocketData, req []byte, msgType int) {
//g.Log().Debugf(ctx, "收到消息:%v,type=%v,conn=%v", string(req), msgType, conn) s.Type = 2
//s.Send(conn.Uuid, msg) var cmd int
//s.SendAll(msg) var msg []byte
msgStr := string(req) //uid := conn.Uid
msg := msgStr[8:] for _, v := range Pb2Bytes {
cmd := gconv.Int(msgStr[:8]) cmd, msg = v(req)
//GetRouter(cmd, conn.Uid, msg) }
g.Log("cmd").Debugf(gctx.New(), fmt.Sprintf("from|%d|%d|%v", cmd, conn.Uid, gjson.MustEncodeString(req)))
//msgStr := string(req)
//cmd = gconv.Int(msgStr[:8])
//msg = []byte(msgStr[8:])
handler, exist := handlers[cmd] handler, exist := handlers[cmd]
if exist { if exist {
//匹配上路由器 //匹配上路由器
handler(conn.Ctx, msg) err := handler(conn, msg)
if err != nil {
g.Log().Error(conn.Ctx, err)
}
} else { } else {
//fmt.Println("未注册的路由器ID:", cmd) //fmt.Println("未注册的路由器ID:", cmd)
s.Send(conn.Uuid, []byte("未注册的协议号:"+msgStr[:8])) //s.Send(20000000, conn.Uid, []byte("未注册的协议号:"+strconv.Itoa(cmd)))
s.OnClose(conn.Uuid, conn.Ws) s.OnClose(conn)
return return
} }
} }
//绑定用户编号
func (s *SocketV1) BindUid(conn *WebsocketData, uid int64) {
lock.Lock()
defer lock.Unlock()
cacheKey := fmt.Sprintf("socket:uid:%d", uid)
g.Redis().Set(nil, cacheKey, conn.Uuid)
if conn.Uid == 0 {
conn.Uid = uid
}
}
//解绑用户
func (s *SocketV1) UnBindUid(uid int64) {
lock.Lock()
defer lock.Unlock()
cacheKey := fmt.Sprintf("socket:uid:%d", uid)
g.Redis().Del(nil, cacheKey)
}
// Uid2Uuid 用户编号转uuid唯一标识
func (s *SocketV1) Uid2Uuid(uid int64) (uuid string) {
cacheKey := fmt.Sprintf("socket:uid:%d", uid)
get, _ := g.Redis().Get(nil, cacheKey)
if get.IsNil() {
return
}
uuid = get.String()
//如果不在线了
if !m.Contains(uuid) {
// 解绑用户编号
s.UnBindUid(uid)
return
}
return
}
// SendUuid
//
// @Description:
// @receiver s
// @param uid
// @param data
func (s *SocketV1) SendUuid(cmd int32, uuidStr string, req proto.Message) {
if !m.Contains(uuidStr) {
return
}
//格式化数据
var data, err = proto.Marshal(req)
if err != nil {
g.Log().Error(gctx.New(), err)
return
}
conn := m.Get(uuidStr).(*WebsocketData)
//前置方法
for _, v := range Byte2Pb {
temp := v(cmd, data, 0, "")
data, _ = proto.Marshal(temp)
}
conn.Ws.WriteMessage(s.Type, data)
return
}
// Send // Send
// //
// @Description: // @Description:
// @receiver s // @receiver s
// @param uid // @param uid
// @param data // @param data
// @return err func (s *SocketV1) Send(cmd int32, uid int64, req proto.Message) {
func (s *SocketV1) Send(id uuid.UUID, data []byte) (err error) { g.Log("cmd").Debugf(gctx.New(), fmt.Sprintf("to|%d|%d|%v", cmd, uid, gjson.MustEncodeString(req)))
if !m.Contains(id) {
uuid := s.Uid2Uuid(uid)
if uuid == "" {
return return
} }
s.SendUuid(cmd, uuid, req)
conn := m.Get(id).(*WebsocketData)
conn.Ws.WriteMessage(1, data)
return return
} }
// 批量发送 // 批量发送
func (s *SocketV1) SendAll(data []byte) { func (s *SocketV1) SendAll(cmd int32, req proto.Message) {
g.Log("cmd").Debugf(gctx.New(), fmt.Sprintf("all:%d|-1|%v", cmd, gjson.MustEncodeString(req)))
//格式化数据
var data, err = proto.Marshal(req)
if err != nil {
g.Log().Error(gctx.New(), err)
return
}
for _, v := range Byte2Pb {
temp := v(cmd, data, 0, "")
data, _ = proto.Marshal(temp)
}
m.Iterator(func(k interface{}, v interface{}) bool { m.Iterator(func(k interface{}, v interface{}) bool {
//fmt.Printf("%v:%v ", k, v)
conn := v.(*WebsocketData) conn := v.(*WebsocketData)
conn.Ws.WriteMessage(1, data) conn.Ws.WriteMessage(s.Type, data)
return true return true
}) })
} }
//加入群组
func (s *SocketV1) JoinGroup(conn *WebsocketData, group string) {
conn.Groups = append(conn.Groups, group)
cacheKey := "websocket:group:" + group
get, _ := aycache.New("redis").Get(conn.Ctx, cacheKey)
var list = make(map[int64]string)
if !get.IsNil() {
get.Scan(&list)
}
list[conn.Uid] = conn.Uuid
cache.Set(conn.Ctx, cacheKey, list, time.Hour*24*7)
}
// 退出群组
func (s *SocketV1) LeaveGroup(conn *WebsocketData, group string) {
conn.Groups = tools.RemoveSlice[string](conn.Groups, group)
cacheKey := "websocket:group:" + group
get, _ := cache.Get(conn.Ctx, cacheKey)
var list = make(map[int64]string)
if !get.IsNil() {
get.Scan(&list)
}
delete(list, conn.Uid)
cache.Set(conn.Ctx, cacheKey, list, time.Hour*24*7)
}
//群组广播
func (s *SocketV1) SendGroup(cmd int32, group string, req proto.Message) {
cacheKey := "websocket:group:" + group
get, _ := cache.Get(gctx.New(), cacheKey)
var list = make(map[int64]string)
if !get.IsNil() {
get.Scan(&list)
}
for uid, v := range list {
if m.Contains(v) {
s.Send(cmd, uid, req)
}
}
}
// OnClose // OnClose
// //
// @Description: // @Description:
// @receiver s // @receiver s
// @param conn // @param conn
func (s *SocketV1) OnClose(id uuid.UUID, conn *websocket.Conn) { func (s *SocketV1) OnClose(conn *WebsocketData) {
// 在此处编写断开连接后的处理逻辑 // 在此处编写断开连接后的处理逻辑
g.Log().Debugf(gctx.New(), "WebSocket connection from %s has been closed.", conn.RemoteAddr()) //g.Log().Debugf(gctx.New(), "WebSocket connection from %s has been closed.", conn.RemoteAddr())
//用户登录钩子执行
for _, connect := range OnCloseHandlers {
connect(conn)
}
uid := conn.Uid
if uid > 0 {
s.UnBindUid(uid)
for _, v := range conn.Groups {
s.LeaveGroup(conn, v)
}
}
// 可能的后续操作: // 可能的后续操作:
// 1. 更新连接状态或从连接池移除 // 1. 更新连接状态或从连接池移除
// 2. 发送通知或清理关联资源 // 2. 发送通知或清理关联资源
// 3. 执行特定于业务的断开处理 // 3. 执行特定于业务的断开处理
m.Remove(id) conn.Ws.Close()
conn.Close() m.Remove(conn.Uuid)
}
// 是否在线
func (s *SocketV1) IsOnline(uid int64) bool {
uuid := s.Uid2Uuid(uid)
if m.Contains(uuid) {
return true
}
return false
} }

View File

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

View File

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

View File

@@ -53,12 +53,52 @@ func (m *randMod) RandomAll(data map[int]int, n int) []int {
return result return result
} }
func RandomAll[t Any](data map[t]int, n int) []t {
if n > len(data) {
n = len(data)
}
rand.Seed(time.Now().UnixNano())
// 复制权重映射,避免修改原始数据
remainingWeights := make(map[t]int)
for k, v := range data {
remainingWeights[k] = v
}
result := make([]t, 0, n)
for i := 0; i < n; i++ {
totalWeight := 0
// 计算剩余元素的总权重
for _, weight := range remainingWeights {
totalWeight += weight
}
if totalWeight == 0 {
break
}
// 生成一个 0 到总权重之间的随机数
randomNum := rand.Intn(totalWeight)
currentWeight := 0
for key, weight := range remainingWeights {
currentWeight += weight
if randomNum < currentWeight {
// 将选中的元素添加到结果切片中
result = append(result, key)
// 从剩余权重映射中移除选中的元素
delete(remainingWeights, key)
break
}
}
}
return result
}
// RandByArrInt 根据传入的 interface 切片中的整数值按权重随机返回一个索引 // RandByArrInt 根据传入的 interface 切片中的整数值按权重随机返回一个索引
// 参数 s: 一个包含整数的 interface 切片,切片中的每个元素代表一个权重 // 参数 s: 一个包含整数的 interface 切片,切片中的每个元素代表一个权重
// 返回值: 随机选中的元素的索引 // 返回值: 随机选中的元素的索引
func RandByArrInt(s []interface{}) int { func (m *randMod) RandByArrInt(_s interface{}) int {
// 初始化总权重为 0 // 初始化总权重为 0
sv := 0 sv := 0
s := gconv.Ints(_s)
// 遍历切片,累加每个元素的权重 // 遍历切片,累加每个元素的权重
for i := range s { for i := range s {
sv += gconv.Int(s[i]) sv += gconv.Int(s[i])
@@ -73,7 +113,7 @@ func RandByArrInt(s []interface{}) int {
var all int var all int
// 再次遍历切片,累加权重 // 再次遍历切片,累加权重
for i := range s { for i := range s {
all += gconv.Int(s[i]) all += s[i]
// 如果当前累加的权重大于随机数,则返回当前索引 // 如果当前累加的权重大于随机数,则返回当前索引
if all > r { if all > r {
return i return i

View File

@@ -124,7 +124,7 @@ func (m *timeMod) GetDailyTimeList(time1 time.Time, time2 time.Time) (timeList [
// ExcelTime2Time excel时间转时间 (12/10/24 02:03转为时间) // ExcelTime2Time excel时间转时间 (12/10/24 02:03转为时间)
func (m *timeMod) ExcelTime2Time(excelTime string) time.Time { func (m *timeMod) ExcelTime2Time(excelTime string) time.Time {
layout := "01/02/06 15:04" // 月/日/年(最后两位) 小时:分钟 (24小时制) layout := "1/2/06 15:04" // 月/日/年(最后两位) 小时:分钟 (24小时制)
timeNew, _ := time.ParseInLocation(layout, excelTime, time.Local) timeNew, _ := time.ParseInLocation(layout, excelTime, time.Local)
return timeNew return timeNew
} }

View File

@@ -20,9 +20,9 @@ type Number interface {
int | int64 | int32 | int16 | uint64 | uint32 | uint16 | float32 | float64 int | int64 | int32 | int16 | uint64 | uint32 | uint16 | float32 | float64
} }
//type Any interface { type Any interface {
// interface{} | string | int | int64 | int32 | int16 | uint64 | uint32 | uint16 | float32 | float64 string | int | int64 | int32 | int16 | uint64 | uint32 | uint16 | float32 | float64
//} }
type toolsInterface interface { type toolsInterface interface {
Load() Load()
@@ -128,11 +128,11 @@ func (m *tools) Items2Map(items [][]int64) (list map[int64]int64) {
// 该函数通过遍历切片,从后向前检查每个元素,如果找到与指定值相等的元素,则将其从切片中移除。 // 该函数通过遍历切片,从后向前检查每个元素,如果找到与指定值相等的元素,则将其从切片中移除。
// 这种从后向前的遍历方法可以避免因移除元素而导致的数组重新排列带来的额外计算。 // 这种从后向前的遍历方法可以避免因移除元素而导致的数组重新排列带来的额外计算。
// RemoveSlice 删除切片中的某个值 // RemoveSlice 删除切片中的某个值
func RemoveSlice[t Number](slice []t, value ...t) []t { func RemoveSlice[t Any](slice []t, value ...t) []t {
// 从后向前遍历切片 // 从后向前遍历切片
for i := len(slice) - 1; i >= 0; i-- { for i := len(slice) - 1; i >= 0; i-- {
// 检查当前元素是否等于需要移除的值 // 检查当前元素是否等于需要移除的值
if InArray(slice[i], value) { if InArray[t](value, slice[i]) {
// 如果相等,移除该元素 // 如果相等,移除该元素
// 使用append和切片操作符来实现移除操作将i之前和i之后的元素合并到一起 // 使用append和切片操作符来实现移除操作将i之前和i之后的元素合并到一起
slice = append(slice[:i], slice[i+1:]...) slice = append(slice[:i], slice[i+1:]...)
@@ -149,7 +149,7 @@ func RemoveSlice[t Number](slice []t, value ...t) []t {
// @param value 需要查找的值 // @param value 需要查找的值
// @param array 进行查找的切片 // @param array 进行查找的切片
// @return bool 返回是否存在 // @return bool 返回是否存在
func InArray[t Number](value t, array []t) bool { func InArray[t Any](array []t, value t) bool {
for _, v := range array { for _, v := range array {
if v == value { if v == value {
return true return true

View File

@@ -4,6 +4,7 @@ import (
"github.com/ayflying/utility_go/config" "github.com/ayflying/utility_go/config"
"github.com/ayflying/utility_go/internal/boot" "github.com/ayflying/utility_go/internal/boot"
_ "github.com/ayflying/utility_go/internal/logic" _ "github.com/ayflying/utility_go/internal/logic"
"github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gctx" "github.com/gogf/gf/v2/os/gctx"
) )
@@ -13,12 +14,11 @@ var (
) )
func init() { func init() {
go func() { g.Log().Debug(ctx, "utility_go init启动完成")
// 初始化配置 // 初始化配置
var err = boot.Boot() var err = boot.Boot()
if err != nil { if err != nil {
panic(err) panic(err)
} }
}()
} }

View File

@@ -1,9 +1,9 @@
package utility_go_test package utility_go_test
import ( import (
"github.com/ayflying/utility_go/internal/boot" //_ "github.com/ayflying/utility_go/internal/logic"
_ "github.com/ayflying/utility_go/internal/logic"
"github.com/ayflying/utility_go/internal/boot"
"github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/frame/g"
"github.com/gogf/gf/v2/os/gctx" "github.com/gogf/gf/v2/os/gctx"
"testing" "testing"