noodles_fasta/io/reader/
records.rs

1use std::io::{self, BufRead};
2
3use crate::{record::Sequence, Record};
4
5use super::Reader;
6
7/// An iterator over records of a FASTA reader.
8///
9/// This is created by calling [`Reader::records`].
10pub struct Records<'a, R> {
11    inner: &'a mut Reader<R>,
12    line_buf: String,
13}
14
15impl<'a, R> Records<'a, R>
16where
17    R: BufRead,
18{
19    pub(crate) fn new(inner: &'a mut Reader<R>) -> Self {
20        Self {
21            inner,
22            line_buf: String::new(),
23        }
24    }
25}
26
27impl<R> Iterator for Records<'_, R>
28where
29    R: BufRead,
30{
31    type Item = io::Result<Record>;
32
33    fn next(&mut self) -> Option<Self::Item> {
34        self.line_buf.clear();
35
36        match self.inner.read_definition(&mut self.line_buf) {
37            Ok(0) => return None,
38            Ok(_) => {}
39            Err(e) => return Some(Err(e)),
40        }
41
42        let definition = match self.line_buf.parse() {
43            Ok(d) => d,
44            Err(e) => return Some(Err(io::Error::new(io::ErrorKind::InvalidData, e))),
45        };
46
47        let mut sequence_buf = Vec::new();
48
49        match self.inner.read_sequence(&mut sequence_buf) {
50            Ok(_) => {
51                let record = Record::new(definition, Sequence::from(sequence_buf));
52                Some(Ok(record))
53            }
54            Err(e) => Some(Err(e)),
55        }
56    }
57}
noodles_fasta/io/reader/records.rs

noodles_fasta/io/reader/
records.rs
