-
Notifications
You must be signed in to change notification settings - Fork 180
/
userRolesModels.go
87 lines (77 loc) · 2.08 KB
/
userRolesModels.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
package models
import (
"github.com/hzwy23/dbobj"
"github.com/hzwy23/hauth/utils"
"github.com/hzwy23/hauth/utils/logs"
)
type UserRolesModel struct {
User_id string `json:"user_id"`
Role_id string `json:"role_id"`
Code_number string `json:"code_number"`
Role_name string `json:"role_name"`
Role_status string `json:"role_status"`
}
// 根据用户id,获取这个用户已经拥有的角色
func (UserRolesModel) GetRolesByUser(user_id string) ([]UserRolesModel, error) {
rows, err := dbobj.Query(sys_rdbms_094, user_id)
if err != nil {
logs.Error(err)
return nil, err
}
var rst []UserRolesModel
err = dbobj.Scan(rows, &rst)
return rst, err
}
// 获取这个用户id,还没有获取的角色信息
func (UserRolesModel) GetOtherRoles(user_id string) ([]UserRolesModel, error) {
rows, err := dbobj.Query(sys_rdbms_095, user_id)
if err != nil {
logs.Error(err)
return nil, err
}
var rst []UserRolesModel
err = dbobj.Scan(rows, &rst)
return rst, err
}
// 对这个域中的用户进行授权
func (UserRolesModel) Auth(data []UserRolesModel, user_id string) (string, error) {
tx, err := dbobj.Begin()
if err != nil {
logs.Error(err)
return "error_user_role_begin", err
}
for _, val := range data {
uuid := utils.JoinCode(val.User_id, val.Role_id)
_, err = tx.Exec(sys_rdbms_096, uuid, val.Role_id, val.User_id, user_id)
if err != nil {
logs.Info("用户【", val.User_id, "】已经拥有了角色【", val.Role_id, "】,无需重复授权。")
}
}
err = tx.Commit()
if err != nil {
logs.Error(err)
return "error_user_role_commit", err
}
return "success", nil
}
// 移除这个用户拥有的角色信息
func (UserRolesModel) Revoke(rst []UserRolesModel) (string, error) {
tx, err := dbobj.Begin()
if err != nil {
return "error_sql_begin", err
}
for _, val := range rst {
_, err := dbobj.Exec(sys_rdbms_097, val.User_id, val.Role_id)
if err != nil {
logs.Error(err)
tx.Rollback()
return "error_user_role_commit", err
}
}
err = tx.Commit()
if err != nil {
logs.Error(err)
return "error_user_role", err
}
return "success", nil
}