go_dreamfactory/modules/hero/model_hero.go

package hero

import (
	"fmt"
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/log"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"
	"math"

	mengine "github.com/dengsgo/math-engine/engine"
	"go.mongodb.org/mongo-driver/bson/primitive"
)

type ModelHero struct {
	modules.MCompModel
	moduleHero *Hero
}

func (this *ModelHero) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
	err = this.MCompModel.Init(service, module, comp, options)
	this.moduleHero = module.(*Hero)
	this.TableName = "hero"
	return
}

//初始化英雄
func (this *ModelHero) initHero(uid string, heroCfgId int32) *pb.DBHero {
	heroCfg := this.moduleHero.configure.GetHero(heroCfgId)
	if heroCfg == nil {
		log.Errorf("%v hero not found from config %v", heroCfgId)
		return nil
	}
	objId := primitive.NewObjectID().Hex()
	newHero := &pb.DBHero{
		Id:          objId,
		Uid:         uid,
		HeroID:      heroCfg.Hid,
		Star:        heroCfg.Star,      //初始星级
		Lv:          1,                 //初始等级
		NormalSkill: []*pb.SkillData{}, //初始技能
		IsOverlying: true,              //是否允许叠加,
		Block:       false,             //未锁定
		CardType:    heroCfg.Type,      //卡片类型
		Skins:       []int32{},
		EquipID:     make([]string, 6), //初始装备
		SameCount:   1,                 //默认叠加数量
		AddProperty: make(map[string]int32),
		Energy:      make(map[int32]int32),
		Property:    make(map[string]int32),
	}
	return newHero
}

//创建一个指定的英雄
func (this *ModelHero) createOneHero(uid string, heroCfgId int32) (err error) {
	hero := this.initHero(uid, heroCfgId)
	if hero != nil {
		if err = this.moduleHero.modelHero.AddList(uid, hero.Id, hero); err != nil {
			log.Errorf("%v", err)
			return
		}
	}
	return nil
}

//创建多个指定的英雄 heroCfgIds可填入多个英雄ID
func (this *ModelHero) createMultiHero(uid string, heroCfgIds ...int32) error {
	heroes, err := this.moduleHero.modelHero.getHeroList(uid)
	if err != nil {
		return err
	}

	if len(heroes) == 0 {
		for _, v := range heroCfgIds {
			if err := this.createOneHero(uid, v); err != nil {
				return err
			}
		}
	} else {

		findHero := func(heroId int32) (*pb.DBHero, bool) {
			for _, h := range heroes {
				if h.HeroID == heroId {
					return h, true
				}
			}
			return nil, false
		}
		for _, v := range heroCfgIds {
			if h, ok := findHero(v); ok {
				h.SameCount++
				data := map[string]interface{}{
					"sameCount": h.SameCount, //叠加数
				}
				if err := this.modifyHeroData(uid, h.Id, data); err != nil {
					return err
				}
			} else {
				if err := this.createOneHero(uid, v); err != nil {
					return err
				}
			}
		}
	}

	return nil
}

//获取一个英雄
func (this *ModelHero) getOneHero(uid, heroId string) *pb.DBHero {
	hero := &pb.DBHero{}
	err := this.moduleHero.modelHero.GetListObj(uid, heroId, hero)
	if err != nil {
		return nil
	}
	return hero
}

//消耗一张英雄卡
func (this *ModelHero) consumeOneHeroCard(uid, heroId string, count int32) (err error) {
	for i := 0; i < int(count); i++ {
		if err := this.moduleHero.modelHero.DelListlds(uid, heroId); err != nil {
			log.Errorf("%v", err)
			break
		}
	}
	return
}

//更新英雄数据
func (this *ModelHero) modifyHeroData(uid, heroId string, data map[string]interface{}) error {
	return this.moduleHero.modelHero.ChangeList(uid, heroId, data)
}

//获取玩家的英雄列表
func (this *ModelHero) getHeroList(uid string) ([]*pb.DBHero, error) {
	heroes := make([]*pb.DBHero, 0)
	err := this.GetList(uid, &heroes)
	if err != nil {
		return nil, err
	}
	return heroes, nil
}

