如何优雅地检查三个值的相等性?

How do I check the equality of three values elegantly?

假设我有值 abc。我想知道它们是否相等。如果我这样做

if a == b == c{...}

然后我得到一个编译错误

invalid operation: a == b == c (mismatched types bool and TypeOfABandC)

这很明显,因为它解析为:

(a == b) == c

(a == b) 是一个布尔值。

当然可以:

if a == b && a == c {...}

但是,这看起来不太好看,而且让人感到困惑。还有别的办法吗?

事先说明:

您最后提出的解决方案比较3个值是否相等的最短、最清晰和最有效的方法:

if a == b && a == c {
    fmt.Println("Clearest: all 3 are equal")
}

或者(根据您的喜好):

if a == b && b == c {
    fmt.Println("Clearest: all 3 are equal")
}

这个答案的其余部分(接下来)只是在玩弄语言规范和语言的功能,展示我觉得有趣和有创意的东西。他们不会尝试提供更好的解决方案。


Go Playground. The examples build on the terms and the result of the comparisons which are defined in Spec: Comparison operators 上尝试以下所有示例。

一般说明:在下面的示例中,我使用了 interface{} 类型,无论您的值是什么类型(abc),它都可以工作,但是例如,如果您知道它们是 int 类型,您也可以使用该特定类型(这将提高效率并缩短示例的长度)。

map为一组

if len(map[interface{}]int{a: 0, b: 0, c: 0}) == 1 {
    fmt.Println("Map set: all 3 are equal")
}

实际上我们将所有可比较的值作为键放入一个map中,如果都相等,则映射中将只有1对,因此映射的"length"将为1 . map 的值类型在这里没有任何作用,它可以是任何东西。我使用 int 因为这会导致最短的 composite literal(定义 map 值)。

此解决方案非常灵活,因为您可以使用任意数量的值来测试是否所有值都相等,而不仅仅是 3 个。

有数组

数组是可比较的(与切片不同):

if [2]interface{}{a, b} == [2]interface{}{b, c} {
    fmt.Println("Arrays: all 3 are equal")
}

如果a == bb == c(如果对应的元素相等),数组比较的结果将是true

请注意,您也可以将此方法应用于任意数量的值。对于 5 个值,它看起来像这样:

if [4]interface{}{a, b, c, d} == [4]interface{}{b, c, d, e} {
    fmt.Println("Arrays: all 5 are equal")
}

有取巧map

if map[interface{}]bool{a: b == c}[b] {
    fmt.Println("Tricky map: all 3 are equal")
}

这个复合字面量会将b == c的比较结果赋给键a。然后我们询问与键 b 关联的值。如果a == b,索引表达式的结果将是b == c的结果,即3个值是否都相等。如果 a != b,则值类型的零值将是结果,在 bool 的情况下为 false,正确地表明所有 3 个值都不相等。

匿名 structs

struct 值也具有可比性,所以:

if struct{ a, b interface{} }{a, b} == struct{ a, b interface{} }{b, c} {
    fmt.Println("Anon structs: all 3 are equal")
}

与(命名)structs

如果我们预先定义一个简单的struct

type P struct{ a, b interface{} }

比较会紧凑很多:

if (P{a, b} == P{b, c}) {
    fmt.Println("Structs: all 3 are equal")
}

(注意if语句的表达式必须放在括号内,避免——否则编译时错误!)

有切片

切片不可比较,所以我们需要向reflect.DeepEqual()借用一些帮助:

if reflect.DeepEqual([]interface{}{a, b}, []interface{}{b, c}) {
    fmt.Println("Slices: all 3 are equal")
}

有辅助功能

func AllEquals(v ...interface{}) bool {
    if len(v) > 1 {
        a := v[0]
        for _, s := range v {
            if a != s {
                return false
            }
        }
    }
    return true
}

并使用它:

if AllEquals(a, b, c) {
    fmt.Println("Helper function: all 3 are equal")
}

此解决方案也很灵活,您可以使用任意数量的值调用 AllEquals()

请注意,如果我们也愿意在这里调用 reflect.DeepEqual(),我们可以使 AllEquals() 函数更紧凑:

func AllEquals2(v ...interface{}) bool {
    if len(v) < 2 {
        return true
    }
    return reflect.DeepEqual(v[:len(v)-1], v[1:])
}