diff options
| author | bors <bors@rust-lang.org> | 2016-09-14 20:29:15 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2016-09-14 20:29:15 -0700 |
| commit | 16ff9e22cdb552fd10e6cee2eb22f0c5da6d7e79 (patch) | |
| tree | 3d42f5b6742c237ff7a6a7157f67c616080cb015 /src/libcollections | |
| parent | 6ffdda1ba183c981d57e63b59c88184be449eee4 (diff) | |
| parent | ebda77072a56a43d33d5723196a5ae37544a1ab9 (diff) | |
Auto merge of #36347 - knight42:str-replacen, r=alexcrichton
Implement std::str::replacen
Replaces first N matches of a pattern with another string.
```
assert_eq!("acaaa".replacen(a, "b", 3), "bcbba")
```
Diffstat (limited to 'src/libcollections')
| -rw-r--r-- | src/libcollections/str.rs | 43 |
1 files changed, 43 insertions, 0 deletions
diff --git a/src/libcollections/str.rs b/src/libcollections/str.rs index 999c84ba705..6a6b450e518 100644 --- a/src/libcollections/str.rs +++ b/src/libcollections/str.rs @@ -1594,6 +1594,49 @@ impl str { result } + /// Replaces first N matches of a pattern with another string. + /// + /// `replacen` creates a new [`String`], and copies the data from this string slice into it. + /// While doing so, it attempts to find matches of a pattern. If it finds any, it + /// replaces them with the replacement string slice at most `N` times. + /// + /// [`String`]: string/struct.String.html + /// + /// # Examples + /// + /// Basic usage: + /// + /// ``` + /// # #![feature(str_replacen)] + /// let s = "foo foo 123 foo"; + /// assert_eq!("new new 123 foo", s.replacen("foo", "new", 2)); + /// assert_eq!("faa fao 123 foo", s.replacen('o', "a", 3)); + /// assert_eq!("foo foo new23 foo", s.replacen(char::is_numeric, "new", 1)); + /// ``` + /// + /// When the pattern doesn't match: + /// + /// ``` + /// # #![feature(str_replacen)] + /// let s = "this is old"; + /// assert_eq!(s, s.replacen("cookie monster", "little lamb", 10)); + /// ``` + #[unstable(feature = "str_replacen", + issue = "36436", + reason = "only need to replace first N matches")] + pub fn replacen<'a, P: Pattern<'a>>(&'a self, pat: P, to: &str, count: usize) -> String { + // Hope to reduce the times of re-allocation + let mut result = String::with_capacity(32); + let mut last_end = 0; + for (start, part) in self.match_indices(pat).take(count) { + result.push_str(unsafe { self.slice_unchecked(last_end, start) }); + result.push_str(to); + last_end = start + part.len(); + } + result.push_str(unsafe { self.slice_unchecked(last_end, self.len()) }); + result + } + /// Returns the lowercase equivalent of this string slice, as a new [`String`]. /// /// 'Lowercase' is defined according to the terms of the Unicode Derived Core Property |