//更新装备
func (this *ModelHero) setEquipment(uid, heroId string, equipIds []string) pb.ErrorCode {
	hero := this.getOneHero(uid, heroId)
	if hero == nil {
		return pb.ErrorCode_HeroNoExist
	}

	hero.EquipID = equipIds
	return pb.ErrorCode_Success
}

//合并属性即属性值累加
func (this *ModelHero) mergeMainProperty(uid, heroId string, data map[string]int32) {
	hero := this.getOneHero(uid, heroId)
	if hero == nil {
		return
	}

	hero.Property[comm.PropertyHp] += data[comm.PropertyHp]
	hero.Property[comm.PropertyAtk] += data[comm.PropertyAtk]
	hero.Property[comm.PropertyDef] += data[comm.PropertyDef]
}

//合并附加属性
func (this *ModelHero) mergeAddProperty(uid, heroId string, data map[string]int32) {
	hero := this.getOneHero(uid, heroId)
	if hero == nil {
		return
	}
	hero.AddProperty[comm.PropertyHp] += data[comm.PropertyHp]
	hero.AddProperty[comm.PropertyAtk] += data[comm.PropertyAtk]
	hero.AddProperty[comm.PropertyDef] += data[comm.PropertyDef]
}

//属性计算 - 暂时放在modelHero里实现
//英雄基础属性 + 英雄等级基础属性 * 英雄成长系数 + 英雄星级对应等级属性 * 英雄品质系数
func (this *ModelHero) PropertyCompute(uid, heroId string) map[string]int32 {
	hero := this.getOneHero(uid, heroId)
	if hero == nil {
		return nil
	}

	//英雄等级基础属性levelup
	heroLvCfg := this.moduleHero.configure.GetHeroLv(hero.Lv)
	if heroLvCfg == nil {
		return nil
	}

	//英雄基础配置 newhero
	heroCfg := this.moduleHero.configure.GetHero(hero.HeroID)
	if heroCfg == nil {
		return nil
	}

	//品质系数
	stargrowCfg := this.moduleHero.configure.GetHeroStar(heroCfg.Star)
	if stargrowCfg == nil {
		return nil
	}

	//英雄星级对应等级属性
	heroStarCfg := this.moduleHero.configure.GetHeroLv(stargrowCfg.Level)
	if heroStarCfg == nil {
		return nil
	}

	//成长系数
	lvGrow := this.moduleHero.configure.GetHeroLvgrow(hero.HeroID)
	if lvGrow == nil {
		return nil
	}

	curHp := hero.Property[comm.PropertyHp]
	exprHp := fmt.Sprintf("%v + %v * %v/1000 + %v * %v",
		(curHp + lvGrow.Hp), heroLvCfg.Hp, lvGrow.Hpgrow, heroStarCfg.Hp, stargrowCfg.StarupHp)
	hp, _ := mengine.ParseAndExec(exprHp)

	curAtk := hero.Property[comm.PropertyAtk]
	exprAtk := fmt.Sprintf("%v +%v * %v/1000 + %v * %v",
		(curAtk + lvGrow.Atk), heroLvCfg.Atk, lvGrow.Atkgrow, heroStarCfg.Atk, stargrowCfg.StarupAtk)
	atk, _ := mengine.ParseAndExec(exprAtk)

	curDef := hero.Property[comm.PropertyDef]
	exprDef := fmt.Sprintf("%v +%v * %v/1000 + %v * %v",
		(curDef + lvGrow.Def), heroLvCfg.Def, lvGrow.Defgrow, heroStarCfg.Def, stargrowCfg.StarupDef)
	def, _ := mengine.ParseAndExec(exprDef)

	return map[string]int32{
		comm.PropertyHp:  int32(math.Floor(hp)),
		comm.PropertyAtk: int32(math.Floor(atk)),
		comm.PropertyDef: int32(math.Floor(def)),
	}
}

//推送用户指定英雄属性
func (this *ModelHero) PushHeroProperty(session comm.IUserSession, heroId string) (err error) {
	m := this.PropertyCompute(session.GetUserId(), heroId)
	update := map[string]interface{}{
		"property": m,
	}

	if err = this.ChangeList(session.GetUserId(), heroId, update); err != nil {
		return
	}
	return session.SendMsg("push", "property", &pb.HeroProperty{Property: m})
}