Golang比较两个slice是否相等

Compare two string slices in GoLang

开发中经常会遇到需要比较两个slice包含的元素是否完全相等的情况,一般来说有两个思路:

  • reflect比较的方法
  • 循环遍历比较的方法

这里用检查两个字符串slice是否相等的例子来测试一下这两种思路的效率<strike>我当然知道你知道reflect方法效率更差啦</strike>

reflect比较的方法

func StringSliceReflectEqual(a, b []string) bool {
    return reflect.DeepEqual(a, b)
}

这个写法很简单,就是直接使用reflect包的reflect.DeepEqual方法来比较ab是否相等

这是我最初完成这个需求的方式,年轻嘛,比较天真,觉得reflect啊,高端大气,而且一行代码搞定,简洁有力,给自己默默的点个赞

循环遍历比较的方法

func StringSliceEqual(a, b []string) bool {
    if len(a) != len(b) {
        return false
    }

    if (a == nil) != (b == nil) {
        return false
    }

    for i, v := range a {
        if v != b[i] {
            return false
        }
    }

    return true
}

以上是我们项目中的使用的用来比较字符串slice是否相等的一个函数,代码逻辑很简单;先比较长度是否相等,false;再比较两个slice是否都为nil或都不为nil,false;再比较对应索引处两个slice的元素是否相等,false;前面都为true

需要注意

if (a == nil) != (b == nil) {
    return false
}

这段代码是必须的,虽然如果没有这段代码,在大多数情况下,上面的函数可以正常工作,但是增加这段代码的作用是与reflect.DeepEqual的结果保持一致:[]int{} != []int(nil)

Benchmark测试效率

我们都知道Golang中reflect效率很低,所以虽然循环遍历的方法看起来很啰嗦,但是如果真的效率比reflect方法高很多,就只能忍痛放弃reflect了

使用Benchmark来简单的测试下二者的效率

Benchmark StringSliceEqual

func BenchmarkEqual(b *testing.B) {
    sa := []string{"q", "w", "e", "r", "t"}
    sb := []string{"q", "w", "a", "s", "z", "x"}
    b.ResetTimer()
    for n := 0; n < b.N; n++ {
        StringSliceEqual(sa, sb)
    }
}

Benchmark StringSliceReflectEqual

func BenchmarkDeepEqual(b *testing.B) {
    sa := []string{"q", "w", "e", "r", "t"}
    sb := []string{"q", "w", "a", "s", "z", "x"}
    b.ResetTimer()
    for n := 0; n < b.N; n++ {
        StringSliceReflectEqual(sa, sb)
    }
}

上面两个函数中,b.ResetTimer()一般用于准备时间比较长的时候重置计时器减少准备时间带来的误差,这里可用可不用

在测试文件所在目录执行go test -bench=.命令

Benchmark对比测试结果

在我的电脑,使用循环遍历的方式,3.43纳秒完成一次比较;使用reflect的方式,208纳米完成一次操作,效率对比十分明显

BCE优化

Golang提供BCE特性,即Bounds-checking elimination,关于Golang中的BCE,推荐一篇大牛博客Bounds Check Elimination (BCE) In Golang 1.7

func StringSliceEqualBCE(a, b []string) bool {
    if len(a) != len(b) {
        return false
    }

    if (a == nil) != (b == nil) {
        return false
    }

    b = b[:len(a)]
    for i, v := range a {
        if v != b[i] {
            return false
        }
    }

    return true
}

上述代码通过b = b[:len(a)]处的bounds check能够明确保证v != b[i]中的b[i]不会出现越界错误,从而避免了b[i]中的越界检查从而提高效率

类似的,完成Benchmark函数

func BenchmarkEqualBCE(b *testing.B) {
    sa := []string{"q", "w", "e", "r", "t"}
    sb := []string{"q", "w", "a", "s", "z", "x"}
    b.ResetTimer()
    for n := 0; n < b.N; n++ {
        StringSliceEqualBCE(sa, sb)
    }
}

在测试文件所在目录执行go test -bench=.命令

Benchmark对比测试结果

看起来提升并不明显啊,而且在运行多次Benchmard测试的过程中还出现过BenchmarkEqualBCE效率低于BenchmarkEqual的情况(╯‵□′)╯︵┴─┴

可能是我对BCE的理解姿势有问题亦或是Golang BCE自身的问题,总之这个如果我有了更深入的理解会再次更新

但是随着Golang的优化,应该会越来越明显吧 ┬─┬ ノ( ' - 'ノ)

结论

推荐使用StringSliceEqualBCE形式的比较函数<strike>反正不用reflect比较就不会被领导骂被同事喷</strike>

代码已经上传到了我的Github仓库可以下载测试

最后当然是希望喜欢这篇文章的朋友为我的个人博客增加点人气啦

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 官方中文版原文链接 感谢社区中各位的大力支持,译者再次奉上一点点福利:阿里云产品券,享受所有官网优惠,并抽取幸运大...
    HetfieldJoe阅读 6,582评论 3 22
  • Go语言做Web编程非常方便,并且在开发效率和程序运行效率方面都非常优秀。相比于Java,其最大的优势就是简便易用...
    暗黑破坏球嘿哈阅读 9,075评论 6 66
  • 1. Java基础部分 基础部分的顺序:基本语法,类相关的语法,内部类的语法,继承相关的语法,异常的语法,线程的语...
    子非鱼_t_阅读 31,779评论 18 399
  • 很幸运,当我想自己认识男朋友时,不想让别人介绍男朋友时,他出现了。自打在北大第一次见面后,我就有了等他的念头。很自...
    伊艾阅读 247评论 0 0
  • 群猫争宠,你算哪一只?
    云朵儿z阅读 566评论 4 3