JDK14新特性之switch表达式的实现

简介

switch的新特性可是源远流长，早在JDK 12就以预览功能被引入了，最终在JDK 14成为了正式版本的功能：JEP 361: Switch Expressions (Standard)。

其实Switch新增的功能有两个，一个就是可以连写case，一个就是switch可以带返回值了。

写在前面

就在我兴致勃勃的想要创建一个以switch命名的package的时候，突然间发现在IDEA中居然创建不java类了。

经过我的再三尝试，反复改名，终于被我发现了隐藏在里面的小秘密：

java key word是不能被用在package名字中的。好吧，一直以来package的名字就那么多个，现在想创建一个拉风一点的，居然发现还有这样一个规则。

那么java key word有哪些呢？ 下面就是了。

连写case

先看一个老版本的例子：

 @Test
 public void useOldSwitch(){
  switch (MONDAY) {
   case MONDAY:
   case FRIDAY:
   case SUNDAY:
    System.out.println(6);
    break;
   case TUESDAY:
    System.out.println(7);
    break;
   case THURSDAY:
   case SATURDAY:
    System.out.println(8);
    break;
   case WEDNESDAY:
    System.out.println(9);
    break;
  }
 }

上面的例子中，我们想要匹配所有的星期，然后打印出相应的结果。写了很多个case语句，不美观。

再看一下新版本的例子：

 @Test
 public void useNewSwitch(){
  switch (MONDAY) {
   case MONDAY, FRIDAY, SUNDAY -> System.out.println(6);
   case TUESDAY    -> System.out.println(7);
   case THURSDAY, SATURDAY  -> System.out.println(8);
   case WEDNESDAY    -> System.out.println(9);
  }
 }

一个漂亮的连写，将一切都带走。

注意这里switch语句没有返回值，所以并不需要default语句。

switch返回值

考虑一个在switch中赋值的情况：

 @Test
 public void oldSwitchWithReturnValue(){
  int numLetters;
  switch (MONDAY) {
   case MONDAY:
   case FRIDAY:
   case SUNDAY:
    numLetters = 6;
    break;
   case TUESDAY:
    numLetters = 7;
    break;
   case THURSDAY:
   case SATURDAY:
    numLetters = 8;
    break;
   case WEDNESDAY:
    numLetters = 9;
    break;
   default:
    throw new IllegalStateException("这天没发见人！");
  }
 }

传统方式我们需要定义一个局部变量，并在case中给这个局部变量赋值。

我们看下怎么使用新版的switch替换：

 @Test
 public void newSwitchWithReturnValue(){
  int numLetters = switch (MONDAY) {
   case MONDAY, FRIDAY, SUNDAY -> 6;
   case TUESDAY    -> 7;
   case THURSDAY, SATURDAY  -> 8;
   case WEDNESDAY    -> 9;
   default -> throw new IllegalStateException("这天没发见人!");
  };
 }

是不是非常简单。

注意，这里需要一个default操作，否则会报编译错误。因为可能存在未遍历的值。

yield

上面的switch返回值的情况，如果case后面的表达式比较复杂，那么就需要使用大括号来围起来。这种情况我们需要使用到yield来返回要返回的值。

 @Test
 public void withYield(){
  int result = switch (MONDAY) {
   case MONDAY: {
    yield 1;
   }
   case TUESDAY: {
    yield 2;
   }
   default: {
    System.out.println("不是MONDAY，也不是TUESDAY！");
    yield 0;
   }
  };
 }

1.case L -> 标签取消了贯穿功能 case后面的常量可以使用逗号来分割多个常量

看下代码

public class SwitchTest {
  public static void main(String[] args) {
    Weekday day = Weekday.MON; //初始化一个枚举变量
    //case L:标签具有贯穿功能
    switch (day){//不需要break;
      case MON, TUE, WEN -> {
        System.out.println("上半周");
        String s = "abc"; //局部变量只在当前代码块有效
      }
      case THU, FRI   -> {
        System.out.println("下半周");
        String s = "def";
      }
      case SAT, SUN   -> {
        System.out.println("周末");
      }
    }
    System.out.println("12");
}

//定义枚举
enum Weekday{
  MON,TUE,WEN,THU,FRI,SAT,SUN
}

运行结果：
上半周
12

2.JDK14中扩展了switch可以作为表达式进行赋值或打印操作

public class SwitchTest {
  public static void main(String[] args) {
    Weekday day = Weekday.MON; //初始化一个枚举变量

    //JDK14中扩展了switch可以作为表达式
    //将表达式的值打印出来
    System.out.println(switch(day){
      case MON, TUE, WEN -> "上半周";
      case THU, FRI   -> "下半周";
      case SAT, SUN   -> "周末";
    });

    //将表达式的值赋值给一个变量
    String text = switch(day){
      case MON, TUE, WEN -> "上半周";
      case THU, FRI   -> "下半周";
      case SAT, SUN   -> "周末";
    };

    //通过yield产生一个新的值
    day = Weekday.SAT;
    int x = switch(day){
      case MON, TUE, WEN -> 1;
      case THU, FRI   -> 2;
      case SAT, SUN   -> {
        int rnd = (int)(Math.random()*10); //生成一个(0,10)内的随机整数
        System.out.println("rnd = " + rnd);
        yield rnd; //通过yield产生一个新的值
      };
    };

    System.out.println("x = " + x);
  }
}

//定义枚举
enum Weekday{
  MON,TUE,WEN,THU,FRI,SAT,SUN
}

本文介绍了JDK14中的switch新特性。也是唯一正式版本的新特性....

本文的例子https://github.com/ddean2009/learn-java-base-9-to-20