一种避免大量If-else代码的新思路

开发 后端
今天,我要给大家带来一个超级无敌霹雳的编码新招式,只要看完,保证你的代码像用了某某洗发水一样,不仅去屑还更柔顺。

哈喽,各位代码战士们,我是Jensen,一个梦想着和大家一起在代码的海洋里遨游,顺便捡起那些散落的知识点的程序员小伙伴。

今天,我要给大家带来一个超级无敌霹雳的编码新招式,只要看完,保证你的代码像用了某某洗发水一样,不仅去屑还更柔顺。

咱们要聊的是那些让人又爱又恨的技术点:自定义异常、全局异常捕获、断言。

一、控制异常流程

首先,让我们来聊聊自定义异常。

你知道的,在Java的世界里,我们通常用if-else语句来检查那些让人头疼的条件。

比如用户登录:

// 伪代码
if (验证码 != 8888) {
    return "验证码错误";
}
// 处理其他业务逻辑
return "登录成功";

但是,你有没有想过,如果不用if-else,而是用异常来控制流程,会怎样呢?

比如,你的登录方法里,验证码不正确,不是返回一个错误信息,而是“砰”地一声抛出个异常。

// 伪代码
if (验证码 != 8888) {
    throw new ServiceException("验证码错误");
}
// 处理其他业务逻辑
return "登录成功";

听起来是不是有点疯狂?但这就是自定义异常的魅力所在,它能让你的代码看起来像是在演电影,每个异常都是一个剧情转折。

这里封装了一个业务类异常类ServiceException,该异常类继承了运行时异常RuntimeException:

/**
 * 服务异常,可用于控制业务异常流程,抛出后由统一异常增强类捕获,返回友好提示
 *
 * @author Jensen
 * @公众号 架构师修行录
 */
@Data
public class ServiceException extends RuntimeException {
    protected Integer code;
    public ServiceException(Integer code, String message) {
        super(message);
        this.code = code;
    }
    public ServiceException() {
        this(500, "请求成功但是服务异常");
    }
    public ServiceException(String message) {
        this(500, message);
    }
    public ServiceException(Throwable e) {
        this(e.getMessage());
    }
}

使用自定义异常来代替传统的if-else语句进行错误处理和流程控制,有其优缺点:

优点:

  • 提高代码可读性:自定义异常可以让你更清晰地表达错误情况,使得代码的意图更容易理解。
  • 集中错误处理:通过抛出和捕获异常,你可以将错误处理逻辑集中到特定的位置,而不是分散在代码的多个地方。
  • 易于维护:当需要修改或扩展错误处理逻辑时,使用异常机制可以更容易地进行维护和升级。
  • 支持多路选择:在某些情况下,异常可以用于实现多路选择结构,通过不同的异常类型来区分不同的执行路径。
  • 资源管理:异常处理机制通常与资源管理(如文件关闭、数据库连接释放等)结合使用,确保资源在使用完毕后得到正确释放。

缺点:

  • 性能开销:异常处理通常比条件判断要慢,因为它涉及到栈展开和异常对象的创建。
  • 滥用风险:异常机制有时会被滥用,例如用于正常的流程控制,这可能会导致性能问题和难以理解的代码。
  • 调试困难:异常可能会使调试变得更加困难,特别是在复杂的异常传播和处理过程中。
  • 异常类型管理:随着项目的发展,可能会有大量的自定义异常类型,管理这些异常类型并确保它们的适当使用可能会变得复杂。
  • 语言特性依赖:使用异常处理可能依赖于特定的编程语言特性,这可能会限制代码的移植性。

总的来说,自定义异常在处理错误和异常情况时提供了一种强大而灵活的机制,我觉得可以使用,但不能滥用,常规的还是要使用if-else控制,涉及API的中断流程可以快速跳出内部,并且R对象可无须传递进Service层。

二、全局异常捕获

想象一下,你的业务代码里抛出了一大堆异常,你总不能把这些异常原封不动地扔给API的使用者吧。

这时候,@RestControllerAdvice就像是那个超级英雄,跳出来拯救世界,把所有的异常都捕获起来,然后优雅地包装成一个个友好的响应。

@RestControllerAdvice 是 Spring MVC 中的一个注解,用于定义一个全局的异常处理类,该类可以捕获和处理 Spring 应用中发生的特定类型的异常。这个注解通常与 @ExceptionHandler 注解一起使用,来定义处理特定异常的方法。

@RestControllerAdvice 的主要作用和特点包括:

  • 全局异常处理:通过在类上使用 @RestControllerAdvice 注解,你可以定义一个类来全局处理控制器(Controller)中抛出的异常。
  • 减少重复代码:你不需要在每个控制器中编写相同的异常处理逻辑。通过定义一个全局的异常处理类,可以统一处理特定类型的异常,从而减少代码重复。
  • 自定义错误响应:可以自定义异常的响应格式,例如返回 JSON 对象,包含错误信息、状态码等,以提供更友好的错误提示给前端。
  • 支持多控制器:一个 @RestControllerAdvice 类可以为多个控制器提供异常处理支持,无论这些控制器是否位于同一个包路径下。
  • 组合使用:可以定义多个 @RestControllerAdvice 类,并通过 basePackages 或 basePackageClasses 属性来指定它们所影响的控制器包路径。
  • 支持继承:如果多个控制器有相似的异常处理需求,可以通过继承一个基础的异常处理类来实现代码复用。

