建站
咨詢
售后
建站咨詢
新聞中心
環(huán)境：SpringBoot2.7.12
公司主營(yíng)業(yè)務(wù)：成都網(wǎng)站設(shè)計(jì)、成都做網(wǎng)站、移動(dòng)網(wǎng)站開(kāi)發(fā)等業(yè)務(wù)。幫助企業(yè)客戶真正實(shí)現(xiàn)互聯(lián)網(wǎng)宣傳，提高企業(yè)的競(jìng)爭(zhēng)能力。創(chuàng)新互聯(lián)是一支青春激揚(yáng)、勤奮敬業(yè)、活力青春激揚(yáng)、勤奮敬業(yè)、活力澎湃、和諧高效的團(tuán)隊(duì)。公司秉承以“開(kāi)放、自由、嚴(yán)謹(jǐn)、自律”為核心的企業(yè)文化，感謝他們對(duì)我們的高要求，感謝他們從不同領(lǐng)域給我們帶來(lái)的挑戰(zhàn)，讓我們激情的團(tuán)隊(duì)有機(jī)會(huì)用頭腦與智慧不斷的給客戶帶來(lái)驚喜。創(chuàng)新互聯(lián)推出都安免費(fèi)做網(wǎng)站回饋大家。
前言
在Spring MVC框架中，HandlerMapping是用于將HTTP請(qǐng)求映射到處理器的方法的組件。當(dāng)一個(gè)請(qǐng)求到達(dá)時(shí)，HandlerMapping會(huì)根據(jù)請(qǐng)求的URL和其他屬性來(lái)確定哪個(gè)處理器方法應(yīng)該處理該請(qǐng)求。在Spring MVC中，我們可以自定義HandlerMapping來(lái)滿足特定的匹配需求。其中一個(gè)方法是使用getCustomMethodCondition()方法來(lái)自定義匹配條件。
本文將詳細(xì)介紹如何使用getCustomMethodCondition()方法來(lái)自定義HandlerMapping的匹配條件。通過(guò)閱讀本文，您將了解如何擴(kuò)展HandlerMapping的默認(rèn)行為，并使用自定義條件來(lái)匹配請(qǐng)求和處理器方法。
需求：我們希望根據(jù)請(qǐng)求header中的x-token值來(lái)匹配具體的接口。所有的接口都必須使用了自定義的注解標(biāo)注。
1. 自定義請(qǐng)求匹配
在SpringMVC中可以通過(guò)自定義RequestMappingHandlerMapping#getCustomMethodCondition來(lái)實(shí)現(xiàn)此功能。
自定義請(qǐng)求匹配通過(guò)實(shí)現(xiàn)RequestCondition接口自定義規(guī)則
系統(tǒng)默認(rèn)提供了以下RequestCondition實(shí)現(xiàn)
圖片
2. 自定義匹配條件
public class CustomRequestCondition implements RequestCondition {
private static final String X_TOKEN_NAME = "x-token" ;
private Method method ;
public CustomRequestCondition(Method method) {
this.method = method ;
}
// 當(dāng)接口上有多個(gè)匹配規(guī)則時(shí)，進(jìn)行合并操作
@Override
public CustomRequestCondition combine(CustomRequestCondition other) {
return new CustomRequestCondition(other.method) ;
}
// 核心方法：根據(jù)匹配的條件進(jìn)行判斷是否匹配，如果匹配則返回當(dāng)前的對(duì)象，不匹配則返回null
@Override
public CustomRequestCondition getMatchingCondition(HttpServletRequest request) {
AKF akf = method.getAnnotation(AKF.class) ;
return akf != null ? buildToken(request, akf) : null ;
}
// 當(dāng)有多個(gè)都滿足條件的時(shí)候，進(jìn)行比較具體使用哪個(gè)
@Override
public int compareTo(CustomRequestCondition other, HttpServletRequest request) {
return 0 ;
}
// 判斷請(qǐng)求header中的信息與注解中配置的信息是否一致
private CustomRequestCondition buildToken(HttpServletRequest request, AKF akf) {
String xToken = request.getHeader(X_TOKEN_NAME) ;
if (xToken == null || xToken.length() == 0) {
return null ;
}
return xToken.equals(akf.value()) ? this : null ;
}
} 3. 配置自定義HandlerMapping
public class CustomMethodConditionRequestHandlerMapping extends RequestMappingHandlerMapping {
@Override
protected RequestCondition getCustomMethodCondition(Method method) {
return new CustomRequestCondition(method) ;
}
}配置自定義的HandlerMapping
@Component
public class CustomEndpointConfig implements WebMvcRegistrations {
public RequestMappingHandlerMapping getRequestMappingHandlerMapping() {
return new CustomMethodConditionRequestHandlerMapping() ;
}
}通過(guò)實(shí)現(xiàn)WebMvcRegistrations中的getRequestMappingHandlerMapping方法覆蓋系統(tǒng)默認(rèn)的RequestMappingHandlerMapping配置實(shí)現(xiàn)。當(dāng)然這種方式你可能失去了某些功能。這里我們可以參考默認(rèn)實(shí)現(xiàn)來(lái)完善自定義的實(shí)現(xiàn)。
4. 測(cè)試接口
@RestController
@RequestMapping("/conditions")
public class CustomMethodConditionController {
@GetMapping("/index")
public Object index() {
return "custom method condition success" ;
}
@GetMapping("/index")
@AKF
public Object x() {
return "x method invoke" ;
}
@GetMapping("/index")
@AKF("x1")
public Object x1() {
return "x1 method invoke" ;
}
@GetMapping("/index")
@AKF("x2")
public Object x2() {
return "x2 method invoke" ;
}
}上面的接口與通常的開(kāi)發(fā)配置是一致的，只是有些有接口使用了@AKF注解。這些接口中，沒(méi)有@AKF注解或者沒(méi)有設(shè)置@AKF值的，都不能訪問(wèn)，只有設(shè)置值了，且請(qǐng)求中攜帶了x-token并匹配上值了才會(huì)訪問(wèn)到接口。
當(dāng)訪問(wèn)其它沒(méi)有@AKF注解的接口，返回404。
5. 原理
根據(jù)請(qǐng)求查找HandlerMethod
public abstract class AbstractHandlerMethodMapping extends AbstractHandlerMapping {
protected HandlerMethod getHandlerInternal(HttpServletRequest request) throws Exception {
String lookupPath = initLookupPath(request);
try {
// 根據(jù)請(qǐng)求查找匹配d餓HandlerMethod
HandlerMethod handlerMethod = lookupHandlerMethod(lookupPath, request);
return (handlerMethod != null ? handlerMethod.createWithResolvedBean() : null);
}
}
protected HandlerMethod lookupHandlerMethod(String lookupPath, HttpServletRequest request) throws Exception {
List matches = new ArrayList<>();
// 根據(jù)請(qǐng)求的uri，獲取相應(yīng)的RequestMappingInfo（該對(duì)象對(duì)應(yīng)的Controller中的每一個(gè)接口）
List directPathMatches = this.mappingRegistry.getMappingsByDirectPath(lookupPath);
if (directPathMatches != null) {
// 根據(jù)請(qǐng)求找到了相應(yīng)的RequestMappingInfo，則進(jìn)行匹配執(zhí)行相應(yīng)的條件
addMatchingMappings(directPathMatches, matches, request);
}
// ...
}
private void addMatchingMappings(Collection mappings, List matches, HttpServletRequest request) {
for (T mapping : mappings) {
// 執(zhí)行相應(yīng)的條件進(jìn)行匹配，比如：你在@RequestMapping中配置了header，params等相應(yīng)的值
T match = getMatchingMapping(mapping, request);
if (match != null) {
matches.add(new Match(match, this.mappingRegistry.getRegistrations().get(mapping)));
}
}
}
}
public abstract class RequestMappingInfoHandlerMapping extends AbstractHandlerMethodMapping {
protected RequestMappingInfo getMatchingMapping(RequestMappingInfo info, HttpServletRequest request) {
return info.getMatchingCondition(request);
}
}
// RequestMappingInfo
public final class RequestMappingInfo {
// 該方法中就會(huì)根據(jù)請(qǐng)求request對(duì)象，判斷是否當(dāng)前對(duì)象符合條件
public RequestMappingInfo getMatchingCondition(HttpServletRequest request) {
RequestMethodsRequestCondition methods = this.methodsCondition.getMatchingCondition(request);
if (methods == null) {
return null;
}
ParamsRequestCondition params = this.paramsCondition.getMatchingCondition(request);
if (params == null) {
return null;
}
HeadersRequestCondition headers = this.headersCondition.getMatchingCondition(request);
if (headers == null) {
return null;
}
// ...
// 我們配置了自定義的，這里就會(huì)執(zhí)行我們自定義的條件（必須有@AKF注解）
RequestConditionHolder custom = this.customConditionHolder.getMatchingCondition(request);
if (custom == null) {
// 返回null 則表示當(dāng)前的RequestMappingInfo沒(méi)有匹配。
// 最終如果都是返回的null，則最終返回客戶端將是404
return null;
}
return new RequestMappingInfo(this.name, pathPatterns, patterns,
methods, params, headers, consumes, produces, custom, this.options);
}
} 在本文中，介紹了如何自定義RequestMappingHandlerMapping。通過(guò)自定義getCustomMethodCondition()方法，我們可以根據(jù)特定的需求擴(kuò)展HandlerMapping的行為，并使用自定義條件來(lái)匹配請(qǐng)求和處理器方法。通過(guò)這種方式，我們可以更好地控制請(qǐng)求的處理邏輯。
網(wǎng)站欄目：玩轉(zhuǎn)SpringMVC自定義請(qǐng)求匹配規(guī)則
路徑分享：http://fisionsoft.com.cn/article/ccdjppo.html
