-
Notifications
You must be signed in to change notification settings - Fork 22
/
user.go
62 lines (53 loc) · 1.86 KB
/
user.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
package service
import (
"github.com/tmnhs/crony/admin/internal/model/request"
"github.com/tmnhs/crony/common/models"
"github.com/tmnhs/crony/common/pkg/dbclient"
"github.com/tmnhs/crony/common/pkg/utils"
"gorm.io/gorm"
)
type UserService struct {
}
var DefaultUserService = new(UserService)
func (us *UserService) Login(username, password string) (u *models.User, err error) {
u = new(models.User)
err = dbclient.GetMysqlDB().Select("id", "username", "email", "role", "created", "updated").Table(models.CronyUserTableName).Where("username = ? And password = ?", username, utils.MD5(password)).Find(u).Error
return
}
func (us *UserService) FindByUserName(username string) (u *models.User, err error) {
u = new(models.User)
err = dbclient.GetMysqlDB().Table(models.CronyUserTableName).Where("username = ? ", username).First(u).Error
if err == gorm.ErrRecordNotFound {
return nil, nil
}
return
}
func (us *UserService) ChangePassword(userId int, oldPassword, newPassword string) error {
return dbclient.GetMysqlDB().Table(models.CronyUserTableName).Where("id = ? And password =? ", userId, utils.MD5(oldPassword)).Update("password", utils.MD5(newPassword)).Error
}
func (us *UserService) Search(s *request.ReqUserSearch) ([]models.User, int64, error) {
db := dbclient.GetMysqlDB().Table(models.CronyUserTableName)
if len(s.UserName) > 0 {
db = db.Where("username like ?", s.UserName+"%")
}
if len(s.Email) > 0 {
db.Where("email = ?", s.Email)
}
if s.Role > 0 {
db.Where("role = ?", s.Role)
}
if s.ID > 0 {
db.Where("id = ?", s.ID)
}
users := make([]models.User, 2)
var total int64
err := db.Count(&total).Error
if err != nil {
return nil, 0, err
}
err = db.Select("id", "username", "email", "role", "created", "updated").Limit(s.PageSize).Offset((s.Page - 1) * s.PageSize).Find(&users).Error
if err != nil {
return nil, 0, err
}
return users, total, nil
}