88542e266c
Exponentially easier to use moq-transport as there's no message handling required. This is a BREAKING CHANGE.
36 lines
1005 B
Rust
36 lines
1005 B
Rust
// A helper class to guard sending control messages behind a Mutex.
|
|
|
|
use std::{fmt, sync::Arc};
|
|
|
|
use tokio::sync::Mutex;
|
|
use webtransport_quinn::{RecvStream, SendStream};
|
|
|
|
use crate::{message::Message, Error};
|
|
|
|
#[derive(Debug, Clone)]
|
|
pub(crate) struct Control {
|
|
send: Arc<Mutex<SendStream>>,
|
|
recv: Arc<Mutex<RecvStream>>,
|
|
}
|
|
|
|
impl Control {
|
|
pub fn new(send: SendStream, recv: RecvStream) -> Self {
|
|
Self {
|
|
send: Arc::new(Mutex::new(send)),
|
|
recv: Arc::new(Mutex::new(recv)),
|
|
}
|
|
}
|
|
|
|
pub async fn send<T: Into<Message> + fmt::Debug>(&self, msg: T) -> Result<(), Error> {
|
|
let mut stream = self.send.lock().await;
|
|
log::info!("sending message: {:?}", msg);
|
|
msg.into().encode(&mut *stream).await.map_err(|_e| Error::Write)
|
|
}
|
|
|
|
// It's likely a mistake to call this from two different tasks, but it's easier to just support it.
|
|
pub async fn recv(&self) -> Result<Message, Error> {
|
|
let mut stream = self.recv.lock().await;
|
|
Message::decode(&mut *stream).await.map_err(|_e| Error::Read)
|
|
}
|
|
}
|