gobyexample-collection-functions

来源：https://github.com/xg-wang/gobyexample/tree/master/examples

//我们经常需要程序在数据集上执行操作，比如选择满足给定条件的所有项，或者将所有项通过一个自定义函数映射到一个新的集合上
//在某些语言中，会习惯使用【泛型】。Go不支持泛型，在 Go 中，当你的程序或者数据类型需要时，通常是通过组合的方式来提供操作函数
//有时候，直接使用内联组合操作代码会更清晰，而不是创建并调用一个帮助函数
package main

import (
    "fmt"
    "strings"
)

//返回目标字符串`t`出现的第一个索引位置，或者在没有匹配值时返回 -1
func Index(vs []string, t string) int {
    for i, v := range vs {
        if v == t {
            return i
        }
    }
    return -1
}

//如果目标字符串`t`在这个切片中则返回`true`
func Include(vs []string, t string) bool {
    return Index(vs, t) >= 0
}

//如果这些切片中的字符串有一个满足条件`f`则返回`true`
func Any(vs []string, f func(string) bool) bool {
    for _, v := range vs {
        if f(v) {
            return true
        }
    }
    return false
}

//如果切片中的所有字符串都满足条件`f`则返回`true`
func All(vs []string, f func(string) bool) bool {
    for _, v := range vs {
        if !f(v) {
            return false
        }
    }
    return true
}

//返回一个包含所有切片中满足条件`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
}

//返回一个对原始切片所有字符串执行函数`f`后的新切片
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() {

    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")
    }))

    //上面的都是匿名函数，但是也可以使用类型正确的命名函数
    fmt.Println(Map(strs, strings.ToUpper))
}

输出结果为：

2
false
true
false
[peach apple pear]
[PEACH APPLE PEAR PLUM]