65 lines
1.5 KiB
Go
65 lines
1.5 KiB
Go
|
package test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"encoding/json"
|
||
|
|
"testing"
|
||
|
|
|
||
|
|
"github.com/gogf/gf/v2/container/gvar"
|
||
|
|
"github.com/samber/lo"
|
||
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/dao"
|
||
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/model"
|
||
|
|
"github.com/tiger1103/gfast/v3/internal/app/system/model/do"
|
||
|
|
)
|
||
|
|
|
||
|
|
const userID = 145
|
||
|
|
|
||
|
|
// TestGetUserRoles 获取用户角色
|
||
|
|
func TestGetUserRoles(t *testing.T) {
|
||
|
|
var a []model.AppMenusInfoRes
|
||
|
|
// SELECT m.menu_id, m.menu_name
|
||
|
|
// FROM app_menus m
|
||
|
|
// JOIN app_role_menus rm ON m.menu_id = rm.menu_id
|
||
|
|
// JOIN app_user_roles ur ON rm.role_id = ur.role_id
|
||
|
|
// WHERE ur.user_id = 145
|
||
|
|
|
||
|
|
ctx := context.Background()
|
||
|
|
|
||
|
|
dao.AppMenus.Ctx(ctx).As("m").
|
||
|
|
InnerJoin("app_role_menus rm", "m.menu_id = rm.menu_id").
|
||
|
|
InnerJoin("app_user_roles ur", "rm.role_id = ur.role_id").
|
||
|
|
Where("ur.user_id = ?", userID).
|
||
|
|
Fields("m.menu_id, m.menu_name").
|
||
|
|
Scan(&a)
|
||
|
|
|
||
|
|
data, _ := json.Marshal(a)
|
||
|
|
println(string(data))
|
||
|
|
}
|
||
|
|
|
||
|
|
// 遍历所有用户，给用户添加角色
|
||
|
|
func TestAddUserRole(t *testing.T) {
|
||
|
|
ctx := context.Background()
|
||
|
|
// 获取所有用户
|
||
|
|
users, err := dao.SysUser.Ctx(ctx).Fields("id").Array()
|
||
|
|
if err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
|
||
|
|
userIds := lo.FilterMap(users, func(item *gvar.Var, _ int) (int, bool) {
|
||
|
|
return item.Int(), true
|
||
|
|
})
|
||
|
|
|
||
|
|
for _, userId := range userIds {
|
||
|
|
if userId == 144 || userId == 145 {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
|
||
|
|
if _, err := dao.AppUserRoles.Ctx(ctx).Data(do.AppUserRoles{
|
||
|
|
UserId: userId,
|
||
|
|
RoleId: 2,
|
||
|
|
}).Insert(); err != nil {
|
||
|
|
t.Fatal(err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|