backend/internal/repository/user_repository.go

package repository

import (
	"carrotskin/internal/model"
	"errors"

	"gorm.io/gorm"
)

// userRepository UserRepository的实现
type userRepository struct {
	db *gorm.DB
}

// NewUserRepository 创建UserRepository实例
func NewUserRepository(db *gorm.DB) UserRepository {
	return &userRepository{db: db}
}

func (r *userRepository) Create(user *model.User) error {
	return r.db.Create(user).Error
}

func (r *userRepository) FindByID(id int64) (*model.User, error) {
	var user model.User
	err := r.db.Where("id = ? AND status != -1", id).First(&user).Error
	return handleNotFoundResult(&user, err)
}

func (r *userRepository) FindByUsername(username string) (*model.User, error) {
	var user model.User
	err := r.db.Where("username = ? AND status != -1", username).First(&user).Error
	return handleNotFoundResult(&user, err)
}

func (r *userRepository) FindByEmail(email string) (*model.User, error) {
	var user model.User
	err := r.db.Where("email = ? AND status != -1", email).First(&user).Error
	return handleNotFoundResult(&user, err)
}

func (r *userRepository) Update(user *model.User) error {
	return r.db.Save(user).Error
}

func (r *userRepository) UpdateFields(id int64, fields map[string]interface{}) error {
	return r.db.Model(&model.User{}).Where("id = ?", id).Updates(fields).Error
}

func (r *userRepository) Delete(id int64) error {
	return r.db.Model(&model.User{}).Where("id = ?", id).Update("status", -1).Error
}

func (r *userRepository) CreateLoginLog(log *model.UserLoginLog) error {
	return r.db.Create(log).Error
}

func (r *userRepository) CreatePointLog(log *model.UserPointLog) error {
	return r.db.Create(log).Error
}

func (r *userRepository) UpdatePoints(userID int64, amount int, changeType, reason string) error {
	return r.db.Transaction(func(tx *gorm.DB) error {
		var user model.User
		if err := tx.Where("id = ?", userID).First(&user).Error; err != nil {
			return err
		}

		balanceBefore := user.Points
		balanceAfter := balanceBefore + amount

		if balanceAfter < 0 {
			return errors.New("积分不足")
		}

		if err := tx.Model(&user).Update("points", balanceAfter).Error; err != nil {
			return err
		}

		log := &model.UserPointLog{
			UserID:        userID,
			ChangeType:    changeType,
			Amount:        amount,
			BalanceBefore: balanceBefore,
			BalanceAfter:  balanceAfter,
			Reason:        reason,
		}

		return tx.Create(log).Error
	})
}

// handleNotFoundResult 处理记录未找到的情况
func handleNotFoundResult[T any](result *T, err error) (*T, error) {
	if err != nil {
		if errors.Is(err, gorm.ErrRecordNotFound) {
			return nil, nil
		}
		return nil, err
	}
	return result, nil
}