use std::io::Write;
use std::io::Result as IoResult;
use result::WebSocketResult;
use ws::dataframe::DataFrame;
use stream::WebSocketStream;
use stream::Shutdown;
use ws;
pub struct Sender<W> {
inner: W,
mask: bool,
}
impl<W> Sender<W> {
pub fn new(writer: W, mask: bool) -> Sender<W> {
Sender {
inner: writer,
mask: mask,
}
}
pub fn get_ref(&self) -> &W {
&self.inner
}
pub fn get_mut(&mut self) -> &mut W {
&mut self.inner
}
}
impl Sender<WebSocketStream> {
pub fn shutdown(&mut self) -> IoResult<()> {
self.inner.shutdown(Shutdown::Write)
}
pub fn shutdown_all(&mut self) -> IoResult<()> {
self.inner.shutdown(Shutdown::Both)
}
}
impl<W: Write> ws::Sender for Sender<W> {
fn send_dataframe<D>(&mut self, dataframe: &D) -> WebSocketResult<()>
where D: DataFrame {
dataframe.write_to(&mut self.inner, self.mask)
}
}