bl/modules/player/service/cdk_reward.go

package service

import (
	"blazing/common/data"
	baseservice "blazing/modules/base/service"
	configservice "blazing/modules/config/service"
	"blazing/modules/player/model"
	"fmt"
	"time"
)

type CdkRewardPet struct {
	PetID     uint32 `json:"pet_id"`
	CatchTime uint32 `json:"catch_time"`
}

type CdkRewardResult struct {
	CdkID    uint32          `json:"cdk_id"`
	Items    []data.ItemInfo `json:"items,omitempty"`
	Pets     []CdkRewardPet  `json:"pets,omitempty"`
	TitleIDs []uint32        `json:"title_ids,omitempty"`
	Coins    int64           `json:"coins,omitempty"`
	Gold     int64           `json:"gold,omitempty"`
	FreeGold int64           `json:"free_gold,omitempty"`
	ExpPool  int64           `json:"exp_pool,omitempty"`
	EVPool   int64           `json:"ev_pool,omitempty"`
}

type CdkSpecialRewardAdder func(itemID uint32, count int64) bool

// GrantConfigReward 按 cdk 配置 ID 发放奖励，不处理兑换码次数和领取资格校验。
// 当传入 specialAdder 时，赛尔豆/累计经验/金豆/学习力会优先走在线玩家加成逻辑。
func (s *CdkService) GrantConfigReward(cdkID uint32, specialAdders ...CdkSpecialRewardAdder) (*CdkRewardResult, error) {
	cfg := configservice.NewCdkService().GetByID(cdkID)
	if cfg == nil {
		return nil, fmt.Errorf("绑定的CDK不存在")
	}
	if cfg.Type != configservice.CDKTypeReward {
		return nil, fmt.Errorf("当前CDK不是游戏内奖励类型")
	}
	if cfg.BindUserId != 0 && cfg.BindUserId != s.userid {
		return nil, fmt.Errorf("CDK已绑定其他用户")
	}
	if !cfg.ValidEndTime.IsZero() && cfg.ValidEndTime.Before(time.Now()) {
		return nil, fmt.Errorf("绑定的CDK已过期")
	}

	result := &CdkRewardResult{CdkID: cdkID}
	infoService := NewInfoService(s.userid)
	playerInfo := infoService.GetLogin()
	if playerInfo == nil {
		return nil, fmt.Errorf("玩家角色不存在")
	}
	var specialAdder CdkSpecialRewardAdder
	if len(specialAdders) > 0 {
		specialAdder = specialAdders[0]
	}

	var (
		infoDirty bool
		bagItems  []data.ItemInfo
	)

	appendRewardItem := func(itemID uint32, count int64) error {
		if itemID == 0 || count <= 0 {
			return nil
		}
		switch itemID {
		case 1:
			result.Coins += count
			if specialAdder != nil {
				if !specialAdder(itemID, count) {
					return fmt.Errorf("在线发放赛尔豆失败")
				}
			} else {
				playerInfo.Coins += count
				infoDirty = true
			}
		case 3:
			result.ExpPool += count
			if specialAdder != nil {
				if !specialAdder(itemID, count) {
					return fmt.Errorf("在线发放经验池失败")
				}
			} else {
				playerInfo.ExpPool += count
				infoDirty = true
			}
		case 5:
			result.Gold += count
			if specialAdder != nil {
				if !specialAdder(itemID, count) {
					return fmt.Errorf("在线发放金豆失败")
				}
			}
		case 9:
			result.EVPool += count
			if specialAdder != nil {
				if !specialAdder(itemID, count) {
					return fmt.Errorf("在线发放学习力失败")
				}
			} else {
				playerInfo.EVPool += count
				infoDirty = true
			}
		default:
			bagItems = append(bagItems, data.ItemInfo{ItemId: int64(itemID), ItemCnt: count})
		}
		return nil
	}

	for _, rewardID := range cfg.ItemRewardIds {
		itemInfo := configservice.NewItemService().GetItemCount(rewardID)
		if err := appendRewardItem(uint32(itemInfo.ItemId), itemInfo.ItemCnt); err != nil {
			return nil, err
		}
	}

	if result.Gold != 0 && specialAdder == nil {
		baseservice.NewBaseSysUserService().UpdateGold(s.userid, result.Gold*100)
	}
	if result.FreeGold != 0 {
		baseservice.NewBaseSysUserService().UpdateFreeGold(s.userid, result.FreeGold*100)
	}
	if len(bagItems) > 0 {
		items, err := NewItemService(s.userid).AddItems(bagItems)
		if err != nil {
			return nil, err
		}
		result.Items = items
	}

	for _, rewardID := range cfg.ElfRewardIds {
		pet := configservice.NewPetRewardService().Get(rewardID)
		if pet == nil {
			continue
		}
		petInfo := model.GenPetInfo(int(pet.MonID), int(pet.DV), int(pet.Nature), int(pet.Effect), int(pet.Lv), nil, 0)
		catchTime, err := NewPetService(s.userid).PetAdd(petInfo, 0)
		if err != nil {
			return nil, err
		}
		result.Pets = append(result.Pets, CdkRewardPet{
			PetID:     uint32(pet.MonID),
			CatchTime: catchTime,
		})
	}

	if cfg.TitleRewardIds != 0 {
		NewTitleService(s.userid).Give(cfg.TitleRewardIds)
		result.TitleIDs = append(result.TitleIDs, cfg.TitleRewardIds)
	}

	if infoDirty {
		infoService.Save(*playerInfo)
	}

	return result, nil
}