package store import ( "github.com/Xhofe/go-cache" "github.com/alist-org/alist/v3/internal/model" "github.com/alist-org/alist/v3/pkg/singleflight" "github.com/pkg/errors" ) var userCache = cache.NewMemCache(cache.WithShards[*model.User](2)) var userG singleflight.Group[*model.User] func ExistAdmin() bool { return db.Take(&model.User{Role: model.ADMIN}).Error != nil } func ExistGuest() bool { return db.Take(&model.User{Role: model.GUEST}).Error != nil } func GetUserByName(name string) (*model.User, error) { user, ok := userCache.Get(name) if ok { return user, nil } user, err, _ := userG.Do(name, func() (*model.User, error) { user := model.User{Name: name} if err := db.Where(user).First(&user).Error; err != nil { return nil, errors.Wrapf(err, "failed select user") } userCache.Set(name, &user) return &user, nil }) return user, err } func GetUserById(id uint) (*model.User, error) { var u model.User if err := db.First(&u, id).Error; err != nil { return nil, errors.Wrapf(err, "failed get old user") } return &u, nil } func CreateUser(u *model.User) error { return errors.WithStack(db.Create(u).Error) } func UpdateUser(u *model.User) error { old, err := GetUserById(u.ID) if err != nil { return err } userCache.Del(old.Name) return errors.WithStack(db.Save(u).Error) } func GetUsers(pageIndex, pageSize int) ([]model.User, int64, error) { userDB := db.Model(&model.User{}) var count int64 if err := userDB.Count(&count).Error; err != nil { return nil, 0, errors.Wrapf(err, "failed get users count") } var users []model.User if err := userDB.Offset((pageIndex - 1) * pageSize).Limit(pageSize).Find(&users).Error; err != nil { return nil, 0, errors.Wrapf(err, "failed get find users") } return users, count, nil } func DeleteUserById(id uint) error { old, err := GetUserById(id) if err != nil { return err } userCache.Del(old.Name) return errors.WithStack(db.Delete(&model.User{}, id).Error) }