golang 通用Contains方法

package main
import (
 "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 := 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))
}

有疑问加站长微信联系(非本文作者)

• 请尽量让自己的回复能够对别人有帮助
• 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
• 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
• 图片支持拖拽、截图粘贴等方式上传