spring boot 中统一异常处理
什么是异常?
通俗的说就是,让你感觉不爽的,阻碍你的事都算异常,也就是说不让我们程序正常运行的情况。
为什么要统一处理异常?
方便集中管理,集中定位问题
异常实例
举个例子,还用之前的学生信息那个案例,我们添加一个小于18岁的学生,调用接口,控制台报错如下:
再看接口返回信息,如下图:
添加失败 添加成功
暂且先不说控制台报错,对比下,我们添加成功的接口信息返回情况,明显这给客户端调用我们程序的同学,有些不便,那么我们这里做下优化。
1、统一格式化输出json
强迫症的我,这里有必要做下统一格式的输出,那么具体怎么做呢?
增加一个外层对象,用于包裹里面对象,具体代码示例如下:
package com.rongrong.springboot.demo.domain; import lombok.Data; /** * @description: 最外层对象 * @author rongrong * @version 1.0 * @date 2020/1/9 21:51 */ @Data public class Result<T> { private Integer code; private String msg; private T data; }
针对成功、失败,定制统一的工具类,具体示例代码如下:
package com.rongrong.springboot.demo.utils; import com.rongrong.springboot.demo.domain.Result; /** * @description: 统一格式化输出json * @author rongrong * @version 1.0 * @date 2020/1/9 21:55 */ public class ResultUtils { public static Result success(Object obj){ Result result = new Result(); result.setCode(0); result.setMsg("success"); result.setData(obj); return result; } public static Result success(){ return success(null); } public static Result error(String msg){ Result result = new Result(); result.setCode(-1); result.setMsg(msg); //result.setMsg("unknown error"); return result; } }
接着我们需要对添加学生的接口进行改造,将我们封装好的工具类引入,达到统一输出的效果,具体代码如下:
/** * 新增一个学生 * * @return */ @PostMapping("/studentAdd") public Result<Student> sudentAdd(@Valid Student student, BindingResult bindingResult) { if(bindingResult.hasFieldErrors()){ Result result = ResultUtils.error(bindingResult.getFieldError().getDefaultMessage()); //输出错误信息 //System.out.println(bindingResult.getFieldError().getDefaultMessage()); return result; } student.setName(student.getName()); student.setAge(student.getAge()); student.setSex(student.getSex()); student.setEmail(student.getEmail()); Result result = ResultUtils.success(studentResponstory.save(student)); //保存和更新都用该方法 return result; }
我们调用接口服务,再来看接口返回,如下图:
再来看下,明显舒服好多了。
2、多个异常情况的统一
现在我们实现这样一组功能,获取学生的年龄并判断,小于10岁,返回“应该上小学”,大于10岁且小于16岁,返回“应该上初中了”
我们需要在StudentService中写逻辑,供controller调用,具体代码如下:
/** * 查询学生年龄 * * @param id * @throws Exception */ public void getStudnetAge(Integer id) throws Exception { Student student = studentResponstory.findOne(id); Integer age = student.getAge(); //小于10岁,返回“应该上小学”,大于10岁且小于16岁,返回“应该上初中了” if (age <= 10) { throw new Exception("应该上小学"); } else if (age > 10 && age < 16) { throw new Exception("应该上小学"); } }
接着我们在StudentController中调用,具体代码示例如下:
/** * 获取学生年龄 * @param id * @throws Exception */ @GetMapping("/students/getAge/{id}") public void getAge(@PathVariable("id") Integer id) throws Exception { studentService.getStudnetAge(id); }
数据库中学生的信息如下:
我们先来查询id为13、15、16的学生,查看接口返回信息如下:
异常不一样,我们需要再次进行统一化管理了,输出统一格式化后的json。
3、使用 @ControllerAdvice 实现全局异常处理
显然我们需要把message中的信息及code组合外部对象,在包装内部返回data对象,这时需要我们使用 @ControllerAdvice 进行全局异常处理,配合@ExceptionHandle注解使用,@ExceptionHandle注解可以自动捕获controller层出现的指定类型异常,并对该异常进行相应的异常处理。
我们先来建立一个统一的异常类,继承RuntimeException,因为对于spring boot框架中,只有RuntimeException类的异常才会进行事务回滚,具体示例代码如下:
package com.rongrong.springboot.demo.exception; /** * @author rongrong * @version 1.0 * @description: * @date 2020/1/10 0:24 */ public class StudentException extends RuntimeException{ //code码 private Integer code; //错误信息 private String msg; public StudentException(Integer code, String msg) { this.code = code; this.msg = msg; } public void setCode(Integer code) { this.code = code; } public void setMsg(String msg) { this.msg = msg; } public Integer getCode() { return code; } public String getMsg() { return msg; } }
注意:此处必须用getSet方法,不能lombok插件,否则会报错,没有定义getSet方法。
接着我们再来编写全局异常处理,并针对异常类型做出判断,具体示例代码如下:
package com.rongrong.springboot.demo.handle; import com.rongrong.springboot.demo.domain.Result; import com.rongrong.springboot.demo.exception.StudentException; import com.rongrong.springboot.demo.utils.ResultUtils; import org.springframework.web.bind.annotation.ControllerAdvice; import org.springframework.web.bind.annotation.ExceptionHandler; import org.springframework.web.bind.annotation.ResponseBody; /** * @description: 全局异常处理 * @author rongrong * @version 1.0 * @date 2020/1/10 0:17 */ @ControllerAdvice public class ExceptionHandle { @ResponseBody @ExceptionHandler(Exception.class) public Result error(Exception e) { if(e instanceof StudentException){ StudentException studentException=(StudentException)e; return ResultUtils.error(studentException.getCode(),studentException.getMsg()); }else { return ResultUtils.error(-1, "unknown error!"); } } }
同样的,我们需要对StudentService中作出调整,修改为我们自定义的异常,具体示例代码如下:
/** * 查询学生年龄 * * @param id * @throws Exception */ public void getStudnetAge(Integer id) throws Exception { Student student = studentResponstory.findOne(id); Integer age = student.getAge(); //小于10岁,返回“应该上小学”,大于10岁且小于16岁,返回“应该上初中了” if (age <= 10) { throw new StudentException(100,"应该上小学"); } else if (age > 10 && age < 16) { throw new StudentException(101,"应该上初中了"); } }
重新启动项目,再次调用查询学生年龄接口,查看返回结果如下所示证明成功。
4、对code码的统一管理维护
很明显,现在两个报错对应两个code和msg,那么如果有多种code和msg对应的话这里感觉维护起来就很难了,所以我们要把它拿出来统一集中管理就好,这里使用枚举,来实现code和msg的映射。
具体示例代码如下:
package com.rongrong.springboot.demo.exceptionenum; /** * @author rongrong * @version 1.0 * @description: * @date 2020/1/9 23:11 */ public enum ResultEnum { UNKNOW_ERROR(-1,"unknown error!"), HIGH_SCHOOL(10001,"应该上小学啦!"), PRIMARY_SCHOOL(10002,"应该上初中啦!"), SUCCESS(0,"success"); //code码 private Integer code; //错误信息 private String msg; public Integer getCode() { return code; } public void setCode(Integer code) { this.code = code; } public String getMsg() { return msg; } public void setMsg(String msg) { this.msg = msg; } ResultEnum(Integer code, String msg) { this.code = code; this.msg = msg; } }
接下来,需要我们在对StudentService中作出调整,修改为我们自定义的异常,传参为我们的枚举对象,具体示例代码如下:
/** * 查询学生年龄 * * @param id * @throws Exception */ public void getStudnetAge(Integer id) throws Exception { Student student = studentResponstory.findOne(id); Integer age = student.getAge(); //小于10岁,返回“应该上小学”,大于10岁且小于16岁,返回“应该上初中了”,其他正常输出 if (age <= 10) { throw new StudentException(ResultEnum.PRIMARY_SCHOOL); } else if (age > 10 && age < 16) { throw new StudentException(ResultEnum.HIGH_SCHOOL); }else { throw new StudentException(ResultEnum.SUCCESS); } }
接着在对,StudentException这个异常构造器,做下调整,具体代码如下:
package com.rongrong.springboot.demo.exception; import com.rongrong.springboot.demo.exceptionenum.ResultEnum; /** * @author rongrong * @version 1.0 * @description: * @date 2020/1/10 0:24 */ public class StudentException extends RuntimeException{ //code码 private Integer code; //错误信息 private String msg; public StudentException(ResultEnum resultEnum) { this.code = resultEnum.getCode(); this.msg = resultEnum.getMsg(); } public void setCode(Integer code) { this.code = code; } public void setMsg(String msg) { this.msg = msg; } public Integer getCode() { return code; } public String getMsg() { return msg; } }
最后,我们再来启动项目,调用下接口,返回如下信息,证明修改成功!
5、单元测试
为了程序能够更好的运行,我们必须要做测试,所以要养成写完程序进行单元测试的好习惯。
那么在这里我们需要对Service、API进行测试。
5.1、对service进行单元测试
可以通过自定义创建类,来编写单元测试,也可以通过idea向导来创建,具体操作如下图所示:
具体示例代码如下:
package com.rongrong.springboot.demo.controller; import com.rongrong.springboot.demo.domain.Student; import com.rongrong.springboot.demo.responstory.StudentResponstory; import org.junit.Assert; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.test.context.junit4.SpringRunner; /** * @description: 对service进行单元测试 * @author rongrong * @version 1.0 * @date 2020/1/10 20:52 */ @RunWith(SpringRunner.class) @SpringBootTest public class StudentControllerTest { @Autowired StudentResponstory studentResponstory; @Test public void sudentFindOne() { Student student = studentResponstory.findOne(13); Assert.assertEquals(new Integer(25), student.getAge()); } }
5.2、对API进行测试
使用@AutoConfigureMockMvc注解,配合MockMvcRequestBuilders、MockMvcResultMatchers来测试,具体示例代码如下:
package com.rongrong.springboot.demo.controller; import com.rongrong.springboot.demo.responstory.StudentResponstory; import org.junit.Test; import org.junit.runner.RunWith; import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc; import org.springframework.boot.test.context.SpringBootTest; import org.springframework.test.context.junit4.SpringRunner; import org.springframework.test.web.servlet.MockMvc; import org.springframework.test.web.servlet.request.MockMvcRequestBuilders; import org.springframework.test.web.servlet.result.MockMvcResultMatchers; /** * @description: 对API进行单元测试 * @author rongrong * @version 1.0 * @date 2020/1/10 21:12 */ @RunWith(SpringRunner.class) @SpringBootTest @AutoConfigureMockMvc public class StudentApiTest { @Autowired MockMvc mockMvc; @Test public void testStudentApiTest() throws Exception { mockMvc.perform(MockMvcRequestBuilders.get("/students")) .andExpect(MockMvcResultMatchers.status().isOk()) .andExpect(MockMvcResultMatchers.content().string("student")); } }
运行测试,结果如下:
到此,spring boot 中统一异常处理,AutoConfigureMockMvc这个注解,感觉与powermock很像,其中各种APi,有兴趣的同学自己可以去尝试。
学习他人的优点,对比自己的不足!
优秀不够,你是否无可替代
软件测试交流QQ群:721256703,期待你的加入!!
欢迎关注我的微信公众号:软件测试君
![](https://www.cnblogs.com/images/cnblogs_com/longronglang/1061549/o_QQ%E6%88%AA%E5%9B%BE20190728134401.jpg)