- 精通Rust(第2版)
- (印)拉胡尔·沙玛 (芬)韦萨·凯拉维塔
- 559字
- 2021-02-22 15:36:18
1.4 改进字符计数器
掌握了前面的基础知识后,是时候学以致用了!在这里,我们有一个程序来统计文本文件中的单词实例,并将文件作为参数传递给它。这个程序已经快要完成了,但是有一些编译器捕获的错误和瑕疵,以下是该程序代码:
// word_counter.rs
use std::env;
use std::fs::File;
use std::io::prelude::BufRead;
use std::io::BufReader;
#[derive(Debug)]
struct WordCounter(HashMap<String, u64>);
impl WordCounter {
fn new() -> WordCounter {
WordCounter(HashMap::new());
}
fn increment(word: &str) {
let key = word.to_string();
let count = self.0.entry(key).or_insert(0);
*count += 1;
}
fn display(self) {
for (key, value) in self.0.iter() {
println!("{}: {}", key, value);
}
}
}
fn main() {
let arguments: Vec<String> = env::args().collect();
let filename = arguments[1];
println!("Processing file: {}", filename);
let file = File::open(filenam).expect("Could not open file");
let reader = BufReader::new(file);
let mut word_counter = WordCounter::new();
for line in reader.lines() {
let line = line.expect("Could not read line");
let words = line.split(" ");
for word in words {
if word == "" {
continue
} else {
word_counter.increment(word);
}
}
}
word_counter.display();
}
根据上述代码继续完善该程序,并将它另存为一个文件;尝试在编译器的帮助下修复所有错误。每次尝试修复一个错误,并根据编译器重新编译代码来获得反馈。除了本章介绍的主题之外,本练习的目的是让你学会利用编译器的错误提示信息,这是了解编译器及其如何分析代码的更多信息的重要练习。你可能会惊讶地发现编译器在帮助你从代码中剔除错误方面非常有用。
完善上述代码之后,你可以尝试通过以下练习来进一步提高自己的水平。
将filter参数添加到WordCounter的display()方法,以根据计数过滤输出结果。换句话说,仅当只大于该过滤值时才显示键/值对。
因为HashMap随机存储值,所以每次运行程序的输出结果也是随机的。尝试对输出结果排序,HashMap的values()方法可能会很有用。
看一下display()方法的self参数。如果你将self前面的&运算符删除再编译运行会发生什么?