47 lines
1.2 KiB
Go
47 lines
1.2 KiB
Go
package service
|
|
|
|
import (
|
|
"github.com/pkg/errors"
|
|
|
|
"github.com/zhilv666/navsite/internal/model"
|
|
)
|
|
|
|
func GetUserByID(id uint) (*model.User, error) {
|
|
var user model.User
|
|
if err := DB().First(&user, id).Error; err != nil {
|
|
return nil, errors.Wrapf(err, "failed get old user")
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func GetUserByName(username string) (*model.User, error) {
|
|
user := model.User{Username: username}
|
|
if err := DB().Where(user).First(&user).Error; err != nil {
|
|
return nil, errors.Wrapf(err, "failed get old user")
|
|
}
|
|
return &user, nil
|
|
}
|
|
|
|
func GetUsers(pageIndex, pageSize int) (users []*model.User, count int64, err error) {
|
|
userDB := DB().Model(&model.User{})
|
|
if err := userDB.Count(&count).Error; err != nil {
|
|
return nil, 0, errors.Wrapf(err, "failed get counts")
|
|
}
|
|
if err := userDB.Order(columnName("id")).Offset((pageIndex - 1) * pageSize).Limit(pageSize).Find(&users).Error; err != nil {
|
|
return nil, 0, errors.Wrapf(err, "failed get users")
|
|
}
|
|
return users, count, nil
|
|
}
|
|
|
|
func AddUser(user *model.User) error {
|
|
return errors.WithStack(DB().Create(&user).Error)
|
|
}
|
|
|
|
func UpdateUser(user *model.User) error {
|
|
return errors.WithStack(DB().Save(&user).Error)
|
|
}
|
|
|
|
func DeleteUser(id uint) error {
|
|
return errors.WithStack(DB().Delete(&model.User{}, id).Error)
|
|
}
|