go_dreamfactory/modules/hero/module.go

package hero

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/lego/sys/event"
	"go_dreamfactory/lego/sys/log"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"
)

func NewModule() core.IModule {
	m := new(Hero)
	return m
}

type Hero struct {
	modules.ModuleBase
	api         *apiComp
	configure   *configureComp
	modelHero   *ModelHero
	modelRecord *ModelRecord
}

//模块名
func (this *Hero) GetType() core.M_Modules {
	return comm.ModuleHero
}

//模块初始化接口 注册用户创建角色事件
func (this *Hero) Init(service core.IService, module core.IModule, options core.IModuleOptions) (err error) {
	err = this.ModuleBase.Init(service, module, options)
	return
}

//装备组件
func (this *Hero) OnInstallComp() {
	this.ModuleBase.OnInstallComp()
	this.api = this.RegisterComp(new(apiComp)).(*apiComp)
	this.modelHero = this.RegisterComp(new(ModelHero)).(*ModelHero)
	this.modelRecord = this.RegisterComp(new(ModelRecord)).(*ModelRecord)
	this.configure = this.RegisterComp(new(configureComp)).(*configureComp)

}
func (this *Hero) Start() (err error) {
	err = this.ModuleBase.Start()
	event.RegisterGO(comm.EventUserOffline, this.EventUserOffline)
	return
}

//创建新英雄
func (this *Hero) CreateHeroes(uid string, heroCfgId ...string) error {
	return this.modelHero.createMultiHero(uid, heroCfgId...)
}

//创建单个叠加英雄
func (this *Hero) CreateRepeatHero(session comm.IUserSession, heroCfgId string, num int32, bPush bool) (code pb.ErrorCode) {
	_hero, err := this.modelHero.createHeroOverlying(session.GetUserId(), heroCfgId, num)
	if err == nil {
		go func(uid string, heroCfgId string) { // 携程处理 图鉴数据
			if result, err1 := this.ModuleUser.GetUserExpand(uid); err1 == nil {
				initUpdate := map[string]interface{}{}
				sz := result.GetTujian()
				if len(sz) == 0 {
					sz = make(map[string]bool, 0)
				}

				if _, ok := result.GetTujian()[heroCfgId]; !ok {
					sz[heroCfgId] = true
					initUpdate["tujian"] = sz
				}

				if len(initUpdate) != 0 {
					this.ModuleUser.ChangeUserExpand(uid, initUpdate)
				}
			}
		}(session.GetUserId(), heroCfgId)
		if bPush { //推送
			session.SendMsg("hero", "change", &pb.HeroChangePush{List: []*pb.DBHero{_hero}})
		}
		return
	}
	code = pb.ErrorCode_HeroCreate
	return
}

//获取英雄
func (this *Hero) GetHeroByObjID(uid, heroId string) (*pb.DBHero, pb.ErrorCode) {
	hero := this.modelHero.getOneHero(uid, heroId)
	if hero == nil {
		return nil, pb.ErrorCode_HeroNoExist
	}
	return hero, pb.ErrorCode_Success
}

//佩戴装备
func (this *Hero) UpdateEquipment(session comm.IUserSession, hero *pb.DBHero, equip []*pb.DB_Equipment) (code pb.ErrorCode) {

	if hero == nil {
		code = pb.ErrorCode_HeroNoExist
		return
	}

	list := make([]*pb.DBHero, 0)
	if newHero, err := this.modelHero.setEquipment(hero); err != nil {
		code = pb.ErrorCode_HeroEquipUpdate
		return
	} else {
		if newHero != nil {
			list = append(list, newHero)
		}
	}
	list = append(list, hero)

	this.modelHero.setEquipProperty(hero, equip)
	session.SendMsg("hero", "change", &pb.HeroChangePush{List: list})

	return
}

//英雄列表
func (this *Hero) GetHeroList(uid string) []*pb.DBHero {
	return this.modelHero.getHeroList(uid)
}

//查询英雄数量
func (this *Hero) QueryHeroAmount(uId string, heroCfgId string) (amount uint32) {
	heroes := this.GetHeroList(uId)
	for _, v := range heroes {
		if v.HeroID == heroCfgId {
			amount++
		}
	}
	return amount
}

// 删除指定卡牌
func (this *Hero) DelCard(udi string, hero *pb.DBHero, amount int32) (code pb.ErrorCode) {
	err := this.modelHero.consumeHeroCard(udi, hero, amount)
	if err != nil {
		code = pb.ErrorCode_DBError
		return
	}
	return
}

// 清空数据
func (this *Hero) CleanData(uid string) {
	this.modelHero.cleanData(uid)
}

// 创建一些特殊的英雄
func (this *Hero) GetSpecifiedHero(session comm.IUserSession, heroConfId string, star, lv, amount int32) (code pb.ErrorCode) {
	if session.GetUserId() == "" || heroConfId == "" || star == 0 || lv == 0 || amount == 0 {
		return pb.ErrorCode_ReqParameterError
	}
	hero, err := this.modelHero.createOneHero(session.GetUserId(), heroConfId)
	if err != nil {
		return pb.ErrorCode_HeroCreate
	}
	hero.Lv = lv
	hero.Star = star
	hero.SameCount = amount
	_heroMap := map[string]interface{}{
		"lv":          hero.Lv,
		"star":        hero.Star,
		"isOverlying": false,
		"sameCount":   amount,
	}
	// 保存数据
	err = this.modelHero.ChangeList(session.GetUserId(), hero.Id, _heroMap)
	if err != nil {
		log.Errorf("GetSpecified failed:%v", err)
		return
	}
	// push change
	session.SendMsg("hero", "change", &pb.HeroChangePush{List: []*pb.DBHero{hero}})
	return
}

