102 lines
2.7 KiB
Go
102 lines
2.7 KiB
Go
package repository
|
||
|
||
import (
|
||
"dianshang/internal/model"
|
||
|
||
"gorm.io/gorm"
|
||
)
|
||
|
||
// AdminRepository 管理员仓库
|
||
type AdminRepository struct {
|
||
db *gorm.DB
|
||
}
|
||
|
||
// NewAdminRepository 创建管理员仓库
|
||
func NewAdminRepository(db *gorm.DB) *AdminRepository {
|
||
return &AdminRepository{db: db}
|
||
}
|
||
|
||
// Create 创建管理员
|
||
func (r *AdminRepository) Create(admin *model.AdminUser) error {
|
||
return r.db.Create(admin).Error
|
||
}
|
||
|
||
// GetByID 根据ID获取管理员
|
||
func (r *AdminRepository) GetByID(id uint) (*model.AdminUser, error) {
|
||
var admin model.AdminUser
|
||
err := r.db.First(&admin, id).Error
|
||
return &admin, err
|
||
}
|
||
|
||
// GetByIDWithRole 根据ID获取管理员（包含角色信息）
|
||
func (r *AdminRepository) GetByIDWithRole(id uint) (*model.AdminUser, error) {
|
||
var admin model.AdminUser
|
||
err := r.db.Preload("Role").First(&admin, id).Error
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
// 不返回密码
|
||
admin.Password = ""
|
||
return &admin, nil
|
||
}
|
||
|
||
// GetByUsername 根据用户名获取管理员
|
||
func (r *AdminRepository) GetByUsername(username string) (*model.AdminUser, error) {
|
||
var admin model.AdminUser
|
||
err := r.db.Where("username = ?", username).First(&admin).Error
|
||
return &admin, err
|
||
}
|
||
|
||
// GetList 获取管理员列表
|
||
func (r *AdminRepository) GetList(page, pageSize int, keyword string) ([]model.AdminUser, int64, error) {
|
||
var admins []model.AdminUser
|
||
var total int64
|
||
|
||
query := r.db.Model(&model.AdminUser{}).Preload("Role")
|
||
|
||
// 关键词搜索
|
||
if keyword != "" {
|
||
query = query.Where("username LIKE ? OR nickname LIKE ? OR email LIKE ?",
|
||
"%"+keyword+"%", "%"+keyword+"%", "%"+keyword+"%")
|
||
}
|
||
|
||
// 获取总数
|
||
if err := query.Count(&total).Error; err != nil {
|
||
return nil, 0, err
|
||
}
|
||
|
||
// 分页查询
|
||
offset := (page - 1) * pageSize
|
||
err := query.Offset(offset).Limit(pageSize).Order("created_at DESC").Find(&admins).Error
|
||
|
||
// 清除密码字段
|
||
for i := range admins {
|
||
admins[i].Password = ""
|
||
}
|
||
|
||
return admins, total, err
|
||
}
|
||
|
||
// Update 更新管理员
|
||
func (r *AdminRepository) Update(id uint, updates map[string]interface{}) error {
|
||
return r.db.Model(&model.AdminUser{}).Where("id = ?", id).Updates(updates).Error
|
||
}
|
||
|
||
// Delete 删除管理员（软删除）
|
||
func (r *AdminRepository) Delete(id uint) error {
|
||
return r.db.Delete(&model.AdminUser{}, id).Error
|
||
}
|
||
|
||
// GetAdminCount 获取管理员总数
|
||
func (r *AdminRepository) GetAdminCount() (int64, error) {
|
||
var count int64
|
||
err := r.db.Model(&model.AdminUser{}).Count(&count).Error
|
||
return count, err
|
||
}
|
||
|
||
// GetActiveAdminCount 获取活跃管理员总数
|
||
func (r *AdminRepository) GetActiveAdminCount() (int64, error) {
|
||
var count int64
|
||
err := r.db.Model(&model.AdminUser{}).Where("status = ?", 1).Count(&count).Error
|
||
return count, err
|
||
} |