浏览代码

small fix

Radu Boncea 7 月之前
父节点
当前提交
7067f0855d
共有 1 个文件被更改,包括 8 次插入10 次删除
  1. 8 10
      src/main.rs

+ 8 - 10
src/main.rs

@@ -98,16 +98,14 @@ mod charcounter {
     pub fn walk_dir(path: &std::path::Path, counter: &mut Counter) {
         if path.is_dir() {
             counter.increment(IncType::Directory);
-            for entry in std::fs::read_dir(path).expect("read_dir call failed") {
-                if let Ok(entry) = entry {
-                    let path = entry.path();
-                    if path.is_dir() {
-                        walk_dir(&path, counter);
-                    } else if path.is_file() {
-                        do_counting(&path, counter);
-                    } else {
-                        continue;
-                    }
+            for entry in std::fs::read_dir(path).expect("read_dir call failed").flatten() {
+                let path = entry.path();
+                if path.is_dir() {
+                    walk_dir(&path, counter);
+                } else if path.is_file() {
+                    do_counting(&path, counter);
+                } else {
+                    continue;
                 }
             }
         }