Compare commits
9 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
504836d165 | ||
|
|
e9962f1b74 | ||
|
|
1153313384 | ||
|
|
750d5c56ce | ||
|
|
c24ac78b91 | ||
|
|
569937c67f | ||
|
|
d6bfe1c2fb | ||
|
|
16da554a60 | ||
|
|
7f6635fb91 |
@@ -119,8 +119,6 @@ var (
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
continue
|
|
||||||
|
|
||||||
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],
|
||||||
|
|||||||
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,16 @@ 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()
|
err = service.GameKv().SavesV1()
|
||||||
|
err = service.GameAct().Saves(ctx)
|
||||||
|
return err
|
||||||
})
|
})
|
||||||
|
|
||||||
//初始化自启动方法
|
//初始化自启动方法
|
||||||
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"
|
||||||
@@ -8,6 +9,7 @@ import (
|
|||||||
service2 "github.com/ayflying/utility_go/service"
|
service2 "github.com/ayflying/utility_go/service"
|
||||||
"github.com/ayflying/utility_go/tools"
|
"github.com/ayflying/utility_go/tools"
|
||||||
"github.com/gogf/gf/v2/container/gset"
|
"github.com/gogf/gf/v2/container/gset"
|
||||||
|
"github.com/gogf/gf/v2/container/gvar"
|
||||||
"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"
|
||||||
"github.com/gogf/gf/v2/os/gtime"
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
@@ -62,8 +64,11 @@ func (s *sGameAct) Info(uid int64, actId int) (data *g.Var, err error) {
|
|||||||
Uid: uid,
|
Uid: uid,
|
||||||
ActId: actId,
|
ActId: actId,
|
||||||
}).Fields("action").OrderDesc("updated_at").Value()
|
}).Fields("action").OrderDesc("updated_at").Value()
|
||||||
getDb.Scan(&data)
|
//getDb.Scan(&data)
|
||||||
|
if getDb == nil || getDb.IsEmpty() {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
data = gvar.New(getDb)
|
||||||
if data == nil || data.IsEmpty() {
|
if data == nil || data.IsEmpty() {
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
@@ -105,16 +110,24 @@ 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) {
|
||||||
|
getCache, _ := pkg.Cache("redis").Get(nil, "cron:game_act")
|
||||||
|
//如果没有执行过,设置时间戳
|
||||||
|
if getCache.Int64() > 0 {
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
pkg.Cache("redis").Set(nil, "cron:game_act", gtime.Now().Unix(), time.Hour)
|
||||||
|
}
|
||||||
|
|
||||||
//遍历执行
|
//遍历执行
|
||||||
ActList.Iterator(func(i interface{}) bool {
|
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 +162,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,13 +1,17 @@
|
|||||||
package gameKv
|
package gameKv
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
|
"github.com/ayflying/utility_go/pkg"
|
||||||
"github.com/ayflying/utility_go/service"
|
"github.com/ayflying/utility_go/service"
|
||||||
"github.com/ayflying/utility_go/tools"
|
"github.com/ayflying/utility_go/tools"
|
||||||
"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"
|
||||||
|
"github.com/gogf/gf/v2/os/gtime"
|
||||||
"strconv"
|
"strconv"
|
||||||
"strings"
|
"strings"
|
||||||
"sync"
|
"sync"
|
||||||
|
"time"
|
||||||
)
|
)
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -25,10 +29,6 @@ func New() *sGameKv {
|
|||||||
|
|
||||||
func init() {
|
func init() {
|
||||||
service.RegisterGameKv(New())
|
service.RegisterGameKv(New())
|
||||||
|
|
||||||
//支付钩子
|
|
||||||
//task.Task.Trigger(tasks.TaskType_PAY, service.GameKv().HookPay)
|
|
||||||
//task.Task.Trigger(tasks.TaskType_WARDROBE_LEVEL, service.GameKv().HookLevelRwd)
|
|
||||||
}
|
}
|
||||||
|
|
||||||
// SavesV1 方法
|
// SavesV1 方法
|
||||||
@@ -37,6 +37,14 @@ func init() {
|
|||||||
// @receiver s: sGameKv的实例。
|
// @receiver s: sGameKv的实例。
|
||||||
// @return err: 错误信息,如果操作成功,则为nil。
|
// @return err: 错误信息,如果操作成功,则为nil。
|
||||||
func (s *sGameKv) SavesV1() (err error) {
|
func (s *sGameKv) SavesV1() (err error) {
|
||||||
|
getCache, err := pkg.Cache("redis").Get(nil, "cron:game_kv")
|
||||||
|
//如果没有执行过,设置时间戳
|
||||||
|
if getCache.Int64() > 0 {
|
||||||
|
return
|
||||||
|
} else {
|
||||||
|
pkg.Cache("redis").Set(nil, "cron:game_kv", gtime.Now().Unix(), time.Hour)
|
||||||
|
}
|
||||||
|
|
||||||
// 从Redis列表中获取所有用户KV索引的键
|
// 从Redis列表中获取所有用户KV索引的键
|
||||||
//keys, err := utils.RedisScan("user:kv:*")
|
//keys, err := utils.RedisScan("user:kv:*")
|
||||||
err = tools.Redis.RedisScanV2("user:kv:*", func(keys []string) (err error) {
|
err = tools.Redis.RedisScanV2("user:kv:*", func(keys []string) (err error) {
|
||||||
@@ -69,12 +77,17 @@ func (s *sGameKv) SavesV1() (err error) {
|
|||||||
//if user.UpdatedAt.Seconds < gtime.Now().Add(consts.ActSaveTime).Unix() {
|
//if user.UpdatedAt.Seconds < gtime.Now().Add(consts.ActSaveTime).Unix() {
|
||||||
// continue
|
// continue
|
||||||
//}
|
//}
|
||||||
|
//如果有活跃,跳过持久化
|
||||||
|
if getBool, _ := pkg.Cache("redis").
|
||||||
|
Contains(ctx, fmt.Sprintf("act:update:%d", uid)); getBool {
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
get, _ := g.Redis().Get(ctx, cacheKey)
|
get, _ := g.Redis().Get(ctx, cacheKey)
|
||||||
var data interface{}
|
var data interface{}
|
||||||
get.Scan(&data)
|
get.Scan(&data)
|
||||||
list = append(list, &ListData{
|
list = append(list, &ListData{
|
||||||
Uid: int64(uid),
|
Uid: uid,
|
||||||
Kv: data,
|
Kv: data,
|
||||||
})
|
})
|
||||||
|
|
||||||
|
|||||||
@@ -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"
|
||||||
@@ -12,7 +11,8 @@ 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
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -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
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -12,34 +12,37 @@ import (
|
|||||||
"math"
|
"math"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// Mod 定义缓存模块结构体,包含一个 gcache.Cache 客户端实例
|
||||||
type Mod struct {
|
type Mod struct {
|
||||||
client *gcache.Cache
|
client *gcache.Cache
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
// QPSCount 记录缓存的 QPS 计数
|
||||||
QPSCount int
|
var QPSCount int
|
||||||
QPS = promauto.NewGauge(
|
// QPS 是一个 Prometheus 指标,用于记录当前缓存的 QPS 数量
|
||||||
prometheus.GaugeOpts{
|
var QPS = promauto.NewGauge(
|
||||||
Name: "Cache_QPS",
|
prometheus.GaugeOpts{
|
||||||
Help: "当前缓存QPS数量",
|
Name: "Cache_QPS",
|
||||||
},
|
Help: "当前缓存QPS数量",
|
||||||
)
|
},
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// init 函数在包被导入时执行,用于初始化定时任务以更新 QPS 指标
|
||||||
func init() {
|
func init() {
|
||||||
boot.AddFunc(func() {
|
boot.AddFunc(func() {
|
||||||
//初始化指标
|
// 初始化指标,每分钟计算一次平均 QPS 并重置计数器
|
||||||
service.SystemCron().AddCron(v1.CronType_MINUTE, func() error {
|
service.SystemCron().AddCron(v1.CronType_MINUTE, func() error {
|
||||||
QPS.Set(math.Round(float64(QPSCount) / 60))
|
QPS.Set(math.Round(float64(QPSCount) / 60))
|
||||||
QPSCount = 0
|
QPSCount = 0
|
||||||
return nil
|
return nil
|
||||||
})
|
})
|
||||||
})
|
})
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// New 根据传入的名称创建不同类型的缓存适配器
|
||||||
|
// 如果未传入名称,默认使用 "cache" 类型
|
||||||
|
// 支持的类型包括 "cache"(内存缓存)、"redis"(Redis 缓存)、"file"(文件缓存)和 "es"(Elasticsearch 缓存)
|
||||||
func New(_name ...string) gcache.Adapter {
|
func New(_name ...string) gcache.Adapter {
|
||||||
|
|
||||||
var cacheAdapterObj gcache.Adapter
|
var cacheAdapterObj gcache.Adapter
|
||||||
var name = "cache"
|
var name = "cache"
|
||||||
if len(_name) > 0 {
|
if len(_name) > 0 {
|
||||||
@@ -47,18 +50,23 @@ func New(_name ...string) gcache.Adapter {
|
|||||||
}
|
}
|
||||||
switch name {
|
switch name {
|
||||||
case "cache":
|
case "cache":
|
||||||
|
// 创建内存缓存适配器
|
||||||
cacheAdapterObj = drive2.NewAdapterMemory()
|
cacheAdapterObj = drive2.NewAdapterMemory()
|
||||||
case "redis":
|
case "redis":
|
||||||
|
// 创建 Redis 缓存适配器
|
||||||
cacheAdapterObj = drive2.NewAdapterRedis()
|
cacheAdapterObj = drive2.NewAdapterRedis()
|
||||||
case "file":
|
case "file":
|
||||||
|
// 创建文件缓存适配器,指定缓存目录为 "runtime/cache"
|
||||||
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
cacheAdapterObj = drive2.NewAdapterFile("runtime/cache")
|
||||||
case "es":
|
case "es":
|
||||||
|
// 创建 Elasticsearch 缓存适配器,需要传入额外参数
|
||||||
cacheAdapterObj = drive.NewAdapterElasticsearch(_name[1])
|
cacheAdapterObj = drive.NewAdapterElasticsearch(_name[1])
|
||||||
}
|
}
|
||||||
|
|
||||||
//var client = gcache.New()
|
//var client = gcache.New()
|
||||||
//client.SetAdapter(cacheAdapterObj)
|
//client.SetAdapter(cacheAdapterObj)
|
||||||
|
|
||||||
|
// 每次创建适配器时,QPS 计数加 1
|
||||||
QPSCount++
|
QPSCount++
|
||||||
return cacheAdapterObj
|
return cacheAdapterObj
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -18,26 +18,40 @@ import (
|
|||||||
var (
|
var (
|
||||||
//ApolloCfg *apolloConfig.AppConfig
|
//ApolloCfg *apolloConfig.AppConfig
|
||||||
ApolloCfg *apollo.Config
|
ApolloCfg *apollo.Config
|
||||||
|
// ApolloListener 存储需要监听的 Apollo 配置项名称
|
||||||
ApolloListener []string
|
ApolloListener []string
|
||||||
|
// Item2Obj 存储配置项名称和对应的加载器对象的映射
|
||||||
Item2Obj = map[string]Load{}
|
Item2Obj = map[string]Load{}
|
||||||
)
|
)
|
||||||
|
|
||||||
// load接口定义了Load方法,用于加载数据
|
// Load 接口定义了 Load 方法,用于加载数据
|
||||||
type Load interface {
|
type Load interface {
|
||||||
|
// Load 方法用于加载配置数据,支持传入可选的配置参数
|
||||||
Load(cfg ...string)
|
Load(cfg ...string)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// NewV1 创建一个新的 Cfg 实例
|
||||||
func NewV1() *Cfg {
|
func NewV1() *Cfg {
|
||||||
return &Cfg{}
|
return &Cfg{}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Cfg 结构体包含配置操作的相关方法
|
||||||
type Cfg struct {
|
type Cfg struct {
|
||||||
|
// Lock 用于保证并发安全的互斥锁
|
||||||
Lock sync.Mutex
|
Lock sync.Mutex
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetDbFile 从数据库中获取指定名称的配置文件
|
||||||
|
// 参数:
|
||||||
|
// name - 配置文件的名称
|
||||||
|
// 返回值:
|
||||||
|
// *g.Var - 存储配置数据的变量
|
||||||
|
// error - 操作过程中遇到的错误
|
||||||
func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
|
func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
|
||||||
|
// 从数据库的 game_config 表中查询指定名称的配置数据
|
||||||
get2, err := g.Model("game_config").
|
get2, err := g.Model("game_config").
|
||||||
Where("name", name).Master().Value("data")
|
Where("name", name).Master().Value("data")
|
||||||
|
// 将查询结果扫描到 res 变量中
|
||||||
err = get2.Scan(&res)
|
err = get2.Scan(&res)
|
||||||
if res == nil {
|
if res == nil {
|
||||||
res = &gvar.Var{}
|
res = &gvar.Var{}
|
||||||
@@ -45,107 +59,81 @@ func (c *Cfg) GetDbFile(name string) (res *g.Var, err error) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c *Cfg) GetFile(filename string, obj ...Load) (jsonObj *gjson.Json, err error) {
|
// GetFile 从文件系统或资源文件中加载 JSON 配置
|
||||||
|
// 参数:
|
||||||
|
// filename - 需要加载的配置文件名(不带扩展名)
|
||||||
|
// _pathStr - 可选参数,指定配置文件目录路径,默认"manifest/game/"
|
||||||
|
// 返回值:
|
||||||
|
// *gjson.Json - 解析后的 JSON 对象
|
||||||
|
// error - 文件加载或解析过程中遇到的错误
|
||||||
|
func (c *Cfg) GetFile(filename string, _pathStr ...string) (jsonObj *gjson.Json, err error) {
|
||||||
|
// 处理路径参数,使用默认路径或传入参数
|
||||||
pathStr := "manifest/game/"
|
pathStr := "manifest/game/"
|
||||||
|
if len(_pathStr) > 0 {
|
||||||
|
pathStr = _pathStr[0]
|
||||||
|
}
|
||||||
|
// 拼接完整的文件路径
|
||||||
filePath := pathStr + filename + ".json"
|
filePath := pathStr + filename + ".json"
|
||||||
//err := gres.Load(pathStr + filename)
|
|
||||||
|
|
||||||
//载入静态资源到文件对象
|
// 载入静态资源到文件对象
|
||||||
err = gres.Load(filePath)
|
err = gres.Load(filePath)
|
||||||
var bytes []byte
|
var bytes []byte
|
||||||
|
|
||||||
|
// 优先从文件系统读取,不存在时从资源文件读取
|
||||||
if gfile.IsFile(filePath) {
|
if gfile.IsFile(filePath) {
|
||||||
bytes = gfile.GetBytes(filePath)
|
bytes = gfile.GetBytes(filePath) // 读取物理文件内容
|
||||||
} else {
|
} else {
|
||||||
bytes = gres.GetContent(filePath)
|
bytes = gres.GetContent(filePath) // 从打包资源中获取内容
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 解析 JSON 内容并返回结果
|
||||||
jsonObj, err = gjson.DecodeToJson(bytes)
|
jsonObj, err = gjson.DecodeToJson(bytes)
|
||||||
//g.Dump(filePath, jsonObj)
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// getUrlFile 获取远程配置
|
// GetUrlFile 获取远程配置
|
||||||
|
// 参数:
|
||||||
|
// name - 配置文件的名称
|
||||||
|
// 返回值:
|
||||||
|
// *gjson.Json - 解析后的 JSON 对象
|
||||||
|
// error - 请求或解析过程中遇到的错误
|
||||||
func (c *Cfg) GetUrlFile(name string) (jsonObj *gjson.Json, err error) {
|
func (c *Cfg) GetUrlFile(name string) (jsonObj *gjson.Json, err error) {
|
||||||
|
// 拼接远程配置文件的 URL
|
||||||
urlStr := fmt.Sprintf("http://sdf.sdfs.sdf/%s.json", name)
|
urlStr := fmt.Sprintf("http://sdf.sdfs.sdf/%s.json", name)
|
||||||
|
// 发送 HTTP 请求获取远程配置数据
|
||||||
getUrl, err := g.Client().Discovery(nil).Get(nil, urlStr)
|
getUrl, err := g.Client().Discovery(nil).Get(nil, urlStr)
|
||||||
|
// 读取响应内容
|
||||||
bytes := getUrl.ReadAll()
|
bytes := getUrl.ReadAll()
|
||||||
|
// 解析 JSON 内容并返回结果
|
||||||
jsonObj, err = gjson.DecodeToJson(bytes)
|
jsonObj, err = gjson.DecodeToJson(bytes)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 获取阿波罗
|
// GetApollo 从 Apollo 配置中心获取指定名称的配置
|
||||||
//func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
|
// 参数:
|
||||||
// jsonObj, err = c.GetApolloV2(name, obj)
|
// name - 配置文件的名称
|
||||||
// return
|
// obj - 实现了 Load 接口的加载器对象
|
||||||
//
|
// 返回值:
|
||||||
// //c.Lock.Lock()
|
// *gjson.Json - 解析后的 JSON 对象
|
||||||
// //defer c.Lock.Unlock()
|
// error - 操作过程中遇到的错误
|
||||||
// //
|
|
||||||
// //Item2Obj[name+".json"] = obj
|
|
||||||
// //var cfg = apolloConfig.AppConfig{
|
|
||||||
// // AppID: ApolloCfg.AppID,
|
|
||||||
// // Cluster: ApolloCfg.Cluster,
|
|
||||||
// // IP: ApolloCfg.IP,
|
|
||||||
// // NamespaceName: name + ".json",
|
|
||||||
// // Secret: ApolloCfg.Secret,
|
|
||||||
// // IsBackupConfig: ApolloCfg.IsBackupConfig,
|
|
||||||
// // BackupConfigPath: ApolloCfg.BackupConfigPath,
|
|
||||||
// // SyncServerTimeout: 60,
|
|
||||||
// // MustStart: true,
|
|
||||||
// //}
|
|
||||||
// ////cfg.NamespaceName = name + ".json"
|
|
||||||
// //
|
|
||||||
// //client, err := agollo.StartWithConfig(func() (*apolloConfig.AppConfig, error) {
|
|
||||||
// // return ApolloCfg, nil
|
|
||||||
// //})
|
|
||||||
// //if client == nil {
|
|
||||||
// // return
|
|
||||||
// //}
|
|
||||||
// //var getStr string
|
|
||||||
// //var getApollo *storage.Config
|
|
||||||
// //for range 5 {
|
|
||||||
// // getApollo = client.GetConfig(cfg.NamespaceName)
|
|
||||||
// // if getApollo != nil {
|
|
||||||
// // break
|
|
||||||
// // }
|
|
||||||
// // time.Sleep(time.Second * 5)
|
|
||||||
// //}
|
|
||||||
// //
|
|
||||||
// //if getApollo != nil {
|
|
||||||
// // getStr = getApollo.GetValue("content")
|
|
||||||
// // if getStr != "" {
|
|
||||||
// // //写入配置
|
|
||||||
// // gfile.PutContents(path.Join("manifest", "game", name+".json"), getStr)
|
|
||||||
// // }
|
|
||||||
// //} else {
|
|
||||||
// // jsonObj, err = c.GetFile(name)
|
|
||||||
// //}
|
|
||||||
// //jsonObj, err = gjson.DecodeToJson(getStr)
|
|
||||||
// ////首次运行加入监听器
|
|
||||||
// //if !gstr.InArray(ApolloListener, name) {
|
|
||||||
// // c2 := &CustomChangeListener{}
|
|
||||||
// // client.AddChangeListener(c2)
|
|
||||||
// // ApolloListener = append(ApolloListener, name)
|
|
||||||
// //}
|
|
||||||
// //return
|
|
||||||
//}
|
|
||||||
|
|
||||||
func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
|
func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error) {
|
||||||
|
// 将配置项名称和对应的加载器对象存入映射
|
||||||
Item2Obj[name+".json"] = obj
|
Item2Obj[name+".json"] = obj
|
||||||
|
|
||||||
// 接入阿波罗配置
|
// 接入 Apollo 配置
|
||||||
ApolloCfg.NamespaceName = name + ".json"
|
ApolloCfg.NamespaceName = name + ".json"
|
||||||
|
// 创建 Apollo 配置适配器
|
||||||
adapter, err := apollo.New(nil, *ApolloCfg)
|
adapter, err := apollo.New(nil, *ApolloCfg)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
// 配置适配器创建失败,记录致命错误日志
|
||||||
g.Log().Fatalf(nil, `%+v`, err)
|
g.Log().Fatalf(nil, `%+v`, err)
|
||||||
}
|
}
|
||||||
// Change the adapter of default configuration instance.
|
// 更改默认配置实例的适配器
|
||||||
g.Cfg(name).SetAdapter(adapter)
|
g.Cfg(name).SetAdapter(adapter)
|
||||||
|
|
||||||
//首次运行加入监听器
|
// 首次运行加入监听器
|
||||||
if !gstr.InArray(ApolloListener, name+".json") {
|
if !gstr.InArray(ApolloListener, name+".json") {
|
||||||
//放置监听器
|
// 启动 Apollo 客户端
|
||||||
client, _ := agollo.StartWithConfig(func() (*apolloConfig.AppConfig, error) {
|
client, _ := agollo.StartWithConfig(func() (*apolloConfig.AppConfig, error) {
|
||||||
return &apolloConfig.AppConfig{
|
return &apolloConfig.AppConfig{
|
||||||
AppID: ApolloCfg.AppID,
|
AppID: ApolloCfg.AppID,
|
||||||
@@ -159,31 +147,40 @@ func (c *Cfg) GetApollo(name string, obj Load) (jsonObj *gjson.Json, err error)
|
|||||||
MustStart: ApolloCfg.MustStart,
|
MustStart: ApolloCfg.MustStart,
|
||||||
}, nil
|
}, nil
|
||||||
})
|
})
|
||||||
|
// 创建自定义监听器实例
|
||||||
c2 := &CustomChangeListener{}
|
c2 := &CustomChangeListener{}
|
||||||
|
// 为 Apollo 客户端添加监听器
|
||||||
client.AddChangeListener(c2)
|
client.AddChangeListener(c2)
|
||||||
|
// 将配置项名称添加到监听器列表
|
||||||
ApolloListener = append(ApolloListener, name+".json")
|
ApolloListener = append(ApolloListener, name+".json")
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 从配置中心获取指定配置项的值
|
||||||
cfg, err := g.Cfg(name).Get(nil, "content")
|
cfg, err := g.Cfg(name).Get(nil, "content")
|
||||||
|
// 将配置值扫描到 jsonObj 中
|
||||||
cfg.Scan(&jsonObj)
|
cfg.Scan(&jsonObj)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 阿波罗监听器
|
// CustomChangeListener 是 Apollo 配置变化的自定义监听器
|
||||||
type CustomChangeListener struct {
|
type CustomChangeListener struct {
|
||||||
|
// wg 用于等待所有处理任务完成
|
||||||
wg sync.WaitGroup
|
wg sync.WaitGroup
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// OnChange 当 Apollo 配置发生变化时触发
|
||||||
func (c *CustomChangeListener) OnChange(changeEvent *storage.ChangeEvent) {
|
func (c *CustomChangeListener) OnChange(changeEvent *storage.ChangeEvent) {
|
||||||
|
// 记录配置变化的日志
|
||||||
g.Log().Debugf(nil, "当前Namespace变化了:%v", changeEvent.Namespace)
|
g.Log().Debugf(nil, "当前Namespace变化了:%v", changeEvent.Namespace)
|
||||||
|
// 获取变化的配置项名称
|
||||||
filename := changeEvent.Namespace
|
filename := changeEvent.Namespace
|
||||||
if obj, ok := Item2Obj[filename]; ok {
|
if obj, ok := Item2Obj[filename]; ok {
|
||||||
//重载配置文件
|
// 重载配置文件
|
||||||
obj.Load(changeEvent.Changes["content"].NewValue.(string))
|
obj.Load(changeEvent.Changes["content"].NewValue.(string))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// OnNewestChange 当获取到最新配置时触发,当前为空实现
|
||||||
func (c *CustomChangeListener) OnNewestChange(event *storage.FullChangeEvent) {
|
func (c *CustomChangeListener) OnNewestChange(event *storage.FullChangeEvent) {
|
||||||
//write your code here
|
//write your code here
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|||||||
174
pkg/s3/s3.go
174
pkg/s3/s3.go
@@ -16,26 +16,32 @@ import (
|
|||||||
"github.com/minio/minio-go/v7/pkg/credentials"
|
"github.com/minio/minio-go/v7/pkg/credentials"
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// ctx 全局上下文,用于在整个包中传递请求范围的数据
|
||||||
var (
|
var (
|
||||||
//client *minio.Client
|
//client *minio.Client
|
||||||
ctx = gctx.New()
|
ctx = gctx.New()
|
||||||
)
|
)
|
||||||
|
|
||||||
|
// DataType 定义了 S3 配置的数据结构,用于存储访问 S3 所需的各种信息
|
||||||
type DataType struct {
|
type DataType struct {
|
||||||
AccessKey string `json:"access_key"`
|
AccessKey string `json:"access_key"` // 访问 S3 的密钥 ID
|
||||||
SecretKey string `json:"secret_key"`
|
SecretKey string `json:"secret_key"` // 访问 S3 的密钥
|
||||||
Address string `json:"address"`
|
Address string `json:"address"` // S3 服务的地址
|
||||||
Ssl bool `json:"ssl"`
|
Ssl bool `json:"ssl"` // 是否使用 SSL 加密连接
|
||||||
Url string `json:"url"`
|
Url string `json:"url"` // S3 服务的访问 URL
|
||||||
BucketName string `json:"bucket_name"`
|
BucketName string `json:"bucket_name"` // 默认存储桶名称
|
||||||
BucketNameCdn string `json:"bucket_name_cdn"`
|
BucketNameCdn string `json:"bucket_name_cdn"` // CDN 存储桶名称
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// Mod 定义了 S3 模块的结构体,包含一个 S3 客户端实例和配置信息
|
||||||
type Mod struct {
|
type Mod struct {
|
||||||
client *minio.Client
|
client *minio.Client // Minio S3 客户端实例
|
||||||
cfg DataType
|
cfg DataType // S3 配置信息
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// New 根据配置创建一个新的 S3 模块实例
|
||||||
|
// 如果未提供名称,则从配置中获取默认的 S3 类型
|
||||||
|
// 配置错误时会触发 panic
|
||||||
func New(_name ...string) *Mod {
|
func New(_name ...string) *Mod {
|
||||||
var name string
|
var name string
|
||||||
if len(_name) > 0 {
|
if len(_name) > 0 {
|
||||||
@@ -52,7 +58,7 @@ func New(_name ...string) *Mod {
|
|||||||
var cfg DataType
|
var cfg DataType
|
||||||
get.Scan(&cfg)
|
get.Scan(&cfg)
|
||||||
|
|
||||||
// 使用minio-go创建S3客户端
|
// 使用 minio-go 创建 S3 客户端
|
||||||
obj, err := minio.New(
|
obj, err := minio.New(
|
||||||
cfg.Address,
|
cfg.Address,
|
||||||
&minio.Options{
|
&minio.Options{
|
||||||
@@ -72,183 +78,111 @@ func New(_name ...string) *Mod {
|
|||||||
return mod
|
return mod
|
||||||
}
|
}
|
||||||
|
|
||||||
//func (s *Mod) Load() {
|
// GetCfg 获取当前 S3 模块的配置信息
|
||||||
// //导入配置
|
|
||||||
// get, err := g.Cfg().Get(ctx, "s3.type")
|
|
||||||
// cfgType := get.String()
|
|
||||||
// if cfgType == "" {
|
|
||||||
// cfgType = "default"
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// cfgData, err := g.Cfg().Get(ctx, "s3."+cfgType)
|
|
||||||
// if cfgData.IsEmpty() {
|
|
||||||
// panic("当前配置中未配置s3:" + cfgType)
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// get, err = g.Cfg().Get(ctx, "s3."+cfgType)
|
|
||||||
// err = get.Scan(&Cfg)
|
|
||||||
//
|
|
||||||
// // 使用minio-go创建S3客户端
|
|
||||||
// obj, err := minio.New(
|
|
||||||
// Cfg.Address,
|
|
||||||
// &minio.Options{
|
|
||||||
// Creds: credentials.NewStaticV4(Cfg.AccessKey, Cfg.SecretKey, ""),
|
|
||||||
// Secure: Cfg.Ssl,
|
|
||||||
// },
|
|
||||||
// )
|
|
||||||
// if err != nil {
|
|
||||||
// log.Fatalln(err)
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// client = obj
|
|
||||||
//}
|
|
||||||
//
|
|
||||||
//func (s *Mod) S3(name string) {
|
|
||||||
// get, err := g.Cfg().Get(ctx, "s3."+name)
|
|
||||||
// if err != nil {
|
|
||||||
// panic(err)
|
|
||||||
// }
|
|
||||||
// get.Scan(&Cfg)
|
|
||||||
//
|
|
||||||
// // 使用minio-go创建S3客户端
|
|
||||||
// obj, err := minio.New(
|
|
||||||
// Cfg.Address,
|
|
||||||
// &minio.Options{
|
|
||||||
// Creds: credentials.NewStaticV4(Cfg.AccessKey, Cfg.SecretKey, ""),
|
|
||||||
// Secure: Cfg.Ssl,
|
|
||||||
// },
|
|
||||||
// )
|
|
||||||
// if err != nil {
|
|
||||||
// log.Fatalln(err)
|
|
||||||
// }
|
|
||||||
//
|
|
||||||
// client = obj
|
|
||||||
//
|
|
||||||
//}
|
|
||||||
|
|
||||||
// GetCfg 获取配置
|
|
||||||
func (s *Mod) GetCfg() *DataType {
|
func (s *Mod) GetCfg() *DataType {
|
||||||
return &s.cfg
|
return &s.cfg
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetFileUrl 生成指向S3存储桶中指定文件的预签名URL
|
// GetFileUrl 生成指向 S3 存储桶中指定文件的预签名 URL
|
||||||
//
|
// 预签名 URL 可用于在有限时间内访问 S3 存储桶中的文件
|
||||||
// @Description: 生成一个具有有限有效期的预签名URL,可用于访问S3存储桶中的文件。
|
// 支持从缓存中获取预签名 URL,以减少重复请求
|
||||||
// @receiver s: S3的实例,用于执行S3操作。
|
|
||||||
// @param name: 要获取预签名URL的文件名。
|
|
||||||
// @param bucketName: 文件所在的存储桶名称。
|
|
||||||
// @return presignedURL: 生成的预签名URL,可用于访问文件。
|
|
||||||
// @return err: 在获取预签名URL过程中遇到的任何错误。
|
|
||||||
func (s *Mod) GetFileUrl(name string, bucketName string, _expires ...time.Duration) (presignedURL *url.URL, err error) {
|
func (s *Mod) GetFileUrl(name string, bucketName string, _expires ...time.Duration) (presignedURL *url.URL, err error) {
|
||||||
// 设置预签名URL的有效期为1小时
|
// 设置预签名 URL 的有效期为 1 小时,可通过参数覆盖
|
||||||
expires := time.Hour * 1
|
expires := time.Hour * 1
|
||||||
if len(_expires) > 0 {
|
if len(_expires) > 0 {
|
||||||
expires = _expires[0]
|
expires = _expires[0]
|
||||||
}
|
}
|
||||||
|
// 生成缓存键
|
||||||
cacheKey := fmt.Sprintf("s3:%v:%v", name, bucketName)
|
cacheKey := fmt.Sprintf("s3:%v:%v", name, bucketName)
|
||||||
|
// 尝试从缓存中获取预签名 URL
|
||||||
get, _ := gcache.Get(ctx, cacheKey)
|
get, _ := gcache.Get(ctx, cacheKey)
|
||||||
//g.Dump(get.Vars())
|
|
||||||
if !get.IsEmpty() {
|
if !get.IsEmpty() {
|
||||||
|
// 将缓存中的值转换为 *url.URL 类型
|
||||||
err = gconv.Struct(get.Val(), &presignedURL)
|
err = gconv.Struct(get.Val(), &presignedURL)
|
||||||
//presignedURL =
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
//expires := time.Duration(604800)
|
// 调用 S3 客户端生成预签名 URL
|
||||||
// 调用s3().PresignedGetObject方法生成预签名URL
|
|
||||||
presignedURL, err = s.client.PresignedGetObject(ctx, bucketName, name, expires, nil)
|
presignedURL, err = s.client.PresignedGetObject(ctx, bucketName, name, expires, nil)
|
||||||
|
// 将生成的预签名 URL 存入缓存
|
||||||
err = gcache.Set(ctx, cacheKey, presignedURL, expires)
|
err = gcache.Set(ctx, cacheKey, presignedURL, expires)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// PutFileUrl 生成一个用于上传文件到指定bucket的预签名URL
|
// PutFileUrl 生成一个用于上传文件到指定存储桶的预签名 URL
|
||||||
//
|
// 预签名 URL 的有效期默认为 10 分钟
|
||||||
// @Description:
|
|
||||||
// @receiver s
|
|
||||||
// @param name 文件名
|
|
||||||
// @param bucketName 存储桶名称
|
|
||||||
// @return presignedURL 预签名的URL,用于上传文件
|
|
||||||
// @return err 错误信息,如果在生成预签名URL时发生错误
|
|
||||||
func (s *Mod) PutFileUrl(name string, bucketName string) (presignedURL *url.URL, err error) {
|
func (s *Mod) PutFileUrl(name string, bucketName string) (presignedURL *url.URL, err error) {
|
||||||
// 设置预签名URL的有效期
|
// 设置预签名 URL 的有效期为 10 分钟
|
||||||
//expires := time.Now().Add(time.Minute * 30).Unix() // 例如:有效期30分钟
|
|
||||||
//expires2 := time.Duration(expires)
|
|
||||||
expires := time.Minute * 10
|
expires := time.Minute * 10
|
||||||
// 生成预签名URL
|
// 调用 S3 客户端生成预签名 URL
|
||||||
presignedURL, err = s.client.PresignedPutObject(ctx, bucketName, name, expires)
|
presignedURL, err = s.client.PresignedPutObject(ctx, bucketName, name, expires)
|
||||||
|
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// 获取储存桶列表
|
// ListBuckets 获取当前 S3 客户端可访问的所有存储桶列表
|
||||||
|
// 出错时返回 nil
|
||||||
func (s *Mod) ListBuckets() []minio.BucketInfo {
|
func (s *Mod) ListBuckets() []minio.BucketInfo {
|
||||||
buckets, err := s.client.ListBuckets(ctx)
|
buckets, err := s.client.ListBuckets(ctx)
|
||||||
//g.Dump(buckets)
|
|
||||||
if err != nil {
|
if err != nil {
|
||||||
//fmt.Println(err)
|
|
||||||
return nil
|
return nil
|
||||||
}
|
}
|
||||||
return buckets
|
return buckets
|
||||||
}
|
}
|
||||||
|
|
||||||
// PutObject 上传文件到指定的存储桶中。
|
// PutObject 上传文件到指定的存储桶中
|
||||||
//
|
// 支持指定文件大小,未指定时将读取文件直到结束
|
||||||
// @Description: 上传一个文件到指定的存储桶。
|
|
||||||
// @receiver s *Mod: 表示调用此方法的Mod实例。
|
|
||||||
// @param f io.Reader: 文件的读取器,用于读取待上传的文件内容。
|
|
||||||
// @param name string: 待上传文件的名称。
|
|
||||||
// @param bucketName string: 存储桶的名称。
|
|
||||||
// @param _size ...int64: 可选参数,指定上传文件的大小。如果未提供,则默认为-1,表示读取文件直到EOF。
|
|
||||||
// @return res minio.UploadInfo: 上传成功后返回的上传信息。
|
|
||||||
// @return err error: 如果上传过程中出现错误,则返回错误信息。
|
|
||||||
func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...int64) (res minio.UploadInfo, err error) {
|
func (s *Mod) PutObject(f io.Reader, name string, bucketName string, _size ...int64) (res minio.UploadInfo, err error) {
|
||||||
// 初始化文件大小为-1,表示将读取文件至结束。
|
// 初始化文件大小为 -1,表示将读取文件至结束
|
||||||
var size = int64(-1)
|
var size = int64(-1)
|
||||||
// 如果提供了文件大小,则使用提供的大小值。
|
|
||||||
if len(_size) > 0 {
|
if len(_size) > 0 {
|
||||||
size = _size[0]
|
size = _size[0]
|
||||||
}
|
}
|
||||||
|
// 调用 S3 客户端上传文件,设置内容类型为 "application/octet-stream"
|
||||||
// 调用client的PutObject方法上传文件,并设置内容类型为"application/octet-stream"。
|
|
||||||
res, err = s.client.PutObject(ctx, bucketName, name, f, size, minio.PutObjectOptions{ContentType: "application/octet-stream"})
|
res, err = s.client.PutObject(ctx, bucketName, name, f, size, minio.PutObjectOptions{ContentType: "application/octet-stream"})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
|
// 记录上传错误日志
|
||||||
g.Log().Error(ctx, err)
|
g.Log().Error(ctx, err)
|
||||||
}
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// RemoveObject 删除文件
|
// RemoveObject 从指定存储桶中删除指定名称的文件
|
||||||
func (s *Mod) RemoveObject(name string, bucketName string) (err error) {
|
func (s *Mod) RemoveObject(name string, bucketName string) (err error) {
|
||||||
opts := minio.RemoveObjectOptions{
|
opts := minio.RemoveObjectOptions{
|
||||||
//GovernanceBypass: true,
|
//GovernanceBypass: true,
|
||||||
//VersionID: "myversionid",
|
//VersionID: "myversionid",
|
||||||
}
|
}
|
||||||
|
// 调用 S3 客户端删除文件
|
||||||
err = s.client.RemoveObject(ctx, bucketName, name, opts)
|
err = s.client.RemoveObject(ctx, bucketName, name, opts)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// ListObjects 文件列表
|
// ListObjects 获取指定存储桶中指定前缀的文件列表
|
||||||
|
// 返回一个包含文件信息的通道
|
||||||
func (s *Mod) ListObjects(bucketName string, prefix string) (res <-chan minio.ObjectInfo, err error) {
|
func (s *Mod) ListObjects(bucketName string, prefix string) (res <-chan minio.ObjectInfo, err error) {
|
||||||
|
// 调用 S3 客户端获取文件列表
|
||||||
res = s.client.ListObjects(ctx, bucketName, minio.ListObjectsOptions{
|
res = s.client.ListObjects(ctx, bucketName, minio.ListObjectsOptions{
|
||||||
Prefix: prefix,
|
Prefix: prefix,
|
||||||
})
|
})
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// SetBucketPolicy 设置bucket或对象前缀的访问权限
|
// SetBucketPolicy 设置指定存储桶或对象前缀的访问策略
|
||||||
|
// 目前使用固定的策略,可根据需求修改
|
||||||
func (s *Mod) SetBucketPolicy(bucketName string, prefix string) (err error) {
|
func (s *Mod) SetBucketPolicy(bucketName string, prefix string) (err error) {
|
||||||
|
// 定义访问策略
|
||||||
policy := `{"Version": "2012-10-17","Statement": [{"Action": ["s3:GetObject"],"Effect": "Allow","Principal": {"AWS": ["*"]},"Resource": ["arn:aws:s3:::my-bucketname/*"],"Sid": ""}]}`
|
policy := `{"Version": "2012-10-17","Statement": [{"Action": ["s3:GetObject"],"Effect": "Allow","Principal": {"AWS": ["*"]},"Resource": ["arn:aws:s3:::my-bucketname/*"],"Sid": ""}]}`
|
||||||
|
// 调用 S3 客户端设置存储桶策略
|
||||||
err = s.client.SetBucketPolicy(ctx, bucketName, policy)
|
err = s.client.SetBucketPolicy(ctx, bucketName, policy)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
// GetUrl 获取文件访问地址
|
// GetUrl 获取文件的访问地址
|
||||||
|
// 支持返回默认文件地址,根据 SSL 配置生成不同格式的 URL
|
||||||
func (s *Mod) GetUrl(filePath string, defaultFile ...string) (url string) {
|
func (s *Mod) GetUrl(filePath string, defaultFile ...string) (url string) {
|
||||||
bucketName := s.cfg.BucketNameCdn
|
bucketName := s.cfg.BucketNameCdn
|
||||||
get := s.cfg.Url
|
get := s.cfg.Url
|
||||||
|
|
||||||
//如果没有图片,返回默认的图片地址
|
// 如果没有指定文件路径,且提供了默认文件路径,则使用默认路径
|
||||||
if filePath == "" && len(defaultFile) > 0 {
|
if filePath == "" && len(defaultFile) > 0 {
|
||||||
filePath = defaultFile[0]
|
filePath = defaultFile[0]
|
||||||
}
|
}
|
||||||
@@ -262,6 +196,7 @@ func (s *Mod) GetUrl(filePath string, defaultFile ...string) (url string) {
|
|||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// GetPath 从文件访问 URL 中提取文件路径
|
||||||
func (s *Mod) GetPath(url string) (filePath string) {
|
func (s *Mod) GetPath(url string) (filePath string) {
|
||||||
bucketName := s.cfg.BucketNameCdn
|
bucketName := s.cfg.BucketNameCdn
|
||||||
get := s.cfg.Url
|
get := s.cfg.Url
|
||||||
@@ -269,21 +204,22 @@ func (s *Mod) GetPath(url string) (filePath string) {
|
|||||||
return url[len(get+bucketName)+1:]
|
return url[len(get+bucketName)+1:]
|
||||||
}
|
}
|
||||||
|
|
||||||
// 复制文件
|
// CopyObject 在指定存储桶内复制文件
|
||||||
|
// 支持指定源文件和目标文件路径
|
||||||
func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
|
func (s *Mod) CopyObject(bucketName string, dstStr string, srcStr string) (err error) {
|
||||||
|
// 定义目标文件选项
|
||||||
// 原始文件
|
|
||||||
var dst = minio.CopyDestOptions{
|
var dst = minio.CopyDestOptions{
|
||||||
Bucket: bucketName,
|
Bucket: bucketName,
|
||||||
Object: dstStr,
|
Object: dstStr,
|
||||||
}
|
}
|
||||||
|
|
||||||
// 新文件
|
// 定义源文件选项
|
||||||
var src = minio.CopySrcOptions{
|
var src = minio.CopySrcOptions{
|
||||||
Bucket: bucketName,
|
Bucket: bucketName,
|
||||||
Object: srcStr,
|
Object: srcStr,
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// 调用 S3 客户端复制文件
|
||||||
_, err = s.client.CopyObject(ctx, dst, src)
|
_, err = s.client.CopyObject(ctx, dst, src)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -14,6 +14,7 @@ import (
|
|||||||
"github.com/gogf/gf/v2/util/guid"
|
"github.com/gogf/gf/v2/util/guid"
|
||||||
"github.com/gorilla/websocket"
|
"github.com/gorilla/websocket"
|
||||||
"google.golang.org/protobuf/proto"
|
"google.golang.org/protobuf/proto"
|
||||||
|
"net/http"
|
||||||
"sync"
|
"sync"
|
||||||
"time"
|
"time"
|
||||||
)
|
)
|
||||||
@@ -34,9 +35,10 @@ type WebsocketData struct {
|
|||||||
Ws *websocket.Conn `json:"ws" dc:"websocket连接池"`
|
Ws *websocket.Conn `json:"ws" dc:"websocket连接池"`
|
||||||
Uuid string `json:"uuid" dc:"用户唯一标识"`
|
Uuid string `json:"uuid" dc:"用户唯一标识"`
|
||||||
Uid int64 `json:"uid" dc:"用户编号"`
|
Uid int64 `json:"uid" dc:"用户编号"`
|
||||||
|
Guid string `json:"guid" dc:"用户凭证"`
|
||||||
Groups []string `json:"groups" dc:"群组"`
|
Groups []string `json:"groups" dc:"群组"`
|
||||||
Ctx context.Context `json:"ctx" dc:""`
|
Ctx context.Context `json:"ctx" dc:""`
|
||||||
RoomId int `json:"roomId" dc:"房间编号"`
|
RoomId string `json:"roomId" dc:"房间编号"`
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewV1() *SocketV1 {
|
func NewV1() *SocketV1 {
|
||||||
@@ -57,14 +59,21 @@ type SocketInterface interface {
|
|||||||
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)
|
||||||
@@ -92,7 +101,7 @@ func (s *SocketV1) OnConnect(ctx context.Context, ws *websocket.Conn) {
|
|||||||
Ws: ws,
|
Ws: ws,
|
||||||
Ctx: ctx,
|
Ctx: ctx,
|
||||||
Groups: make([]string, 0),
|
Groups: make([]string, 0),
|
||||||
RoomId: -1,
|
RoomId: "",
|
||||||
}
|
}
|
||||||
m.Set(id, conn)
|
m.Set(id, conn)
|
||||||
|
|
||||||
@@ -161,7 +170,7 @@ func (s *SocketV1) OnMessage(conn *WebsocketData, req []byte, msgType int) {
|
|||||||
}
|
}
|
||||||
|
|
||||||
//绑定用户编号
|
//绑定用户编号
|
||||||
func (s *SocketV1) BindUid(conn *WebsocketData, uid int64) {
|
func (s *SocketV1) BindUid(conn *WebsocketData, uid int64, guid string) {
|
||||||
lock.Lock()
|
lock.Lock()
|
||||||
defer lock.Unlock()
|
defer lock.Unlock()
|
||||||
|
|
||||||
@@ -171,6 +180,7 @@ func (s *SocketV1) BindUid(conn *WebsocketData, uid int64) {
|
|||||||
if conn.Uid == 0 {
|
if conn.Uid == 0 {
|
||||||
conn.Uid = uid
|
conn.Uid = uid
|
||||||
}
|
}
|
||||||
|
conn.Guid = guid
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -337,8 +347,8 @@ func (s *SocketV1) OnClose(conn *WebsocketData) {
|
|||||||
// 1. 更新连接状态或从连接池移除
|
// 1. 更新连接状态或从连接池移除
|
||||||
// 2. 发送通知或清理关联资源
|
// 2. 发送通知或清理关联资源
|
||||||
// 3. 执行特定于业务的断开处理
|
// 3. 执行特定于业务的断开处理
|
||||||
m.Remove(conn.Uuid)
|
|
||||||
conn.Ws.Close()
|
conn.Ws.Close()
|
||||||
|
m.Remove(conn.Uuid)
|
||||||
}
|
}
|
||||||
|
|
||||||
// 是否在线
|
// 是否在线
|
||||||
|
|||||||
@@ -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)
|
||||||
}
|
}
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ func init() {
|
|||||||
g.Log().Debug(ctx, "utility_go init启动完成")
|
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)
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user