86 lines
2.2 KiB
Go
86 lines
2.2 KiB
Go
package pagoda
|
|
|
|
import (
|
|
"context"
|
|
"go_dreamfactory/comm"
|
|
"go_dreamfactory/lego/core"
|
|
"go_dreamfactory/lego/sys/redis"
|
|
"go_dreamfactory/modules"
|
|
"go_dreamfactory/pb"
|
|
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
"go.mongodb.org/mongo-driver/mongo/options"
|
|
)
|
|
|
|
type ModelRank struct {
|
|
modules.MCompModel
|
|
moduleUser *Pagoda
|
|
}
|
|
|
|
func (this *ModelRank) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
|
|
this.TableName = comm.TablePagodaRank
|
|
err = this.MCompModel.Init(service, module, comp, options)
|
|
this.moduleUser = module.(*Pagoda)
|
|
return
|
|
}
|
|
|
|
//获取用户
|
|
func (this *ModelRank) getUserSession(uid string) (cuser *pb.CacheUser) {
|
|
cuser = &pb.CacheUser{}
|
|
if err := this.Get(uid, cuser); err != nil {
|
|
this.moduleUser.Errorf("GetUserSession err:%v", err)
|
|
return
|
|
}
|
|
return
|
|
}
|
|
|
|
func (this *ModelRank) AddRank(uId string, data *pb.DBPagodaRank) (err error) {
|
|
if err = this.Add(uId, data); err != nil {
|
|
//this.Errorf("err:%v", err)
|
|
return
|
|
}
|
|
return nil
|
|
}
|
|
|
|
//获取用户通过扩展表
|
|
func (this *ModelRank) GetUserRandData(uid string) (result *pb.DBPagodaRank, err error) {
|
|
result = &pb.DBPagodaRank{}
|
|
if err = this.Get(uid, result); err != nil && redis.RedisNil != err {
|
|
return
|
|
}
|
|
err = nil
|
|
return result, err
|
|
}
|
|
|
|
func (this *ModelRank) ChangeUserRank(uid string, value map[string]interface{}) (err error) {
|
|
if len(value) == 0 {
|
|
return nil
|
|
}
|
|
return this.Change(uid, value)
|
|
}
|
|
|
|
// 临时测试
|
|
func (this *ModelRank) GetRankData(floorId int32, pagodaType int32) (data []*pb.DBPagodaRank, err error) {
|
|
if floorId == 0 {
|
|
if _data, err := this.DB.Find(comm.TablePagodaRank, bson.M{"type": pagodaType}, options.Find().SetSort(bson.M{"pagodaId": -1})); err == nil {
|
|
for _data.Next(context.TODO()) {
|
|
temp := &pb.DBPagodaRank{}
|
|
if err = _data.Decode(temp); err == nil {
|
|
data = append(data, temp)
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
if _data, err := this.DB.Find(comm.TablePagodaRank, bson.M{"pagodaId": bson.M{"$gte": floorId}, "type": pagodaType}, options.Find().SetSort(bson.M{"pagodaId": -1})); err == nil {
|
|
for _data.Next(context.TODO()) {
|
|
temp := &pb.DBPagodaRank{}
|
|
if err = _data.Decode(temp); err == nil {
|
|
data = append(data, temp)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
return
|
|
}
|