forked from razonyang/clevergo-examples
-
Notifications
You must be signed in to change notification settings - Fork 0
/
service.go
99 lines (85 loc) · 2.21 KB
/
service.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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
package user
import (
"context"
"database/sql"
"strconv"
"github.com/Masterminds/squirrel"
"github.com/casbin/casbin/v2"
"github.com/clevergo/demo/internal/models"
"github.com/clevergo/demo/internal/oldmodels"
"github.com/clevergo/demo/internal/utils"
"github.com/clevergo/demo/pkg/sqlex"
"github.com/volatiletech/sqlboiler/v4/boil"
)
type User struct {
oldmodels.User
}
type Service interface {
Count() (uint64, error)
Query(limit, offset uint64, qps *QueryParams) ([]oldmodels.User, error)
Create(form *CreateForm) (*models.User, error)
}
func NewService(db *sqlex.DB, enforcer *casbin.Enforcer) Service {
return &service{db: db, enforcer: enforcer}
}
type service struct {
db *sqlex.DB
enforcer *casbin.Enforcer
}
func (s *service) Count() (count uint64, err error) {
sql, args, err := squirrel.Select("count(*)").From("users").ToSql()
if err != nil {
return 0, err
}
err = s.db.Get(&count, sql, args...)
return
}
func (s *service) Query(limit, offset uint64, qps *QueryParams) (users []oldmodels.User, err error) {
query := squirrel.Select("*").From("users")
if qps.Username != "" {
query = query.Where(squirrel.Like{"username": "%" + qps.Username + "%"})
}
if qps.Email != "" {
query = query.Where(squirrel.Like{"email": "%" + qps.Email + "%"})
}
if qps.State != "" {
query = query.Where(squirrel.Eq{"state": qps.State})
}
if orderBy := qps.OrderBy(); orderBy != "" {
query = query.OrderBy(orderBy)
}
sql, args, err := query.ToSql()
if err != nil {
return nil, err
}
users = []oldmodels.User{}
err = s.db.Select(&users, sql, args...)
return
}
func (s *service) Create(form *CreateForm) (u *models.User, err error) {
ctx := context.TODO()
hashedPassword, err := utils.GeneratePassword(form.Password)
if err != nil {
return
}
err = utils.Tx(ctx, func(tx *sql.Tx) (err error) {
u = &models.User{}
u.Email = form.Email
u.Username = form.Username
u.HashedPassword = hashedPassword
u.State = form.State
err = u.Insert(ctx, tx, boil.Infer())
if err != nil {
return
}
userID := "user_" + strconv.FormatInt(u.ID, 10)
for _, role := range form.Roles {
_, err = s.enforcer.AddRoleForUser(userID, role)
if err != nil {
return
}
}
return nil
})
return
}