小强哥博客

小强哥,小强哥博客,技术大咖

springboot自定义http反馈状态码

image

最近在开发一些http server类型程序,通过spring boot构建一些web程序,这些web程序之间通过http进行数据访问、共享,如下图,

image

假设现在client发起一次保存数据的请求到server,server可能会返回如下类似的数据

{
	"status":1,
	"message":"xxxxxx"
}

然后client通过解析json获得status来判断当前的请求操作是否成功,开发过程中通过都是这么做的,但是这样在restful设计中不怎么好,其实这个status字段的表达完全可以通过http status来表示,类似404、500、502这种都有明确的定义并且相互理解、沟通起来也方便。

文章主要记录一下我是如何在spring boot中实现自定反馈状态码的,以及我找到的三种实现方式。

第一种,使用**@ResponseStatus**。这是一个注解,可以作用在方法和类上面,如下使用,

在方法上使用方式,

@RequestMapping(value = "/user", method = RequestMethod.GET)
	@ResponseStatus(code=HttpStatus.INTERNAL_SERVER_ERROR,reason="server error")
	public String getUser(){
		return "im zhangsan";
	}

启动web程序,通过postman访问http://127.0.0.1:8100/user,会出现下面结果,

{
    "timestamp": 1497850427325,
    "status": 500,
    "error": "Internal Server Error",
    "message": "server error",
    "path": "/user"
}

这里我一开始觉得很奇怪,为什么我的getUser方法中没有错误,结果还是出现了500错误?原因就是@ResponseStatus注解的问题,我后面猜测它会强制的将映射转化成500的状态码。这种应用场景我想不太明白在什么地方会用到。

在类中使用方式,

@ResponseStatus(code=HttpStatus.INTERNAL_SERVER_ERROR,reason="111")
public class ServerException extends Exception {

}

这种使用方式就是将自定义异常和状态码结合在一起,合理使用自定义异常机制可以最大化的提高程序的健壮性,下面看如何使用,

@RequestMapping(value = "/user", method = RequestMethod.GET)
public String getUser(@RequestParam String userName) throws ServerException{
	if(StringUtils.isEmpty(userName)){
		throw new ServerException();
	}
	return "im zhangsan";
}

这段代码的意思是当userName字段为null的时候会抛出ServerException异常,但是ServerException类被标记了@ResponseStatus注解,因此会直接报500错误,如果觉得500不适合还可以定义其它的错误代码。

这种方式看着已经很好了,可以按照逻辑自定义反馈码,程序够健壮。这种方式也有不好地方,如果反馈码太多需要定义太多的异常类,并且错误内容reason还是不能手动定义。

到这里,我基本上放弃了@ResponseStatus的使用了。

第二种,使用HttpServletResponse,HttpServletResponse是javax.servlet下的一个接口,如下使用,

@RequestMapping(value = "/user", method = RequestMethod.GET)
public void getUser(HttpServletResponse response) throws IOException{
	response.setStatus(500);
	response.getWriter().append("server error");
}

这种方式可以很好的实现同时满足自定义反馈码+消息内容,一般的实现方式也都是这样。但是这样也不是太好,

  1. 在括号内创建了一个response内置变量,这样显得不够美观,反而有些多余。
  2. 在方法中调用了源生的方法来设置反馈码和消息体,并且如果需要返回json格式数据还需要设置response.setContentType("application/json");和response.setCharacterEncoding("UTF-8");,这样做有些多余,重复的工作太多,虽然可以进行封装。
  3. 最严重的问题这个方法必须是void类型,否则就会和@ResponseBody出现冲突,其次就是不能利用@ResponseBody自动封装json的特性,在spring mvc框架中如果在方法上加上@ResponseBody是可以对返回值自动进行json封装的。

再找找其他的,如果没有找到,估计也只能接受这个不完美的东西了。

后来在翻阅spring boot文档的时候找到了ResponseEntity这么一个东西,这就是我要说的第三种方式。

第三种,使用ResponseEntity

不多说,直接上代码,

@RequestMapping(value = "/user", method = RequestMethod.GET)
public ResponseEntity<Map<String,Object>> getUser() throws IOException{
	Map<String,Object> map = new HashMap<String,Object>();
	map.put("name", "zhangsan");
	return new ResponseEntity<Map<String,Object>>(map,HttpStatus.OK);
}

通过postman查看返回结果,如下,

{
    "name": "zhangsan"
}

可以直接将map对象帮我转化成json对象,并且可以获得自定义状态码,很好,很强大。

这种方式很和我意,

  1. 不需要多于的HttpServletResponse,看着很干净。
  2. 可以充分利用@ResponseBody注解,直接将我的返回值帮我转化成json对象。
  3. 在设置返回值的时候同时还可以设置http反馈码,HttpStatus是springframework提供的一个枚举类,里面封装了所有的http反馈码,方便使用命名统一,不会有任何歧义。

相比于前面两种,这种方式很对我胃口。

仔细看了ResponseEntity的说明,发现spring mvc其它很多地方也都有使用,如下,下面内容摘自org.springframework.http.ResponseEntity文件注释,

In RestTemplate, this class is returned by getForEntity() and exchange():

ResponseEntity<String> entity = template.getForEntity("http://example.com", String.class);
 String body = entity.getBody();
 MediaType contentType = entity.getHeaders().getContentType();
 HttpStatus statusCode = entity.getStatusCode();

Can also be used in Spring MVC, as the return value from a @Controller method:

@RequestMapping("/handle")
 public ResponseEntity<String> handle() {
   URI location = ...;
   HttpHeaders responseHeaders = new HttpHeaders();
   responseHeaders.setLocation(location);
   responseHeaders.set("MyResponseHeader", "MyValue");
   return new ResponseEntity<String>("Hello World", responseHeaders, HttpStatus.CREATED);
 }

这就是上面说过的。

Or, by using a builder accessible via static methods:

@RequestMapping("/handle")
 public ResponseEntity<String> handle() {
   URI location = ...;
   return ResponseEntity.created(location).header("MyResponseHeader", "MyValue").body("Hello World");
 }

自定义http反馈码在设计优良的restful api中起到关键作用,http反馈码是业内统一、共识的,建议在尽量不要通过解析json来获得status判断操作结果。

完。