fix: 修复道具扣除和宠物融合事务处理问题

modules/player/service/pet_fusion_tx.go

@@ -0,0 +1,269 @@
+package service
+
+import (
+	"blazing/common/socket/errorcode"
+	"blazing/cool"
+	baseservice "blazing/modules/base/service"
+	"blazing/modules/player/model"
+	"context"
+	"errors"
+
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+var (
+	errPetFusionInsufficientItems = errors.New("pet fusion insufficient items")
+	errPetFusionPetNotFound       = errors.New("pet fusion pet not found")
+	errPetFusionPlayerNotFound    = errors.New("pet fusion player not found")
+)
+
+type PetFusionTxResult struct {
+	NewPet       *model.PetInfo
+	CostItemUsed bool
+	UpdatedAux   *model.PetInfo
+}
+
+func (s *UserService) PetFusionTx(
+	currentInfo model.PlayerInfo,
+	masterCatchTime uint32,
+	auxCatchTime uint32,
+	materialCounts map[uint32]int,
+	goldItemIDs []uint32,
+	keepAuxItemID uint32,
+	failureItemID uint32,
+	cost int64,
+	newPet *model.PetInfo,
+	failedAux *model.PetInfo,
+) (*PetFusionTxResult, errorcode.ErrorCode) {
+	if s == nil || s.Pet == nil {
+		return nil, errorcode.ErrorCodes.ErrSystemError
+	}
+
+	userID := s.Pet.userid
+	nextInfo := currentInfo
+	if nextInfo.Coins < cost {
+		return nil, errorcode.ErrorCodes.ErrSunDouInsufficient10016
+	}
+	nextInfo.Coins -= cost
+
+	result := &PetFusionTxResult{}
+	err := g.DB().Transaction(context.TODO(), func(ctx context.Context, tx gdb.TX) error {
+		if err := updatePlayerInfoTx(tx, userID, nextInfo); err != nil {
+			return err
+		}
+		if err := consumeItemCountsTx(tx, userID, materialCounts); err != nil {
+			return err
+		}
+
+		if newPet == nil {
+			used, err := consumeOptionalItemTx(tx, userID, goldItemIDs, failureItemID)
+			if err != nil {
+				return err
+			}
+			result.CostItemUsed = used
+			if !used && failedAux != nil {
+				if err := updatePetDataTx(tx, userID, auxCatchTime, *failedAux); err != nil {
+					return err
+				}
+				auxCopy := *failedAux
+				result.UpdatedAux = &auxCopy
+			}
+			return nil
+		}
+
+		if err := deletePetTx(tx, userID, masterCatchTime); err != nil {
+			return err
+		}
+
+		used, err := consumeOptionalItemTx(tx, userID, goldItemIDs, keepAuxItemID)
+		if err != nil {
+			return err
+		}
+		result.CostItemUsed = used
+		if !used {
+			if err := deletePetTx(tx, userID, auxCatchTime); err != nil {
+				return err
+			}
+		}
+
+		catchTime, err := addPetTx(tx, s.Pet, newPet, 0)
+		if err != nil {
+			return err
+		}
+		petCopy := *newPet
+		petCopy.CatchTime = catchTime
+		result.NewPet = &petCopy
+		return nil
+	})
+	if err == nil {
+		return result, 0
+	}
+
+	switch {
+	case errors.Is(err, errPetFusionInsufficientItems):
+		return nil, errorcode.ErrorCodes.ErrInsufficientItems
+	case errors.Is(err, errPetFusionPetNotFound):
+		return nil, errorcode.ErrorCodes.ErrPokemonNotFusionReady2
+	case errors.Is(err, errPetFusionPlayerNotFound):
+		return nil, errorcode.ErrorCodes.ErrSystemError
+	default:
+		cool.Logger.Error(context.TODO(), "pet fusion tx failed", userID, err)
+		return nil, errorcode.ErrorCodes.ErrSystemError
+	}
+}
+
+func updatePlayerInfoTx(tx gdb.TX, userID uint32, info model.PlayerInfo) error {
+	res, err := tx.Model(model.NewPlayer()).Where("player_id", userID).Data("data", info).Update()
+	if err != nil {
+		return err
+	}
+	affected, err := res.RowsAffected()
+	if err != nil {
+		return err
+	}
+	if affected == 0 {
+		return errPetFusionPlayerNotFound
+	}
+	return nil
+}
+
+func consumeItemCountsTx(tx gdb.TX, userID uint32, itemCounts map[uint32]int) error {
+	for itemID, count := range itemCounts {
+		if err := updateItemCountTx(tx, userID, itemID, -count); err != nil {
+			return err
+		}
+	}
+	return nil
+}
+
+func consumeOptionalItemTx(tx gdb.TX, userID uint32, itemIDs []uint32, target uint32) (bool, error) {
+	for _, itemID := range itemIDs {
+		if itemID != target {
+			continue
+		}
+		if err := updateItemCountTx(tx, userID, target, -1); err != nil {
+			return false, err
+		}
+		return true, nil
+	}
+	return false, nil
+}
+
+func updateItemCountTx(tx gdb.TX, userID uint32, id uint32, count int) error {
+	if cool.Config.ServerInfo.IsVip != 0 && count < 0 {
+		return nil
+	}
+	if id == 0 {
+		return nil
+	}
+
+	baseModel := tx.Model(model.NewPlayerBag()).
+		Where("player_id", userID).
+		Where("is_vip", cool.Config.ServerInfo.IsVip)
+
+	ok, err := baseModel.Where("item_id", id).Exist()
+	if err != nil {
+		return err
+	}
+
+	if ok {
+		updateModel := tx.Model(model.NewPlayerBag()).
+			Where("player_id", userID).
+			Where("is_vip", cool.Config.ServerInfo.IsVip).
+			Where("item_id", id)
+		if count < 0 {
+			updateModel = updateModel.Where("item_cnt + ? > 0", count)
+		}
+
+		result, err := updateModel.Increment("item_cnt", count)
+		if err != nil {
+			return err
+		}
+		affected, err := result.RowsAffected()
+		if err != nil {
+			return err
+		}
+		if affected == 0 {
+			return errPetFusionInsufficientItems
+		}
+		return nil
+	}
+
+	if count <= 0 {
+		return errPetFusionInsufficientItems
+	}
+
+	_, err = tx.Model(model.NewPlayerBag()).Data(g.Map{
+		"player_id": userID,
+		"item_id":   id,
+		"item_cnt":  count,
+		"is_vip":    cool.Config.ServerInfo.IsVip,
+	}).Insert()
+	return err
+}
+
+func updatePetDataTx(tx gdb.TX, userID uint32, catchTime uint32, pet model.PetInfo) error {
+	res, err := tx.Model(model.NewPet()).
+		Where("player_id", userID).
+		Where("is_vip", cool.Config.ServerInfo.IsVip).
+		Where("catch_time", catchTime).
+		Data("data", pet).
+		Update()
+	if err != nil {
+		return err
+	}
+	affected, err := res.RowsAffected()
+	if err != nil {
+		return err
+	}
+	if affected == 0 {
+		return errPetFusionPetNotFound
+	}
+	return nil
+}
+
+func deletePetTx(tx gdb.TX, userID uint32, catchTime uint32) error {
+	res, err := tx.Model(model.NewPet()).
+		Where("player_id", userID).
+		Where("is_vip", cool.Config.ServerInfo.IsVip).
+		Where("catch_time", catchTime).
+		Delete()
+	if err != nil {
+		return err
+	}
+	affected, err := res.RowsAffected()
+	if err != nil {
+		return err
+	}
+	if affected == 0 {
+		return errPetFusionPetNotFound
+	}
+	return nil
+}
+
+func addPetTx(tx gdb.TX, petService *PetService, petInfo *model.PetInfo, saleCount uint32) (uint32, error) {
+	if petInfo == nil {
+		return 0, nil
+	}
+
+	catchTime, err := petService.nextCatchTime(tx.Model(baseservice.NewBaseSysUserService().Model))
+	if err != nil {
+		return 0, err
+	}
+
+	petCopy := *petInfo
+	petCopy.CatchTime = catchTime
+	playerPet := model.Pet{
+		PlayerID:  petService.userid,
+		Data:      petCopy,
+		CatchTime: catchTime,
+		Free:      0,
+		SaleCount: saleCount,
+	}
+	playerPet.IsVip = cool.Config.ServerInfo.IsVip
+	if _, err := tx.Model(model.NewPet()).Insert(playerPet); err != nil {
+		return 0, err
+	}
+	return catchTime, nil
+}