Created

inx/in.go

@@ -0,0 +1,37 @@
+package inx
+
+import (
+	"strings"
+)
+
+// In Check value in values, return true if in values, otherwise return false.
+// Value T is a generic value
+func In[T comparable](value T, values []T) bool {
+	if values == nil || len(values) == 0 {
+		return false
+	}
+	for _, v := range values {
+		if v == value {
+			return true
+		}
+	}
+	return false
+}
+
+// StringIn 判断 s 是否在 ss 中（忽略大小写）
+func StringIn(s string, ss ...string) bool {
+	if len(ss) == 0 {
+		return false
+	}
+	for _, s2 := range ss {
+		if strings.EqualFold(s, s2) {
+			return true
+		}
+	}
+	return false
+}
+
+// IntIn 判断 i 是否在 ii 中
+func IntIn(i int, ii ...int) bool {
+	return In(i, ii)
+}

inx/in_test.go

@@ -0,0 +1,67 @@
+package inx
+
+import (
+	"github.com/stretchr/testify/assert"
+	"sort"
+	"strconv"
+	"testing"
+)
+
+func TestIn(t *testing.T) {
+	assert.Equal(t, true, In(1, []int{1, 2, 3, 4}), "int1")
+	assert.Equal(t, false, In(1, []int{2, 3, 4, 5}), "int2")
+	assert.Equal(t, false, In(1, nil), "int3")
+	assert.Equal(t, false, In(1, []int{}), "int4")
+	assert.Equal(t, true, In(1, []float64{1.0, 2.0, 3.0}), "float1")
+	assert.Equal(t, false, In(1.1, []float64{1.0, 2.0, 3.0}), "float2")
+	assert.Equal(t, true, In(true, []bool{true, false, false}), "bool1")
+	assert.Equal(t, false, In(true, []bool{false, false, false}), "bool2")
+}
+
+func BenchmarkIn(b *testing.B) {
+	b.StopTimer()
+	ss := make([]string, 100000)
+	for i := 0; i < 100000; i++ {
+		ss[i] = strconv.Itoa(100000 - i)
+	}
+	sort.Strings(ss)
+	b.StartTimer()
+	In("1", ss)
+}
+
+func BenchmarkStringIn(b *testing.B) {
+	b.StopTimer()
+	ss := make([]string, 100000)
+	for i := 0; i < 100000; i++ {
+		ss[i] = strconv.Itoa(100000 - i)
+	}
+	b.StartTimer()
+	StringIn("1", ss...)
+}
+
+func TestIntIn(t *testing.T) {
+	testCases := []struct {
+		tag      string
+		i        int
+		ii       []int
+		expected bool
+	}{
+		{"t1", 1, []int{1, 2, 3, 4, 4}, true},
+		{"t2", 1, []int{2, 3, 4, 5}, false},
+		{"t3", 1, nil, false},
+		{"t4", 1, []int{}, false},
+	}
+	for _, testCase := range testCases {
+		assert.Equal(t, testCase.expected, IntIn(testCase.i, testCase.ii...), testCase.tag)
+	}
+}
+
+func BenchmarkIntIn(b *testing.B) {
+	b.StopTimer()
+	ii := make([]int, 100000)
+	for i := 0; i < 100000; i++ {
+		ii[i] = 100000 - i
+	}
+	b.StartTimer()
+	IntIn(1, ii...)
+}