update

internal/repository/user_repository.go

@@ -0,0 +1,71 @@
+package repository
+
+import (
+	"context"
+	"errors"
+
+	"fashionapi/internal/model"
+
+	"gorm.io/gorm"
+)
+
+// UserRepository 用户数据访问接口。
+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)
+}
+
+type userRepository struct {
+	db *gorm.DB
+}
+
+// NewUserRepository 创建用户仓储。
+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
+	return wrapUser(&u, err)
+}
+
+func wrapUser(u *model.User, err error) (*model.User, error) {
+	if err != nil {
+		if errors.Is(err, gorm.ErrRecordNotFound) {
+			return nil, ErrNotFound
+		}
+		return nil, err
+	}
+	return u, nil
+}