快捷導航

golang 通用Contains方法分享

更新時間：2021年04月30日 09:10:16 作者：hongmingover

這篇文章主要介紹了golang 通用Contains方法分享，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

今天在群里看見有人寫了一個InAarry方法，主要作用是判斷一個元素是否在slice中，看完后，我感覺方法應(yīng)該還有擴展的空間

于是自己重新寫了一個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
}

以上為個人經(jīng)驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。

您可能感興趣的文章: