上一節(jié)我們實現(xiàn)了通過Command調(diào)用subprocess。這一節(jié)，我們將通過自己的代碼去實現(xiàn)一個簡單的grep。當(dāng)然了，這種基礎(chǔ)的工具你是能找到源碼的，而我們的實現(xiàn)也并不像真正的grep那樣注重效率，本節(jié)的主要作用就在于演示標(biāo)準(zhǔn)庫API的使用。

首先，我們需要對當(dāng)前目錄進(jìn)行遞歸，遍歷，每當(dāng)查找到文件的時候，我們回調(diào)一個函數(shù)。

于是，我們就有了這么個函數(shù)：

use std::env::args;
use std::io;
use std::fs::{self, File, DirEntry};
use std::path::Path;

fn visit_dirs(dir: &Path, pattern: &String, cb: &Fn(&DirEntry, &String)) -> io::Result<()> {
    if try!(fs::metadata(dir)).is_dir() {
        for entry in try!(fs::read_dir(dir)) {
            let entry = try!(entry);
            if try!(fs::metadata(entry.path())).is_dir() {
                try!(visit_dirs(&entry.path(), pattern, cb));
            } else {
                cb(&entry, pattern);
            }
        }
    }else{
        let entry = try!(try!(fs::read_dir(dir)).next().unwrap());
        cb(&entry, pattern);
    }
    Ok(())
}

我們有了這樣的一個函數(shù)，有同學(xué)可能覺得這代碼眼熟。這不是標(biāo)準(zhǔn)庫里的例子改了一下么？

是啊！

好了，繼續(xù)，我們需要讀取每個查到的文件，同時判斷每一行里有沒有所查找的內(nèi)容。我們用一個BufferIO去讀取各個文件，同時用String的自帶方法來判斷內(nèi)容是否存在。

fn call_back(de: &DirEntry, pt: &String) {
    let mut f = File::open(de.path()).unwrap();
    let mut buf = io::BufReader::new(f);
    for line in io::BufRead::lines(buf) {
        let line = line.unwrap_or("".to_string());
        if line.contains(pt) {
            println!("{}", &line);
        }
    }
}

最后，我們將整個函數(shù)調(diào)用起來，如下：

use std::env::args;
use std::io;
use std::fs::{self, File, DirEntry};
use std::path::Path;

fn visit_dirs(dir: &Path, pattern: &String, cb: &Fn(&DirEntry, &String)) -> io::Result<()> {
    if try!(fs::metadata(dir)).is_dir() {
        for entry in try!(fs::read_dir(dir)) {
            let entry = try!(entry);
            if try!(fs::metadata(entry.path())).is_dir() {
                try!(visit_dirs(&entry.path(), pattern, cb));
            } else {
                cb(&entry, pattern);
            }
        }
    }else{
        let entry = try!(try!(fs::read_dir(dir)).next().unwrap());
        cb(&entry, pattern);
    }
    Ok(())
}

fn call_back(de: &DirEntry, pt: &String) {
    let mut f = File::open(de.path()).unwrap();
    let mut buf = io::BufReader::new(f);
    for line in io::BufRead::lines(buf) {
        let line = line.unwrap_or("".to_string());
        if line.contains(pt) {
            println!("{}", &line);
        }
    }
}

// 實現(xiàn)調(diào)用grep命令搜索文件
fn main() {
    let mut arg_iter = args();
    arg_iter.next();
    // panic if there is no one
    let pattern = arg_iter.next().unwrap_or("main".to_string());
    let pt =  arg_iter.next().unwrap_or("./".to_string());
    let pt = Path::new(&pt);
    visit_dirs(&pt, &pattern, &call_back).unwrap();
}

調(diào)用如下：

?  demo git:(master) ? ./target/debug/demo "fn main()" ../
fn main() {
fn main() { }
fn main() {
    pub fn main() {
    pub fn main() {}
fn main() {
    pub fn main() {
    pub fn main() {}

上一篇：標(biāo)準(zhǔn)庫示例下一篇：輸入輸出流