【升级到Java 17】switch 表达式

提到 switch, 所有 Java 开发人员应该都不陌生。switch 根据变量的值来执行不同的逻辑，用来替代多分支的if/else。

不过 switch 语句在使用时存在一些常见问题。

首先是 switch 语句中的 case 分支的执行，默认是往下进行的，需要添加 break 来跳出 switch 的执行。这种行为很容易产生bug。

比如，下面代码中的 trafficControl 方法，在参数值是 RED 时，会执行全部的 case 分支，因为并没有添加 break 语句。

public class SwitchStatement {

  enum TrafficColor {
    RED, YELLOW, GREEN
  }

  public void trafficControl(TrafficColor color) {
    switch (color) {
      case RED:
        this.stop();
      case GREEN:
        this.go();
      default:
        this.watchOut();
    }
  }

  private void go() {
    System.out.println("Go!");
  }

  private void stop() {
    System.out.println("Stop!");
  }

  private void watchOut() {
    System.out.println("Be careful!");
  }
}

switch 语句的另外一个常见用法是在不同的分支中对同一个变量进行赋值。如下面的代码所示。这种写法比较繁琐。

public class SwitchStatementAssignment {

  enum TrafficColor {
    RED, YELLOW, GREEN
  }

  public String getMessage(TrafficColor color) {
    String message;
    switch (color) {
      case RED:
        message = "Stop!";
        break;
      case GREEN:
        message = "Go!";
        break;
      default:
        message = "Be careful!";
    }
    return message;
  }
}

为了解决这两个问题，可以用到 switch 表达式和箭头标签。

switch 表达式在 Java 12 中以预览功能的形式引入，在 Java 13 中再次预览，在 Java 14 中成为正式功能。

switch 可以作为表达式来使用，因此可以有自己的值。这就简化了赋值操作。switch 表达式的值由分支来确定。下面的代码展示了 switch 表达式的基本用法。

public class SwitchExpression {

  public String formatGifts(int number) {
    return switch (number) {
      case 0 -> "no gifts";
      case 1 -> "only one gift";
      default -> number + " gifts";
    };
  }
}

在上述代码中，switch 表达式使用的不是传统的冒号，而是箭头。在使用了箭头之后，代码执行不会转到下一个分支，相当于添加了 break。在下面的代码中，同样的 trafficControl 方法，使用箭头标签之后，就不再需要使用 break。

public class SwitchLabel {

  enum TrafficColor {
    RED, YELLOW, GREEN
  }

  public void trafficControl(TrafficColor color) {
    switch (color) {
      case RED -> System.out.println("Stop!");
      case YELLOW -> System.out.println("Be careful!");
      case GREEN -> System.out.println("Go!");
    }
  }
}

大多数情况下，箭头标签之后使用单个表达式就可以满足需求。如果有复杂的逻辑，可以使用代码块。这个时候就需要用 yield 来提供值。

public class YieldValue {

  public String formatGifts(int number) {
    return switch (number) {
      case 0 -> "no gifts";
      case 1 -> "only one gift";
      default -> {
        if (number < 0) {
          yield "no gifts";
        } else {
          yield number + " gifts";
        }
      }
    };
  }
}

在使用传统标签的 switch 语句中，也是可以使用 yield。

public class YieldValue2 {

  public String formatGifts(int number) {
    return switch (number) {
      case 0:
        yield "no gifts";
      case 1:
        yield "only one gift";
      default: {
        if (number < 0) {
          yield "no gifts";
        } else {
          yield number + " gifts";
        }
      }
    };
  }
}

关于 switch 表达式的内容，就介绍到这里。相应的视频见我的B站。

推一下我的新书《Quarkus云原生微服务开发实战》，京东淘宝搜索书名即可。

【升级到Java 17】switch 表达式

猜你喜欢