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, objLibrary []*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

		}
		objLibrary = append(objLibrary, obj)
	}

	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(uid string, heroConfId string) {
	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)
				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)
			}
		}
	}
	return
}

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