当前位置 博文首页 > golang 通用Contains方法分享

    golang 通用Contains方法分享

    作者:hongmingover 时间:2021-05-31 17:55

    今天在群里看见有人写了一个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))
    }

    补充:golang中的strings.ContainsAny

    golang中的字符串操作strings.ContainsAny

    package main 
    import (
            "fmt"
            "strings"
    )
     
    //golang字符串操作
    func main(){
            s := "hello world hello world"
            str := "wo"
     
            //判断字符串s中是否包含个子串str中的任何一个字符。包含则返回true,
            //如果str为空则返回false
            index := strings.ContainsAny(s,str)
            fmt.Println(index) //true
    }

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持站长博客。如有错误或未考虑完全的地方,望不吝赐教。

    js
    下一篇:没有了