feat():learning后台管理项目初始化
This commit is contained in:
103
internal/storage/user_level_config_storage.go
Normal file
103
internal/storage/user_level_config_storage.go
Normal file
@@ -0,0 +1,103 @@
|
||||
package storage
|
||||
|
||||
import (
|
||||
"goalfymax-admin/internal/models"
|
||||
"gorm.io/gorm"
|
||||
)
|
||||
|
||||
// UserLevelConfigStorage 用户等级配置存储接口
|
||||
type UserLevelConfigStorage interface {
|
||||
Create(config *models.UserLevelConfig) error
|
||||
GetByID(id uint) (*models.UserLevelConfig, error)
|
||||
GetByLevelCode(levelCode string) (*models.UserLevelConfig, error)
|
||||
Update(config *models.UserLevelConfig) error
|
||||
Delete(id uint) error
|
||||
List(req *models.UserLevelConfigListRequest) ([]models.UserLevelConfig, int64, error)
|
||||
UpdateStatus(id uint, status int) error
|
||||
GetAll() ([]models.UserLevelConfig, error)
|
||||
}
|
||||
|
||||
type userLevelConfigStorage struct {
|
||||
db *gorm.DB
|
||||
}
|
||||
|
||||
// NewUserLevelConfigStorage 创建用户等级配置存储实例
|
||||
func NewUserLevelConfigStorage() UserLevelConfigStorage {
|
||||
return &userLevelConfigStorage{db: DB}
|
||||
}
|
||||
|
||||
// Create 创建用户等级配置
|
||||
func (s *userLevelConfigStorage) Create(config *models.UserLevelConfig) error {
|
||||
return s.db.Create(config).Error
|
||||
}
|
||||
|
||||
// GetByID 根据ID获取用户等级配置
|
||||
func (s *userLevelConfigStorage) GetByID(id uint) (*models.UserLevelConfig, error) {
|
||||
var config models.UserLevelConfig
|
||||
err := s.db.First(&config, id).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &config, nil
|
||||
}
|
||||
|
||||
// GetByLevelCode 根据等级代码获取配置
|
||||
func (s *userLevelConfigStorage) GetByLevelCode(levelCode string) (*models.UserLevelConfig, error) {
|
||||
var config models.UserLevelConfig
|
||||
err := s.db.Where("level_code = ?", levelCode).First(&config).Error
|
||||
if err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return &config, nil
|
||||
}
|
||||
|
||||
// Update 更新用户等级配置
|
||||
func (s *userLevelConfigStorage) Update(config *models.UserLevelConfig) error {
|
||||
return s.db.Save(config).Error
|
||||
}
|
||||
|
||||
// Delete 删除用户等级配置
|
||||
func (s *userLevelConfigStorage) Delete(id uint) error {
|
||||
return s.db.Delete(&models.UserLevelConfig{}, id).Error
|
||||
}
|
||||
|
||||
// List 获取用户等级配置列表
|
||||
func (s *userLevelConfigStorage) List(req *models.UserLevelConfigListRequest) ([]models.UserLevelConfig, int64, error) {
|
||||
var configs []models.UserLevelConfig
|
||||
var total int64
|
||||
|
||||
query := s.db.Model(&models.UserLevelConfig{})
|
||||
|
||||
// 构建查询条件
|
||||
if req.LevelName != "" {
|
||||
query = query.Where("level_name LIKE ?", "%"+req.LevelName+"%")
|
||||
}
|
||||
if req.Status != nil {
|
||||
query = query.Where("status = ?", *req.Status)
|
||||
}
|
||||
|
||||
// 获取总数
|
||||
if err := query.Count(&total).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
// 分页查询
|
||||
offset := (req.Page - 1) * req.Size
|
||||
if err := query.Order("sort_order ASC, id DESC").Offset(offset).Limit(req.Size).Find(&configs).Error; err != nil {
|
||||
return nil, 0, err
|
||||
}
|
||||
|
||||
return configs, total, nil
|
||||
}
|
||||
|
||||
// UpdateStatus 更新状态
|
||||
func (s *userLevelConfigStorage) UpdateStatus(id uint, status int) error {
|
||||
return s.db.Model(&models.UserLevelConfig{}).Where("id = ?", id).Update("status", status).Error
|
||||
}
|
||||
|
||||
// GetAll 获取所有用户等级配置(不分页)
|
||||
func (s *userLevelConfigStorage) GetAll() ([]models.UserLevelConfig, error) {
|
||||
var configs []models.UserLevelConfig
|
||||
err := s.db.Where("status = ?", 1).Order("sort_order ASC").Find(&configs).Error
|
||||
return configs, err
|
||||
}
|
||||
Reference in New Issue
Block a user