什么是异常?
通俗的说就是,让你感觉不爽的,阻碍你的事都算异常,也就是说不让我们程序正常运行的情况。
为什么要统一处理异常?
方便集中管理,集中定位问题
异常实例
举个例子,还用之前的学生信息那个案例,我们添加一个小于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,有兴趣的同学自己可以去尝试。
学习他人的优点,对比自己的不足! |