Compare commits
12 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
569937c67f | ||
|
|
d6bfe1c2fb | ||
|
|
16da554a60 | ||
|
|
7f6635fb91 | ||
|
|
0605302db6 | ||
|
|
5f2fe5dcb2 | ||
|
|
452aefe3d0 | ||
|
|
97bf7fc390 | ||
|
|
08f2b2d9bc | ||
|
|
366ddb45ea | ||
|
|
09a9f14a29 | ||
|
|
e0afb55bb2 |
31
cmd/make.go
31
cmd/make.go
@@ -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
10
cmd/make/socket
Normal file
@@ -0,0 +1,10 @@
|
|||||||
|
package {name}
|
||||||
|
|
||||||
|
type {name} struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func New() *{name} {
|
||||||
|
return &{name}{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() {}
|
||||||
19
cmd/make/socket2
Normal file
19
cmd/make/socket2
Normal 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
|
||||||
|
}
|
||||||
@@ -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 {
|
||||||
|
|||||||
5
go.mod
5
go.mod
@@ -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
|
||||||
|
|||||||
@@ -1,6 +1,7 @@
|
|||||||
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/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"github.com/gogf/gf/v2/os/gctx"
|
"github.com/gogf/gf/v2/os/gctx"
|
||||||
@@ -15,15 +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)
|
||||||
})
|
})
|
||||||
|
|
||||||
//初始化自启动方法
|
//初始化自启动方法
|
||||||
for _, v := range _func {
|
for _, v := range _func {
|
||||||
v()
|
v()
|
||||||
}
|
}
|
||||||
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -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
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -1,7 +1,6 @@
|
|||||||
package ip2region
|
package ip2region
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"github.com/ayflying/utility_go/internal/boot"
|
|
||||||
"github.com/ayflying/utility_go/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"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"
|
||||||
@@ -13,6 +12,7 @@ import (
|
|||||||
|
|
||||||
var (
|
var (
|
||||||
ctx = gctx.New()
|
ctx = gctx.New()
|
||||||
|
wait = false
|
||||||
)
|
)
|
||||||
|
|
||||||
type sIp2region struct {
|
type sIp2region struct {
|
||||||
@@ -27,9 +27,9 @@ func New() *sIp2region {
|
|||||||
func init() {
|
func init() {
|
||||||
service.RegisterIp2Region(New())
|
service.RegisterIp2Region(New())
|
||||||
|
|
||||||
boot.AddFunc(func() {
|
//boot.AddFunc(func() {
|
||||||
service.Ip2Region().Load()
|
// service.Ip2Region().Load()
|
||||||
})
|
//})
|
||||||
}
|
}
|
||||||
|
|
||||||
// Load 加载到内存中
|
// Load 加载到内存中
|
||||||
@@ -39,10 +39,16 @@ func init() {
|
|||||||
func (s *sIp2region) Load() {
|
func (s *sIp2region) Load() {
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
var url = "https://github.com/ayflying/resource/raw/refs/heads/main/attachment/ip2region.xdb"
|
|
||||||
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库文件")
|
g.Log().Debug(ctx, "等待下载ip库文件")
|
||||||
//下载文件
|
//下载文件
|
||||||
putData, err2 := g.Client().Discovery(nil).Get(ctx, url)
|
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) {
|
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
|
||||||
|
|||||||
@@ -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 {
|
||||||
|
|||||||
@@ -14,7 +14,7 @@ import (
|
|||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
ctx = gctx.New()
|
//ctx = gctx.New()
|
||||||
startTime *gtime.Time
|
startTime *gtime.Time
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -23,39 +23,44 @@ 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
|
||||||
|
}
|
||||||
|
|
||||||
|
//每年任务
|
||||||
|
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 {
|
if err != nil {
|
||||||
g.Log().Error(ctx, err)
|
g.Log().Error(ctx, err)
|
||||||
}
|
}
|
||||||
|
}()
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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)
|
||||||
|
}
|
||||||
|
|||||||
@@ -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()
|
|
||||||
id, _ := uuid.NewUUID()
|
|
||||||
ip := conn.RemoteAddr().String()
|
|
||||||
|
|
||||||
data := &WebsocketData{
|
|
||||||
Uuid: id,
|
Uuid: id,
|
||||||
Ws: conn,
|
Ws: ws,
|
||||||
Ctx: ctx,
|
Ctx: ctx,
|
||||||
|
Groups: make([]string, 0),
|
||||||
|
RoomId: -1,
|
||||||
}
|
}
|
||||||
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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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)
|
||||||
|
|||||||
@@ -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)
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -53,6 +53,44 @@ 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 切片,切片中的每个元素代表一个权重
|
||||||
// 返回值: 随机选中的元素的索引
|
// 返回值: 随机选中的元素的索引
|
||||||
|
|||||||
@@ -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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -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,7 +128,7 @@ 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-- {
|
||||||
// 检查当前元素是否等于需要移除的值
|
// 检查当前元素是否等于需要移除的值
|
||||||
@@ -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](array []t, value 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
|
||||||
|
|||||||
Reference in New Issue
Block a user