package service import ( "blazing/common/data" "blazing/cool" "blazing/modules/config/model" "context" "encoding/json" "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 if method == "Delete" { return nil } 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 并扫描返回值 dbm(s.Model). Wheref(`bind_elf_ids @> ?::jsonb`, id). Wheref(`jsonb_typeof(bind_elf_ids) = ?`, "array"). Where("is_enabled", 1).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 { if cool.Config.ServerInfo.IsVip == 0 { 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 并扫描返回值 dbm(s.Model). Wheref(`bind_elf_ids @> ?::jsonb`, id). Wheref(`jsonb_typeof(bind_elf_ids) = ?`, "array"). Where("is_enabled", 1).Scan(&ret) if len(ret) == 0 { return nil } v := ret[grand.Intn(len(ret))] var t data.GlowFilter r := json.Unmarshal([]byte(v.Color), &t) if r == nil { if cool.Config.ServerInfo.IsVip == 0 { m := cool.DBM(s.Model).Where("id", v.ID) m.Increment("usage_count", 1) } return &t } return nil }