ServerIterator now iterates over IoResults.

examples/multithreaded.rs

@@ -22,7 +22,7 @@ fn main() {
     // We won't use a wrapper here because we don't need the added functionality.
     spawn(move || { 
         for msg in server.iter() {
-            print!("{}", msg.into_string());
+            print!("{}", msg.unwrap().into_string());
         }
     });
 }

examples/simple.rs

@@ -19,6 +19,7 @@ fn main() {
     let server = Wrapper::new(&irc_server);     
     server.identify().unwrap();
     for message in server.iter() {
+        let message = message.unwrap();
         print!("{}", message.into_string());
         if message.command[] == "PRIVMSG" {
             if let Some(msg) = message.suffix {

examples/simple_ssl.rs

@@ -20,6 +20,7 @@ fn main() {
     let server = Wrapper::new(&irc_server);
     server.identify().unwrap();
     for message in server.iter() {
+        let message = message.unwrap();
         print!("{}", message.into_string());
         if message.command[] == "PRIVMSG" {
             if let Some(msg) = message.suffix {

examples/tweeter.rs

@@ -22,7 +22,7 @@ fn main() {
     let server = Wrapper::new(&*irc_server2);
     server.identify().unwrap();
     // Let's set up a loop that just prints the messages.
-    spawn(move || { irc_server.iter().map(|m| print!("{}", m.into_string())).count(); });
+    spawn(move || { irc_server.iter().map(|m| print!("{}", m.unwrap().into_string())).count(); });
     loop {
         server.send_privmsg("#vana", "TWEET TWEET").unwrap();
         sleep(Duration::seconds(10))

src/server/mod.rs

@@ -1,7 +1,7 @@
 //! Interface for working with IRC Servers
 #![experimental]
 use std::collections::HashMap;
-use std::io::{BufferedReader, BufferedWriter, IoResult};
+use std::io::{BufferedReader, BufferedWriter, IoError, IoErrorKind, IoResult};
 use std::sync::{Mutex, RWLock};
 use conn::{Connection, NetStream};
 use data::{Command, Config, Message, Response, User};
@@ -217,15 +217,24 @@ impl<'a, T: IrcReader, U: IrcWriter> ServerIterator<'a, T, U> {
     }
 }
 
-impl<'a, T: IrcReader, U: IrcWriter> Iterator<Message> for ServerIterator<'a, T, U> {
+impl<'a, T: IrcReader, U: IrcWriter> Iterator<IoResult<Message>> for ServerIterator<'a, T, U> {
-    fn next(&mut self) -> Option<Message> {
+    fn next(&mut self) -> Option<IoResult<Message>> {
-        match self.get_next_line() {
+        let res = self.get_next_line().and_then(|msg|
-            Err(_) => None,
+             match from_str(msg[]) {
-            Ok(msg) => {
+                Some(msg) => {
-                let message = from_str(msg[]);
+                    self.server.handle_message(&msg);
-                self.server.handle_message(message.as_ref().unwrap());
+                    Ok(msg)
-                message
+                },
+                None => Err(IoError {
+                    kind: IoErrorKind::InvalidInput,
+                    desc: "Failed to parse message.",
+                    detail: Some(msg)
+                })
             }   
+        );
+        match res {
+            Err(ref err) if err.kind == IoErrorKind::EndOfFile => None,
+            _ => Some(res)
         }
     }
 }
@@ -265,7 +274,7 @@ mod test {
         ));
         let mut messages = String::new();
         for message in server.iter() {
-            messages.push_str(message.into_string()[]);
+            messages.push_str(message.unwrap().into_string()[]);
         }
         assert_eq!(messages[], exp);
     }