96 lines
2.6 KiB
Go
96 lines
2.6 KiB
Go
package dreamwarorder
|
|
|
|
import (
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/lego/sys/mgo"
|
|
"go_dreamfactory/modules"
|
|
"go_dreamfactory/pb"
|
|
"go_dreamfactory/sys/configure"
|
|
cfg "go_dreamfactory/sys/configure/structs"
|
|
|
|
"go.mongodb.org/mongo-driver/bson/primitive"
|
|
"go.mongodb.org/mongo-driver/mongo"
|
|
"go.mongodb.org/mongo-driver/x/bsonx"
|
|
)
|
|
|
|
type ModelDreamWarorder struct {
|
|
modules.MCompModel
|
|
module *DreamWarorder
|
|
}
|
|
|
|
func (this *ModelDreamWarorder) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
|
|
err = this.MCompModel.Init(service, module, comp, options)
|
|
this.TableName = comm.TablekfDreamwarorder
|
|
this.module = module.(*DreamWarorder)
|
|
this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
|
|
Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
|
|
})
|
|
return
|
|
}
|
|
|
|
// 获取用户全部的埋点数据
|
|
func (this *ModelDreamWarorder) getUserDreamwarorder(uid string) (results *pb.DBDreamWarorder, err error) {
|
|
results = &pb.DBDreamWarorder{}
|
|
if err = this.Get(uid, results); err != nil && err != mgo.MongodbNil {
|
|
this.module.Errorln(err)
|
|
return
|
|
}
|
|
if err == mgo.MongodbNil {
|
|
results = &pb.DBDreamWarorder{
|
|
Id: primitive.NewObjectID().Hex(),
|
|
Uid: uid,
|
|
Daytasks: make([]int32, 0),
|
|
Weektasks: make([]int32, 0),
|
|
Completetasks: make([]int32, 0),
|
|
}
|
|
|
|
err = this.Add(uid, results)
|
|
}
|
|
return
|
|
}
|
|
|
|
//刷新日常任务
|
|
func (this *ModelDreamWarorder) refreshDayTask(info *pb.DBDreamWarorder, confs []*cfg.GamePassCheckTaskData, num int32) {
|
|
var (
|
|
tasks []*cfg.GamePassCheckTaskData = make([]*cfg.GamePassCheckTaskData, 0)
|
|
weight []int32 = make([]int32, 0)
|
|
indexs []int = make([]int, 0)
|
|
)
|
|
for _, v := range confs {
|
|
if v.Page == 1 {
|
|
tasks = append(tasks, v)
|
|
weight = append(weight, v.Pro)
|
|
}
|
|
}
|
|
|
|
indexs = comm.GetRandWs(weight, num)
|
|
|
|
for _, i := range indexs {
|
|
info.Daytasks = append(info.Daytasks, tasks[i].Id)
|
|
}
|
|
info.Daytime = configure.Now().Unix()
|
|
}
|
|
|
|
//刷新日常任务
|
|
func (this *ModelDreamWarorder) refreshWeekTask(info *pb.DBDreamWarorder, confs []*cfg.GamePassCheckTaskData, num int32) {
|
|
var (
|
|
tasks []*cfg.GamePassCheckTaskData = make([]*cfg.GamePassCheckTaskData, 0)
|
|
weight []int32 = make([]int32, 0)
|
|
indexs []int = make([]int, 0)
|
|
)
|
|
for _, v := range confs {
|
|
if v.Page == 2 {
|
|
tasks = append(tasks, v)
|
|
weight = append(weight, v.Pro)
|
|
}
|
|
}
|
|
|
|
indexs = comm.GetRandWs(weight, num)
|
|
|
|
for _, i := range indexs {
|
|
info.Weektasks = append(info.Daytasks, tasks[i].Id)
|
|
}
|
|
info.Weektime = configure.Now().Unix()
|
|
}
|