package mainline import ( "go_dreamfactory/comm" "go_dreamfactory/lego/core" "go_dreamfactory/modules" "go_dreamfactory/pb" "sort" "go.mongodb.org/mongo-driver/bson/primitive" ) type Mainline struct { modules.ModuleBase modelMainline *ModelMainline service core.IService api *apiComp configure *configureComp battle comm.IBattle } func NewModule() core.IModule { return &Mainline{} } func (this *Mainline) GetType() core.M_Modules { return comm.ModuleMainline } func (this *Mainline) Init(service core.IService, module core.IModule, options core.IModuleOptions) (err error) { err = this.ModuleBase.Init(service, module, options) this.service = service return } func (this *Mainline) OnInstallComp() { this.ModuleBase.OnInstallComp() this.api = this.RegisterComp(new(apiComp)).(*apiComp) this.modelMainline = this.RegisterComp(new(ModelMainline)).(*ModelMainline) this.configure = this.RegisterComp(new(configureComp)).(*configureComp) } // 接口信息 给其他模块调用 用来修改主线关卡信息 func (this *Mainline) ModifyMainlineData(uid string, id int32) (code pb.ErrorCode) { conf := this.configure.GetMainlineConfigData(id, 1) list, err := this.modelMainline.getMainlineList(uid) if err != nil { code = pb.ErrorCode_DBError return } for _, v := range list { if v.ChapterId == conf.Chapter { data := make(map[string]interface{}, 0) data["mainlineId"] = id data["chapterId"] = v.ChapterId this.modelMainline.modifyMainlineData(uid, v.Id, data) if err != nil { code = pb.ErrorCode_DBError } return } } //add _data := &pb.DBMainline{} _data.Id = primitive.NewObjectID().Hex() _data.ChapterId = conf.Chapter _data.MainlineId = conf.Id _mData := make(map[string]interface{}, 0) _data.Uid = uid _data.Intensity = 1 // 难度1 _mData[_data.Id] = _data this.modelMainline.addNewChapter(uid, _mData) return } func (this *Mainline) GetUsermainLineData(uid string) (mainlineId int32) { _szData, err := this.modelMainline.getMainlineList(uid) if err == nil { sort.SliceStable(_szData, func(i, j int) bool { // 排序 return _szData[i].ChapterId > _szData[j].ChapterId }) } return } func (this *Mainline) Start() (err error) { err = this.ModuleBase.Start() var module core.IModule if module, err = this.service.GetModule(comm.ModuleBattle); err != nil { return } this.battle = module.(comm.IBattle) return }