canglan
4193a1a153
- 密码哈希从 MD5+SHA1 升级为 bcrypt - 删除 super_admins/users 表中的 salt 字段 - 删除旧版升级文件（upgrade.php, check_upgrade, execute_upgrade, sql/upgrades/） - 删除 PASSWORD_SALT 配置项 - 清理所有'兼容 Python 版'注释 - 新项目独立，无历史包袱
102 lines
2.9 KiB
Go
102 lines
2.9 KiB
Go
// ===========================================
|
|
// 多班级版班级管理系统 - Go 后端
|
|
//
|
|
// 开发者: Canglan
|
|
// 联系方式: admin@sea-studio.top
|
|
// 版权归属: Sea Network Technology Studio
|
|
// 许可证: Apache License 2.0
|
|
//
|
|
// 版权所有 © Sea Network Technology Studio
|
|
// ===========================================
|
|
|
|
package repository
|
|
|
|
import (
|
|
"gorm.io/gorm"
|
|
"gorm.io/gorm/clause"
|
|
|
|
"hz-gitea.sea-studio.top/canglan/SharedClassManager/internal/model"
|
|
)
|
|
|
|
// SuperAdminRepo 超级管理员数据访问层
|
|
type SuperAdminRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
// NewSuperAdminRepo 创建超级管理员 Repository
|
|
func NewSuperAdminRepo(db *gorm.DB) *SuperAdminRepo {
|
|
return &SuperAdminRepo{db: db}
|
|
}
|
|
|
|
// GetByUsername 根据用户名获取超级管理员
|
|
func (r *SuperAdminRepo) GetByUsername(username string) (*model.SuperAdmin, error) {
|
|
var admin model.SuperAdmin
|
|
if err := r.db.Where("username = ? AND status = 1", username).First(&admin).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin, nil
|
|
}
|
|
|
|
// GetByID 根据ID获取超级管理员
|
|
func (r *SuperAdminRepo) GetByID(id int) (*model.SuperAdmin, error) {
|
|
var admin model.SuperAdmin
|
|
if err := r.db.Where("id = ?", id).First(&admin).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return &admin, nil
|
|
}
|
|
|
|
// Create 创建超级管理员
|
|
func (r *SuperAdminRepo) Create(admin *model.SuperAdmin) (int, error) {
|
|
if err := r.db.Create(admin).Error; err != nil {
|
|
return 0, err
|
|
}
|
|
return admin.ID, nil
|
|
}
|
|
|
|
// UpdatePassword 更新超级管理员密码并清除强制改密标记
|
|
func (r *SuperAdminRepo) UpdatePassword(id int, passwordHash string) error {
|
|
return r.db.Model(&model.SuperAdmin{}).
|
|
Where("id = ?", id).
|
|
Updates(map[string]interface{}{
|
|
"password_hash": passwordHash,
|
|
"need_change_password": 0,
|
|
}).Error
|
|
}
|
|
|
|
// CheckUsernameExists 检查用户名是否存在
|
|
func (r *SuperAdminRepo) CheckUsernameExists(username string) (bool, error) {
|
|
var count int64
|
|
if err := r.db.Model(&model.SuperAdmin{}).Where("username = ?", username).Count(&count).Error; err != nil {
|
|
return false, err
|
|
}
|
|
return count > 0, nil
|
|
}
|
|
|
|
// List 获取所有超级管理员
|
|
func (r *SuperAdminRepo) List() ([]model.SuperAdmin, error) {
|
|
var admins []model.SuperAdmin
|
|
if err := r.db.Order("id").Find(&admins).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return admins, nil
|
|
}
|
|
|
|
// UpdateStatus 更新超级管理员状态
|
|
func (r *SuperAdminRepo) UpdateStatus(id int, status int8) error {
|
|
return r.db.Model(&model.SuperAdmin{}).
|
|
Where("id = ?", id).
|
|
Update("status", status).Error
|
|
}
|
|
|
|
// EnsureDefaultAdmin 确保默认超级管理员存在（使用 INSERT IGNORE 避免并发竞态）
|
|
func (r *SuperAdminRepo) EnsureDefaultAdmin(username, passwordHash, realName string) error {
|
|
admin := model.SuperAdmin{
|
|
Username: username,
|
|
PasswordHash: passwordHash,
|
|
RealName: realName,
|
|
Status: 1,
|
|
}
|
|
return r.db.Clauses(clause.OnConflict{DoNothing: true}).Create(&admin).Error
|
|
}
|