尋找golang中最高效的字符串連接函數(shù)

以下五個(gè)字符串連接函數(shù)，你認(rèn)為哪一個(gè)最快？

func StrConcat1(strs []string) (string) {
    var str string
    for _, value := range strs {
        str += value
    }
    return str
}

func StrConcat2(strs []string) (string) {
    if len(strs) == 0 {
        return ""
    }
    b := bytes.Buffer{}
    for _, s := range strs {
        b.WriteString(s)
    }
    return b.String()
}

func StrConcat3(strs []string) (string) {
    return strings.Join(strs, "")
}

func StrConcat4(strs []string) (string) {
    var data []byte
    for _, value := range strs {
        data = append(data, value...)
    }
    return string(data)
}

func StrConcat5(strs []string) (string) {
    var length int
    for _, s := range strs {
        length += len(s)
    }
    bs := make([]byte, length)
    var i int
    for _, value := range strs {
        i += copy(bs[i:], value)
    }
    return string(bs[:])
}

編寫如下的測(cè)試文件

import "testing"

var TestStr = []string{"hello", "world", "my", "god", "this", "is", "big", "content"}

func Benchmark_StrConcat1(b *testing.B)  {
    for i := 0; i < b.N ; i++ {
        StrConcat1(TestStr)
    }
}

func Benchmark_StrConcat2(b *testing.B)  {
    for i := 0; i < b.N ; i++ {
        StrConcat2(TestStr)
    }
}

func Benchmark_StrConcat3(b *testing.B)  {
    for i := 0; i < b.N ; i++ {
        StrConcat3(TestStr)
    }
}

func Benchmark_StrConcat4(b *testing.B)  {
    for i := 0; i < b.N ; i++ {
        StrConcat4(TestStr)
    }
}

func Benchmark_StrConcat5(b *testing.B)  {
    for i := 0; i < b.N ; i++ {
        StrConcat5(TestStr)
    }
}

然后執(zhí)行測(cè)試命令

go test -test.bench=".*"

執(zhí)行結(jié)果如下：

Benchmark_StrConcat1-4       5000000           366 ns/op
Benchmark_StrConcat2-4      10000000           178 ns/op
Benchmark_StrConcat3-4      10000000           127 ns/op
Benchmark_StrConcat4-4      10000000           178 ns/op
Benchmark_StrConcat5-4      20000000           102 ns/op

可見(jiàn)，第5個(gè)函數(shù)的效率是最高的，雖然它的代碼量最多。
其次是第3個(gè)函數(shù)，其實(shí)我們查看第3個(gè)函數(shù)的源代碼就可以發(fā)現(xiàn)，它與第5個(gè)函數(shù)本質(zhì)上是一樣的，但是因?yàn)槎嗔藢?duì)分隔符的拷貝操作，執(zhí)行時(shí)間增加了。我認(rèn)為這個(gè)系統(tǒng)函數(shù)可以再優(yōu)化，判斷分隔符是不是空的，如果是空的，就按第5個(gè)函數(shù)那個(gè)處理，這樣效率還能提升15%。

附，strings.Join函數(shù)源代碼：

// Join concatenates the elements of a to create a single string. The separator string
// sep is placed between elements in the resulting string.
func Join(a []string, sep string) string {
    switch len(a) {
    case 0:
        return ""
    case 1:
        return a[0]
    case 2:
        // Special case for common small values.
        // Remove if golang.org/issue/6714 is fixed
        return a[0] + sep + a[1]
    case 3:
        // Special case for common small values.
        // Remove if golang.org/issue/6714 is fixed
        return a[0] + sep + a[1] + sep + a[2]
    }
    n := len(sep) * (len(a) - 1)
    for i := 0; i < len(a); i++ {
        n += len(a[i])
    }

    b := make([]byte, n)
    bp := copy(b, a[0])
    for _, s := range a[1:] {
        bp += copy(b[bp:], sep)
        bp += copy(b[bp:], s)
    }
    return string(b)
}