feat():learning后台管理项目初始化

internal/storage/role_storage.go

@@ -0,0 +1,93 @@
+package storage
+
+import (
+	"goalfymax-admin/internal/models"
+	"gorm.io/gorm"
+)
+
+// RoleStorage 角色存储接口
+type RoleStorage interface {
+	Create(role *models.Role) error
+	GetByID(id uint) (*models.Role, error)
+	GetByName(name string) (*models.Role, error)
+	Update(role *models.Role) error
+	Delete(id uint) error
+	List(req *models.RoleListRequest) ([]models.Role, int64, error)
+	UpdateStatus(id uint, status int) error
+}
+
+type roleStorage struct {
+	db *gorm.DB
+}
+
+// NewRoleStorage 创建角色存储实例
+func NewRoleStorage() RoleStorage {
+	return &roleStorage{db: DB}
+}
+
+// Create 创建角色
+func (s *roleStorage) Create(role *models.Role) error {
+	return s.db.Create(role).Error
+}
+
+// GetByID 根据ID获取角色
+func (s *roleStorage) GetByID(id uint) (*models.Role, error) {
+	var role models.Role
+	err := s.db.First(&role, id).Error
+	if err != nil {
+		return nil, err
+	}
+	return &role, nil
+}
+
+// GetByName 根据名称获取角色
+func (s *roleStorage) GetByName(name string) (*models.Role, error) {
+	var role models.Role
+	err := s.db.Where("name = ?", name).First(&role).Error
+	if err != nil {
+		return nil, err
+	}
+	return &role, nil
+}
+
+// Update 更新角色
+func (s *roleStorage) Update(role *models.Role) error {
+	return s.db.Save(role).Error
+}
+
+// Delete 删除角色
+func (s *roleStorage) Delete(id uint) error {
+	return s.db.Delete(&models.Role{}, id).Error
+}
+
+// List 获取角色列表
+func (s *roleStorage) List(req *models.RoleListRequest) ([]models.Role, int64, error) {
+	var roles []models.Role
+	var total int64
+
+	query := s.db.Model(&models.Role{})
+
+	// 构建查询条件
+	if req.Name != "" {
+		query = query.Where("name LIKE ?", "%"+req.Name+"%")
+	}
+	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
+	err := query.Offset(offset).Limit(req.Size).Find(&roles).Error
+
+	return roles, total, err
+}
+
+// UpdateStatus 更新角色状态
+func (s *roleStorage) UpdateStatus(id uint, status int) error {
+	return s.db.Model(&models.Role{}).Where("id = ?", id).Update("status", status).Error
+}