ws_stream 0.0.0

Provide AsyncRead/AsyncWrite over WebSockets (both tokio and 0.3)
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
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67

use
{
	crate     :: { import::*, WsErr, WsErrKind, Accept } ,
};


/// A stream of connections.
//
pub struct Connections
{
	incoming: Compat01As03<Incoming>,
}


impl Connections
{
	// See pin-utils documentation for safety:
	// To make using this macro safe, three things need to be ensured:
	//
   // - If the struct implements Drop, the drop method is not allowed to move the value of the field.
   // - If the struct wants to implement Unpin, it has to do so conditionally:
   //   The struct can only implement Unpin if the field's type is Unpin.
   // - The struct must not be #[repr(packed)].
   //
	pin_utils::unsafe_pinned!( incoming: Compat01As03<Incoming> );


	/// A new Connections stream
	//
	pub fn new( incoming: Compat01As03<Incoming> ) -> Self
	{
		Self{ incoming }
	}
}


impl Stream for Connections
{
	type Item = Result< Compat01As03<Accept<TcpStream>>, WsErr >;


	fn poll_next( self: Pin<&mut Self>, cx: &mut Context ) -> Poll< Option<Self::Item> >
	{
		match self.incoming().poll_next( cx )
		{
			Poll::Pending => Poll::Pending,

			Poll::Ready( incoming ) =>
			{
				match incoming
				{
					Some( Ok(conn) ) =>
					{
						Poll::Ready(Some(Ok( Accept::new(conn).compat() )))
					}

					Some( Err(_) ) => Poll::Ready(Some(Err( WsErrKind::TcpConnection.into() ))),

					None           => Poll::Ready( None ),
				}
			}
		}
	}
}