1. 程式人生 > >我去,你寫的 switch 語句也太老土了吧

我去,你寫的 switch 語句也太老土了吧

昨天早上通過遠端的方式 review 了兩名新來同事的程式碼,大部分程式碼都寫得很漂亮,嚴謹的同時註釋也很到位,這令我非常滿意。但當我看到他們當中有一個人寫的 switch 語句時,還是忍不住破口大罵:“我擦,小王,你丫寫的 switch 語句也太老土了吧!”

來看看小王寫的程式碼吧,看完不要罵我裝逼啊。

private static String createPlayer(PlayerTypes playerType) {
    switch (playerType) {
        case TENNIS:
            return "網球運動員費德勒";
        case FOOTBALL:
            return "足球運動員C羅";
        case BASKETBALL:
            return "籃球運動員詹姆斯";
        case UNKNOWN:
            throw new IllegalArgumentException("未知");
        default:
            throw new IllegalArgumentException(
                    "運動員型別: " + playerType);

    }
}

看完上述程式碼後,你是不是會發出這樣的感慨——“程式碼寫得很好,沒有任何問題啊!”是不是覺得我在無事生非,錯怪了小王!但此時我要送上《了不起的蓋茨比》中的一句話:

我年紀還輕,閱歷不深的時候,我父親教導過我一句話,我至今還念念不忘。 “每逢你想要批評任何人的時候, ”他對我說,“你就記住,這個世界上所有的人,並不是個個都有過你擁有的那些優越條件。”

哈哈,這句話不光是讓你看的,也是給我看的。是時候冷靜下來談談上述 switch 語句的老土問題了。

看到上圖了吧,當不小心刪掉 default 語句後,編譯器就會報錯,提示:“沒有返回語句”,為了解決這個問題,我們可以新建一個 player 變數作為返回結果,就像下面這樣:

private static String createPlayer(PlayerTypes playerType) {
    String player = null;
    switch (playerType) {
        case TENNIS:
            player = "網球運動員費德勒";
            break;
        case FOOTBALL:
            player = "足球運動員C羅";
            break;
        case BASKETBALL:
            player = "籃球運動員詹姆斯";
            break;
        case UNKNOWN:
            throw new IllegalArgumentException("未知");
    }

    return player;
}

當添加了 player 變數後,case 語句中就需要新增上 break 關鍵字;另外在 switch 語句結束後,返回 player。這時候,編譯器並不會提示任何錯誤,說明 default 語句在這種情況下是可以省略的。

從 JDK 12 開始(本例使用的是 JDK 13),switch 語句升級了,不僅可以像傳統的 switch 語句那樣作為條件的判斷,還可以直接作為一個返回結果。來對小王的程式碼進行改造,如下所示:

private static String createPlayer(PlayerTypes playerType) {
   return switch (playerType) {
        case TENNIS -> "網球運動員費德勒";
        case FOOTBALL -> "足球運動員C羅";
        case BASKETBALL -> "籃球運動員詹姆斯";
        case UNKNOWN ->  throw new IllegalArgumentException("未知");
    };
}

夠 fashion 吧?不僅 switch 關鍵字之前加了 return 關鍵字,case 中還見到了 Lambda 表示式的影子,中劃線和箭頭替代了冒號,意味著箭頭右側的程式碼只管執行無須 break。

並且,default 語句變成了可選項,可有可無,不信?你也動手試試。

新的 switch 語句足夠的智慧化,除了有上述的 3 個優勢,還可以對列舉型別的條件進行校驗。假如在 PlayerTypes 中增加了新的型別 PINGPANG(乒乓球):

public enum PlayerTypes {
    TENNIS,
    FOOTBALL,
    BASKETBALL,
    PINGPANG,
    UNKNOWN
}

此時編譯器會發出以下警告:

意思就是 switch 中的 case 條件沒有完全覆蓋列舉中可能存在的值。好吧,那就把 PINGPANG 的條件加上吧。來看一下完整的程式碼:

