✨ feat: byteutil - add new util func Cut() for split []byte

gookit · May 30, 2023 · d3c8d4b · d3c8d4b
1 parent 359d163
commit d3c8d4b
Show file tree

Hide file tree

Showing 2 changed files with 23 additions and 0 deletions.
diff --git a/byteutil/byteutil.go b/byteutil/byteutil.go
@@ -103,3 +103,13 @@ func AppendAny(dst []byte, v any) []byte {
 	}
 	return dst
 }
+
+// Cut bytes. like the strings.Cut()
+func Cut(bs []byte, sep byte) (before, after []byte, found bool) {
+	if i := bytes.IndexByte(bs, sep); i >= 0 {
+		return bs[:i], bs[i+1:], true
+	}
+
+	before = bs
+	return
+}
diff --git a/byteutil/byteutil_test.go b/byteutil/byteutil_test.go
@@ -41,3 +41,16 @@ func TestAppendAny(t *testing.T) {
 	assert.Eq(t, []byte("1<nil>"), byteutil.AppendAny([]byte("1"), nil))
 	assert.Eq(t, "3600000000000", string(byteutil.AppendAny([]byte{}, timex.OneHour)))
 }
+
+func TestCut(t *testing.T) {
+	// test for byteutil.Cut()
+	b, a, ok := byteutil.Cut([]byte("age=123"), '=')
+	assert.True(t, ok)
+	assert.Eq(t, []byte("age"), b)
+	assert.Eq(t, []byte("123"), a)
+
+	b, a, ok = byteutil.Cut([]byte("age=123"), 'x')
+	assert.False(t, ok)
+	assert.Eq(t, []byte("age=123"), b)
+	assert.Empty(t, a)
+}