spring security獲取用戶信息的實(shí)現(xiàn)代碼
前言
我們?cè)谑褂胹pring security的時(shí)候可以通過(guò)好幾種方法獲取用戶信息, 但是今天這篇文章介紹的是一個(gè)筆者覺(jué)得最優(yōu)雅的實(shí)現(xiàn); 借鑒現(xiàn)有的spring security controller自動(dòng)注入?yún)?shù)的方法, 我們來(lái)進(jìn)一步的實(shí)現(xiàn)更適合我們業(yè)務(wù)的用戶信息獲取方法;
思路
現(xiàn)在spring security會(huì)在controller自動(dòng)注入Authentication/Userdetails等參數(shù), 我們拿到這些對(duì)象之后還需要一些處理才可以拿到我們需要的信息, 例如用戶ID; 那獲取用戶ID這個(gè)步驟其實(shí)可以切片的, 我們直接在controller的參數(shù)綁定之前, 獲取到我們需要的用戶信息, 然后添加到request的param里面, 就可以實(shí)現(xiàn)獲取用戶信息, controller里面使用參數(shù)名可以直接接收參數(shù);
少啰嗦, 看代碼
首先我們這個(gè)功能的實(shí)現(xiàn)遇到額第一個(gè)障礙就是默認(rèn)的HttpServletRequest是沒(méi)有提供修改Parameter的方法的, 那么我們即使獲取到用戶信息也無(wú)法寫(xiě)入request; 解決這個(gè)問(wèn)題就需要自己實(shí)現(xiàn)一個(gè)HttpServletRequestWrapper, 再使用一個(gè)Filter替換原來(lái)的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);
//將參數(shù)表，賦予給當(dāng)前的Map以便于持有request中的參數(shù)
this.params.putAll(request.getParameterMap());
}
/**
* 在獲取所有的參數(shù)名,必須重寫(xiě)此方法，否則對(duì)象中參數(shù)值映射不上
*/
@Override
public Enumeration<String> getParameterNames() {
return new Vector<>(params.keySet()).elements();
}
/**
* 重寫(xiě)getParameter方法
*
* @param name 參數(shù)名
* @return 返回參數(shù)值
*/
@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;
}
/**
* 增加參數(shù)
*
* @param name 參數(shù)名
* @param value 參數(shù)值
*/
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)});
}
}
}
}
這段代碼使用了樂(lè)傻驢用戶的代碼, 在此表示感謝; 然后使用Filter將原有的request替換;
@Component
public class UserInfoFilter extends OncePerRequestFilter {
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain) throws ServletException, IOException {
filterChain.doFilter(new UserInfoRequest(request), response);
}
}
現(xiàn)在我們可以獲取用戶信息然后寫(xiě)入request的parameter了, 這個(gè)邏輯是在filter里實(shí)現(xiàn)還是在interceptor里實(shí)現(xiàn)就看讀者自己的想法了; 筆者系統(tǒng)里面有多種用戶, 獲取用戶信息的邏輯有所不同, 所以筆者選擇使用interceptor來(lái)實(shí)現(xiàn), 可以通過(guò)自定義注解來(lái)控制注入哪種用戶信息;
@Component
public 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) {
// 獲取用戶信息的邏輯 自由發(fā)揮
Long userId = ((Admin) ((OAuth2Authentication) request.getUserPrincipal()).getPrincipal()).getId();
// 將用戶信息寫(xiě)入request的parameter
((UserInfoRequest)request).addParameter("userId", userId);
return true;
}
EmployeeUserInfo employeeUserInfo = method.getDeclaredAnnotation(EmployeeUserInfo.class);
if (employeeUserInfo != null) {
Long userId = ((Employee) ((OAuth2Authentication) request.getUserPrincipal()).getPrincipal()).getId();
((UserInfoRequest)request).addParameter("userId", userId);
return true;
}
return true;
}
}
上面我自己寫(xiě)了兩個(gè)注解, 這兩個(gè)注解的代碼我就不貼出來(lái)了, 寫(xiě)這兩個(gè)注解完全就是為了注入不同的用戶信息; 大家可以各自發(fā)揮, 注解不是必須的, 如果大家系統(tǒng)里面只有一種用戶或者由于其他原因可以直接注入parameter; 接下來(lái)配置interceptor
@Configuration
public class WebMvcConfig implements WebMvcConfigurer {
private final UserInfoInterceptor userInfoInterceptor;
public WebMvcConfig(UserInfoInterceptor userInfoInterceptor) {
this.userInfoInterceptor = userInfoInterceptor;
}
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(userInfoInterceptor);
}
}
代碼寫(xiě)到這里功能已經(jīng)做完了, 我們可以在controller里面這樣獲取用戶信息
@EmployeeUserInfo // 自定義注解
@GetMapping
public void testObtainUserInfo(Long userId) {
System.out.println("userId = " + userId);
}
寫(xiě)EmployeeUserInfo注解注入的就是employee的用戶信息, 寫(xiě)AdminUserInfo注解注入的就是admin的用戶信息
以上就是本文的全部?jī)?nèi)容，希望對(duì)大家的學(xué)習(xí)有所幫助，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要給大家介紹了關(guān)于java如何獲取視頻文件的編解碼器的相關(guān)資料,文中通過(guò)代碼介紹的非常詳細(xì),對(duì)大家學(xué)習(xí)或者使用java具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2024-05-05
Java獲取漢字拼音的全拼和首拼實(shí)現(xiàn)代碼分享
這篇文章主要介紹了Java獲取漢字拼音的全拼和首拼實(shí)現(xiàn)代碼分享,本文直接給出實(shí)現(xiàn)代碼,需要的朋友可以參考下2015-06-06
spring?mvc?AOP切面方法未執(zhí)行的一種情況的分析和處理過(guò)程
這篇文章主要介紹了spring?mvc?AOP切面方法未執(zhí)行的一種情況的分析和處理,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-05-05
springboot bean掃描路徑的實(shí)現(xiàn)
這篇文章主要介紹了springboot bean掃描路徑的實(shí)現(xiàn)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-01-01
Mybatis中#{}和${}傳參的區(qū)別及#和$的區(qū)別小結(jié)
這篇文章主要介紹了Mybatis中#{}和${}傳參的區(qū)別及#和$的區(qū)別小結(jié) 的相關(guān)資料,非常不錯(cuò)，具有參考借鑒價(jià)值，需要的朋友可以參考下2016-07-07
IDEA創(chuàng)建SpringBoot的maven項(xiàng)目的方法步驟
這篇文章主要介紹了IDEA創(chuàng)建SpringBoot的maven項(xiàng)目的方法步驟,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，需要的朋友們下面隨著小編來(lái)一起學(xué)習(xí)學(xué)習(xí)吧2021-04-04
本教程主要介紹如何擴(kuò)展Spring的xml配置，讓Spring能夠識(shí)別我們自定義的Schema和Annotation，,需要的朋友可以參考下2016-01-01
IDEA里找不到Maven的有效解決辦法(小白超詳細(xì))
這篇文章主要給大家介紹了關(guān)于IDEA里找不到Maven的有效解決辦法,文中通過(guò)圖文將解決的辦法介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或者工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下2023-07-07
SpringBoot?Webflux創(chuàng)建TCP/UDP?server并使用handler解析數(shù)據(jù)
這篇文章主要介紹了SpringBoot?Webflux創(chuàng)建TCP/UDP?server并使用handler解析數(shù)據(jù)，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教2022-02-02
Java中將File轉(zhuǎn)化為MultipartFile的操作
這篇文章主要介紹了Java中將File轉(zhuǎn)化為MultipartFile的操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2020-10-10