cargo fmt
This commit is contained in:
@@ -1,4 +1,4 @@
|
||||
#[path= "lib/shared.rs"]
|
||||
#[path = "lib/shared.rs"]
|
||||
mod share;
|
||||
|
||||
// Test data generator
|
||||
@@ -11,10 +11,14 @@ fn main() {
|
||||
let mut args = std::env::args().skip(1);
|
||||
|
||||
let (number, out_file) = match args.next() {
|
||||
None => { eprintln!("writer <number of nodes> [<out_file>]"); std::process::exit(1); },
|
||||
Some(number) => {
|
||||
(number.parse::<usize>().expect("invalid number"), args.next())
|
||||
None => {
|
||||
eprintln!("writer <number of nodes> [<out_file>]");
|
||||
std::process::exit(1);
|
||||
}
|
||||
Some(number) => (
|
||||
number.parse::<usize>().expect("invalid number"),
|
||||
args.next(),
|
||||
),
|
||||
};
|
||||
|
||||
let iterator = share::NodeDataIterator::new().take(number);
|
||||
@@ -24,12 +28,11 @@ fn main() {
|
||||
|
||||
let mut buf = Vec::new();
|
||||
|
||||
for node in iterator{
|
||||
for node in iterator {
|
||||
node.write(&mut buf).expect("Failed to write data");
|
||||
}
|
||||
|
||||
let mut file = std::fs::File::create(&out_file_path)
|
||||
.expect("Failed to create output file");
|
||||
let mut file = std::fs::File::create(&out_file_path).expect("Failed to create output file");
|
||||
|
||||
file.write_all(&buf[..])
|
||||
.expect("Failed to write data to file");
|
||||
@@ -38,4 +41,4 @@ fn main() {
|
||||
println!("{:?}", n);
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user