Added set_nonblocking methods to Server, Receiver<WebSocketStream> and

Sender<WebSocketStream>.

Author: Paul-Andre

src/receiver.rs

@@ -50,6 +50,14 @@ impl Receiver<WebSocketStream> {
     pub fn shutdown_all(&mut self) -> IoResult<()> {
         self.inner.get_mut().shutdown(Shutdown::Both)
     }
+
+    /// Changes whether the receiver is in nonblocking mode.
+    ///
+    /// If it is in nonblocking mode and there is no incoming message, trying to receive a message
+    /// will return an error instead of blocking.
+    pub fn set_nonblocking(&self, nonblocking: bool) -> IoResult<()> {
+        self.inner.get_ref().set_nonblocking(nonblocking)
+    }
 }
 
 impl<R: Read> ws::Receiver<DataFrame> for Receiver<R> {

src/sender.rs

@@ -45,6 +45,11 @@ impl Sender<WebSocketStream> {
     pub fn shutdown_all(&mut self) -> IoResult<()> {
         self.inner.shutdown(Shutdown::Both)
     }
+
+    /// Changes whether the sender is in nonblocking mode.
+    pub fn set_nonblocking(&self, nonblocking: bool) -> IoResult<()> {
+        self.inner.set_nonblocking(nonblocking)
+    }
 }
 
 impl<W: Write> ws::Sender for Sender<W> {

src/server/mod.rs

@@ -126,6 +126,14 @@ impl<'a> Server<'a> {
 		};
 		Ok(Connection(try!(wsstream.try_clone()), try!(wsstream.try_clone())))
 	}
+
+    /// Changes whether the Server is in nonblocking mode.
+    ///
+    /// If it is in nonblocking mode, accept() will return an error instead of blocking when there
+    /// are no incoming connections.
+    pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
+        self.inner.set_nonblocking(nonblocking)
+    }
 }
 
 impl<'a> Iterator for Server<'a> {

src/stream.rs

@@ -83,4 +83,12 @@ impl WebSocketStream {
 			WebSocketStream::Ssl(ref inner) => WebSocketStream::Ssl(try!(inner.try_clone())),
 		})
 	}
+
+    /// Changes whether the stream is in nonblocking mode.
+    pub fn set_nonblocking(&self, nonblocking: bool) -> io::Result<()> {
+        match *self {
+			WebSocketStream::Tcp(ref inner) => inner.set_nonblocking(nonblocking),
+			WebSocketStream::Ssl(ref inner) => inner.get_ref().set_nonblocking(nonblocking),
+        }
+    }
 }