捕获业务异常ServiceException后,封装R对象返回给接口调用方:

@RestControllerAdvice
public class GlobalRestExceptionAdvice {


    @ExceptionHandler({ServiceException.class})
    public R<Map<String, String>> serviceException(HttpServletRequest request, ServiceException e) {
        return R.fail(e.getCode(), e.getMessage(), null);
    }
}

到这里,我们就可以通过抛出ServiceException来实现异常流程的控制了,并且方法上注解了@Transactional实现的事务控制,在抛出异常后也是能正常回滚的。

三、封装使用业务断言

在Java中,断言就像是那个总是在后台默默支持你的好朋友,它在开发和测试阶段帮你检查那些逻辑错误。

但是,别忘了,它也是有脾气的,一旦到了生产环境,它就罢工了。所以,记得在需要的时候启用它,不需要的时候,就让它好好休息。

assert condition : "This is an error message.";

断言的特点:

  • 条件检查:断言用于检查程序的预期条件,如果条件不满足,则抛出异常。
  • 调试时有用:断言在开发和测试阶段非常有用,可以帮助开发者快速定位问题。
  • 性能考虑:由于断言可能会影响程序性能,因此在生产环境中通常不启用断言。
  • 非正式错误处理:断言不是错误处理的一部分,它不适用于处理那些预期会发生的情况。

使用断言的注意事项:

  • 性能影响:断言会增加额外的性能开销,因为每次断言都会计算布尔表达式的值。
  • 生产环境:在生产环境中,断言应该被禁用,以避免不必要的性能损耗。
  • 异常处理:断言抛出的是 AssertionError,这是一种特殊的 Error,通常不应该被应用程序捕获或处理。

断言是Java语言的一个特性,它的使用应该谨慎,主要限于开发和测试阶段,以确保生产环境中的程序性能和稳定性。

我们借鉴断言这种设计思想,针对业务异常进一步封装,形成 “业务断言”,主要目标只有一个:

管理我们对代码的期望。

先定义一个业务断言工具类BizAssert:

/**
 * 业务断言类,断言不通过将抛出ServiceException
 */
@UtilityClass
public class BizAssert {
  
    public <T> T notNull(T dontNull) {
        if (dontNull == null) {
            throw new ServiceException("this object must not be null");
        }
        return dontNull;
    }


    public void notBlank(String dontBlank) {
        if (dontBlank == null || dontBlank.length() == 0) {
            throw new ServiceException("this string must not be null or blank");
        }
    }


    public <E, T extends Iterable<E>> T notEmpty(T dontEmpty) {
        if (dontEmpty == null || !dontEmpty.iterator().hasNext()) {
            throw new ServiceException("this collection must not be null or empty");
        }
        return dontEmpty;
    }


    public <T> T isOk(IR r) {
        if (r == null || !r.isOk()) {
            throw new ServiceException(r == null ? "this result must not be null" : r.getMsg());
        }
        return r.getData();
    }


    public <T> T notNull(IR r) {
        if (r == null || !r.isOk() || r.getData() == null) {
            throw new ServiceException(r == null ? "this result must not be null" : r.getMsg());
        }
        return r.getData();
    }
}

使用业务断言:

// 从数据库中查询出用户
User user = UserQuery.builder().id(userId).build().getOne();
// 业务断言,我希望查出来的用户不能为null,否则后续业务会报错
BizAssert.notNull(user);
// 其他针对user的业务处理逻辑
Map<String, String> data = new HashMap<>();
data.put("girl-friend", user.getName());

四、写在最后

上述提到的业务断言方法,仅展示了其中一部分,除了简单的判断,我们还可以加上日期比较、数值大小比较、equals比较、集合的contains方法等等,我把它的完整版集成到了我的D3Boot开源基础框架内,大家需要可以移步Gitee抄作业。

Gitee源码地址:

https://gitee.com/jensvn/d3boot(例行赊Star)

D3boot基础框架具体的使用方式见源码的README.md文件,这里不再赘述。

能写一行代码的事,绝不写3行5行,适度封装,让代码更优雅!

责任编辑:姜华 来源: 架构师修行录
相关推荐

2024-04-30 08:12:05

CRUD方法JavaAC架构

2021-11-04 08:53:00

if-else代码Java

2024-03-25 10:00:00

C++编程else

2023-06-02 07:30:24

If-else结构流程控制

2023-11-14 08:00:00

Angular前端开发

2022-06-23 07:05:46

跳板机服务器PAM

2020-05-13 14:15:25

if-else代码前端

2018-04-18 07:34:58

2016-10-26 09:12:58

2023-09-17 23:16:46

缓存数据库

2013-03-06 10:28:57

ifJava

2020-03-11 08:00:12

if-else优化运算符

2022-08-17 09:07:09

低代码LCDP编码

2020-04-20 15:40:03

if-elseJava优化

2020-11-27 14:45:57

开发服务器代码

2021-04-13 06:39:13

代码重构code

2021-03-10 07:20:43

if-else静态代码

2017-08-24 15:02:01

前端增量式更新

2016-10-13 10:57:55

phptcp专栏

2019-11-22 09:21:17

技术研发数据
点赞
收藏

51CTO技术栈公众号