go_dreamfactory/modules/realarena/modelrecord.go
2024-03-13 10:17:55 +08:00

55 lines
1.4 KiB
Go

package realarena
import (
"context"
"go_dreamfactory/comm"
"go_dreamfactory/lego/core"
"go_dreamfactory/modules"
"go_dreamfactory/pb"
"go.mongodb.org/mongo-driver/bson"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/x/bsonx"
)
// /装备 数据组件
type modelRecordComp struct {
modules.MCompModel
module *RealArena
}
// 组件初始化接口
func (this *modelRecordComp) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
this.TableName = comm.TableRealarenaRecord
this.MCompModel.Init(service, module, comp, opt)
this.module = module.(*RealArena)
//创建uid索引
_, err = this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
})
return
}
func (this *modelRecordComp) getmodel(session comm.IUserSession) (records []*pb.DBRealArenaRecord, err error) {
var (
c *mongo.Cursor
)
if c, err = this.DBModel.DB.Find(core.SqlTable(this.TableName), bson.M{"$or": bson.M{"reduid": session.GetUserId(), "buleuid": session.GetUserId()}}); err != nil {
return
} else {
for c.Next(context.TODO()) {
temp := &pb.DBRealArenaRecord{}
if err = c.Decode(temp); err != nil {
return
}
records = append(records, temp)
}
}
return
}
func (this *modelRecordComp) addmodel(record *pb.DBRealArenaRecord) (err error) {
_, err = this.DBModel.DB.InsertOne(core.SqlTable(this.TableName), record)
return
}