HouYunFei
95 lines
2.1 KiB
Go
95 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"errors"
|
|
|
|
"github.com/basketikun/infinite-canvas/model"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
// ListUsers 分页查询用户。
|
|
func ListUsers(q model.Query) ([]model.User, int64, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return nil, 0, err
|
|
}
|
|
q.Normalize()
|
|
tx := db.Model(&model.User{})
|
|
|
|
var total int64
|
|
if err := tx.Count(&total).Error; err != nil {
|
|
return nil, 0, err
|
|
}
|
|
|
|
var users []model.User
|
|
err = tx.Order("created_at desc").Offset(q.Offset()).Limit(q.PageSize).Find(&users).Error
|
|
return users, total, err
|
|
}
|
|
|
|
// CountUsers 返回用户总数。
|
|
func CountUsers() (int64, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
var total int64
|
|
return total, db.Model(&model.User{}).Count(&total).Error
|
|
}
|
|
|
|
// HasAdmin 判断系统中是否存在管理员。
|
|
func HasAdmin() (bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return false, err
|
|
}
|
|
var total int64
|
|
err = db.Model(&model.User{}).Where("role = ?", model.UserRoleAdmin).Count(&total).Error
|
|
return total > 0, err
|
|
}
|
|
|
|
// GetUserByID 根据 ID 查询用户。
|
|
func GetUserByID(id string) (model.User, bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return model.User{}, false, err
|
|
}
|
|
return findUser(db, "id = ?", id)
|
|
}
|
|
|
|
// GetUserByUsername 根据用户名查询用户。
|
|
func GetUserByUsername(username string) (model.User, bool, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return model.User{}, false, err
|
|
}
|
|
return findUser(db, "username = ?", username)
|
|
}
|
|
|
|
// SaveUser 保存用户信息。
|
|
func SaveUser(user model.User) (model.User, error) {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return user, err
|
|
}
|
|
return user, db.Save(&user).Error
|
|
}
|
|
|
|
// DeleteUser 删除指定用户。
|
|
func DeleteUser(id string) error {
|
|
db, err := DB()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return db.Delete(&model.User{}, "id = ?", id).Error
|
|
}
|
|
|
|
// findUser 查询单个用户，并将未命中转换为 ok=false。
|
|
func findUser(db *gorm.DB, query string, args ...any) (model.User, bool, error) {
|
|
user := model.User{}
|
|
err := db.Where(query, args...).First(&user).Error
|
|
if errors.Is(err, gorm.ErrRecordNotFound) {
|
|
return model.User{}, false, nil
|
|
}
|
|
return user, err == nil, err
|
|
}
|