gstrings

snail007 · Dec 26, 2023 · a0b61cd · a0b61cd
1 parent 5554cae
commit a0b61cd
Show file tree

Hide file tree

Showing 2 changed files with 36 additions and 0 deletions.
diff --git a/util/strings/strings.go b/util/strings/strings.go
@@ -31,6 +31,30 @@ func HasSuffixAny(str string, suffix ...string) bool {
 	return false
 }
 
+func ContainsAny(str string, sub ...string) bool {
+	if len(sub) == 0 {
+		return true
+	}
+	for _, v := range sub {
+		if strings.Contains(str, v) {
+			return true
+		}
+	}
+	return false
+}
+
+func ContainsAll(str string, sub ...string) bool {
+	if len(sub) == 0 {
+		return true
+	}
+	for _, v := range sub {
+		if !strings.Contains(str, v) {
+			return false
+		}
+	}
+	return true
+}
+
 func HasHTTPPrefix(str string) bool {
 	return HasPrefixAny(strings.ToLower(str), "http://", "https://")
 }

diff --git a/util/strings/strings_test.go b/util/strings/strings_test.go
@@ -51,3 +51,15 @@ func TestReplace(t *testing.T) {
 		}
 	}
 }
+
+func TestContainsAny(t *testing.T) {
+	assert.False(t, ContainsAny("abcd", "aa", "aaa", "aaa"))
+	assert.True(t, ContainsAny("abcd", "ab", "bc", "cde"))
+	assert.True(t, ContainsAny("abcd"))
+}
+
+func TestContainsAll(t *testing.T) {
+	assert.False(t, ContainsAll("abcd", "a", "aaa", "aaa"))
+	assert.True(t, ContainsAll("abcd", "ab", "bc", "cd"))
+	assert.True(t, ContainsAll("abcd"))
+}