go_dreamfactory/modules/stonehenge/modelStonehenge.go
2023-07-27 17:04:36 +08:00

104 lines
2.7 KiB
Go

package stonehenge
import (
"go_dreamfactory/comm"
"go_dreamfactory/lego/core"
"go_dreamfactory/lego/sys/event"
"go_dreamfactory/lego/sys/mgo"
"go_dreamfactory/modules"
"go_dreamfactory/pb"
"go_dreamfactory/utils"
"sync"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/x/bsonx"
)
const (
StoneBossKey = "StoneBossKey"
)
// 石阵秘境
type MStonehenge struct {
modules.MCompModel
module *Stonehenge
lock sync.RWMutex
bossStage map[int32]*pb.StageData
}
//组件初始化接口
func (this *MStonehenge) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
this.TableName = comm.TableStonehenge
this.MCompModel.Init(service, module, comp, opt)
this.module = module.(*Stonehenge)
this.bossStage = make(map[int32]*pb.StageData)
//创建uid索引
this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
})
return
}
func (this *MStonehenge) Start() (err error) {
err = this.MCompModel.Start()
event.RegisterGO(core.Event_ServiceStartEnd, func() {
err = this.reLoadStoneBoos()
})
return
}
func (this *MStonehenge) GetStonehengeData(uid string) *pb.DBStonehenge {
stone := &pb.DBStonehenge{}
if err := this.Get(uid, stone); err != nil && mgo.MongodbNil == err { // 创建一条初始的数据
stone.Id = primitive.NewObjectID().Hex()
stone.Uid = uid
stone.Rooms = nil // 注意初始房间为空
stone.Userbuff = make(map[int32]int32, 0)
stone.Reward = make(map[int32]bool, 0)
stone.Addweight = make(map[int32]int32, 0)
stone.Etime = utils.WeekIntervalTime(0)
stone.Hero = make(map[string]*pb.BattleRole)
this.Add(uid, stone)
}
return stone
}
// 修改石阵信息
func (this *MStonehenge) ChangeStonehengeData(uid string, update map[string]interface{}) (err error) {
return this.Change(uid, update)
}
func (this *MStonehenge) reLoadStoneBoos() (err error) {
s := &pb.DBStoneBoss{}
this.module.ModuleTools.GetGlobalData(StoneBossKey, s)
if len(s.Bossstage) == 0 || utils.WeekIntervalTime(0) != s.Rtime {
this.lock.Lock()
this.bossStage = this.module.configure.CheckStage()
this.lock.Unlock()
this.module.ModuleTools.UpdateGlobalData(StoneBossKey, map[string]interface{}{
"BossStage": this.bossStage,
"rtime": utils.WeekIntervalTime(0),
})
} else {
this.lock.Lock()
this.bossStage = s.Bossstage
this.lock.Unlock()
}
return
}
func (this *MStonehenge) GetStoneBoosData(id int32) (bossstage *pb.StageData) {
return this.bossStage[id]
}
func (this *MStonehenge) GetAllStoneBoosData() (bossstage map[int32]*pb.StageData) {
return this.bossStage
}