From: Scott Worley Date: Thu, 29 Aug 2024 09:26:09 +0000 (-0700) Subject: Prefer &str over String X-Git-Tag: v0.3.0~19 X-Git-Url: http://git.scottworley.com/tablify/commitdiff_plain/116fc215a7f859e20b60e91bcb24d314a9d911f9?hp=70436f237f896aa7651ef259c16a19302e85df38 Prefer &str over String At this intermediate step in this refactor, we leak() everything(!) --- diff --git a/src/lib.rs b/src/lib.rs index 370eb5d..199b8ab 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -76,36 +76,36 @@ impl std::fmt::Display for HTML { } #[derive(Debug, PartialEq, Eq, Hash)] -struct Entry { - col: String, - instance: Option, +struct Entry<'a> { + col: &'a str, + instance: Option<&'a str>, } -impl From<&str> for Entry { - fn from(value: &str) -> Entry { +impl<'a> From<&'a str> for Entry<'a> { + fn from(value: &'a str) -> Entry<'a> { match value.split_once(':') { None => Entry { - col: String::from(value), + col: value, instance: None, }, Some((col, instance)) => Entry { - col: String::from(col.trim()), - instance: Some(String::from(instance.trim())), + col: col.trim(), + instance: Some(instance.trim()), }, } } } #[derive(Debug, PartialEq, Eq)] -struct RowInput { - label: String, - entries: Vec, +struct RowInput<'a> { + label: &'a str, + entries: Vec>, } -struct Reader>> { +struct Reader<'a, Input: Iterator>> { input: std::iter::Enumerate, - row: Option, + row: Option>, } -impl>> Reader { +impl<'a, Input: Iterator>> Reader<'a, Input> { fn new(input: Input) -> Self { Self { input: input.enumerate(), @@ -113,14 +113,15 @@ impl>> Reader { } } } -impl>> Iterator for Reader { - type Item = Result; +impl<'a, Input: Iterator>> Iterator for Reader<'a, Input> { + type Item = Result, std::io::Error>; fn next(&mut self) -> Option { loop { match self .input .next() - .map(|(n, r)| (n, r.map(|line| String::from(line.trim_end())))) + // TODO: Don't leak + .map(|(n, r)| (n, r.map(|line| String::from(line).leak().trim_end()))) { None => return Ok(std::mem::take(&mut self.row)).transpose(), Some((_, Err(e))) => return Some(Err(e)), @@ -152,7 +153,9 @@ impl>> Iterator for Reader } } -fn read_rows(input: impl std::io::Read) -> impl Iterator> { +fn read_rows( + input: impl std::io::Read, +) -> impl Iterator, std::io::Error>> { Reader::new(std::io::BufReader::new(input).lines()) } @@ -167,7 +170,7 @@ fn column_counts(rows: &[RowInput]) -> Vec<(usize, String)> { .into_iter() }) .fold(HashMap::new(), |mut cs, col| { - cs.entry(String::from(col)) + cs.entry(String::from(*col)) .and_modify(|n| *n += 1) .or_insert(1); cs @@ -270,22 +273,22 @@ mod tests { assert_eq!( Entry::from("foo"), Entry { - col: String::from("foo"), + col: "foo", instance: None } ); assert_eq!( Entry::from("foo:bar"), Entry { - col: String::from("foo"), - instance: Some(String::from("bar")) + col: "foo", + instance: Some("bar") } ); assert_eq!( Entry::from("foo: bar"), Entry { - col: String::from("foo"), - instance: Some(String::from("bar")) + col: "foo", + instance: Some("bar") } ); } @@ -295,14 +298,14 @@ mod tests { assert_eq!( read_rows(&b"foo"[..]).flatten().collect::>(), vec![RowInput { - label: String::from("foo"), + label: "foo", entries: vec![] }] ); assert_eq!( read_rows(&b"bar"[..]).flatten().collect::>(), vec![RowInput { - label: String::from("bar"), + label: "bar", entries: vec![] }] ); @@ -310,11 +313,11 @@ mod tests { read_rows(&b"foo\nbar\n"[..]).flatten().collect::>(), vec![ RowInput { - label: String::from("foo"), + label: "foo", entries: vec![] }, RowInput { - label: String::from("bar"), + label: "bar", entries: vec![] } ] @@ -322,7 +325,7 @@ mod tests { assert_eq!( read_rows(&b"foo\n bar\n"[..]).flatten().collect::>(), vec![RowInput { - label: String::from("foo"), + label: "foo", entries: vec![Entry::from("bar")] }] ); @@ -331,7 +334,7 @@ mod tests { .flatten() .collect::>(), vec![RowInput { - label: String::from("foo"), + label: "foo", entries: vec![Entry::from("bar"), Entry::from("baz")] }] ); @@ -341,11 +344,11 @@ mod tests { .collect::>(), vec![ RowInput { - label: String::from("foo"), + label: "foo", entries: vec![] }, RowInput { - label: String::from("bar"), + label: "bar", entries: vec![] } ] @@ -356,11 +359,11 @@ mod tests { .collect::>(), vec![ RowInput { - label: String::from("foo"), + label: "foo", entries: vec![] }, RowInput { - label: String::from("bar"), + label: "bar", entries: vec![] } ] @@ -370,7 +373,7 @@ mod tests { .flatten() .collect::>(), vec![RowInput { - label: String::from("foo"), + label: "foo", entries: vec![Entry::from("bar")] }] ); @@ -426,7 +429,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![] } ), @@ -436,7 +439,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![Entry::from("bar")] } ), @@ -446,7 +449,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![Entry::from("foo")] } ), @@ -456,7 +459,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![Entry::from("foo"), Entry::from("foo")] } ), @@ -466,7 +469,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![Entry::from("foo: 5"), Entry::from("foo: 10")] } ), @@ -476,7 +479,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![Entry::from("foo: 5"), Entry::from("foo")] } ), @@ -486,7 +489,7 @@ mod tests { render_cell( "heart", &RowInput { - label: String::from("nope"), + label: "nope", entries: vec![Entry::from("heart: <3")] } ), @@ -496,7 +499,7 @@ mod tests { render_cell( "foo", &RowInput { - label: String::from("bob's"), + label: "bob's", entries: vec![Entry::from("foo")] } ),