本文主要是介绍golang 通用Contains方法，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！
golang 通用Contains方法
今天在群里看见有人写了一个InAarry方法，主要作用是判断一个元素是否在slice中，看完后，我感觉方法应该还有扩展的空间
于是自己重新写了一个Contains方法，可以支持 slice,array,map等类型
package mainimport ("errors""fmt""reflect"
)// 判断obj是否在target中，target支持的类型arrary,slice,map
func 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 := 1b := [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.1f := [4]float64{1.2, 1.3, 1.1, 1.4}fmt.Println(Contain(e, f))g := 1h := [4]interface{}{2, 4, 6, 1}fmt.Println(Contain(g, h))i := [4]int64{}fmt.Println(Contain(a, i))
}func testSlice() {a := 1b := []int{1, 2, 3}fmt.Println(Contain(a, b))c := "a"d := []string{"b", "c", "d", "a"}fmt.Println(Contain(c, d))e := 1.1f := []float64{1.2, 1.3, 1.1, 1.4}fmt.Println(Contain(e, f))g := 1h := []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))
}
这篇关于golang 通用Contains方法的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
