diff --git a/src/main/java/com/zhangmeng/online/exam/controller/UserController.java b/src/main/java/com/zhangmeng/online/exam/controller/UserController.java index 89ea97c..8f8abf6 100644 --- a/src/main/java/com/zhangmeng/online/exam/controller/UserController.java +++ b/src/main/java/com/zhangmeng/online/exam/controller/UserController.java @@ -1,5 +1,6 @@ package com.zhangmeng.online.exam.controller; +import com.zhangmeng.jwt.dto.LoginUser; import com.zhangmeng.online.exam.dao.PermissionDao; import com.zhangmeng.online.exam.dao.RoleDao; import com.zhangmeng.online.exam.dao.UserDao; @@ -13,6 +14,7 @@ import org.springframework.data.domain.Page; import org.springframework.data.domain.PageRequest; import org.springframework.data.domain.Pageable; import org.springframework.data.domain.Sort; +import org.springframework.security.core.context.SecurityContextHolder; import org.springframework.web.bind.annotation.*; import javax.persistence.EntityManager; @@ -175,4 +177,22 @@ public class UserController { childMap.setChildren(new ArrayList<>()); return childMap; } + + @GetMapping("/getUserInfo") + public Result getUserInfo() { + LoginUser principal =(LoginUser) SecurityContextHolder.getContext().getAuthentication().getPrincipal(); + User user = userDao.findById(principal.getId()).get(); + Map map = new HashMap<>(); + + Set roles = user.getRoles(); + + map.put("id", user.getId()); + map.put("username", user.getUsername()); + map.put("email", user.getEmail()); + map.put("phone", user.getPhone()); + map.put("password", user.getPassword()); + map.put("type",roles.stream().findFirst().get().getType()); + return Result.success(map); + + } } \ No newline at end of file