1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123
use tokio::io::{self, AsyncWrite, AsyncWriteExt};
use crate::Record;
/// An async FASTQ writer.
pub struct Writer<W> {
inner: W,
}
impl<W> Writer<W>
where
W: AsyncWrite + Unpin,
{
/// Creates an async FASTQ writer.
///
/// # Examples
///
/// ```
/// use noodles_fastq as fastq;
/// let writer = fastq::AsyncWriter::new(Vec::new());
/// ```
pub fn new(inner: W) -> Self {
Self { inner }
}
/// Returns a reference to the underlying writer.
///
/// # Examples
///
/// ```
/// use noodles_fastq as fastq;
/// let writer = fastq::AsyncWriter::new(Vec::new());
/// assert!(writer.get_ref().is_empty());
/// ```
pub fn get_ref(&self) -> &W {
&self.inner
}
/// Returns the underlying writer.
///
/// # Examples
///
/// ```
/// use noodles_fastq as fastq;
/// let writer = fastq::AsyncWriter::new(Vec::new());
/// assert!(writer.into_inner().is_empty());
/// ```
pub fn into_inner(self) -> W {
self.inner
}
/// Writes a FASTQ record.
///
/// # Examples
///
/// ```
/// # #[tokio::main]
/// # async fn main() -> std::io::Result<()> {
/// use noodles_fastq::{self as fastq, record::Definition};
///
/// let mut writer = fastq::AsyncWriter::new(Vec::new());
///
/// let record = fastq::Record::new(Definition::new("r0", ""), "ATCG", "NDLS");
/// writer.write_record(&record).await?;
///
/// assert_eq!(writer.get_ref(), b"@r0\nATCG\n+\nNDLS\n");
/// # Ok(())
/// # }
/// ```
pub async fn write_record(&mut self, record: &Record) -> io::Result<()> {
write_record(&mut self.inner, record).await
}
}
async fn write_record<W>(writer: &mut W, record: &Record) -> io::Result<()>
where
W: AsyncWrite + Unpin,
{
writer.write_all(b"@").await?;
writer.write_all(record.name()).await?;
if !record.description().is_empty() {
writer.write_all(b" ").await?;
writer.write_all(record.description()).await?;
}
writer.write_all(b"\n").await?;
writer.write_all(record.sequence()).await?;
writer.write_all(b"\n").await?;
writer.write_all(b"+\n").await?;
writer.write_all(record.quality_scores()).await?;
writer.write_all(b"\n").await?;
Ok(())
}
#[cfg(test)]
mod tests {
use super::*;
use crate::record::Definition;
#[tokio::test]
async fn test_write_record() -> io::Result<()> {
let mut buf = Vec::new();
let mut record = Record::new(Definition::new("r0", ""), "ACGT", "NDLS");
write_record(&mut buf, &record).await?;
let expected = b"@r0\nACGT\n+\nNDLS\n";
assert_eq!(buf, expected);
record.description_mut().extend_from_slice(b"LN:4");
buf.clear();
write_record(&mut buf, &record).await?;
let expected = b"@r0 LN:4\nACGT\n+\nNDLS\n";
assert_eq!(buf, expected);
Ok(())
}
}