Go高級特性探究之對象比較詳解

更新時間：2023年06月02日 12:00:10 作者：tracy小貓

在go語言中，要比較兩個對象是否完全相同，我們可以使用三種方法，這篇文章主要為大家介紹了這三種方法的具體實現(xiàn)，需要的可以參考一下

如何比較兩個go對象完全相同

在go語言中，要比較兩個對象是否完全相同，我們可以使用以下三種方法：

方法一：使用reflect.DeepEqual

reflect.DeepEqual是go語言內(nèi)置的深度比較函數(shù)，它可以遞歸比較任意類型的值，包括結(jié)構(gòu)體、切片、map等。

import (
? ? "reflect"
)
func main() {
? ? a := []int{1, 2, 3}
? ? b := []int{1, 2, 3}
? ? equal := reflect.DeepEqual(a, b)
? ? fmt.Println(equal) // true
}

但需要注意的是，使用reflect.DeepEqual比較struct類型時，必須保證結(jié)構(gòu)體內(nèi)的字段順序和類型完全相同。否則比較結(jié)果會是不正確的。以下是一個錯誤的例子：

import (
? ? "reflect"
)
 type person struct {
? ? Name string
? ? Age int
}
func main() {
? ? a := person{Name: "Alice", Age: 18}
? ? b := person{Age: 18, Name: "Alice"}
? ? equal := reflect.DeepEqual(a, b)
? ? fmt.Println(equal) // false
}

上述例子中，結(jié)構(gòu)體a和b的字段順序不同，導(dǎo)致比較結(jié)果為false。

方法二：使用json.Marshal進(jìn)行序列化

我們可以使用json.Marshal將兩個對象序列化成JSON字符串，然后比較兩個字符串是否相等，以判斷兩個對象是否完全相同。

import (
? ? "encoding/json"
)
type person struct {
? ? Name string
? ? Age int
}
func main() {
? ? a := person{Name: "Alice", Age: 18}
? ? b := person{Name: "Alice", Age: 18}
? ? aa, _ := json.Marshal(a)
? ? bb, _ := json.Marshal(b)
? ? equal := string(aa) == string(bb)
? ? fmt.Println(equal) // true
}

需要注意的是，使用此方法比較struct類型時，struct內(nèi)的字段類型必須是json支持的類型，否則無法進(jìn)行序列化比較。同時，使用此方法比較效率相對較低。

方法三：遞歸比較

我們可以直接編寫遞歸函數(shù)比較兩個對象是否完全相同，這樣可以保證對各種類型的支持，比較靈活，效率也比較高。

以下是一個遞歸比較的例子：

import (
? ? "reflect"
)
func IsEqual(a, b interface{}) bool {
? ? if reflect.DeepEqual(a, b) {
? ? ? ? return true
? ? }
? ? va, vb := reflect.ValueOf(a), reflect.ValueOf(b)
? ? if va.Kind() != vb.Kind() {
? ? ? ? return false
? ? }
? ? switch va.Kind() {
? ? case reflect.Ptr:
? ? ? ? return IsEqual(va.Elem().Interface(), vb.Elem().Interface())
? ? case reflect.Array, reflect.Slice:
? ? ? ? if va.Len() != vb.Len() {
? ? ? ? ? ? return false
? ? ? ? }
? ? ? ? for i := 0; i < va.Len(); i++ {
? ? ? ? ? ? if !IsEqual(va.Index(i).Interface(), vb.Index(i).Interface()) {
? ? ? ? ? ? ? ? return false
? ? ? ? ? ? }
? ? ? ? }
? ? ? ? return true
? ? case reflect.Map:
? ? ? ? if va.Len() != vb.Len() {
? ? ? ? ? ? return false
? ? ? ? }
? ? ? ? for _, key := range va.MapKeys() {
? ? ? ? ? ? if !IsEqual(va.MapIndex(key).Interface(), vb.MapIndex(key).Interface()) {
? ? ? ? ? ? ? ? return false
? ? ? ? ? ? }
? ? ? ? }
? ? ? ? return true
? ? case reflect.Struct:
? ? ? ? for i := 0; i < va.NumField(); i++ {
? ? ? ? ? ? if !IsEqual(va.Field(i).Interface(), vb.Field(i).Interface()) {
? ? ? ? ? ? ? ? return false
? ? ? ? ? ? }
? ? ? ? }
? ? ? ? return true
? ? default:
? ? ? ? return false
? ? }
}
type person struct {
? ? Name string
? ? Age int
}
func main() {
? ? a := []person{{"Alice", 18}, {"Bob", 20}}
? ? b := []person{{"Alice", 18}, {"Bob", 20}}
? ? equal := IsEqual(a, b)
? ? fmt.Println(equal) // true
}