public class OldSwitchDemo {
    public enum PlayerTypes {
        TENNIS,
        FOOTBALL,
        BASKETBALL,
        PINGPANG,
        UNKNOWN
    }

    public static void main(String[] args) {
        System.out.println(createPlayer(PlayerTypes.BASKETBALL));
    }

    private static String createPlayer(PlayerTypes playerType) {
        return switch (playerType) {
            case TENNIS -> "網球運動員費德勒";
            case FOOTBALL -> "足球運動員C羅";
            case BASKETBALL -> "籃球運動員詹姆斯";
            case PINGPANG -> "乒乓球運動員馬龍";
            case UNKNOWN -> throw new IllegalArgumentException("未知");
        };
    }
}

switch 語句變成了強大的 switch 表示式,美滋滋啊!那假如一個運動員既會打籃球又會打乒乓球呢?

private static String createPlayer(PlayerTypes playerType) {
    return switch (playerType) {
        case TENNIS -> "網球運動員費德勒";
        case FOOTBALL -> "足球運動員C羅";
        case BASKETBALL,PINGPANG -> "牛逼運動員沉默王二";
        case UNKNOWN -> throw new IllegalArgumentException("未知");
    };
}

就像上述程式碼那樣,使用英文逗號“,”把條件分割開就行了,666 啊!

不服氣?switch 表示式還有更厲害的,-> 右側還可以是 {} 括起來的程式碼塊,就像 Lambda 表示式那樣。

private static String createPlayer(PlayerTypes playerType) {
    return switch (playerType) {
        case TENNIS -> {
            System.out.println("網球");
            yield "網球運動員費德勒";
        }
        case FOOTBALL -> {
            System.out.println("足球");
            yield "足球運動員C羅";
        }
        case BASKETBALL -> {
            System.out.println("籃球");
            yield "籃球運動員詹姆斯";
        }
        case PINGPANG -> {
            System.out.println("乒乓球");
            yield "乒乓球運動員馬龍";
        }
        case UNKNOWN -> throw new IllegalArgumentException("未知");
    };
}

細心的同學會發現一個之前素未謀面的關鍵字 yield,它和傳統的 return、break 有什麼區別呢?

先來看官方的解釋:

A yield statement transfers control by causing an enclosing switch expression to produce a specified value.

意思就是說 yield 語句通過使一個封閉的 switch 表示式產生一個指定值來轉移控制權。為了進一步地瞭解 yield 關鍵字,我們可以反編譯一下位元組碼:

private static String createPlayer(NewSwitchDemo3.PlayerTypes playerType) {
    String var10000;
    switch(playerType) {
        case TENNIS:
            System.out.println("網球");
            var10000 = "網球運動員費德勒";
            break;
        case FOOTBALL:
            System.out.println("足球");
            var10000 = "足球運動員C羅";
            break;
        case BASKETBALL:
            System.out.println("籃球");
            var10000 = "籃球運動員詹姆斯";
            break;
        case PINGPANG:
            System.out.println("乒乓球");
            var10000 = "乒乓球運動員馬龍";
            break;
        case UNKNOWN:
            throw new IllegalArgumentException("未知");
        default:
            throw new IncompatibleClassChangeError();
    }

    return var10000;
}

編譯器在生成位元組碼的時候對 yield 關鍵字做了自動化轉義,轉成了傳統的 break 語句。這下清楚了吧?

但是,話又說出來,那些看似 fashion 的程式碼也不過是把部分秀技的工作交給了編譯器,還可能存在對舊版本不相容、對隊友不友好的問題——程式碼土點就土點唄,沒準是最實用的。

“不好意思,我為昨天早上的囂張向你道歉。。。。。。”我向小王傳送了一條資訊。

好了,我親愛的讀者朋友,以上就是本文的全部內容了,希望能夠博得你的歡心。對了,我還有一個小小的請求:原創不易,如果覺得有點用的話,請不要吝嗇你手中點讚的權力——因為這將是我寫作的最強動力。