//Event-------------------------------------------------------------------------------------------------玩家离线
func (this *Hero) EventUserOffline(session comm.IUserSession) {
	err := this.modelHero.RemoveUserHeroInfo(session)
	this.Debugf("EventUserOffline:%s err:%v", session, err)
}

// 批量创建多个英雄
func (this *Hero) CreateRepeatHeros(session comm.IUserSession, heros map[string]int32, bPush bool) (code pb.ErrorCode) {
	changeHero := make([]*pb.DBHero, 0)
	for heroCfgId, num := range heros {
		if num == 0 { // 数量为0 不做处理
			continue
		}
		hero, err := this.modelHero.createHeroOverlying(session.GetUserId(), heroCfgId, num)
		if err != nil {
			code = pb.ErrorCode_HeroCreate
			continue
		}
		changeHero = append(changeHero, hero)
	}

	// 添加图鉴
	go func(uid string, heros map[string]int32) { // 携程处理 图鉴数据
		if result, err1 := this.ModuleUser.GetUserExpand(uid); err1 == nil {
			initUpdate := map[string]interface{}{}
			sz := result.GetTujian()
			if len(sz) == 0 {
				sz = make(map[string]bool, 0)
			}
			for k := range heros {
				if _, ok := result.GetTujian()[k]; !ok {
					sz[k] = true
					initUpdate["tujian"] = sz
				}
			}
			if len(initUpdate) != 0 {
				this.ModuleUser.ChangeUserExpand(uid, initUpdate)
			}
		}
	}(session.GetUserId(), heros)
	if bPush && len(changeHero) > 0 { //推送
		session.SendMsg("hero", "change", &pb.HeroChangePush{List: changeHero})
	}

	return
}
func (this *Hero) AddHeroExp(session comm.IUserSession, heroObjID string, exp int32) (code pb.ErrorCode) {
	var (
		_hero       *pb.DBHero
		newhero     *pb.DBHero
		_changeHero []*pb.DBHero // 变化的英雄
	)
	_hero, code = this.GetHeroByObjID(session.GetUserId(), heroObjID)
	if code != pb.ErrorCode_Success {
		return
	}
	newhero, code = this.modelHero.AddCardExp(session.GetUserId(), _hero, exp)
	if code != pb.ErrorCode_Success {
		return
	}
	_changeHero = append(_changeHero, _hero) // 升级后的英雄 hero id 不变
	if newhero != nil {
		_changeHero = append(_changeHero, newhero) // 原来的英雄 只是数量变化了
	}
	_changeHero = append(_changeHero, _hero)
	session.SendMsg(string(this.GetType()), "change", &pb.HeroChangePush{List: _changeHero})
	return
}

// 英雄练功
func (this *Hero) KungFuHero(session comm.IUserSession, heroObjID string, bKongfu bool) (code pb.ErrorCode) {
	var (
		_hero       *pb.DBHero
		newhero     *pb.DBHero
		_changeHero []*pb.DBHero // 变化的英雄
	)
	_hero, code = this.GetHeroByObjID(session.GetUserId(), heroObjID)
	if code != pb.ErrorCode_Success {
		return
	}
	if bKongfu && _hero.Status != pb.HeroType_HeroTypeKongFu {
		code = pb.ErrorCode_HeroAlreadyKongFuStatus // 已经是练功状态
		return
	}
	if !bKongfu {
		_heroMap := map[string]interface{}{
			"status": pb.HeroType_HeroTypeNil,
		}
		err1 := this.modelHero.ChangeList(session.GetUserId(), heroObjID, _heroMap) // 修改英雄信息
		if err1 != nil {
			this.Errorf("update hero skill failed:%v", err1)
			code = pb.ErrorCode_DBError
			return
		}
		return
	}
	if _hero.SameCount > 1 {
		_hero.SameCount -= 1
		newHero := this.modelHero.CloneNewHero(_hero)
		_changeHero = append(_changeHero, newHero)
	}
	_heroMap := map[string]interface{}{
		"status":      pb.HeroType_HeroTypeKongFu,
		"isOverlying": false,
		"sameCount":   1,
	}
	_hero.SameCount = 1
	err1 := this.modelHero.ChangeList(session.GetUserId(), heroObjID, _heroMap) // 修改英雄信息
	if err1 != nil {
		this.Errorf("update hero skill failed:%v", err1)
		code = pb.ErrorCode_DBError
		return
	}
	_changeHero = append(_changeHero, _hero) // 升级后的英雄 hero id 不变
	if newhero != nil {
		_changeHero = append(_changeHero, newhero) // 原来的英雄 只是数量变化了
	}
	_changeHero = append(_changeHero, _hero)
	session.SendMsg(string(this.GetType()), "change", &pb.HeroChangePush{List: _changeHero})
	return
}