快捷導(dǎo)航

Spring?Boot如何實(shí)現(xiàn)統(tǒng)一數(shù)據(jù)返回

更新時(shí)間：2024年07月06日 15:31:25 作者：從零開(kāi)始的-CodeNinja之路

這篇文章主要介紹了Spring?Boot如何實(shí)現(xiàn)統(tǒng)一數(shù)據(jù)返回,本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì),感興趣的朋友跟隨小編一起看看吧

統(tǒng)一數(shù)據(jù)返回

一. 概念

其實(shí)統(tǒng)一數(shù)據(jù)返回是運(yùn)用了AOP（對(duì)某一類事情的集中處理）的思維，簡(jiǎn)單概括就是在我們進(jìn)行前后端數(shù)據(jù)交互的時(shí)候，能夠讓后端的數(shù)據(jù)進(jìn)行統(tǒng)一的打包封裝，返回給前端，便于前后端的數(shù)據(jù)處理。

二.實(shí)現(xiàn)統(tǒng)一數(shù)據(jù)返回

其實(shí)統(tǒng)一數(shù)據(jù)返回非常簡(jiǎn)單，在實(shí)現(xiàn)時(shí)要加入類注解@ControllerAdvice(這是一個(gè)表示控制通知的注解，在接下來(lái)的統(tǒng)一異常處理也要運(yùn)用到)，在添加類注解后還要在實(shí)現(xiàn)ResponseBodyAdvice接口，然后重寫接口的responseAdvice接口和beforeBodyWriter方法即可實(shí)現(xiàn)。

添加類注解@ControllerAdvice，實(shí)現(xiàn)接口ResponseBodyAdvice
重寫responseAdvice方法
重寫beforeBodyWriter方法
特殊數(shù)據(jù)類型特殊處理（String）

2.1 重寫responseAdvice方法

responseAdvice方法如下：

@Override
public boolean supports(MethodParameter returnType, Class converterType) {
	return true;
}

可以看到，重寫該方法主要返回的是Boolean數(shù)據(jù)類型，返回false表示不執(zhí)行統(tǒng)一返回，返回true表示對(duì)于所有的數(shù)據(jù)類型都進(jìn)行數(shù)據(jù)統(tǒng)一返回。

supports方法:判斷是否要執(zhí)行beforeBodyWrite方法.true為執(zhí)行,false不執(zhí)行.通過(guò)該方法可以
選擇哪些類或哪些方法的response要進(jìn)行處理,其他的不進(jìn)行處理

2.2 重寫beforeBodyWriter方法

beforeBodyWriter方法如下：

beforeBodyWrite方法:對(duì)response方法進(jìn)行具體操作處理

其中的參數(shù)body最為重要，如果想要對(duì)統(tǒng)一后的數(shù)據(jù)進(jìn)行格式的轉(zhuǎn)換，直接轉(zhuǎn)換body的格式就行，因?yàn)檫@里的body代指了所有的數(shù)據(jù)格式。

三. 特殊類型-String的處理

在經(jīng)過(guò)多次不同的數(shù)據(jù)類型測(cè)試后發(fā)現(xiàn)，其它類型就可以順利返回，而String類型卻返回不正確。

對(duì)于String的數(shù)據(jù)類型如果想要返回，我們要先引入ObjectMapper，通過(guò)調(diào)用ObjeceMapper的writeValueAsString()方法來(lái)把body的數(shù)據(jù)格式給轉(zhuǎn)換成Json格式，然后再次進(jìn)行統(tǒng)一返回即可。

進(jìn)行數(shù)據(jù)轉(zhuǎn)換的代碼如下：

private static ObjectMapper mapper = new ObjectMapper();
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType,MediaType selectedContentType, Class selectedConverterType,
 ServerHttpRequestrequest, ServerHttpResponse response) {
	//如果返回結(jié)果為String類型, 使?SpringBoot內(nèi)置提供的Jackson來(lái)實(shí)現(xiàn)信息的序列化
	if (body instanceof String){
		return mapper.writeValueAsString(Result.success(body));
	}
}

四. 全部代碼

import com.example.demo.model.Result;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
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;
@Slf4j
@ControllerAdvice
public class ResponseAdvice implements ResponseBodyAdvice {
	private static ObjectMapper mapper = new ObjectMapper();
	@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) {
		//返回結(jié)果更加靈活
		if (body instanceof Result){
			return body;
		}
		//如果返回結(jié)果為String類型, 使?SpringBoot內(nèi)置提供的Jackson來(lái)實(shí)現(xiàn)信息的序列化
		if (body instanceof String){
			return mapper.writeValueAsString(Result.success(body));
		}
		return Result.success(body);
	}
}