golang 通用Contains方法

zsbfree · · 2980 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

今天在群里看见有人写了一个InAarry方法,主要作用是判断一个元素是否在slice中,看完后,我感觉方法应该还有扩展的空间

于是自己重新写了一个Contains方法,可以支持 slice,array,map等类型

 

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

 

本文来自:博客园

感谢作者:zsbfree

查看原文:golang 通用Contains方法

2980 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传