rust学习十二、一个I/O程序练习

本文内容，根本同原书本，只是稍作了一些改动：

• 可以输入 --help展示用法
  
• 如果文件凌驾1MB,则会接纳逐行查找的方法，制止内存不足
  

整个程序包罗个文件:main.rs,lib.rs
一、代码

闲话少叙，直接上代码。
main.rs

1. /**
2. * 完全模仿书本的例子
3. *
4. * 运行示例
5. * cargo run 宋 E:\learning\gitee\rust-org\test-1\minigrep\src\古诗.txt  -- ok
6. * cargo run 宋 古诗.txt  -- ok
7. * cargo run 宋   -- 错误，因为参数太少了
8. */
9. use std::env;
10. use minigrep::{config,search};
12. fn main() {
13.     //1.0 args() 返回的字符是unicode编码的 ,args()本身返回迭代器,collect()方法将迭代器中的元素收集到Vec<String>中                                                                           
14.     let args: Vec<String> = env::args().collect();
16.     //2.0 解析参数,并构建配置
17.     // unwrap_or_else()方法,如果Config::build(&args)返回Ok(config),则执行闭包中的内容,否则执行|err|{...}中的内容
18.     let con=config::build(&args).unwrap_or_else(|err|{
19.         eprintln!("程序异常:{}",err);
20.         std::process::exit(1);
21.     });   
23.     if con.show_help {
24.         config::show_help();
25.         std::process::exit(0);
26.     }
27.    
28.     println!("配置信息:{:?}",&con);
30.     //3.0 根据配置查找文件，并打印查找到的结果
31.     //如果文件比较小，那么就直接在内存处理;否则直接在search中逐行处理
32.     if config::is_too_big(&con.file_name){<br>

复制代码

print!("正在查找大文件...");

1.        let result= search::searh_big_file(&con.file_name,&con.target);
2.        match result{
3.         Ok(_)=>println!("查找完成!"),
4.         Err(e)=>println!("读取文件失败: {:?}", e)
5.        }
6.     }
7.     else{<br>

复制代码

print!("正在查找小文件...");

1.         let result=search::search_small_file(&con);
2.         for line in result.iter() {
3.             println!("{}",line);
4.         }
5.     }
6. }

复制代码

 
 
lib.rs

