122 lines
3.2 KiB
Go
122 lines
3.2 KiB
Go
package entertainment
|
|
|
|
import (
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/lego/sys/mgo"
|
|
"go_dreamfactory/modules"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/sys/configure"
|
|
"go_dreamfactory/sys/db"
|
|
"go_dreamfactory/utils"
|
|
"time"
|
|
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/x/bsonx"
|
|
)
|
|
|
|
type modelComp struct {
|
|
modules.MCompModel
|
|
module *Entertainment
|
|
}
|
|
|
|
func (this *modelComp) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
|
|
this.TableName = comm.TableEntertainm
|
|
err = this.MCompModel.Init(service, module, comp, options)
|
|
this.module = module.(*Entertainment)
|
|
// 通过uid创建索引
|
|
this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
|
|
Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
|
|
})
|
|
this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
|
|
Keys: bsonx.Doc{{Key: "consumeexp", Value: bsonx.Int32(1)}},
|
|
})
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) queryPlayers(uIds []string) (result []*pb.DBXXLData, err error) {
|
|
result = make([]*pb.DBXXLData, 0)
|
|
if _, err = this.Gets(uIds, &result); err != nil && err != mgo.MongodbNil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func (this *modelComp) getEntertainmList(uid string) (result *pb.DBXXLData, err error) {
|
|
var (
|
|
dbModel *db.DBModel
|
|
)
|
|
if db.IsCross() {
|
|
dbModel = this.DBModel
|
|
} else {
|
|
if dbModel, err = this.module.GetCrossDBModel(this.TableName); err != nil {
|
|
return
|
|
}
|
|
}
|
|
|
|
result = &pb.DBXXLData{
|
|
Card: map[string]int32{},
|
|
Skill: map[string]int32{},
|
|
}
|
|
|
|
err = dbModel.Get(uid, result)
|
|
|
|
if mongo.ErrNoDocuments == err { // 没有数据直接创建新的数据
|
|
result.Id = primitive.NewObjectID().Hex()
|
|
result.Uid = uid
|
|
result.Card = make(map[string]int32)
|
|
endSeasonTime := utils.GetMonthEnd(configure.Now().Unix())
|
|
result.Etime = endSeasonTime
|
|
|
|
curMonth := time.Now().Month() // 计算赛季
|
|
curYear := time.Now().Year()
|
|
preMonth := this.module.service.GetOpentime().Month()
|
|
preYear := this.module.service.GetOpentime().Year()
|
|
tmp := int(curYear-preYear)*12 + int(curMonth-preMonth)
|
|
result.Rounds = int32(tmp)
|
|
// 初始化默认获得类型为1的卡片
|
|
for _, v := range this.module.configure.GetInitGameConsumeHero() {
|
|
result.Card[v] = 1
|
|
}
|
|
// 默认技能
|
|
for _, v := range this.module.configure.GetInitGameConsumeSkill() {
|
|
result.Skill[v] = 1
|
|
}
|
|
if user, e := this.module.ModuleUser.GetUser(uid); e == nil {
|
|
// 查询一次 写基本信息
|
|
result.Uinfo = &pb.BaseUserInfo{
|
|
Uid: uid,
|
|
Sid: user.Sid,
|
|
Name: user.Name,
|
|
Gender: user.Gender,
|
|
Skin: user.CurSkin,
|
|
Aframe: user.Curaframe,
|
|
Title: user.Curtitle,
|
|
Lv: user.Lv,
|
|
}
|
|
}
|
|
err = dbModel.Add(uid, result)
|
|
}
|
|
return result, err
|
|
}
|
|
|
|
//支持本服数据变更操作
|
|
func (this *modelComp) modifyEntertainmList(uid string, data map[string]interface{}) error {
|
|
var (
|
|
dbModel *db.DBModel
|
|
err error
|
|
result *pb.DBXXLData
|
|
)
|
|
if db.IsCross() {
|
|
dbModel = this.DBModel
|
|
} else {
|
|
if dbModel, err = this.module.GetCrossDBModel(this.TableName); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
err = dbModel.Get(uid, result)
|
|
return dbModel.Change(uid, data)
|
|
}
|