go_dreamfactory/modules/martialhall/modelMartialhall.go
2022-08-26 17:53:34 +08:00

58 lines
1.5 KiB
Go

package martialhall
import (
"go_dreamfactory/comm"
"go_dreamfactory/lego/core"
"go_dreamfactory/lego/sys/mgo"
"go_dreamfactory/modules"
"go_dreamfactory/pb"
"go.mongodb.org/mongo-driver/bson/primitive"
"go.mongodb.org/mongo-driver/mongo"
"go.mongodb.org/mongo-driver/x/bsonx"
)
///论坛 数据组件
type modelMartialhall struct {
modules.MCompModel
module *Martialhall
}
//组件初始化接口
func (this *modelMartialhall) Init(service core.IService, module core.IModule, comp core.IModuleComp, opt core.IModuleOptions) (err error) {
this.TableName = comm.TableMartialhall
this.MCompModel.Init(service, module, comp, opt)
this.module = module.(*Martialhall)
//创建uid索引
this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{
Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}},
})
return
}
///询用户的练功房信息
func (this *modelMartialhall) queryUserMartialhall(uid string) (result *pb.DBMartialhall, err error) {
result = &pb.DBMartialhall{}
if err = this.Get(uid, result); err != nil && err != mgo.MongodbNil {
this.module.Errorln(err)
}
if err == mgo.MongodbNil {
result = &pb.DBMartialhall{
Id: primitive.NewObjectID().Hex(),
Uid: uid,
Lv: 1,
Pillar1: &pb.DBPillar{Index: 1, Isunlock: true},
Pillar2: &pb.DBPillar{Index: 2},
Pillar3: &pb.DBPillar{Index: 3},
Pillar4: &pb.DBPillar{Index: 4},
Pillar5: &pb.DBPillar{Index: 5},
}
if err = this.Add(uid, result); err != nil {
this.module.Errorln(err)
return
}
}
err = nil
return
}