golang 通用Contains方法

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

今天在群裡看見有人寫了一個InAarry方法，主要作用是判斷一個元素是否在slice中，看完後，我感覺方法應該還有擴充的空間

於是自己重新寫了一個Contains方法，可以支援 slice,array,map等類型

 

package mainimport (    "errors"    "fmt"    "reflect")// 判斷obj是否在target中，target支援的類型arrary,slice,mapfunc Contain(obj interface{}, target interface{}) (bool, error) {    targetValue := reflect.ValueOf(target)    switch reflect.TypeOf(target).Kind() {    case reflect.Slice, reflect.Array:        for i := 0; i < targetValue.Len(); i++ {            if targetValue.Index(i).Interface() == obj {                return true, nil            }        }    case reflect.Map:        if targetValue.MapIndex(reflect.ValueOf(obj)).IsValid() {            return true, nil        }    }    return false, errors.New("not in array")}func main() {    testMap()    testArray()    testSlice()}func testArray() {    a := 1    b := [3]int{1, 2, 3}    fmt.Println(Contain(a, b))    c := "a"    d := [4]string{"b", "c", "d", "a"}    fmt.Println(Contain(c, d))    e := 1.1    f := [4]float64{1.2, 1.3, 1.1, 1.4}    fmt.Println(Contain(e, f))    g := 1    h := [4]interface{}{2, 4, 6, 1}    fmt.Println(Contain(g, h))    i := [4]int64{}    fmt.Println(Contain(a, i))}func testSlice() {    a := 1    b := []int{1, 2, 3}    fmt.Println(Contain(a, b))    c := "a"    d := []string{"b", "c", "d", "a"}    fmt.Println(Contain(c, d))    e := 1.1    f := []float64{1.2, 1.3, 1.1, 1.4}    fmt.Println(Contain(e, f))    g := 1    h := []interface{}{2, 4, 6, 1}    fmt.Println(Contain(g, h))    i := []int64{}    fmt.Println(Contain(a, i))}func testMap() {    var a = map[int]string{1: "1", 2: "2"}    fmt.Println(Contain(3, a))    var b = map[string]int{"1": 1, "2": 2}    fmt.Println(Contain("1", b))    var c = map[string][]int{"1": {1, 2}, "2": {2, 3}}    fmt.Println(Contain("6", c))}