bl/modules/player/service/item.go

package service

import (
	"blazing/common/data"
	"blazing/cool"
	"blazing/modules/player/model"
	"context"
	"strings"

	dictservice "blazing/modules/dict/service"

	"github.com/gogf/gf/v2/database/gdb"
	"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
}

// AddItems 批量添加道具，返回本次实际成功添加的奖励明细（保留原始顺序）。
// AddItemsChecked 写入已完成上限校验的批量道具。
func (s *ItemService) AddItemsChecked(items []data.ItemInfo, currentMap map[uint32]int64) ([]data.ItemInfo, error) {
	if len(items) == 0 {
		return nil, nil
	}

	updateCounts := make(map[uint32]int64, len(items))
	insertData := g.List{}
	successItems := make([]data.ItemInfo, 0, len(items))

	for _, item := range items {
		if item.ItemId <= 0 || item.ItemCnt <= 0 {
			continue
		}

		itemID := uint32(item.ItemId)
		successItems = append(successItems, item)
		if _, ok := currentMap[itemID]; ok {
			updateCounts[itemID] += item.ItemCnt
			continue
		}

		currentMap[itemID] = item.ItemCnt
		insertData = append(insertData, g.Map{
			"player_id": s.userid,
			"item_id":   itemID,
			"item_cnt":  item.ItemCnt,
			"is_vip":    cool.Config.ServerInfo.IsVip,
		})
	}

	err := g.DB().Transaction(context.TODO(), func(ctx context.Context, tx gdb.TX) error {
		if len(updateCounts) > 0 {
			if err := s.batchIncrementItems(tx, updateCounts); err != nil {
				return err
			}
		}

		if len(insertData) > 0 {
			if _, err := tx.Model(s.Model).Data(insertData).Insert(); err != nil {
				return err
			}
		}
		return nil
	})
	if err != nil {
		return nil, err
	}

	return successItems, nil
}

func (s *ItemService) AddItems(items []data.ItemInfo) ([]data.ItemInfo, error) {
	if len(items) == 0 {
		return nil, nil
	}

	var (
		itemIDs = make([]uint32, 0, len(items))
		seenIDs = make(map[uint32]struct{}, len(items))
	)

	for _, item := range items {
		if item.ItemId <= 0 || item.ItemCnt <= 0 {
			continue
		}

		itemID := uint32(item.ItemId)
		if _, ok := seenIDs[itemID]; ok {
			continue
		}

		seenIDs[itemID] = struct{}{}
		itemIDs = append(itemIDs, itemID)
	}

	if len(itemIDs) == 0 {
		return nil, nil
	}

	currentItems := s.CheakItemM(itemIDs...)
	currentMap := make(map[uint32]int64, len(currentItems))
	for _, item := range currentItems {
		currentMap[item.ItemId] = item.ItemCnt
	}
	maxMap := dictservice.NewDictInfoService().GetMaxMap(itemIDs...)

	pendingMap := make(map[uint32]int64, len(itemIDs))
	checkedItems := make([]data.ItemInfo, 0, len(items))

	for _, item := range items {
		if item.ItemId <= 0 || item.ItemCnt <= 0 {
			continue
		}

		itemID := uint32(item.ItemId)
		itemMax := maxMap[itemID]
		if itemMax == 0 {
			continue
		}
		if currentMap[itemID]+pendingMap[itemID]+item.ItemCnt > int64(itemMax) {
			continue
		}

		pendingMap[itemID] += item.ItemCnt
		checkedItems = append(checkedItems, item)
	}

	return s.AddItemsChecked(checkedItems, currentMap)
}

func (s *ItemService) batchIncrementItems(tx gdb.TX, itemCounts map[uint32]int64) error {
	var (
		builder strings.Builder
		args    = make([]any, 0, len(itemCounts)*3+2)
		itemIDs = make([]any, 0, len(itemCounts))
	)

	builder.WriteString("UPDATE ")
	builder.WriteString(s.Model.TableName())
	builder.WriteString(" SET item_cnt = CASE item_id ")

	for itemID, itemCnt := range itemCounts {
		builder.WriteString("WHEN ? THEN item_cnt + ? ")
		args = append(args, itemID, itemCnt)
		itemIDs = append(itemIDs, itemID)
	}

	builder.WriteString("ELSE item_cnt END WHERE player_id = ? AND is_vip = ? AND item_id IN (")
	for i := range itemIDs {
		if i > 0 {
			builder.WriteString(",")
		}
		builder.WriteString("?")
	}
	builder.WriteString(")")

	args = append(args, s.userid, cool.Config.ServerInfo.IsVip)
	args = append(args, itemIDs...)

	_, err := tx.Exec(builder.String(), args...)
	return err
}

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},
					}
				},
			}},
		},
	}

}