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) (obj *pb.DBHeroFetter, 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: make([]int32, 0), Lvprize: make([]int32, 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) (code pb.ErrorCode) { _conf := this.configure.GetLibraryHero(heroConfId) // 配置表中没有这个英雄数据 直接返回 if _conf == nil { return } var ( objFetter *pb.DBHeroFetter // 详细羁绊信息数据 ) rsp := &pb.LibraryChangePush{} _data := this.QueryOneHeroFetter(uid, heroConfId) if _data == nil { objFetter, code = this.createHeroFetter(uid, heroConfId) if code == pb.ErrorCode_Success { rsp.Fetter = append(rsp.Fetter, objFetter) } else { this.Errorf("createHeroFetter failed:%v,uid:%s,heroid:%s", code, uid, heroConfId) } } for _, fid := range _conf.Fid { // 查询是否存在这个羁绊对象 obj := this.GetLibraryListByFid(uid, fid) if obj == nil { // 没有羁绊信息 code, obj = this.CreateLibrary(uid, fid, heroConfId) if code != pb.ErrorCode_Success { this.Errorf("CreateLibrary failed: %v,uid:%s,fid:%d", code, uid, fid) } } 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) break } } } } if len(rsp.Data) != 0 || len(rsp.Fetter) != 0 { this.SendMsgToUser(string(this.GetType()), LibraryChangePush, rsp, uid) } return } func (this *Library) getLibraryByObjID(uid, oid string) *pb.DBLibrary { return this.modelLibrary.getOneLibrary(uid, oid) }