Remove exit variable from WriteStage

src/tpu.rs

@@ -80,13 +80,8 @@ impl Tpu {
             None => RecordStage::new(signal_receiver, &bank.last_id()),
         };
 
-        let (write_stage, blob_receiver) = WriteStage::new(
+        let (write_stage, blob_receiver) =
-            bank.clone(),
+            WriteStage::new(bank.clone(), blob_recycler.clone(), writer, entry_receiver);
-            exit.clone(),
-            blob_recycler.clone(),
-            writer,
-            entry_receiver,
-        );
 
         let tpu = Tpu {
             fetch_stage,

src/write_stage.rs

@@ -11,7 +11,6 @@ use result::{Error, Result};
 use service::Service;
 use std::collections::VecDeque;
 use std::io::Write;
-use std::sync::atomic::{AtomicBool, Ordering};
 use std::sync::mpsc::{channel, Receiver, RecvTimeoutError};
 use std::sync::Arc;
 use std::thread::{self, Builder, JoinHandle};
@@ -46,7 +45,6 @@ impl WriteStage {
     /// Create a new WriteStage for writing and broadcasting entries.
     pub fn new<W: Write + Send + 'static>(
         bank: Arc<Bank>,
-        exit: Arc<AtomicBool>,
         blob_recycler: BlobRecycler,
         writer: W,
         entry_receiver: Receiver<Vec<Entry>>,
@@ -65,15 +63,10 @@ impl WriteStage {
                     ) {
                         match e {
                             Error::RecvTimeoutError(RecvTimeoutError::Disconnected) => break,
-                            Error::SendError => break,
+                            Error::SendError => (), // Ignore when downstream stage exists prematurely.
                             _ => error!("{:?}", e),
                         }
                     };
-
-                    if exit.load(Ordering::Relaxed) {
-                        info!("broadcat_service exiting");
-                        break;
-                    }
                 }
             })
             .unwrap();