noodles_cram/async/io/reader/header/
container.rs

1//! Async CRAM header container reader.
2
3mod block;
4mod header;
5mod sam_header;
6
7use tokio::io::{self, AsyncRead, AsyncReadExt, Take};
8
9use self::block::read_block;
10pub(super) use self::header::read_header;
11
12/// An async CRAM header container reader.
13pub struct Reader<R> {
14    inner: Take<R>,
15}
16
17impl<R> Reader<R>
18where
19    R: AsyncRead + Unpin,
20{
21    pub(super) fn new(inner: R, len: u64) -> Self {
22        Self {
23            inner: inner.take(len),
24        }
25    }
26
27    /// Returns a raw SAM header reader.
28    pub async fn raw_sam_header_reader(
29        &mut self,
30    ) -> io::Result<sam_header::Reader<impl AsyncRead + Unpin + '_>> {
31        let mut reader = read_block(&mut self.inner).await?;
32
33        let len = reader.read_i32_le().await.and_then(|n| {
34            u64::try_from(n).map_err(|e| io::Error::new(io::ErrorKind::InvalidData, e))
35        })?;
36
37        Ok(sam_header::Reader::new(reader, len))
38    }
39
40    /// Discards all input until EOF.
41    pub async fn discard_to_end(&mut self) -> io::Result<u64> {
42        io::copy(&mut self.inner, &mut io::sink()).await
43    }
44}