-
Notifications
You must be signed in to change notification settings - Fork 1
/
user.go
113 lines (90 loc) · 2.72 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
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
100
101
102
103
104
105
106
107
108
109
110
111
112
113
package biz
import (
"context"
"github.com/casbin/casbin/v2"
"github.com/go-kratos/kratos/v2/log"
"github.com/lalifeier/vvgo-mall/pkg/util/convert"
)
type UserRole struct {
Id int64
UserId int64
RoleId int64
}
type UserRoles []*UserRole
type User struct {
Id int64
UserRoles UserRoles
}
type UserListReq struct {
}
type UserPageListReq struct {
PageNum int64
PageSize int64
}
type UserPageListResp struct {
TotalPages int64
CurrentPage int64
PageSize int64
Data []*User
}
type UserRepo interface {
CreateUser(ctx context.Context, user *User) (int64, error)
UpdateUser(ctx context.Context, user *User) error
DeleteUser(ctx context.Context, id int64) error
GetUser(ctx context.Context, id int64) (*User, error)
ListUser(ctx context.Context, req *UserListReq) ([]*User, error)
PageListUser(ctx context.Context, req *UserPageListReq) (*UserPageListResp, error)
}
type UserUseCase struct {
repo UserRepo
log *log.Helper
enforcer *casbin.Enforcer
}
func NewUserUseCase(repo UserRepo, enforcer *casbin.Enforcer, logger log.Logger) *UserUseCase {
return &UserUseCase{repo: repo, enforcer: enforcer, log: log.NewHelper(logger)}
}
func (uc *UserUseCase) CreateUser(ctx context.Context, user *User) (int64, error) {
id, err := uc.repo.CreateUser(ctx, user)
if err != nil {
return 0, err
}
for _, item := range user.UserRoles {
uc.enforcer.AddRoleForUser(convert.Int64ToString(item.UserId), convert.Int64ToString(item.RoleId))
}
return id, nil
}
func (uc *UserUseCase) UpdateUser(ctx context.Context, user *User) error {
err := uc.repo.UpdateUser(ctx, user)
if err != nil {
return err
}
addUserRoles := make([]*UserRole, 0)
delUserRoles := make([]*UserRole, 0)
for _, item := range addUserRoles {
uc.enforcer.AddRoleForUser(convert.Int64ToString(item.UserId), convert.Int64ToString(item.RoleId))
}
for _, item := range delUserRoles {
uc.enforcer.DeleteRoleForUser(convert.Int64ToString(item.UserId), convert.Int64ToString(item.RoleId))
}
return nil
}
func (uc *UserUseCase) DeleteUser(ctx context.Context, id int64) error {
err := uc.repo.DeleteUser(ctx, id)
if err != nil {
return err
}
uc.enforcer.DeleteUser(convert.Int64ToString(id))
return nil
}
func (uc *UserUseCase) GetUser(ctx context.Context, id int64) (*User, error) {
return uc.repo.GetUser(ctx, id)
}
func (uc *UserUseCase) ListUser(ctx context.Context, req *UserListReq) ([]*User, error) {
return uc.repo.ListUser(ctx, req)
}
func (uc *UserUseCase) PageListUser(ctx context.Context, req *UserPageListReq) (*UserPageListResp, error) {
return uc.repo.PageListUser(ctx, req)
}
func (uc *UserUseCase) compareUserRoles(ctx context.Context, oldUserRoles, newUserRoles UserRoles) (addList, delList UserRoles) {
return
}