Created

setx/set.go

@@ -0,0 +1,70 @@
+package setx
+
+import (
+	"strings"
+
+	gox "git.cloudyne.io/go/hiscaler-gox"
+	"git.cloudyne.io/go/hiscaler-gox/inx"
+)
+
+// ToSet change slice to unique values
+func ToSet[T gox.Number | string | bool | byte | rune](values []T) []T {
+	if len(values) <= 1 {
+		return values
+	}
+
+	uniqueValues := make([]T, 0)
+	kv := make(map[T]struct{}, len(values))
+	for _, value := range values {
+		if _, ok := kv[value]; !ok {
+			kv[value] = struct{}{}
+			uniqueValues = append(uniqueValues, value)
+		}
+	}
+	return uniqueValues
+}
+
+func ToStringSet(values []string, caseSensitive bool) []string {
+	if len(values) <= 1 {
+		return values
+	}
+
+	m := make(map[string]string, 0)
+	for _, value := range values {
+		value = strings.TrimSpace(value)
+		if value != "" {
+			fixedValue := value
+			if !caseSensitive {
+				fixedValue = strings.ToLower(fixedValue)
+			}
+			if _, ok := m[fixedValue]; !ok {
+				m[fixedValue] = value
+			}
+		}
+	}
+	if len(m) == 0 {
+		return nil
+	}
+
+	sets := make([]string, len(m))
+	i := 0
+	for _, v := range m {
+		sets[i] = v
+		i++
+	}
+	return sets
+}
+
+func ToIntSet(values []int) []int {
+	if len(values) <= 1 {
+		return values
+	}
+
+	sets := make([]int, 0)
+	for _, value := range values {
+		if !inx.IntIn(value, sets...) {
+			sets = append(sets, value)
+		}
+	}
+	return sets
+}

setx/set_test.go

@@ -0,0 +1,92 @@
+package setx
+
+import (
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+func TestToSet(t *testing.T) {
+	assert.ElementsMatch(t, []int{1, 2, 3}, ToSet([]int{1, 2, 3}), "int1")
+	assert.ElementsMatch(t, []int{1, 2, 3}, ToSet([]int{1, 1, 2, 2, 3, 3}), "int2")
+	assert.ElementsMatch(t, []float32{1, 2, 3}, ToSet([]float32{1, 2, 3}), "float321")
+	assert.ElementsMatch(t, []float64{1, 2, 3}, ToSet([]float64{1, 1, 2, 2, 3, 3}), "float641")
+	assert.ElementsMatch(t, []string{"A", "B", "C"}, ToSet([]string{"A", "B", "C", "C", "B", "A"}), "string1")
+	assert.ElementsMatch(t, []string{"A", " A ", "B", "C"}, ToSet([]string{" A ", "B", "C", "C", "B", "A"}), "string2")
+}
+
+func TestToStringSet(t *testing.T) {
+	testCases := []struct {
+		A      []string
+		B      []string
+		Len    int
+		Values []string
+	}{
+		{[]string{"1", "2", "3"}, []string{"0", "1", "4"}, 5, []string{"0", "1", "2", "3", "4"}},
+		{[]string{"1", "1", "1"}, []string{"0", "1", "2"}, 3, []string{"0", "1", "2"}},
+		{[]string{"   ", "1", "1", "1"}, []string{"0", "1", "2"}, 3, []string{"0", "1", "2"}},
+		{[]string{"\tabc\t", " abc ", "1", "1", "1"}, []string{"0", "1", "2"}, 4, []string{"0", "1", "2", "abc"}},
+		{[]string{"\tabc\t", " abc ", "1", "1", "1", "ABC"}, []string{"0", "1", "2"}, 5, []string{"0", "1", "2", "abc", "ABC"}},
+	}
+
+	for _, testCase := range testCases {
+		c := ToStringSet(append(testCase.A, testCase.B...), true)
+		if len(c) != testCase.Len {
+			t.Errorf("Except %d, actual %d", testCase.Len, len(c))
+		}
+		for _, value := range testCase.Values {
+			exists := false
+			for _, v := range c {
+				if v == value {
+					exists = true
+					break
+				}
+			}
+			if !exists {
+				t.Errorf("%s not in %#v", value, testCase.Values)
+			}
+		}
+	}
+}
+
+func BenchmarkToStringSet(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		ToStringSet([]string{"A", "B", "c", "C", "a", "d", "d", "e", "fgh", "FGH", "fGH", "fgH"}, false)
+	}
+}
+
+func TestIntSliceToSet(t *testing.T) {
+	testCases := []struct {
+		A      []int
+		B      []int
+		Len    int
+		Values []int
+	}{
+		{[]int{1, 2, 3}, []int{0, 1, 4}, 5, []int{0, 1, 2, 3, 4}},
+		{[]int{1, 1, 1}, []int{0, 1, 2}, 3, []int{0, 1, 2}},
+	}
+
+	for _, testCase := range testCases {
+		c := ToIntSet(append(testCase.A, testCase.B...))
+		if len(c) != testCase.Len {
+			t.Errorf("Except %d, actual %d", testCase.Len, len(c))
+		}
+		for _, value := range testCase.Values {
+			exists := false
+			for _, v := range c {
+				if v == value {
+					exists = true
+					break
+				}
+			}
+			if !exists {
+				t.Errorf("%d not in %#v", value, testCase.Values)
+			}
+		}
+	}
+}
+
+func BenchmarkToIntSet(b *testing.B) {
+	for i := 0; i < b.N; i++ {
+		ToIntSet([]int{1, 2, 3, 3, 45, 5, 6, 56, 56, 56, 77, 6, 7, 67, 678, 78, 78, 8, 78})
+	}
+}