Move logging thread's state into a struct

This commit is contained in:
Greg Fitzgerald
2018-03-03 13:52:57 -07:00
parent 8ef864fb39
commit f906112c03

View File

@ -26,6 +26,15 @@ pub enum ExitReason {
RecvDisconnected, RecvDisconnected,
SendDisconnected, SendDisconnected,
} }
pub struct Logger<T> {
pub sender: SyncSender<Entry<T>>,
pub receiver: Receiver<Event<T>>,
pub end_hash: Sha256Hash,
pub num_hashes: u64,
pub num_ticks: u64,
}
fn log_event<T: Serialize + Clone + Debug>( fn log_event<T: Serialize + Clone + Debug>(
sender: &SyncSender<Entry<T>>, sender: &SyncSender<Entry<T>>,
num_hashes: &mut u64, num_hashes: &mut u64,
@ -108,24 +117,28 @@ pub fn create_logger<T: 'static + Serialize + Clone + Debug + Send>(
) -> JoinHandle<(Entry<T>, ExitReason)> { ) -> JoinHandle<(Entry<T>, ExitReason)> {
use std::thread; use std::thread;
thread::spawn(move || { thread::spawn(move || {
let mut end_hash = start_hash; let mut logger = Logger {
let mut num_hashes = 0; receiver: receiver,
let mut num_ticks = 0; sender: sender,
end_hash: start_hash,
num_hashes: 0,
num_ticks: 0,
};
let epoch = SystemTime::now(); let epoch = SystemTime::now();
loop { loop {
if let Err(err) = log_events( if let Err(err) = log_events(
&receiver, &logger.receiver,
&sender, &logger.sender,
&mut num_hashes, &mut logger.num_hashes,
&mut end_hash, &mut logger.end_hash,
epoch, epoch,
&mut num_ticks, &mut logger.num_ticks,
ms_per_tick, ms_per_tick,
) { ) {
return err; return err;
} }
end_hash = hash(&end_hash); logger.end_hash = hash(&logger.end_hash);
num_hashes += 1; logger.num_hashes += 1;
} }
}) })
} }