moq-transport 0.1.1

Media over QUIC
Documentation 
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

// A helper class to guard sending control messages behind a Mutex.

use std::{fmt, sync::Arc};

use tokio::sync::Mutex;
use webtransport_quinn::SendStream;

use crate::{message::Message, Error};

#[derive(Debug, Clone)]
pub(crate) struct Control {
	stream: Arc<Mutex<SendStream>>,
}

impl Control {
	pub fn new(stream: SendStream) -> Self {
		Self {
			stream: Arc::new(Mutex::new(stream)),
		}
	}

	pub async fn send<T: Into<Message> + fmt::Debug>(&self, msg: T) -> Result<(), Error> {
		let mut stream = self.stream.lock().await;
		log::info!("sending message: {:?}", msg);
		msg.into().encode(&mut *stream).await.map_err(|_e| Error::Unknown)
	}
}