昔念
41babda646
feat(config): 新增固定异色功能并重构异色服务 - 新增 FixShiny 方法用于固定获取异色效果 - 重命名 Args 方法为 RandShiny 以明确功能 - 添加 utils 包依赖用于数据转换 feat(model): 扩展宠物异色相关方法 - 为 PetInfo 结构体添加 FixShiny 方法 - 修改 RandShiny 方法逻辑，确保异色信息正确添加 ref
106 lines
2.1 KiB
Go
106 lines
2.1 KiB
Go
package service
|
|
|
|
import (
|
|
"blazing/common/data"
|
|
"blazing/common/utils"
|
|
"blazing/cool"
|
|
"blazing/modules/config/model"
|
|
"context"
|
|
"encoding/json"
|
|
|
|
"github.com/gogf/gf/v2/database/gdb"
|
|
"github.com/gogf/gf/v2/frame/g"
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
"github.com/gogf/gf/v2/util/grand"
|
|
)
|
|
|
|
type ShinyService struct {
|
|
*cool.Service
|
|
}
|
|
|
|
func NewShinyService() *ShinyService {
|
|
return &ShinyService{
|
|
&cool.Service{
|
|
Model: model.NewColorfulSkin(),
|
|
InsertParam: func(ctx context.Context) g.MapStrAny {
|
|
admin := cool.GetAdmin(ctx)
|
|
userId := admin.UserId
|
|
return g.MapStrAny{
|
|
"author": userId,
|
|
}
|
|
},
|
|
},
|
|
}
|
|
}
|
|
func (s *ShinyService) ModifyBefore(ctx context.Context, method string, param g.MapStrAny) (err error) {
|
|
var t data.GlowFilter
|
|
|
|
r := json.Unmarshal([]byte(gconv.String(param["color"])), &t)
|
|
if r != nil {
|
|
return r
|
|
}
|
|
return nil
|
|
}
|
|
func (s *ShinyService) RandShiny(id uint32) *data.GlowFilter {
|
|
var ret []model.ColorfulSkin
|
|
|
|
// 执行 Raw SQL 并扫描返回值
|
|
cool.DBM(s.Model).Wheref(`bind_elf_ids @> ?::jsonb`, id).Wheref(`jsonb_typeof(bind_elf_ids) = ?`, "array").Cache(gdb.CacheOption{
|
|
// Duration: time.Hour,
|
|
|
|
Force: false,
|
|
}).Scan(&ret)
|
|
|
|
for _, v := range ret {
|
|
//print(v.ID)
|
|
|
|
id := v.ID
|
|
|
|
if grand.Meet(int(v.ElfProbability), 1000) {
|
|
var t data.GlowFilter
|
|
|
|
r := json.Unmarshal([]byte(v.Color), &t)
|
|
if r == nil {
|
|
m := cool.DBM(s.Model).Where("id", id)
|
|
m.Increment("refresh_count", 1)
|
|
return &t
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
func (s *ShinyService) FixShiny(id uint32) *data.GlowFilter {
|
|
var ret []model.ColorfulSkin
|
|
|
|
// 执行 Raw SQL 并扫描返回值
|
|
cool.DBM(s.Model).Wheref(`bind_elf_ids @> ?::jsonb`, id).Wheref(`jsonb_typeof(bind_elf_ids) = ?`, "array").Cache(gdb.CacheOption{
|
|
// Duration: time.Hour,
|
|
|
|
Force: false,
|
|
}).Scan(&ret)
|
|
|
|
rets := utils.ToMap(ret, func(t model.ColorfulSkin) uint32 {
|
|
return uint32(t.ID)
|
|
})
|
|
|
|
for _, v := range rets {
|
|
//print(v.ID)
|
|
|
|
id := v.ID
|
|
|
|
var t data.GlowFilter
|
|
|
|
r := json.Unmarshal([]byte(v.Color), &t)
|
|
if r == nil {
|
|
m := cool.DBM(s.Model).Where("id", id)
|
|
m.Increment("refresh_count", 1)
|
|
return &t
|
|
}
|
|
|
|
}
|
|
|
|
return nil
|
|
}
|