在一个前后端分离的项目中,需要对后端RestController里返回的ResponseBody进行统一的封装,让所有的API结果的都是json对象,带有是否成功的标志位,并且将实际的数据放到json的result字段中,例如:
{"result":["age","height"],"success":true}
如果在每个RestController中进行封装,重复的工作很多,因此需要自动封装,比如已经写好了如下Controller
@RestController()
@RequestMapping("bar")
public class BarController {
@GetMapping("foo")
public List<String> getSessionColumns() {
List<String> result = new ArrayList<>();
result.add("age");
result.add("height");
return result;
}
}
自动将结果封装为
{"result":["age","height"],"success":true}
一开始试验了AOP,但是发现总是有ClassCastException的问题,后来又找到了ResponseBodyAdvice,才解决这个问题
public class NovaResponseBodyAdvice implements ResponseBodyAdvice {
@Override
public boolean supports(MethodParameter returnType, Class converterType) {
return true;
}
@Override
public Object beforeBodyWrite(Object body, MethodParameter returnType, MediaType selectedContentType,
Class selectedConverterType, ServerHttpRequest request, ServerHttpResponse response) {
Map<String, Object> result = new HashMap<>();
result.put("result", body);
result.put("success", true);
return result;
}
}