pig-farm-controller/internal/infra/repository/user_action_log_repository.go

package repository

import (
	"context"
	"time"

	"git.huangwc.com/pig/pig-farm-controller/internal/infra/logs"
	"git.huangwc.com/pig/pig-farm-controller/internal/infra/models"

	"gorm.io/gorm"
)

// UserActionLogListOptions 定义了查询用户操作日志时的可选参数
type UserActionLogListOptions struct {
	UserID     *uint
	Username   *string
	ActionType *string
	Status     *models.AuditStatus
	StartTime  *time.Time // 基于 time 字段
	EndTime    *time.Time // 基于 time 字段
	OrderBy    string     // 例如 "time asc"
}

// UserActionLogRepository 定义了与用户操作日志相关的数据库操作接口
type UserActionLogRepository interface {
	Create(ctx context.Context, log *models.UserActionLog) error
	List(ctx context.Context, opts UserActionLogListOptions, page, pageSize int) ([]models.UserActionLog, int64, error)
}

// gormUserActionLogRepository 是 UserActionLogRepository 的 GORM 实现
type gormUserActionLogRepository struct {
	ctx context.Context
	db  *gorm.DB
}

// NewGormUserActionLogRepository 创建一个新的 UserActionLogRepository GORM 实现实例
func NewGormUserActionLogRepository(ctx context.Context, db *gorm.DB) UserActionLogRepository {
	return &gormUserActionLogRepository{ctx: ctx, db: db}
}

// Create 创建一条新的用户操作日志记录
func (r *gormUserActionLogRepository) Create(ctx context.Context, log *models.UserActionLog) error {
	repoCtx := logs.AddFuncName(ctx, r.ctx, "Create")
	return r.db.WithContext(repoCtx).Create(log).Error
}

// List 根据选项查询用户操作日志，并返回总数
func (r *gormUserActionLogRepository) List(ctx context.Context, opts UserActionLogListOptions, page, pageSize int) ([]models.UserActionLog, int64, error) {
	repoCtx := logs.AddFuncName(ctx, r.ctx, "List")
	if page <= 0 || pageSize <= 0 {
		return nil, 0, ErrInvalidPagination
	}

	var logs []models.UserActionLog
	var total int64

	query := r.db.WithContext(repoCtx).Model(&models.UserActionLog{})

	if opts.UserID != nil {
		query = query.Where("user_id = ?", *opts.UserID)
	}
	if opts.Username != nil {
		query = query.Where("username LIKE ?", "%"+*opts.Username+"%")
	}
	if opts.ActionType != nil {
		query = query.Where("action_type = ?", *opts.ActionType)
	}
	if opts.Status != nil {
		query = query.Where("status = ?", *opts.Status)
	}
	if opts.StartTime != nil {
		query = query.Where("time >= ?", *opts.StartTime)
	}
	if opts.EndTime != nil {
		query = query.Where("time <= ?", *opts.EndTime)
	}

	// 统计总数
	if err := query.Count(&total).Error; err != nil {
		return nil, 0, err
	}

	orderBy := "time DESC"
	if opts.OrderBy != "" {
		orderBy = opts.OrderBy
	}
	query = query.Order(orderBy)

	offset := (page - 1) * pageSize
	if err := query.Limit(pageSize).Offset(offset).Find(&logs).Error; err != nil {
		return nil, 0, err
	}

	return logs, total, nil
}