Search contents using iterators

minigrep/src/lib.rs

@@ -50,28 +50,19 @@ pub fn run(config: Config) -> Result<(), Box<dyn Error>> {
 }
 
 pub fn search<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
-    let mut results = Vec::new();
+    contents
-
+        .lines()
-    for line in contents.lines() {
+        .filter(|line| line.contains(query))
-        if line.contains(query) {
+        .collect()
-            results.push(line);
-        }
-    }
-
-    results
 }
 
 pub fn search_case_insensitive<'a>(query: &str, contents: &'a str) -> Vec<&'a str> {
     let query = query.to_lowercase();
-    let mut results = Vec::new();
 
-    for line in contents.lines() {
+    contents
-        if line.to_lowercase().contains(&query) {
+        .lines()
-            results.push(line);
+        .filter(|line| line.to_lowercase().contains(&query))
-        }
+        .collect()
-    }
-
-    results
 }