完成user模块

This commit is contained in:
yumaojun03 2024-12-08 11:04:25 +08:00
parent 33699fad93
commit 5ee814ecd1
3 changed files with 69 additions and 4 deletions

View File

@ -3,7 +3,9 @@ package impl
import ( import (
"context" "context"
"github.com/infraboard/mcube/v2/ioc/config/datasource"
"gitlab.com/go-course-project/go17/vblog/apps/user" "gitlab.com/go-course-project/go17/vblog/apps/user"
"golang.org/x/crypto/bcrypt"
) )
var UserService user.Service = &UserServiceImpl{} var UserService user.Service = &UserServiceImpl{}
@ -14,8 +16,20 @@ type UserServiceImpl struct {
} }
// DescribeUser implements user.Service. // DescribeUser implements user.Service.
func (u *UserServiceImpl) DescribeUser(context.Context, *user.DescribeUserRequest) (*user.User, error) { func (u *UserServiceImpl) DescribeUser(ctx context.Context, in *user.DescribeUserRequest) (*user.User, error) {
panic("unimplemented") query := datasource.DBFromCtx(ctx)
switch in.DescribeBy {
case user.DESCRIBE_BY_ID:
query = query.Where("id = ?", in.Value)
case user.DESCRIBE_BY_USERNAME:
query = query.Where("username = ?", in.Value)
}
ins := &user.User{}
if err := query.Take(ins).Error; err != nil {
return nil, err
}
return ins, nil
} }
// Registry implements user.Service. // Registry implements user.Service.
@ -25,6 +39,24 @@ func (u *UserServiceImpl) Registry(ctx context.Context, in *user.RegistryRequest
return nil, err return nil, err
} }
// 明文密码保持到数据库,是不安全
// 对称加密/非对称, 解密
// 消息摘要, 无法还原
// 怎么知道用户的密码 比对hash 123 -> (xxx)
// md5 sha1/256/512, hmac, ...
// 结果固定
hashPass, err := bcrypt.GenerateFromPassword([]byte(in.Password), bcrypt.DefaultCost)
if err != nil {
return nil, err
}
ins.Password = string(hashPass)
if err := datasource.DBFromCtx(ctx).Create(ins).Error; err != nil {
return nil, err
}
// context.WithValue()
// 无事务的模式 // 无事务的模式
// datasource.DB().Transaction(func(tx *gorm.DB) error { // datasource.DB().Transaction(func(tx *gorm.DB) error {
// ctx := datasource.WithTransactionCtx(ctx) // ctx := datasource.WithTransactionCtx(ctx)

View File

@ -14,10 +14,29 @@ var (
// 我要测试的对象是什么?, 这个服务的具体实现 // 我要测试的对象是什么?, 这个服务的具体实现
// Service的具体实现现在还没实现 // Service的具体实现现在还没实现
// $2a$10$yHVSVuyIpTrQxwiuZUwSMuaJFsnd4YBd6hgA.31xNzuyTu4voD/QW
// $2a$10$fe0lsMhM15i4cjHmWudroOOIIBR27Nb7vwrigwK.9PhWdFld44Yze
// $2a$10$RoR0qK37vfc7pddPV0mpU.nN15Lv8745A40MkCJLe47Q00Ag83Qru
// https://gitee.com/infraboard/go-course/blob/master/day09/go-hash.md#bcrypt
func TestRegistry(t *testing.T) { func TestRegistry(t *testing.T) {
ins, err := impl.UserService.Registry(ctx, &user.RegistryRequest{}) req := user.NewRegistryRequest()
req.Username = "test02"
req.Password = "123456"
ins, err := impl.UserService.Registry(ctx, req)
if err != nil { if err != nil {
t.Fatal(err) t.Fatal(err)
} }
t.Log(ins) t.Log(ins)
} }
func TestDescribeUser(t *testing.T) {
ins, err := impl.UserService.DescribeUser(ctx, &user.DescribeUserRequest{
user.DESCRIBE_BY_USERNAME, "admin",
})
if err != nil {
t.Fatal(err)
}
//
// if ins.Password = in.Password
t.Log(ins.CheckPassword("1234567"))
}

View File

@ -5,7 +5,9 @@ import (
"github.com/infraboard/mcube/v2/exception" "github.com/infraboard/mcube/v2/exception"
"github.com/infraboard/mcube/v2/ioc/config/validator" "github.com/infraboard/mcube/v2/ioc/config/validator"
"github.com/infraboard/mcube/v2/tools/pretty"
"gitlab.com/go-course-project/go17/vblog/utils" "gitlab.com/go-course-project/go17/vblog/utils"
"golang.org/x/crypto/bcrypt"
) )
func New(in *RegistryRequest) (*User, error) { func New(in *RegistryRequest) (*User, error) {
@ -25,17 +27,29 @@ type User struct {
RegistryRequest RegistryRequest
} }
func (r *User) String() string {
return pretty.ToJSON(r)
}
func NewRegistryRequest() *RegistryRequest {
return &RegistryRequest{}
}
type RegistryRequest struct { type RegistryRequest struct {
// 用户名 // 用户名
Username string `json:"username" gorm:"column:username;unique;index" validate:"required"` Username string `json:"username" gorm:"column:username;unique;index" validate:"required"`
// 密码 // 密码
Password string `json:"password" gorm:"column:username;type:varchar(255)" validate:"required"` Password string `json:"password" gorm:"column:password;type:varchar(255)" validate:"required"`
// Profile 信息 // Profile 信息
Profile Profile
// 用户状态 // 用户状态
Status Status
} }
func (r *RegistryRequest) CheckPassword(password string) error {
return bcrypt.CompareHashAndPassword([]byte(r.Password), []byte(password))
}
func (r *RegistryRequest) Validate() error { func (r *RegistryRequest) Validate() error {
return validator.Validate(r) return validator.Validate(r)
} }