diff options
Diffstat (limited to 'src/bytes')
| -rw-r--r-- | src/bytes/bytes.go | 13 | ||||
| -rw-r--r-- | src/bytes/bytes_test.go | 21 |
2 files changed, 34 insertions, 0 deletions
diff --git a/src/bytes/bytes.go b/src/bytes/bytes.go index 98f78f10c3..2bd1284296 100644 --- a/src/bytes/bytes.go +++ b/src/bytes/bytes.go @@ -1422,3 +1422,16 @@ func CutSuffix(s, suffix []byte) (before []byte, found bool) { } return s[:len(s)-len(suffix)], true } + +// CutLast slices s around the last instance of sep, +// returning the text before and after sep. +// The found result reports whether sep appears in s. +// If sep does not appear in s, CutLast returns s, nil, false. +// +// CutLast returns slices of the original slice s, not copies. +func CutLast(s, sep []byte) (before, after []byte, found bool) { + if i := LastIndex(s, sep); i >= 0 { + return s[:i], s[i+len(sep):], true + } + return s, nil, false +} diff --git a/src/bytes/bytes_test.go b/src/bytes/bytes_test.go index 07280464d2..20d333217e 100644 --- a/src/bytes/bytes_test.go +++ b/src/bytes/bytes_test.go @@ -1976,6 +1976,27 @@ func TestCut(t *testing.T) { } } +func TestCutLast(t *testing.T) { + tests := []struct { + s, sep string + before, after string + found bool + }{ + {"a/b/c", "/", "a/b", "c", true}, + {"a//b//c", "//", "a//b", "c", true}, + {"abc", "/", "abc", "", false}, + {"abc", "", "abc", "", true}, + {"", "", "", "", true}, + {"/abc", "/", "", "abc", true}, + {"abc/", "/", "abc", "", true}, + } + for _, tt := range tests { + if before, after, found := CutLast([]byte(tt.s), []byte(tt.sep)); string(before) != tt.before || string(after) != tt.after || found != tt.found { + t.Errorf("CutLast(%q, %q) = %q, %q, %v; want %q, %q, %v", tt.s, tt.sep, before, after, found, tt.before, tt.after, tt.found) + } + } +} + var cutPrefixTests = []struct { s, sep string after string |
