update
This commit is contained in:
@ -10,13 +10,10 @@ import (
|
||||
)
|
||||
|
||||
// UserRepository 用户数据访问接口。
|
||||
//
|
||||
// 当前只保留 FindByID(供 /auth/me 取当前用户)。
|
||||
// ExistsByUsernameOrEmail / Create / FindByAccount 是注册登录专用,已随路由下线一并删除。
|
||||
type UserRepository interface {
|
||||
// ExistsByUsernameOrEmail 判断用户名或邮箱是否已被占用。
|
||||
ExistsByUsernameOrEmail(ctx context.Context, username, email string) (bool, error)
|
||||
// Create 创建用户。
|
||||
Create(ctx context.Context, u *model.User) error
|
||||
// FindByAccount 按邮箱或用户名查找未删除的用户。不存在时返回 ErrNotFound。
|
||||
FindByAccount(ctx context.Context, account string) (*model.User, error)
|
||||
// FindByID 按主键查找用户。不存在时返回 ErrNotFound。
|
||||
FindByID(ctx context.Context, id uint32) (*model.User, error)
|
||||
}
|
||||
@ -30,30 +27,6 @@ func NewUserRepository(db *gorm.DB) UserRepository {
|
||||
return &userRepository{db: db}
|
||||
}
|
||||
|
||||
func (r *userRepository) ExistsByUsernameOrEmail(ctx context.Context, username, email string) (bool, error) {
|
||||
var cnt int64
|
||||
err := r.db.WithContext(ctx).
|
||||
Model(&model.User{}).
|
||||
Where("username = ? OR email = ?", username, email).
|
||||
Count(&cnt).Error
|
||||
if err != nil {
|
||||
return false, err
|
||||
}
|
||||
return cnt > 0, nil
|
||||
}
|
||||
|
||||
func (r *userRepository) Create(ctx context.Context, u *model.User) error {
|
||||
return r.db.WithContext(ctx).Create(u).Error
|
||||
}
|
||||
|
||||
func (r *userRepository) FindByAccount(ctx context.Context, account string) (*model.User, error) {
|
||||
var u model.User
|
||||
err := r.db.WithContext(ctx).
|
||||
Where("is_deleted = 0 AND (email = ? OR username = ?)", account, account).
|
||||
First(&u).Error
|
||||
return wrapUser(&u, err)
|
||||
}
|
||||
|
||||
func (r *userRepository) FindByID(ctx context.Context, id uint32) (*model.User, error) {
|
||||
var u model.User
|
||||
err := r.db.WithContext(ctx).First(&u, id).Error
|
||||
|
||||
Reference in New Issue
Block a user