package venture 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 ModelSign struct { modules.MCompModel module *Venture } func (this *ModelSign) 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.TableVentureSign this.module = module.(*Venture) this.DB.CreateIndex(core.SqlTable(this.TableName), mongo.IndexModel{ Keys: bsonx.Doc{{Key: "uid", Value: bsonx.Int32(1)}}, }) return } // 获取用户全部的埋点数据 func (this *ModelSign) getUserSign(uid string) (results *pb.DBVentureSign, err error) { results = &pb.DBVentureSign{} if err = this.Get(uid, results); err != nil && err != mgo.MongodbNil { this.module.Errorln(err) return } if err == mgo.MongodbNil { results = &pb.DBVentureSign{ Id: primitive.NewObjectID().Hex(), Uid: uid, Sign: map[int32]int32{}, } err = this.Add(uid, results) } return }