Merge pull request #1289 from donutloop/interal-opt-uniq

internal: optimized uniq function

internal/stringy/slice.go

@@ -27,17 +27,14 @@ func Union(s []string, a []string) []string {
 }
 
 func Uniq(s []string) (r []string) {
+outerLoop:
 	for _, entry := range s {
-		found := false
 		for _, existing := range r {
 			if existing == entry {
-				found = true
-				break
+				continue outerLoop
 			}
 		}
-		if !found {
-			r = append(r, entry)
-		}
+		r = append(r, entry)
 	}
 	return
 }

internal/stringy/slice_test.go

@@ -0,0 +1,24 @@
+package stringy_test
+
+import (
+	"github.com/nsqio/nsq/internal/stringy"
+	"testing"
+)
+
+func BenchmarkUniq(b *testing.B) {
+	values := []string{"a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "a", "b"}
+	for i := 0; i < b.N; i++ {
+		values = stringy.Uniq(values)
+		if len(values) != 2 {
+			b.Fatal("values len is incorrect")
+		}
+	}
+}
+
+func TestUniq(t *testing.T) {
+	values := []string{"a", "a", "a", "b", "b", "b", "c", "c", "c"}
+	values = stringy.Uniq(values)
+	if len(values) != 3 {
+		t.Fatal("values len is incorrect")
+	}
+}