golang如何實現(xiàn)三元運算符功能

更新時間：2024年02月06日 08:52:47 作者：波羅學

這篇文章主要介紹了在其他一些編程語言中,如?C?語言,三元運算符是一種可以用一行代碼實現(xiàn)條件選擇的簡便方法,那么在Go語言中如何實現(xiàn)類似功能呢,下面就跟隨小編一起學習一下吧

使用 if-else 語句

三元運算符，本質(zhì)上其實就是 if-else 的簡化版本。通過 if-else 實現(xiàn)自然就是最常用的做法。

var x int
if condition {
    x = a
} else {
    x = b
}

非常簡單且易理解，無心智負擔。畢竟，這就應該是它本來的樣子。

雖然這比三元運算符要長一些，但它更容易理解，也是 Go 所推薦的方式。

一行表達式

三元運算符之所以被人喜愛，我覺得重要的一個原因就是：它足夠簡潔。我們只要一行代碼就實現(xiàn)條件判斷。

在 Go 中，如果想在一行代碼實現(xiàn)，可能嗎？

我們先來看看 rust 和 Python 是如何實現(xiàn)的。

如果了解 rust，你可能看過如下代碼。

let x = {
  if condition {
    a
  } else {
    b
  }
};

如上的代碼中，我們創(chuàng)建了一個代碼塊，它的最后一個表達式會作為 x 的值。這是 rust 所支持的語法。其實現(xiàn)代的不少語言支持這種簡約語法。

或者更簡潔下寫法也可以，如下：

let = if condition {a} else

如果你了解 Python，你可能看到這樣的代碼。

x = a if condition else b

是不是更加簡潔。

Go 不支持這樣的語法，我們要實現(xiàn)類似效果，就只能通過立刻執(zhí)行的匿名函數(shù)實現(xiàn)。

代碼如下：

x := func() int {
  if condition {
     return a
  }
  return b
}()

算了，好丑，太麻煩了！

看起來還是 if-else 好用。但我還是不甘心，還是希望實現(xiàn)一行代碼的效果，怎么辦呢？

If 函數(shù)

前面的示例中，我們通過匿名函數(shù)實現(xiàn)類似于三元運算符的功能。那不是說，預實現(xiàn)一個函數(shù)即可？

讓我們寫一個 If 的函數(shù)來模擬三元運算符。這個函數(shù)接收一個布爾值和兩個可能的返回值。根據(jù)布爾值的真假，它返回其中一個值。

代碼如下所示：

func If(condition bool, a, b int) int {
    if condition {
        return a
    }
    return b
}

x := If(3 > 2, x1, x2)

現(xiàn)在的代碼是不是就清晰了許多呢？

但這種方法還是有個缺點，就是針對不同的類型都要實現(xiàn)一個 If 函數(shù)，如 IfInt()、IfString()、IfFloat() 等等。

不過從 Go 1.18 開始，Go 成功引入泛型。

我們可以通過泛型擴展一個更通用的 If 函數(shù)，不僅僅適用于整數(shù)，還可以用于其他類型。

示例代碼如下：

func If[T any](condition bool, a, b T) T {
    if condition {
        return a
    }
    return b
}

func main() {
    x := 10
    result := If(x > 0, "positive", "negative")
    fmt.Println(result) // 輸出 "positive"
}

當然，我也不是建議這么用。既然官方不支持就算了吧，if-else 多寫幾行就多寫幾行吧。

奇淫巧技：基于 map

在網(wǎng)上，我還發(fā)現(xiàn)了一個奇淫巧技：基于 Map 模擬三元運算法。

代碼如下：

x = map[string]int{
  true: b,
  false: c,
}[a]

基于 true 和 false 實現(xiàn)條件判斷。

這方法看起來挺有創(chuàng)意，但這其實會增加代碼的理解成本，降低可讀性。再者，這種方法的效率是沒有 if-else 的效率高的，因為涉及到了 map 的算法實現(xiàn)，沒有那么直接。

為什么 Go 沒有三元運算符

你是否好奇，為什么 Go 語言沒有三元運算符？

官方認為三元運算符有時會讓代碼變得復雜和難以理解。Go 鼓勵寫出更清晰直接的代碼。

一個 C 語言版本的復雜三元運算符示例代碼：

#include <stdio.h>

int main() {
    int x = 5, y = 10, z = 15;
    char *result;

    result = x > y ? "X" : 
             y > z ? "Y" : 
             z > x ? "Z" : 
             x == y ? "X equals Y" : 
             y == z ? "Y equals Z" : 
             x == z ? "X equals Z" : 
             "All equal";

    printf("%s\n", result);
    return 0;
}

看這個代碼，頭暈沒？

我們看看摘自官方文檔的原文：

The reason ?: is absent from Go is that the language's designers had seen the operation used too often to create impenetrably complex expressions. The if-else form, although longer, is unquestionably clearer. A language needs only one conditional control flow construct.

翻譯內(nèi)容：