Blizzard
115 lines
3.2 KiB
Go
115 lines
3.2 KiB
Go
package user
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"sort"
|
|
|
|
"sundynix-micro-go/app/auth/api/internal/svc"
|
|
sysPb "sundynix-micro-go/app/system/rpc/system"
|
|
|
|
"github.com/zeromicro/go-zero/core/logx"
|
|
)
|
|
|
|
type GetUserInfoLogic struct {
|
|
logx.Logger
|
|
ctx context.Context
|
|
svcCtx *svc.ServiceContext
|
|
}
|
|
|
|
func NewGetUserInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *GetUserInfoLogic {
|
|
return &GetUserInfoLogic{Logger: logx.WithContext(ctx), ctx: ctx, svcCtx: svcCtx}
|
|
}
|
|
|
|
func (l *GetUserInfoLogic) GetUserInfo() (resp interface{}, err error) {
|
|
// 1. 从 JWT claims 中获取 userId
|
|
userId := fmt.Sprintf("%v", l.ctx.Value("userId"))
|
|
if userId == "" || userId == "<nil>" {
|
|
return nil, fmt.Errorf("用户未登录")
|
|
}
|
|
|
|
// 2. 调用 system-rpc 获取用户基础信息
|
|
userResp, err := l.svcCtx.SystemRpc.GetUserById(l.ctx, &sysPb.GetUserByIdReq{Id: userId})
|
|
if err != nil {
|
|
l.Errorf("[GetUserInfo] ❌ 获取用户信息失败 | userId=%s | err=%v", userId, err)
|
|
return nil, fmt.Errorf("获取用户信息失败")
|
|
}
|
|
|
|
// 3. 获取用户角色列表
|
|
rolesResp, err := l.svcCtx.SystemRpc.GetRolesByUserId(l.ctx, &sysPb.GetRolesByUserIdReq{UserId: userId})
|
|
if err != nil {
|
|
l.Errorf("[GetUserInfo] ⚠️ 获取用户角色失败（已降级返回空角色） | userId=%s | err=%v", userId, err)
|
|
rolesResp = &sysPb.GetRolesByUserIdResp{Roles: []*sysPb.RoleInfo{}}
|
|
}
|
|
|
|
// 4. 根据角色获取菜单权限（合并所有角色的菜单）
|
|
menuMap := make(map[string]interface{})
|
|
var allMenus []interface{}
|
|
var roleCodes []string
|
|
|
|
for _, role := range rolesResp.Roles {
|
|
roleCodes = append(roleCodes, role.Code)
|
|
|
|
menusResp, err := l.svcCtx.SystemRpc.GetMenusByRoleId(l.ctx, &sysPb.GetMenusByRoleIdReq{RoleId: role.Id})
|
|
if err != nil {
|
|
l.Errorf("获取角色[%s]菜单失败: %v", role.Name, err)
|
|
continue
|
|
}
|
|
for _, m := range menusResp.Menus {
|
|
if _, exists := menuMap[m.Id]; !exists {
|
|
menuMap[m.Id] = true
|
|
allMenus = append(allMenus, convertMenuInfo(m))
|
|
}
|
|
}
|
|
}
|
|
|
|
// 5. 对顶层菜单进行重新排序，防止不同角色的合并打乱顺序
|
|
sort.Slice(allMenus, func(i, j int) bool {
|
|
m1 := allMenus[i].(map[string]interface{})
|
|
m2 := allMenus[j].(map[string]interface{})
|
|
s1 := m1["sort"].(int32)
|
|
s2 := m2["sort"].(int32)
|
|
return s1 < s2
|
|
})
|
|
|
|
// 5. 组装完整的用户信息返回
|
|
u := userResp.User
|
|
return map[string]interface{}{
|
|
"id": u.Id,
|
|
"name": u.Name,
|
|
"account": u.Account,
|
|
"nickName": u.NickName,
|
|
"phone": u.Phone,
|
|
"avatarId": u.AvatarId,
|
|
"gender": u.Gender,
|
|
"roles": roleCodes,
|
|
"menus": allMenus,
|
|
"createdAt": u.CreatedAt,
|
|
}, nil
|
|
}
|
|
|
|
// convertMenuInfo 将proto MenuInfo转为map
|
|
func convertMenuInfo(m *sysPb.MenuInfo) map[string]interface{} {
|
|
menu := map[string]interface{}{
|
|
"id": m.Id,
|
|
"parentId": m.ParentId,
|
|
"category": m.Category,
|
|
"name": m.Name,
|
|
"title": m.Title,
|
|
"code": m.Code,
|
|
"path": m.Path,
|
|
"permission": m.Permission,
|
|
"locale": m.Locale,
|
|
"icon": m.Icon,
|
|
"sort": m.Sort,
|
|
}
|
|
if len(m.Children) > 0 {
|
|
var children []map[string]interface{}
|
|
for _, child := range m.Children {
|
|
children = append(children, convertMenuInfo(child))
|
|
}
|
|
menu["children"] = children
|
|
}
|
|
return menu
|
|
}
|