遞歸比較函數(shù)中，對各種類型的判斷和比較方式不同，需要根據(jù)實際情況進(jìn)行編寫。使用此方法時，請保證遞歸函數(shù)的正確性，否則會導(dǎo)致比較結(jié)果不正確。

項目中的使用例子

在實際項目中，可能需要比較一些復(fù)雜的對象是否完全相同。例如，在一個電商系統(tǒng)中，可能需要比較兩個訂單是否完全相同。以下是一個訂單比較的代碼示例：

type order struct {
? ? ID string
? ? Items []item
? ? Account string
? ? Price float64
}
type item struct {
? ? Name string
? ? Price float64
? ? Count int
}
func (o *order) Equal(other *order) bool {
? ? if o == other {
? ? ? ? return true
? ? }
? ? if o.ID != other.ID || o.Account != other.Account || o.Price != other.Price {
? ? ? ? return false
? ? }
? ? if len(o.Items) != len(other.Items) {
? ? ? ? return false
? ? }
? ? for i := range o.Items {
? ? ? ? if !o.Items[i].Equal(&other.Items[i]) {
? ? ? ? ? ? return false
? ? ? ? }
? ? }
? ? return true
}
func (i *item) Equal(other *item) bool {
? ? return i.Name == other.Name && i.Price == other.Price && i.Count == other.Count
}

在上述代碼中，我們定義了一個Equal方法，在其中分別比較訂單ID、賬號、價格以及商品列表中每一項商品是否相同。

開源項目例子

開源項目中經(jīng)常會涉及到比較對象是否完全相同的問題。以下是一些比較流行的開源項目中的比較方法：

Kubernetes

Kubernetes中定義了ObjectMeta結(jié)構(gòu)體，其中包含Name、Namespace、Labels等等字段。

type ObjectMeta struct {
? ? Name string `json:"name,omitempty"`
? ? Namespace string `json:"namespace,omitempty"`
? ? Labels map[string]string `json:"labels,omitempty"`
? ? Annotations map[string]string `json:"annotations,omitempty"`
}

為了比較兩個對象是否相同，Kubernetes中重載了Equal方法，代碼如下：

func (a *ObjectMeta) Equal(b *ObjectMeta) bool {
? ? if a == nil && b == nil {
? ? ? ? return true
? ? }
? ? if a == nil || b == nil {
? ? ? ? return false
? ? }
? ? return a.Namespace == b.Namespace && a.Name == b.Name && labels.Equals(a.Labels, b.Labels) && annotations.Equals(a.Annotations, b.Annotations)
}

在Equal方法中，判斷兩個對象的所有字段是否相同。

Etcd

Etcd是一個分布式鍵值存儲系統(tǒng)，用于共享配置和服務(wù)發(fā)現(xiàn)。在Etcd中，定義了pb.Compare結(jié)構(gòu)體，用于比較兩個值是否相等。

type Compare struct {
? ? Target Operand
? ? Result Result
? ? Order? Comparison
}
type Operand interface {
? ? Descriptor() ([]byte, []int)
}
type Result interface {
? ? Descriptor() ([]byte, []int)
}
type Comparison int32
const (
? ? Comparison_EQUAL? ? Comparison = 0
? ? Comparison_GREATER? Comparison = 1
? ? Comparison_LESS ? ? Comparison = 2
? ? Comparison_GREATER_EQUAL Comparison = 3
? ? Comparison_LESS_EQUAL? ? Comparison = 4
)

在Compare結(jié)構(gòu)體中，我們可以看到三個字段，分別表示要比較的值、比較結(jié)果和比較方式。在比較方式相同時，只有要比較的值和比較結(jié)果完全相同，才能認(rèn)為兩個對象完全相同。