44 lines
983 B
Go
44 lines
983 B
Go
package data
|
|
|
|
import (
|
|
"context"
|
|
"ky-go-kratos/app/ucenter/internal/biz"
|
|
"ky-go-kratos/app/ucenter/internal/data/ent"
|
|
|
|
api "ky-go-kratos/api/ucenter/user"
|
|
|
|
"github.com/go-kratos/kratos/v2/log"
|
|
)
|
|
|
|
type userRepo struct {
|
|
db *ent.Client
|
|
log *log.Helper
|
|
}
|
|
|
|
func NewUserRepo(db *ent.Client, logger log.Logger) biz.UserRepo {
|
|
return &userRepo{db: db, log: log.NewHelper(logger)}
|
|
}
|
|
|
|
func (r *userRepo) GetUserInfo(ctx context.Context, req *api.GetUserInfoRequest) ([]*ent.User, error) {
|
|
user, err := r.db.User.Query().All(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
return user, nil
|
|
}
|
|
|
|
func (r *userRepo) CreateUserInfo(ctx context.Context, req *api.CreateUserInfoRequest) (*api.CreateUserInfoReply, error) {
|
|
data, err := r.db.User.Create().
|
|
SetName(req.Name).
|
|
SetNickname(req.Nickname).
|
|
SetEmail(req.Email).
|
|
SetPhone(req.Phone).
|
|
SetPassword("md5").
|
|
Save(ctx)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
log.Info(data)
|
|
return &api.CreateUserInfoReply{Id: data.ID}, nil
|
|
}
|