spring security获取用户信息的实现代码

前言

我们在使用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<String, String[]> params = new HashMap<>();  /**   * Constructs a request object wrapping the given request.   *   * @param request The request to wrap   *   * @throws IllegalArgumentException if the request is null   */  public UserInfoRequest(HttpServletRequest request) {    super(request);    //将参数表，赋予给当前的Map以便于持有request中的参数    this.params.putAll(request.getParameterMap());  }  /**   * 在获取所有的参数名,必须重写此方法，否则对象中参数值映射不上   */  @Override  public Enumeration<String> getParameterNames() {    return new Vector<>(params.keySet()).elements();  }  /**   * 重写getParameter方法   *   * @param name 参数名   * @return 返回参数值   */  @Override  public String getParameter(String name) {    String[] values = params.get(name);    if (values == null || values.length == 0) {      return null;    }    return values[0];  }  @Override  public String[] getParameterValues(String name) {    String[] values = params.get(name);    if (values == null || values.length == 0) {      return null;    }    return values;  }  /**   * 增加参数   *   * @param name 参数名   * @param value 参数值   */  public void addParameter(String name, Object value) {    if (value != null) {      if (value instanceof String[]) {        params.put(name, (String[]) value);      } else if (value instanceof String) {        params.put(name, new String[]{(String) value});      } else {        params.put(name, new String[]{String.valueOf(value)});      }    }  }}

这段代码使用了乐傻驴用户的代码, 在此表示感谢; 然后使用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);  }}