From 8adcd08daf4ff695707d10dbb2449334ec5abe8e Mon Sep 17 00:00:00 2001
From: ntr <ntr@smokestack.io>
Date: Fri, 26 Jul 2019 00:17:45 +1000
Subject: [PATCH] we hawt

---
 server/src/websocket.rs | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/server/src/websocket.rs b/server/src/websocket.rs
index 38569a55..4f1655de 100644
--- a/server/src/websocket.rs
+++ b/server/src/websocket.rs
@@ -32,6 +32,11 @@ struct Connection {
 impl Handler for Connection {
     fn on_open(&mut self, _: ws::Handshake) -> ws::Result<()> {
         info!("connected account={:?}", self.account);
+
+        if let Some(ref a) = self.account {
+            self.ws.send(RpcMessage::AccountState(a.clone())).unwrap();
+        }
+
         Ok(())
     }
 
@@ -104,7 +109,10 @@ pub fn start(pool: PgPool, events: Events) {
                         let response = to_vec(&n).unwrap();
                         out.send(Message::Binary(response)).unwrap();
                     }
-                    Err(_) => (),
+                    // we done
+                    Err(_e) => {
+                        break;
+                    },
                 };
             }
         });