springsecurity中怎么获取用户信息，很多新手对此不是很清楚，为了帮助大家解决这个难题，下面小编将为大家详细讲解，有这方面需求的人可以来学习下，希望你能有所收获。
创新互联专注于企业网络营销推广、网站重做改版、沂源网站定制设计、自适应品牌网站建设、H5建站、商城系统网站开发、集团公司官网建设、成都外贸网站建设公司、高端网站制作、响应式网页设计等建站业务，价格优惠性价比高，为沂源等各大城市提供网站开发制作服务。
前言
我们在使用spring security的时候可以通过好几种方法获取用户信息, 但是今天这篇文章介绍的是一个笔者觉得最优雅的实现; 借鉴现有的spring security controller自动注入参数的方法, 我们来进一步的实现更适合我们业务的用户信息获取方法;
思路
现在spring security会在controller自动注入Authentication/Userdetails等参数, 我们拿到这些对象之后还需要一些处理才可以拿到我们需要的信息, 例如用户ID; 那获取用户ID这个步骤其实可以切片的, 我们直接在controller的参数绑定之前, 获取到我们需要的用户信息, 然后添加到request的param里面, 就可以实现获取用户信息, controller里面使用参数名可以直接接收参数;
少啰嗦, 看代码
首先我们这个功能的实现遇到额第一个障碍就是默认的HttpServletRequest是没有提供修改Parameter的方法的, 那么我们即使获取到用户信息也无法写入request; 解决这个问题就需要自己实现一个HttpServletRequestWrapper, 再使用一个Filter替换原来的request;
import javax.servlet.http.HttpServletRequest;import javax.servlet.http.HttpServletRequestWrapper;import java.util.Enumeration;import java.util.HashMap;import java.util.Map;import java.util.Vector;/** * @author sunhao * @date create in 2019-12-09 14:39:52 */public class UserInfoRequest extends HttpServletRequestWrapper { private Map
这段代码使用了乐傻驴用户的代码, 在此表示感谢; 然后使用Filter将原有的request替换;
@Componentpublic class UserInfoFilter extends OncePerRequestFilter { @Override protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException { filterChain.doFilter(new UserInfoRequest(request), response); }}
现在我们可以获取用户信息然后写入request的parameter了, 这个逻辑是在filter里实现还是在interceptor里实现就看读者自己的想法了; 笔者系统里面有多种用户, 获取用户信息的逻辑有所不同, 所以笔者选择使用interceptor来实现, 可以通过自定义注解来控制注入哪种用户信息;
@Componentpublic class UserInfoInterceptor implements HandlerInterceptor { @Override public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) { Method method = ((HandlerMethod) handler).getMethod(); AdminUserInfo adminUserInfo = method.getDeclaredAnnotation(AdminUserInfo.class); if (adminUserInfo != null) { // 获取用户信息的逻辑 自由发挥 Long userId = ((Admin) ((OAuth3Authentication) request.getUserPrincipal()).getPrincipal()).getId(); // 将用户信息写入request的parameter ((UserInfoRequest)request).addParameter("userId", userId); return true; } EmployeeUserInfo employeeUserInfo = method.getDeclaredAnnotation(EmployeeUserInfo.class); if (employeeUserInfo != null) { Long userId = ((Employee) ((OAuth3Authentication) request.getUserPrincipal()).getPrincipal()).getId(); ((UserInfoRequest)request).addParameter("userId", userId); return true; } return true; }}
上面我自己写了两个注解, 这两个注解的代码我就不贴出来了, 写这两个注解完全就是为了注入不同的用户信息; 大家可以各自发挥, 注解不是必须的, 如果大家系统里面只有一种用户或者由于其他原因可以直接注入parameter; 接下来配置interceptor
@Configurationpublic class WebMvcConfig implements WebMvcConfigurer { private final UserInfoInterceptor userInfoInterceptor; public WebMvcConfig(UserInfoInterceptor userInfoInterceptor) { this.userInfoInterceptor = userInfoInterceptor; } @Override public void addInterceptors(InterceptorRegistry registry) { registry.addInterceptor(userInfoInterceptor); }}
代码写到这里功能已经做完了, 我们可以在controller里面这样获取用户信息
@EmployeeUserInfo // 自定义注解 @GetMapping public void testObtainUserInfo(Long userId) { System.out.println("userId = " + userId); }
写EmployeeUserInfo注解注入的就是employee的用户信息, 写AdminUserInfo注解注入的就是admin的用户信息
看完上述内容是否对您有帮助呢？如果还想对相关知识有进一步的了解或阅读更多相关文章，请关注创新互联行业资讯频道，感谢您对创新互联的支持。
