go_dreamfactory/modules/pagoda/model_rank.go
2024-01-11 20:51:23 +08:00

60 lines
1.4 KiB
Go

package pagoda
import (
"go_dreamfactory/comm"
"go_dreamfactory/lego/core"
"go_dreamfactory/sys/db"
//"go_dreamfactory/lego/sys/redis"
"go_dreamfactory/modules"
"go_dreamfactory/pb"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/x/bsonx"
)
var floorRankKey = "pagoda:floor"
type ModelRank struct {
modules.MCompModel
modulePagoda *Pagoda
}
func (this *ModelRank) Init(service core.IService, module core.IModule, comp core.IModuleComp, options core.IModuleOptions) (err error) {
this.TableName = comm.TablePagodaRecord
err = this.MCompModel.Init(service, module, comp, options)
this.modulePagoda = module.(*Pagoda)
//创建uid索引
this.DB.CreateIndex(core.SqlTable(comm.TablePagodaRecord), mongo.IndexModel{
Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
})
return
}
func (this *ModelRank) getPagodaRankListByFloorid(uid string, cid int32) *pb.DBPagodaRecord {
pagodaRank := make([]*pb.DBPagodaRecord, 0)
if db.IsCross() {
if err := this.GetList(uid, &pagodaRank); err != nil {
return nil
}
} else {
if conn, err := db.Cross(); err == nil {
dbModel := db.NewDBModel(db.CrossTag(), comm.TablePagodaRecord, conn)
if err = dbModel.GetList(uid, &pagodaRank); err != nil {
return nil
}
}
}
for _, v := range pagodaRank {
if conf, err := this.modulePagoda.configure.GetPagodaConfigData(cid); err == nil {
if v.Tab == conf.Tab {
return v
}
}
}
return nil
}