package service import ( "blazing/cool" "blazing/modules/player/model" "context" dictservice "blazing/modules/dict/service" "github.com/gogf/gf/v2/frame/g" ) func (s *ItemService) Exist(itemid uint32) bool { var ttt *model.Item s.dbm(s.Model).Where("item_id", itemid).Scan(&ttt) return ttt != nil } func (s *ItemService) Get(min, max uint32) []model.Item { var ttt []model.Item s.dbm(s.Model).WhereBetween("item_id", min, max).Where("item_cnt >", 0).Scan(&ttt) return ttt } func (s *ItemService) GetUserItemList(min, max, leftTime uint32) []model.SingleItemInfo { var items []model.SingleItemInfo s.dbm(s.Model). Fields("item_id,item_cnt"). WhereBetween("item_id", min, max). Where("item_cnt >", 0). Scan(&items) for i := range items { items[i].LeftTime = leftTime } return items } func (s *ItemService) UPDATE(id uint32, count int) error { if cool.Config.ServerInfo.IsVip != 0 && count < 0 { return nil } if id == 0 { return nil } m := s.dbm(s.Model) ok, err := m.Where("item_id", id).Exist() if err != nil { return err } if ok { _, err := s.dbm(s.Model).Where("item_id", id).Increment("item_cnt", count) if err != nil { return err } } else { m := s.dbm(s.Model) data := g.Map{ "player_id": s.userid, "item_id": id, "item_cnt": count, "is_vip": cool.Config.ServerInfo.IsVip, } _, err := m.Data(data).Insert() return err } return nil } // AddUniqueItems 为一组互不重复的物品各增加 1 个。 // 返回值只包含本次实际成功增加的物品 id。 func (s *ItemService) AddUniqueItems(ids []uint32) ([]uint32, error) { if len(ids) == 0 { return nil, nil } currentItems := s.CheakItemM(ids...) currentMap := make(map[uint32]int64, len(currentItems)) for _, item := range currentItems { currentMap[item.ItemId] = item.ItemCnt } updateIDs := make([]uint32, 0, len(ids)) insertData := g.List{} successIDs := make([]uint32, 0, len(ids)) for _, id := range ids { if id == 0 { continue } itemmax := dictservice.NewDictInfoService().GetMax(int64(id)) if itemmax == 0 { continue } if currentMap[id]+1 > int64(itemmax) { continue } successIDs = append(successIDs, id) if _, ok := currentMap[id]; ok { updateIDs = append(updateIDs, id) continue } insertData = append(insertData, g.Map{ "player_id": s.userid, "item_id": id, "item_cnt": 1, "is_vip": cool.Config.ServerInfo.IsVip, }) } if len(updateIDs) > 0 { if _, err := s.dbm(s.Model).WhereIn("item_id", updateIDs).Increment("item_cnt", 1); err != nil { return nil, err } } if len(insertData) > 0 { if _, err := s.dbm(s.Model).Data(insertData).Insert(); err != nil { return nil, err } } return successIDs, nil } // func (s *ItemService) UPDATEM(ids map[uint32]int) { // if cool.Config.ServerInfo.IsVip != 0 { // return // } // m := s.dbm(s.Model) // data := g.List{} // for k, v := range ids { // data = append(data, g.Map{ // "player_id": s.userid, // "item_id": k, // "item_cnt": v, // "is_vip": cool.Config.ServerInfo.IsVip, // }) // } // m.Data(data).Insert() // } func (s *ItemService) CheakItem(id uint32) int64 { var ttt model.Item m := s.dbm(s.Model) m.Where("item_id", id).Scan(&ttt) return ttt.ItemCnt } func (s *ItemService) CheakItemM(id ...uint32) []model.Item { var ttt []model.Item m := s.dbm(s.Model) m.WhereIn("item_id", id).Scan(&ttt) return ttt } // /添加进来的物品一定是保证存在的 type ItemService struct { BaseService } func NewItemService(id uint32) *ItemService { return &ItemService{ BaseService: BaseService{userid: id, Service: &cool.Service{Model: model.NewPlayerBag(), UniqueKey: map[string]string{ "player_id": "角色名称不能重复", }, PageQueryOp: &cool.QueryOp{ KeyWordField: []string{"player_id"}, FieldEQ: []string{"player_id"}, Where: func(ctx context.Context) [][]interface{} { var ( //admin = cool.GetAdmin(ctx) //userId = admin.UserId ) return [][]interface{}{ // {"player_id", userId, true}, // {"free", 0, true}, } }, }}, }, } }