integer_encoding/
writer.rsuse std::io::{Result, Write};
use crate::fixed::FixedInt;
use crate::varint::VarInt;
#[cfg(feature = "tokio_async")]
use tokio::io::{AsyncWrite, AsyncWriteExt};
#[cfg(feature = "futures_async")]
use futures_util::{io::AsyncWrite, io::AsyncWriteExt};
pub trait VarIntWriter {
fn write_varint<VI: VarInt>(&mut self, n: VI) -> Result<usize>;
}
#[cfg(any(feature = "tokio_async", feature = "futures_async"))]
#[async_trait::async_trait]
pub trait VarIntAsyncWriter {
async fn write_varint_async<VI: VarInt + Send>(&mut self, n: VI) -> Result<usize>;
}
#[cfg(any(feature = "tokio_async", feature = "futures_async"))]
#[async_trait::async_trait]
impl<AW: AsyncWrite + Send + Unpin> VarIntAsyncWriter for AW {
async fn write_varint_async<VI: VarInt + Send>(&mut self, n: VI) -> Result<usize> {
let mut buf = [0 as u8; 10];
let b = n.encode_var(&mut buf);
self.write_all(&buf[0..b]).await?;
Ok(b)
}
}
impl<Inner: Write> VarIntWriter for Inner {
fn write_varint<VI: VarInt>(&mut self, n: VI) -> Result<usize> {
let mut buf = [0 as u8; 10];
let used = n.encode_var(&mut buf[..]);
self.write_all(&buf[0..used])?;
Ok(used)
}
}
pub trait FixedIntWriter {
fn write_fixedint<FI: FixedInt>(&mut self, n: FI) -> Result<usize>;
}
#[cfg(any(feature = "tokio_async", feature = "futures_async"))]
#[async_trait::async_trait]
pub trait FixedIntAsyncWriter {
async fn write_fixedint_async<FI: FixedInt + Send>(&mut self, n: FI) -> Result<usize>;
}
#[cfg(any(feature = "tokio_async", feature = "futures_async"))]
#[async_trait::async_trait]
impl<AW: AsyncWrite + Unpin + Send> FixedIntAsyncWriter for AW {
async fn write_fixedint_async<FI: FixedInt + Send>(&mut self, n: FI) -> Result<usize> {
let mut buf = [0 as u8; 8];
n.encode_fixed(&mut buf[0..FI::required_space()]);
self.write_all(&buf[0..FI::required_space()]).await?;
Ok(FI::REQUIRED_SPACE)
}
}
impl<W: Write> FixedIntWriter for W {
fn write_fixedint<FI: FixedInt>(&mut self, n: FI) -> Result<usize> {
let mut buf = [0 as u8; 8];
n.encode_fixed(&mut buf[0..FI::required_space()]);
self.write_all(&buf[0..FI::required_space()])?;
Ok(FI::REQUIRED_SPACE)
}
}