本篇內(nèi)容介紹了“SpringBoot中的統(tǒng)一功能處理怎么實(shí)現(xiàn)”的有關(guān)知識(shí)，在實(shí)際案例的操作過程中，不少人都會(huì)遇到這樣的困境，接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！
創(chuàng)新互聯(lián)公司提供網(wǎng)站制作、成都網(wǎng)站制作、網(wǎng)頁設(shè)計(jì)，成都品牌網(wǎng)站建設(shè)，1元廣告等致力于企業(yè)網(wǎng)站建設(shè)與公司網(wǎng)站制作，10年的網(wǎng)站開發(fā)和建站經(jīng)驗(yàn),助力企業(yè)信息化建設(shè)，成功案例突破上千多家,是您實(shí)現(xiàn)網(wǎng)站建設(shè)的好選擇.
?戶登錄權(quán)限的發(fā)展從之前每個(gè)?法中??驗(yàn)證?戶登錄權(quán)限，到現(xiàn)在統(tǒng)?的?戶登錄驗(yàn)證處理，它是?個(gè)逐漸完善和逐漸優(yōu)化的過程。
我們先來回顧?下最初?戶登錄驗(yàn)證的實(shí)現(xiàn)?法：
重慶分公司
重慶分公司
@RestController
@RequestMapping("/user")
public class UserController {
/**
* 某?法 1
*/
@RequestMapping("/m1")
public Object method(HttpServletRequest request) {
// 有 session 就獲取，沒有不會(huì)創(chuàng)建
HttpSession session = request.getSession(false);
if (session != null && session.getAttribute("userinfo") != null) {
// 說明已經(jīng)登錄，業(yè)務(wù)處理
return true;
} else {
// 未登錄
return false;
}
}
/**
* 某?法 2
*/
@RequestMapping("/m2")
public Object method2(HttpServletRequest request) {
// 有 session 就獲取，沒有不會(huì)創(chuàng)建
HttpSession session = request.getSession(false);
if (session != null && session.getAttribute("userinfo") != null) {
// 說明已經(jīng)登錄，業(yè)務(wù)處理
return true;
} else {
// 未登錄
return false;
}
}
// 其他?法...
}從上述代碼可以看出，每個(gè)?法中都有相同的?戶登錄驗(yàn)證權(quán)限，它的缺點(diǎn)是：
每個(gè)?法中都要單獨(dú)寫?戶登錄驗(yàn)證的?法，即使封裝成公共?法，也?樣要傳參調(diào)?和在?法中進(jìn)?判斷。
添加控制器越多，調(diào)??戶登錄驗(yàn)證的?法也越多，這樣就增加了后期的修改成本和維護(hù)成本。
這些?戶登錄驗(yàn)證的?法和接下來要實(shí)現(xiàn)的業(yè)務(wù)?何沒有任何關(guān)聯(lián)，但每個(gè)?法中都要寫?遍。
所以提供?個(gè)公共的 AOP ?法來進(jìn)?統(tǒng)?的?戶登錄權(quán)限驗(yàn)證迫在眉睫。
說到統(tǒng)?的?戶登錄驗(yàn)證，我們想到的第?個(gè)實(shí)現(xiàn)?案是 Spring AOP 前置通知或環(huán)繞通知來實(shí)現(xiàn)，具體實(shí)現(xiàn)代碼如下：
import org.aspectj.lang.ProceedingJoinPoint;
import org.aspectj.lang.annotation.*;
import org.springframework.stereotype.Component;
@Aspect
@Component
public class UserAspect {
// 定義切點(diǎn)?法 controller 包下、?孫包下所有類的所有?法
@Pointcut("execution(* com.example.demo.controller..*.*(..))")
public void pointcut(){ }
// 前置?法
@Before("pointcut()")
public void doBefore(){
}
// 環(huán)繞?法
@Around("pointcut()")
public Object doAround(ProceedingJoinPoint joinPoint){
Object obj = null;
System.out.println("Around ?法開始執(zhí)?");
try {
// 執(zhí)?攔截?法
obj = joinPoint.proceed();
} catch (Throwable throwable) {
throwable.printStackTrace();
}
System.out.println("Around ?法結(jié)束執(zhí)?");
return obj;
}
}如果要在以上 Spring AOP 的切?中實(shí)現(xiàn)?戶登錄權(quán)限效驗(yàn)的功能，有以下兩個(gè)問題：
1.沒辦法獲取到 HttpSession 對(duì)象。
2.我們要對(duì)?部分?法進(jìn)?攔截，?另?部分?法不攔截，如注冊(cè)?法和登錄?法是不攔截的，這樣的話排除?法的規(guī)則很難定義，甚?沒辦法定義。
那這樣如何解決呢？
對(duì)于以上問題 Spring 中提供了具體的實(shí)現(xiàn)攔截器：HandlerInterceptor，攔截器的實(shí)現(xiàn)分為以下兩個(gè)步驟：
1.創(chuàng)建?定義攔截器，實(shí)現(xiàn) HandlerInterceptor 接?的 preHandle（執(zhí)?具體?法之前的預(yù)處理）?法。
2.將?定義攔截器加? WebMvcConfigurer 的 addInterceptors ?法中。
具體實(shí)現(xiàn)如下。
補(bǔ)充 過濾器：
過濾器是Web容器提供的。觸發(fā)的時(shí)機(jī)比攔截器更靠前，Spring 初始化前就執(zhí)行了，所以并不能處理用戶登錄權(quán)限效驗(yàn)等問題。
1.3.1 準(zhǔn)備工作
package com.example.demo.controller;
import lombok.extern.slf4j.Slf4j;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpSession;
@RestController
@RequestMapping("/user")
@Slf4j
public class UserController {
@RequestMapping("/login")
public boolean login(HttpServletRequest request,
String username, String password) {
// // 1.非空判斷
// if (username != null && username != "" &&
// password != null && username != "") {
// // 2.驗(yàn)證用戶名和密碼是否正確
// }
// 1.非空判斷
if (StringUtils.hasLength(username) && StringUtils.hasLength(password)) {
// 2.驗(yàn)證用戶名和密碼是否正確
if ("admin".equals(username) && "admin".equals(password)) {
// 登錄成功
HttpSession session = request.getSession();
session.setAttribute("userinfo", "admin");
return true;
} else {
// 用戶名或密碼輸入錯(cuò)誤
return false;
}
}
return false;
}
@RequestMapping("/getinfo")
public String getInfo() {
log.debug("執(zhí)行了 getinfo 方法");
return "執(zhí)行了 getinfo 方法";
}
@RequestMapping("/reg")
public String reg() {
log.debug("執(zhí)行了 reg 方法");
return "執(zhí)行了 reg 方法";
}
}1.3.2 自定義攔截器
接下來使?代碼來實(shí)現(xiàn)?個(gè)?戶登錄的權(quán)限效驗(yàn)，?定義攔截器是?個(gè)普通類，具體實(shí)現(xiàn)代碼如下：
package com.example.demo.config;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
/**
* 登錄攔截器
*/
@Component
@Slf4j
public class LoginInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
// 登錄判斷業(yè)務(wù)
HttpSession session = request.getSession(false);
if (session != null && session.getAttribute("userinfo") != null) {
return true;
}
log.error("當(dāng)前用戶沒有訪問權(quán)限");
response.setStatus(401);
return false;
}
}返回 boolean 類型。
相當(dāng)于一層安保：
為 false 則不能繼續(xù)往下執(zhí)行；為 true 則可以。
1.3.3 將自定義攔截器加入到系統(tǒng)配置
將上?步中的?定義攔截器加?到系統(tǒng)配置信息中，具體實(shí)現(xiàn)代碼如下：
package com.example.demo.config;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.ResourceHandlerRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
@Configuration // 一定不要忘記
public class MyConfig implements WebMvcConfigurer {
@Autowired
private LoginInterceptor loginInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(loginInterceptor)
.addPathPatterns("/**") // 攔截所有請(qǐng)求
.excludePathPatterns("/user/login") // 排除不攔截的 url
// .excludePathPatterns("/**/*.html")
// .excludePathPatterns("/**/*.js")
// .excludePathPatterns("/**/*.css")
.excludePathPatterns("/user/reg"); // 排除不攔截的 url
}
}或者：
package com.example.demo.common;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistration;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
import java.util.ArrayList;
import java.util.List;
@Configuration
public class AppConfig implements WebMvcConfigurer {
// 不攔截的 url 集合
List excludes = new ArrayList() {{
add("/**/*.html");
add("/js/**");
add("/editor.md/**");
add("/css/**");
add("/img/**"); // 放行 static/img 下的所有文件
add("/user/login"); // 放行登錄接口
add("/user/reg"); // 放行注冊(cè)接口
add("/art/detail"); // 放行文章詳情接口
add("/art/list"); // 放行文章分頁列表接口
add("/art/totalpage"); // 放行文章分頁總頁數(shù)接口
}};
@Autowired
private LoginInterceptor loginInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
// 配置攔截器
InterceptorRegistration registration =
registry.addInterceptor(loginInterceptor);
registration.addPathPatterns("/**");
registration.excludePathPatterns(excludes);
}
} 如果不注入對(duì)象的話，addInterceptor() 的參數(shù)也可以直接 new 一個(gè)對(duì)象：
@Configuration // 一定不要忘記
public class MyConfig implements WebMvcConfigurer {
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(new LoginInterceptor())
.addPathPatterns("/**") // 攔截所有請(qǐng)求
.excludePathPatterns("/user/login") // 排除不攔截的 url
// .excludePathPatterns("/**/*.html")
// .excludePathPatterns("/**/*.js")
// .excludePathPatterns("/**/*.css")
// .excludePathPatterns("/**/*.jpg")
// .excludePathPatterns("/**/login")
.excludePathPatterns("/user/reg"); // 排除不攔截的 url
}
}其中：
addPathPatterns：表示需要攔截的 URL，“**”表示攔截任意?法（也就是所有?法）。
excludePathPatterns：表示需要排除的 URL。
說明：以上攔截規(guī)則可以攔截此項(xiàng)?中的使? URL，包括靜態(tài)?件 (圖??件、JS 和 CSS 等?件)。
正常情況下的調(diào)?順序：
然?有了攔截器之后，會(huì)在調(diào)? Controller 之前進(jìn)?相應(yīng)的業(yè)務(wù)處理，執(zhí)?的流程如下圖所示：
1.4.1 實(shí)現(xiàn)原理源碼分析
所有的 Controller 執(zhí)?都會(huì)通過?個(gè)調(diào)度器 DispatcherServlet 來實(shí)現(xiàn)，這?點(diǎn)可以從 Spring Boot 控制臺(tái)的打印信息看出，如下圖所示：
?所有?法都會(huì)執(zhí)? DispatcherServlet 中的 doDispatch 調(diào)度?法，doDispatch 源碼如下：
protected void doDispatch(HttpServletRequest request, HttpServletResponse
response) throws Exception {
HttpServletRequest processedRequest = request;
HandlerExecutionChain mappedHandler = null;
boolean multipartRequestParsed = false;
WebAsyncManager asyncManager = WebAsyncUtils.getAsyncManager(request);
try {
try {
ModelAndView mv = null;
Object dispatchException = null;
try {
processedRequest = this.checkMultipart(request);
multipartRequestParsed = processedRequest != request;
mappedHandler = this.getHandler(processedRequest);
if (mappedHandler == null) {
this.noHandlerFound(processedRequest, response);
return;
}
HandlerAdapter ha = this.getHandlerAdapter(mappedHandler.g
etHandler());
String method = request.getMethod();
boolean isGet = HttpMethod.GET.matches(method);
if (isGet || HttpMethod.HEAD.matches(method)) {
long lastModified = ha.getLastModified(request, mapped
Handler.getHandler());
if ((new ServletWebRequest(request, response)).checkNo
tModified(lastModified) && isGet) {
return;
}
}
// 調(diào)?預(yù)處理【重點(diǎn)】
if (!mappedHandler.applyPreHandle(processedRequest, respon
se)) {
return;
}
// 執(zhí)? Controller 中的業(yè)務(wù)
mv = ha.handle(processedRequest, response, mappedHandler.g
etHandler());
if (asyncManager.isConcurrentHandlingStarted()) {
return;
}
this.applyDefaultViewName(processedRequest, mv);
mappedHandler.applyPostHandle(processedRequest, response,
mv);
} catch (Exception var20) {
dispatchException = var20;
} catch (Throwable var21) {
dispatchException = new NestedServletException("Handler di
spatch failed", var21);
}
this.processDispatchResult(processedRequest, response, mappedH
andler, mv, (Exception)dispatchException);
} catch (Exception var22) {
this.triggerAfterCompletion(processedRequest, response, mapped
Handler, var22);
} catch (Throwable var23) {
this.triggerAfterCompletion(processedRequest, response, mapped
Handler, new NestedServletException("Handler processing failed", var23));
}
} finally {
if (asyncManager.isConcurrentHandlingStarted()) {
if (mappedHandler != null) {
mappedHandler.applyAfterConcurrentHandlingStarted(processe
dRequest, response);
}
} else if (multipartRequestParsed) {
this.cleanupMultipart(processedRequest);
}
}
}從上述源碼可以看出在開始執(zhí)? Controller 之前，會(huì)先調(diào)? 預(yù)處理?法 applyPreHandle，?applyPreHandle ?法的實(shí)現(xiàn)源碼如下：
boolean applyPreHandle(HttpServletRequest request, HttpServletResponse response) throws Exception {
for(int i = 0; i < this.interceptorList.size(); this.interceptorIndex
= i++) {
// 獲取項(xiàng)?中使?的攔截器 HandlerInterceptor
HandlerInterceptor interceptor = (HandlerInterceptor)this.intercep
torList.get(i);
if (!interceptor.preHandle(request, response, this.handler)) {
this.triggerAfterCompletion(request, response, (Exception)null
);
return false;
}
}
return true;
}從上述源碼可以看出，在 applyPreHandle 中會(huì)獲取所有的攔截器 HandlerInterceptor 并執(zhí)?攔截器中的 preHandle ?法，這樣就會(huì)咱們前?定義的攔截器對(duì)應(yīng)上了，如下圖所示：
此時(shí)?戶登錄權(quán)限的驗(yàn)證?法就會(huì)執(zhí)?，這就是攔截器的實(shí)現(xiàn)原理。
1.4.2 攔截器小結(jié)
通過上?的源碼分析，我們可以看出，Spring 中的攔截器也是通過動(dòng)態(tài)代理和環(huán)繞通知的思想實(shí)現(xiàn)的，?體的調(diào)?流程如下：
所有請(qǐng)求地址添加 api 前綴：
@Configuration
public class AppConfig implements WebMvcConfigurer {
// 所有的接?添加 api 前綴
@Override
public void configurePathMatch(PathMatchConfigurer configurer) {
configurer.addPathPrefix("api", c -> true);
}
}其中第?個(gè)參數(shù)是?個(gè)表達(dá)式，設(shè)置為 true 表示啟動(dòng)前綴。
統(tǒng)?異常處理使?的是 @ControllerAdvice + @ExceptionHandler 來實(shí)現(xiàn)的，@ControllerAdvice 表示控制器通知類，@ExceptionHandler 是異常處理器，兩個(gè)結(jié)合表示當(dāng)出現(xiàn)異常的時(shí)候執(zhí)?某個(gè)通知，也就是執(zhí)?某個(gè)?法事件，具體實(shí)現(xiàn)代碼如下：
package com.example.demo.config;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import java.util.HashMap;
/**
* 統(tǒng)一處理異常
*/
@ControllerAdvice
public class ErrorAdive {
@ExceptionHandler(Exception.class)
@ResponseBody
public HashMap exceptionAdvie(Exception e) {
HashMap result = new HashMap<>();
result.put("code", "-1");
result.put("msg", e.getMessage());
return result;
}
@ExceptionHandler(ArithmeticException.class)
@ResponseBody
public HashMap arithmeticAdvie(ArithmeticException e) {
HashMap result = new HashMap<>();
result.put("code", "-2");
result.put("msg", e.getMessage());
return result;
}
} 方法名和返回值可以?定義，重要的是 @ControllerAdvice 和 @ExceptionHandler 注解。
以上?法表示，如果出現(xiàn)了異常就返回給前端?個(gè) HashMap 的對(duì)象，其中包含的字段如代碼中定義的那樣。
我們可以針對(duì)不同的異常，返回不同的結(jié)果，?以下代碼所示：
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
import java.util.HashMap;
@ControllerAdvice
@ResponseBody
public class ExceptionAdvice {
@ExceptionHandler(Exception.class)
public Object exceptionAdvice(Exception e) {
HashMap result = new HashMap<>();
result.put("success", -1);
result.put("message", "總的異常信息：" + e.getMessage());
result.put("data", null);
return result;
}
@ExceptionHandler(NullPointerException.class)
public Object nullPointerexceptionAdvice(NullPointerException e) {
HashMap result = new HashMap<>();
result.put("success", -1);
result.put("message", "空指針異常：" + e.getMessage());
result.put("data", null);
return result;
}
} 當(dāng)有多個(gè)異常通知時(shí)，匹配順序?yàn)楫?dāng)前類及其子類向上依次匹配，案例演示：
在 UserController 中設(shè)置?個(gè)空指針異常，實(shí)現(xiàn)代碼如下：
@RestController
@RequestMapping("/u")
public class UserController {
@RequestMapping("/index")
public String index() {
Object obj = null;
int i = obj.hashCode();
return "Hello,User Index.";
}
}以上程序的執(zhí)?結(jié)果如下：
此時(shí)若出現(xiàn)異常就不會(huì)報(bào)錯(cuò)了，代碼會(huì)繼續(xù)執(zhí)行，但是會(huì)把自定義的異常信息返回給前端！
統(tǒng)一完數(shù)據(jù)返回格式后：
package com.example.demo.common;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;
/**
* 異常類的統(tǒng)一處理
*/
@ControllerAdvice
@ResponseBody
public class ExceptionAdvice {
@ExceptionHandler(Exception.class)
public Object exceptionAdvice(Exception e) {
return AjaxResult.fail(-1, e.getMessage());
}
}統(tǒng)一異常處理不用配置路徑，是攔截整個(gè)項(xiàng)目中的所有異常。
統(tǒng)?數(shù)據(jù)返回格式的優(yōu)點(diǎn)有很多，比如以下幾個(gè)：
?便前端程序員更好的接收和解析后端數(shù)據(jù)接?返回的數(shù)據(jù)。
降低前端程序員和后端程序員的溝通成本，按照某個(gè)格式實(shí)現(xiàn)就?了，因?yàn)樗薪?都是這樣返回的。
有利于項(xiàng)?統(tǒng)?數(shù)據(jù)的維護(hù)和修改。
有利于后端技術(shù)部?的統(tǒng)?規(guī)范的標(biāo)準(zhǔn)制定，不會(huì)出現(xiàn)稀奇古怪的返回內(nèi)容。
統(tǒng)?的數(shù)據(jù)返回格式可以使用 @ControllerAdvice + ResponseBodyAdvice接口 的方式實(shí)現(xiàn)，具體實(shí)現(xiàn)代碼如下：
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyA
dvice;
import java.util.HashMap;
/**
* 統(tǒng)一返回?cái)?shù)據(jù)的處理
*/
@ControllerAdvice
public class ResponseAdvice implements ResponseBodyAdvice {
/**
* 內(nèi)容是否需要重寫（通過此?法可以選擇性部分控制器和?法進(jìn)?重寫）
* 返回 true 表示重寫
*/
@Override
public boolean supports(MethodParameter returnType, Class converterTyp
e) {
return true;
}
/**
* ?法返回之前調(diào)?此?法
*/
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType,
MediaType selectedContentType,
Class selectedConverterType, ServerHttpR
equest request,
ServerHttpResponse response) {
// 構(gòu)造統(tǒng)?返回對(duì)象
HashMap result = new HashMap<>();
result.put("state", 1);
result.put("msg", "");
result.put("data", body);
return result;
}
} 統(tǒng)一處理后，此時(shí)所有返回的都是 json 格式的數(shù)據(jù)了。
若方法的返回類型為 String，統(tǒng)一數(shù)據(jù)返回格式封裝后，返回會(huì)報(bào)錯(cuò)??？
轉(zhuǎn)換器的問題，解決方案：
實(shí)際開發(fā)中這種統(tǒng)一數(shù)據(jù)返回格式的方式并不常用。因?yàn)樗鼤?huì)將所有返回都再次進(jìn)行封裝，過于霸道了 ~
而通常我們會(huì)寫一個(gè)統(tǒng)一封裝的類，讓程序猿在返回時(shí)統(tǒng)一返回這個(gè)類 (軟性約束)，例如：
package com.example.demo.common;
import java.util.HashMap;
/**
* 自定義的統(tǒng)一返回對(duì)象
*/
public class AjaxResult {
/**
* 業(yè)務(wù)執(zhí)行成功時(shí)進(jìn)行返回的方法
*
* @param data
* @return
*/
public static HashMap success(Object data) {
HashMap result = new HashMap<>();
result.put("code", 200);
result.put("msg", "");
result.put("data", data);
return result;
}
/**
* 業(yè)務(wù)執(zhí)行成功時(shí)進(jìn)行返回的方法
*
* @param data
* @return
*/
public static HashMap success(String msg, Object data) {
HashMap result = new HashMap<>();
result.put("code", 200);
result.put("msg", msg);
result.put("data", data);
return result;
}
/**
* 業(yè)務(wù)執(zhí)行失敗返回的數(shù)據(jù)格式
*
* @param code
* @param msg
* @return
*/
public static HashMap fail(int code, String msg) {
HashMap result = new HashMap<>();
result.put("code", code);
result.put("msg", msg);
result.put("data", "");
return result;
}
/**
* 業(yè)務(wù)執(zhí)行失敗返回的數(shù)據(jù)格式
*
* @param code
* @param msg
* @return
*/
public static HashMap fail(int code, String msg, Object data) {
HashMap result = new HashMap<>();
result.put("code", code);
result.put("msg", msg);
result.put("data", data);
return result;
}
} 同時(shí)搭配統(tǒng)一數(shù)據(jù)返回格式：
package com.example.demo.common;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.SneakyThrows;
import org.springframework.core.MethodParameter;
import org.springframework.http.MediaType;
import org.springframework.http.server.ServerHttpRequest;
import org.springframework.http.server.ServerHttpResponse;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.servlet.mvc.method.annotation.ResponseBodyAdvice;
import java.util.HashMap;
/**
* 統(tǒng)一數(shù)據(jù)返回封裝
*/
@ControllerAdvice
public class ResponseAdvice implements ResponseBodyAdvice {
@Override
public boolean supports(MethodParameter returnType, Class converterType) {
return true;
}
@SneakyThrows
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType, Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
if (body instanceof HashMap) { // 本身已經(jīng)是封裝好的對(duì)象
return body;
}
if (body instanceof String) { // 返回類型是 String（特殊）
ObjectMapper objectMapper = new ObjectMapper();
return objectMapper.writeValueAsString(AjaxResult.success(body));
}
return AjaxResult.success(body);
}
}通過對(duì) @ControllerAdvice 源碼的分析我們可以知道上?統(tǒng)?異常和統(tǒng)?數(shù)據(jù)返回的執(zhí)?流程，我們先從 @ControllerAdvice 的源碼看起，點(diǎn)擊 @ControllerAdvice 實(shí)現(xiàn)源碼如下：
從上述源碼可以看出 @ControllerAdvice 派?于 @Component 組件，?所有組件初始化都會(huì)調(diào)用 InitializingBean 接?。
所以接下來我們來看 InitializingBean 有哪些實(shí)現(xiàn)類？在查詢的過程中我們發(fā)現(xiàn)了，其中 Spring MVC中的實(shí)現(xiàn)?類是 RequestMappingHandlerAdapter，它??有?個(gè)?法 afterPropertiesSet() ?法，表示所有的參數(shù)設(shè)置完成之后執(zhí)?的?法，如下圖所示：
?這個(gè)?法中有?個(gè) initControllerAdviceCache ?法，查詢此?法的源碼如下：
我們發(fā)現(xiàn)這個(gè)?法在執(zhí)?是會(huì)查找使?所有的 @ControllerAdvice 類，這些類會(huì)被容器中，但發(fā)?某個(gè)事件時(shí)，調(diào)?相應(yīng)的 Advice ?法，?如返回?cái)?shù)據(jù)前調(diào)?統(tǒng)?數(shù)據(jù)封裝，?如發(fā)?異常是調(diào)?異常的 Advice ?法實(shí)現(xiàn)。
“SpringBoot中的統(tǒng)一功能處理怎么實(shí)現(xiàn)”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識(shí)可以關(guān)注創(chuàng)新互聯(lián)網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實(shí)用文章！
