42 lines
1.2 KiB
Rust
42 lines
1.2 KiB
Rust
use crate::error::RealtimeError;
|
|
use futures_util::Sink;
|
|
use std::fmt::Debug;
|
|
use std::pin::Pin;
|
|
use std::task::{Context, Poll};
|
|
|
|
#[derive(Clone)]
|
|
pub struct UnboundedSenderSink<T>(pub tokio::sync::mpsc::UnboundedSender<T>);
|
|
|
|
impl<T> UnboundedSenderSink<T> {
|
|
pub fn new(tx: tokio::sync::mpsc::UnboundedSender<T>) -> Self {
|
|
Self(tx)
|
|
}
|
|
}
|
|
|
|
impl<T> Sink<T> for UnboundedSenderSink<T>
|
|
where
|
|
T: Send + Sync + 'static + Debug,
|
|
{
|
|
type Error = RealtimeError;
|
|
|
|
fn poll_ready(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
|
// An unbounded channel can always accept messages without blocking, so we always return Ready.
|
|
Poll::Ready(Ok(()))
|
|
}
|
|
|
|
fn start_send(self: Pin<&mut Self>, item: T) -> Result<(), Self::Error> {
|
|
let _ = self.0.send(item);
|
|
Ok(())
|
|
}
|
|
|
|
fn poll_flush(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
|
// There is no buffering in an unbounded channel, so we always return Ready.
|
|
Poll::Ready(Ok(()))
|
|
}
|
|
|
|
fn poll_close(self: Pin<&mut Self>, _cx: &mut Context<'_>) -> Poll<Result<(), Self::Error>> {
|
|
// An unbounded channel is closed by dropping the sender, so we don't need to do anything here.
|
|
Poll::Ready(Ok(()))
|
|
}
|
|
}
|