Initialise capacity

WGUNDERWOOD · Oct 9, 2024 · a20149b · a20149b
1 parent 4516239
commit a20149b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/format.rs b/src/format.rs
@@ -30,7 +30,7 @@ pub fn format_file(
     let old_lines = old_text.lines();
     let mut old_lines = zip(1.., old_lines);
     let mut queue: Vec<(usize, String)> = vec![];
-    let mut new_text = String::with_capacity(text.len());
+    let mut new_text = String::with_capacity(2 * text.len());
     let indent_char = if args.usetabs { "\t" } else { " " };
 
     loop {

diff --git a/src/indent.rs b/src/indent.rs
@@ -47,7 +47,7 @@ fn get_diff(line: &str, pattern: &Pattern) -> i8 {
         };
         diff += 1;
         diff += i8::try_from(
-            LISTS_BEGIN.iter().filter(|&r| line.contains(r)).count(),
+            LISTS_BEGIN.iter().any(|r| line.contains(r)),
         )
         .unwrap();
     } else if pattern.contains_env_end && line.contains(ENV_END) {
@@ -57,7 +57,7 @@ fn get_diff(line: &str, pattern: &Pattern) -> i8 {
         };
         diff -= 1;
         diff -= i8::try_from(
-            LISTS_END.iter().filter(|&r| line.contains(r)).count(),
+            LISTS_END.iter().any(|r| line.contains(r)),
         )
         .unwrap();
     };