noodles_vcf/io/reader/
record_bufs.rs

1use std::io::{self, BufRead};
2
3use super::Reader;
4use crate::{variant::RecordBuf, Header};
5
6/// An iterator over records of a VCF reader.
7///
8/// This is created by calling [`Reader::records`].
9pub struct RecordBufs<'r, 'h, R> {
10    inner: &'r mut Reader<R>,
11    header: &'h Header,
12    record: RecordBuf,
13}
14
15impl<'r, 'h, R> RecordBufs<'r, 'h, R>
16where
17    R: BufRead,
18{
19    pub(crate) fn new(inner: &'r mut Reader<R>, header: &'h Header) -> Self {
20        Self {
21            inner,
22            header,
23            record: RecordBuf::default(),
24        }
25    }
26}
27
28impl<R> Iterator for RecordBufs<'_, '_, R>
29where
30    R: BufRead,
31{
32    type Item = io::Result<RecordBuf>;
33
34    fn next(&mut self) -> Option<Self::Item> {
35        match self.inner.read_record_buf(self.header, &mut self.record) {
36            Ok(0) => None,
37            Ok(_) => Some(Ok(self.record.clone())),
38            Err(e) => Some(Err(e)),
39        }
40    }
41}