44 lines
870 B
Go
44 lines
870 B
Go
package db
|
|
|
|
import (
|
|
"go_dreamfactory/pb"
|
|
|
|
"github.com/liwei1dao/lego/core"
|
|
"go.mongodb.org/mongo-driver/bson"
|
|
)
|
|
|
|
const ( //Redis
|
|
DB_UserTable core.SqlTable = "user" //会话列表
|
|
)
|
|
|
|
func (this *DB) FindUserByAccount(account string) (*pb.DB_UserData, error) {
|
|
filter := bson.D{
|
|
{"account", account},
|
|
}
|
|
sr := this.mgo.FindOne(DB_UserTable, filter)
|
|
user := &pb.DB_UserData{}
|
|
err := sr.Decode(user)
|
|
return user, err
|
|
}
|
|
|
|
func (this *DB) FindUserById(id int) (*pb.DB_UserData, error) {
|
|
filter := bson.D{
|
|
{"_id", id},
|
|
}
|
|
sr := this.mgo.FindOne(DB_UserTable, filter)
|
|
user := &pb.DB_UserData{}
|
|
err := sr.Decode(user)
|
|
return user, err
|
|
}
|
|
|
|
func (this *DB) CreateUser(user *pb.DB_UserData) error {
|
|
_, err := this.mgo.InsertOne(DB_UserTable, user)
|
|
return err
|
|
}
|
|
|
|
//更新用户数据到DB
|
|
func (this *DB) UpdateUser(data *pb.DB_UserData) (err error) {
|
|
|
|
return
|
|
}
|