Prevent a crash of the wavefront backend

This commit prevents a crash of the wavefront backend when the a TCP connection cannot be established on flush. Signed-off-by: Brian L. Troutwine <blt@postmates.com>
postmates · Jul 13, 2016 · 977c93c · 977c93c
1 parent b82e84e
commit 977c93c
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 9 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/src/backends/wavefront.rs b/src/backends/wavefront.rs
@@ -113,12 +113,15 @@ impl Wavefront {
 
 impl Backend for Wavefront {
     fn flush(&mut self) {
-        debug!("wavefront flush");
-        let stats = self.format_stats(None);
-        debug!("wavefront - {}", stats);
-        self.points.clear();
-        let mut stream = TcpStream::connect(self.addr).unwrap();
-        let _ = stream.write(stats.as_bytes());
+        let res = TcpStream::connect(self.addr);
+        if res.is_ok() {
+            let mut stream = res.unwrap();
+            debug!("wavefront flush");
+            let stats = self.format_stats(None);
+            debug!("wavefront - {}", stats);
+            self.points.clear();
+            let _ = stream.write(stats.as_bytes());
+        }
     }
 
     fn deliver(&mut self, point: Rc<Metric>) {