go_dreamfactory/modules/library/module.go

package library

import (
	"go_dreamfactory/comm"
	"go_dreamfactory/lego/core"
	"go_dreamfactory/modules"
	"go_dreamfactory/pb"

	"go.mongodb.org/mongo-driver/bson/primitive"
)

type Library struct {
	modules.ModuleBase
	modelLibrary *modelLibrary
	modelFetter  *modelFetter
	api          *apiComp
	configure    *configureComp
}

func NewModule() core.IModule {
	return &Library{}
}

func (this *Library) GetType() core.M_Modules {
	return comm.ModuleLibrary
}

func (this *Library) Init(service core.IService, module core.IModule, options core.IModuleOptions) (err error) {
	err = this.ModuleBase.Init(service, module, options)

	return
}

func (this *Library) OnInstallComp() {
	this.ModuleBase.OnInstallComp()
	this.api = this.RegisterComp(new(apiComp)).(*apiComp)
	this.modelLibrary = this.RegisterComp(new(modelLibrary)).(*modelLibrary)
	this.modelFetter = this.RegisterComp(new(modelFetter)).(*modelFetter)

	this.configure = this.RegisterComp(new(configureComp)).(*configureComp)
}

// 接口信息 更新藏书馆数据
func (this *Library) ModifyLibraryData(uid string, obj string, data map[string]interface{}) (code pb.ErrorCode) {
	err := this.modelLibrary.modifyLibraryDataByObjId(uid, obj, data)
	if err != nil {
		code = pb.ErrorCode_DBError
	}
	return
}

// 获取藏书馆列表
func (this *Library) GetLibraryList(uid string) []*pb.DBLibrary {
	return this.modelLibrary.getLibraryList(uid)
}

// 通过羁绊id 来查询羁绊信息
func (this *Library) GetLibraryListByFid(uid string, fid int32) *pb.DBLibrary {
	list := this.modelLibrary.getLibraryList(uid)
	for _, v := range list {
		if v.Fid == fid {
			return v
		}
	}
	return nil
}

//通过羁绊id  创建多个羁绊信息
func (this *Library) CreateLibrary(uid string, fid int32, heroConfId string) (code pb.ErrorCode, obj *pb.DBLibrary) {
	obj = &pb.DBLibrary{
		Id:       primitive.NewObjectID().Hex(),
		Uid:      uid,
		Fid:      fid,
		Hero:     map[string]int32{},
		Prize:    map[int32]int32{},
		Fetterlv: 0,
	}

	conf := this.configure.GetLibraryFetter(fid, 1)
	if conf != nil {
		for _, v := range conf.Hid {
			obj.Hero[v] = 0 // 默认值
			if v == heroConfId {
				obj.Hero[heroConfId] = 1 // 获得的英雄好感度等级为1级
			}
		}
		if err := this.modelLibrary.createLibrary(uid, obj); err != nil {
			code = pb.ErrorCode_DBError
		}
	}

	return
}
func (this *Library) ModifyHeroFetterData(uid string, obj string, data map[string]interface{}) (code pb.ErrorCode) {
	err := this.modelFetter.modifyHeroFetterDataByObjId(uid, obj, data)
	if err != nil {
		code = pb.ErrorCode_DBError
	}
	return
}
func (this *Library) GetHeroFetterList(uid string) []*pb.DBHeroFetter {
	return this.modelFetter.getHeroFetterList(uid)
}

// 创建一条英雄羁绊信息
func (this *Library) createHeroFetter(uid string, heroConfId string) (code pb.ErrorCode) {
	this.modelFetter.getHeroFetterList(uid)
	obj := &pb.DBHeroFetter{
		Id:         primitive.NewObjectID().Hex(),
		Uid:        uid,
		Heroid:     heroConfId,
		History:    make([]int32, 0),
		Favorlv:    1,
		Stroyprize: 0,
	}
	if err := this.modelFetter.createHeroFetter(uid, obj); err != nil {
		code = pb.ErrorCode_DBError
	}
	return
}

func (this *Library) QueryHeroFetter(uid string) (data []*pb.DBHeroFetter) {
	data = this.GetHeroFetterList(uid)
	return
}

func (this *Library) QueryOneHeroFetter(uid string, cid string) *pb.DBHeroFetter {
	_data := this.GetHeroFetterList(uid)
	for _, v := range _data {
		if v.Heroid == cid {
			return v
		}
	}
	return nil
}

// 创建一条羁绊信息
func (this *Library) AddHeroFetterData(session comm.IUserSession, heroConfId string) (code pb.ErrorCode) {
	uid := session.GetUserId()
	// 推送
	rsp := &pb.LibraryChangePush{}
	_data := this.QueryOneHeroFetter(uid, heroConfId)
	this.Debugf("%v", _data)
	if _data == nil {
		this.createHeroFetter(uid, heroConfId)
		_conf := this.configure.GetLibraryHero(heroConfId)
		if _conf != nil {
			szFid := _conf.Fid
			for _, fid := range szFid {
				// 查询是否存在这个羁绊对象
				obj := this.GetLibraryListByFid(uid, fid)
				if obj == nil { // 没有羁绊信息
					this.createHeroFetter(uid, heroConfId)
					code, obj = this.CreateLibrary(uid, fid, heroConfId)

				} else { // 羁绊信息中没有这个heroid 也需要加进来
					for k, v := range obj.Hero {
						if v == 0 && k == heroConfId {
							obj.Hero[k] = 1
						}
					}
					// 重新计算最低等级
					var minLv int32
					for _, v := range obj.Hero {
						if minLv < v {
							minLv = v
						}
					}
					obj.Fetterlv = minLv
					// 同步数据
					mapData := make(map[string]interface{}, 0)
					mapData["hero"] = obj.Hero
					mapData["fetterlv"] = obj.Fetterlv
					this.modelLibrary.modifyLibraryDataByObjId(uid, obj.Id, mapData)
				}
				rsp.Data = append(rsp.Data, obj)
			}
			session.SendMsg(string(this.GetType()), LibraryChangePush, rsp)
		}
	}

	return
}

func (this *Library) getLibraryByObjID(uid, oid string) *pb.DBLibrary {
	return this.modelLibrary.getOneLibrary(uid, oid)
}