1. /**
2. * 创建两个子模块
3. */
4. #[derive(Debug)]
5. pub struct Config {
6.     pub file_name: String,
7.     pub target: String,
8.     pub show_help: bool,
9. }
10. /**
11. * 模块执行配置
12. */
13. pub mod config {
14.     use super::Config;
15.     use std::env;
16.     use std::fs;
17.     use std::path::Path;
18.     /**
19.      * 第二个参数: 被查找的字符串
20.      * 第三个参数: 文件路径,可以是全路径或者不包含路径的文件名
21.      */
22.     pub fn build(args: &[String]) -> Result<Config, &str> {
23.         if args.len() < 3 {
24.             if args.len() == 2 {
25.                 //如果target_name=="--help"则打印帮助信息,否则执行文件
26.                 let target_name = args[1].clone();
27.                 if target_name == "--help" {
28.                     return Ok(Config {
29.                         target: target_name,
30.                         file_name: "".to_string(),
31.                         show_help: true,
32.                     });
33.                 }
34.             }
35.             return Err("必须有两个参数");
36.         }
37.         println!("rust返回的程序路径:{}", args[0]);
38.         let target_name = args[1].clone();
39.         let file_path = args[2].clone();
41.         if !file_exists(&file_path) {
42.             //用户输入的是文件名
43.             let exec_path = get_current_exec_path();
44.             println!("当前执行路径：{}", exec_path);
45.             let mut tmp_path = exec_path.clone() + &file_path;
46.             if !file_exists(&tmp_path) {
47.                 tmp_path = exec_path.clone().replace("\\target\\debug\", "\\src\") + &file_path;
48.                 if !file_exists(&tmp_path) {
49.                     tmp_path =
50.                         exec_path.clone().replace("\\target\\release\", "\\src\") + &file_path;
51.                     if !file_exists(&tmp_path) {
52.                         return Err("文件不存在");
53.                     }
54.                 }
55.                 //在src下
56.                 Ok(Config {
57.                     target: target_name,
58.                     file_name: tmp_path,
59.                     show_help: false,
60.                 })
61.             } else {
62.                 //在当前目录下=执行路径+文件名
63.                 Ok(Config {
64.                     target: target_name,
65.                     file_name: tmp_path,
66.                     show_help: false,
67.                 })
68.             }
69.         } else {
70.             //在当前目录下 -- 即用户输入的是全路径
71.             println!("{}", file_path);
72.             Ok(Config {
73.                 target: target_name,
74.                 file_name: file_path,
75.                 show_help: false,
76.             })
77.         }
78.     }
80.     /**
81.      * 判断文件是否存在，如果存在则返回true,否则返回false
82.      */
83.     fn file_exists(file_name: &String) -> bool {
84.         Path::new(&file_name).exists()
85.     }
87.     /**
88.      * 获取当前执行路径，最后会包含分隔符号\
89.      * 例如 d:\soft\rust-test\
90.      * 注意：仅限用于windows平台
91.      */
92.     fn get_current_exec_path() -> String {
93.         let cur_dir = match env::current_exe() {
94.             Ok(exe_path) => exe_path.display().to_string(),
95.             Err(e) => panic!("获取当前路径失败:{}", e),
96.         };
97.         // 获取cur_dir的目录，不需要包含文件名称
98.         let path = Path::new(&cur_dir);
99.         let result = path.parent().unwrap().to_str().unwrap().to_string() + "\";
100.         result
101.     }
103.     pub fn is_too_big(file_name: &String) -> bool {
104.         //判断file_name是否超过1Mb,如果是，则返回true,否则返回false
105.         let file_size = fs::metadata(file_name).unwrap().len();
106.         if file_size > 1 * 1024 * 1024 {
107.             return true;
108.         } else {
109.             false
110.         }
111.     }
113.     pub fn show_help() {
114.         println!("用法:minigrep target path");
115.         println!("     minigrep --help");
116.         println!("参数:");
117.         println!("  target  需要查找的目标字符串");
118.         println!("  path    包含target的文件，可以是全路径，也可以是文件名");
119.         println!("          如果是文件名，则会尝试在当前目录下，或者是在相对路径..\\..\\src");
120.         println!("  --help  查看帮助");
121.         println!("示例:");
122.         println!("  minigrep 宋 E:\\learning\\gitee\\rust-org\\test-1\\minigrep\\src\\古诗.txt ");
123.         println!("  minigrep 宋 古诗.txt");
124.         println!("版本 1.0 for windows, 作者->lzfto");
125.     }
126. }
128. /**
129. * 模块执行搜索
130. */
131. pub mod search {
132.     use super::Config;
133.     use std::fs;
134.     use std::fs::File;
135.     use std::io::{BufRead, BufReader};
136.     use std::path::Path;
137.     /**
138.      * 搜索小于1M的文件，并返回查找的结果
139.      * @return  返回查找的结果 Vec<String>
140.      */
141.     pub fn search_small_file(con: &Config) -> Vec<String> {
142.         let contents = fs::read_to_string(con.file_name.clone()).expect("文件不存在");
143.         println!("文章内容:\n{}", contents);
144.         //把contents根据换行符切割成向量
145.         let mut result: Vec<String> = vec![];
146.         let lines = contents.lines();
147.         for line in lines {
148.             if line.find(&con.target).is_some() {
149.                 result.push(line.to_string());
150.             }
151.         }
152.         return result;
153.     }
154.     /**
155.      * 搜索大于1M的文件
156.      * @return  表示成功与否的信息 Result<()>
157.      */
158.     pub fn searh_big_file(file_name: &String, target: &String) -> std::io::Result<()> {
159.         //打开文件file_name,逐行查找,找到则直接打印
160.         let path = Path::new(file_name);
161.         let file = File::open(&path)?;
162.         let reader = BufReader::new(file);
163.         for line in reader.lines() {
164.             match line {
165.                 Ok(line) => {
166.                     if line.find(target).is_some() {
167.                         println!("{}", line);
168.                     }
169.                 }
170.                 Err(e) => println!("Error reading line: {}", e),
171.             }
172.         }
173.         Ok(())
174.     }
175. }

复制代码

 
 
二、测试

为了便于测试，在src目录下放了两个文件：

古诗.txt  1kb
武林外史_utf8.txt 2120kb
表现帮助

 
查找小文件

 
查找大文件

 
三、小结

本例用到了截止12章节的大部分知识：

• 范例-重要是Vec,String,枚举,struct
  
• 结果处理Result，错误处理
  
• 程序组织，包括模块，use等等
  
• i/o-读取程序参数、读取文件，分行读取
  

必须得承认，rust的代码不好写，也不好看，幸好有优秀的编译器和优秀的ide帮了大忙!
这些语法对于初学者而言，还是比较别扭，充斥着怪异的符号和奇特的语法。也许rust的发明者需要泯灭一些心思再优化下编译器，让工程师能够
以更自然优雅的方式写出不违反初心的代码。工程服从也是很重要的！
在编写的过程中，对引用借用有了更具体的体会！
 

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。