Switch to sync_channel to preserve order

This commit is contained in:
Greg Fitzgerald 2018-02-28 19:33:28 -07:00
parent 3fcc2dd944
commit 4610de8fdd
7 changed files with 39 additions and 40 deletions

View File

@ -1,7 +1,7 @@
[package] [package]
name = "silk" name = "silk"
description = "A silky smooth implementation of the Loom architecture" description = "A silky smooth implementation of the Loom architecture"
version = "0.2.3" version = "0.3.0"
documentation = "https://docs.rs/silk" documentation = "https://docs.rs/silk"
homepage = "http://loomprotocol.com/" homepage = "http://loomprotocol.com/"
repository = "https://github.com/loomprotocol/silk" repository = "https://github.com/loomprotocol/silk"

View File

@ -25,7 +25,6 @@ impl Accountant {
} }
pub fn process_event(self: &mut Self, event: &Event<u64>) { pub fn process_event(self: &mut Self, event: &Event<u64>) {
println!("accountant: Processing event: {:?}", event);
match *event { match *event {
Event::Claim { key, data, .. } => { Event::Claim { key, data, .. } => {
if self.balances.contains_key(&key) { if self.balances.contains_key(&key) {
@ -55,7 +54,6 @@ impl Accountant {
pub fn sync(self: &mut Self) { pub fn sync(self: &mut Self) {
let mut entries = vec![]; let mut entries = vec![];
while let Ok(entry) = self.historian.receiver.try_recv() { while let Ok(entry) = self.historian.receiver.try_recv() {
println!("accountant: got event {:?}", entry.event);
entries.push(entry); entries.push(entry);
} }
// TODO: Does this cause the historian's channel to get blocked? // TODO: Does this cause the historian's channel to get blocked?
@ -99,20 +97,17 @@ impl Accountant {
data: u64, data: u64,
sig: Signature, sig: Signature,
) -> Result<(), SendError<Event<u64>>> { ) -> Result<(), SendError<Event<u64>>> {
println!("accountant: Checking funds (needs sync)...");
if self.get_balance(&from).unwrap() < data { if self.get_balance(&from).unwrap() < data {
// TODO: Replace the SendError result with a custom one. // TODO: Replace the SendError result with a custom one.
println!("Error: Insufficient funds"); println!("Error: Insufficient funds");
return Ok(()); return Ok(());
} }
println!("accountant: Sufficient funds.");
let event = Event::Transaction { let event = Event::Transaction {
from, from,
to, to,
data, data,
sig, sig,
}; };
println!("accountant: Sending Transaction to historian.");
self.historian.sender.send(event) self.historian.sender.send(event)
} }
@ -130,9 +125,7 @@ impl Accountant {
} }
pub fn get_balance(self: &mut Self, pubkey: &PublicKey) -> Result<u64, RecvError> { pub fn get_balance(self: &mut Self, pubkey: &PublicKey) -> Result<u64, RecvError> {
println!("accountant: syncing the log...");
self.sync(); self.sync();
println!("accountant: done syncing.");
Ok(*self.balances.get(pubkey).unwrap_or(&0)) Ok(*self.balances.get(pubkey).unwrap_or(&0))
} }
} }

View File

@ -42,9 +42,7 @@ impl AccountantSkel {
None None
} }
Request::Transfer { from, to, val, sig } => { Request::Transfer { from, to, val, sig } => {
println!("skel: Invoking transfer_signed...");
let _ = self.obj.transfer_signed(from, to, val, sig); let _ = self.obj.transfer_signed(from, to, val, sig);
println!("skel: transfer_signed done.");
None None
} }
Request::GetBalance { key } => { Request::GetBalance { key } => {
@ -62,21 +60,16 @@ impl AccountantSkel {
let listener = TcpListener::bind(addr)?; let listener = TcpListener::bind(addr)?;
let mut buf = vec![0u8; 1024]; let mut buf = vec![0u8; 1024];
loop { loop {
println!("skel: Waiting for incoming connections..."); //println!("skel: Waiting for incoming connections...");
let (mut stream, addr) = listener.accept()?; let (mut stream, _from_addr) = listener.accept()?;
println!("skel: Accepted incoming connection frm {:?}.", addr);
let _sz = stream.read(&mut buf)?; let _sz = stream.read(&mut buf)?;
// TODO: Return a descriptive error message if deserialization fails. // TODO: Return a descriptive error message if deserialization fails.
let req = deserialize(&buf).expect("deserialize request"); let req = deserialize(&buf).expect("deserialize request");
println!("skel: Got request {:?}", req);
println!("skel: Processing request...");
if let Some(resp) = self.process_request(req) { if let Some(resp) = self.process_request(req) {
println!("skel: Writing response...");
stream.write(&serialize(&resp).expect("serialize response"))?; stream.write(&serialize(&resp).expect("serialize response"))?;
} }
println!("skel: Done processing request.");
} }
} }
} }

View File

@ -49,13 +49,8 @@ impl AccountantStub {
) -> io::Result<usize> { ) -> io::Result<usize> {
let req = Request::Transfer { from, to, val, sig }; let req = Request::Transfer { from, to, val, sig };
let data = serialize(&req).unwrap(); let data = serialize(&req).unwrap();
println!("TcpStream::connect()...");
let mut stream = TcpStream::connect(&self.addr)?; let mut stream = TcpStream::connect(&self.addr)?;
println!("Connected."); stream.write(&data)
println!("accountant_stub: Writing transfer message...");
let ret = stream.write(&data);
println!("Done.");
ret
} }
pub fn transfer( pub fn transfer(

View File

@ -10,18 +10,36 @@ fn main() {
let mut acc = AccountantStub::new(addr); let mut acc = AccountantStub::new(addr);
let alice_keypair = generate_keypair(); let alice_keypair = generate_keypair();
let bob_keypair = generate_keypair(); let bob_keypair = generate_keypair();
println!("Depositing..."); let txs = 10_000;
acc.deposit(10_000, &alice_keypair).unwrap(); println!("Depositing {} units in Alice's account...", txs);
acc.deposit(1_000, &bob_keypair).unwrap(); acc.deposit(txs, &alice_keypair).unwrap();
//acc.deposit(1_000, &bob_keypair).unwrap();
println!("Done."); println!("Done.");
sleep(Duration::from_millis(30)); sleep(Duration::from_millis(30));
let alice_pubkey = get_pubkey(&alice_keypair);
let bob_pubkey = get_pubkey(&bob_keypair); let bob_pubkey = get_pubkey(&bob_keypair);
println!("Transferring..."); println!("Transferring 1 unit {} times...", txs);
acc.transfer(500, &alice_keypair, bob_pubkey).unwrap(); for _ in 0..txs {
acc.transfer(1, &alice_keypair, bob_pubkey).unwrap();
}
println!("Done."); println!("Done.");
sleep(Duration::from_millis(30)); sleep(Duration::from_millis(20));
println!("Done. Checking balance."); let mut alice_val = acc.get_balance(&alice_pubkey).unwrap();
println!("Balance {}", acc.get_balance(&bob_pubkey).unwrap()); while alice_val > 0 {
println!("Checking on Alice's Balance {}", alice_val);
sleep(Duration::from_millis(20));
alice_val = acc.get_balance(&alice_pubkey).unwrap();
}
println!("Done. Checking balances.");
println!(
"Alice's Final Balance {}",
acc.get_balance(&alice_pubkey).unwrap()
);
println!(
"Bob's Final Balance {}",
acc.get_balance(&bob_pubkey).unwrap()
);
} }

View File

@ -9,5 +9,6 @@ fn main() {
let zero = Sha256Hash::default(); let zero = Sha256Hash::default();
let acc = Accountant::new(&zero, Some(1000)); let acc = Accountant::new(&zero, Some(1000));
let mut skel = AccountantSkel::new(acc); let mut skel = AccountantSkel::new(acc);
println!("Listening on {}", addr);
skel.serve(addr).unwrap(); skel.serve(addr).unwrap();
} }

View File

@ -6,14 +6,14 @@
//! The resulting stream of entries represents ordered events in time. //! The resulting stream of entries represents ordered events in time.
use std::thread::JoinHandle; use std::thread::JoinHandle;
use std::sync::mpsc::{Receiver, Sender}; use std::sync::mpsc::{Receiver, SyncSender};
use std::time::{Duration, SystemTime}; use std::time::{Duration, SystemTime};
use log::{hash, hash_event, verify_event, Entry, Event, Sha256Hash}; use log::{hash, hash_event, verify_event, Entry, Event, Sha256Hash};
use serde::Serialize; use serde::Serialize;
use std::fmt::Debug; use std::fmt::Debug;
pub struct Historian<T> { pub struct Historian<T> {
pub sender: Sender<Event<T>>, pub sender: SyncSender<Event<T>>,
pub receiver: Receiver<Entry<T>>, pub receiver: Receiver<Entry<T>>,
pub thread_hdl: JoinHandle<(Entry<T>, ExitReason)>, pub thread_hdl: JoinHandle<(Entry<T>, ExitReason)>,
} }
@ -24,13 +24,12 @@ pub enum ExitReason {
SendDisconnected, SendDisconnected,
} }
fn log_event<T: Serialize + Clone + Debug>( fn log_event<T: Serialize + Clone + Debug>(
sender: &Sender<Entry<T>>, sender: &SyncSender<Entry<T>>,
num_hashes: &mut u64, num_hashes: &mut u64,
end_hash: &mut Sha256Hash, end_hash: &mut Sha256Hash,
event: Event<T>, event: Event<T>,
) -> Result<(), (Entry<T>, ExitReason)> { ) -> Result<(), (Entry<T>, ExitReason)> {
*end_hash = hash_event(end_hash, &event); *end_hash = hash_event(end_hash, &event);
println!("historian: logging event {:?}", event);
let entry = Entry { let entry = Entry {
end_hash: *end_hash, end_hash: *end_hash,
num_hashes: *num_hashes, num_hashes: *num_hashes,
@ -45,7 +44,7 @@ fn log_event<T: Serialize + Clone + Debug>(
fn log_events<T: Serialize + Clone + Debug>( fn log_events<T: Serialize + Clone + Debug>(
receiver: &Receiver<Event<T>>, receiver: &Receiver<Event<T>>,
sender: &Sender<Entry<T>>, sender: &SyncSender<Entry<T>>,
num_hashes: &mut u64, num_hashes: &mut u64,
end_hash: &mut Sha256Hash, end_hash: &mut Sha256Hash,
epoch: SystemTime, epoch: SystemTime,
@ -88,7 +87,7 @@ pub fn create_logger<T: 'static + Serialize + Clone + Debug + Send>(
start_hash: Sha256Hash, start_hash: Sha256Hash,
ms_per_tick: Option<u64>, ms_per_tick: Option<u64>,
receiver: Receiver<Event<T>>, receiver: Receiver<Event<T>>,
sender: Sender<Entry<T>>, sender: SyncSender<Entry<T>>,
) -> JoinHandle<(Entry<T>, ExitReason)> { ) -> JoinHandle<(Entry<T>, ExitReason)> {
use std::thread; use std::thread;
thread::spawn(move || { thread::spawn(move || {
@ -116,9 +115,9 @@ pub fn create_logger<T: 'static + Serialize + Clone + Debug + Send>(
impl<T: 'static + Serialize + Clone + Debug + Send> Historian<T> { impl<T: 'static + Serialize + Clone + Debug + Send> Historian<T> {
pub fn new(start_hash: &Sha256Hash, ms_per_tick: Option<u64>) -> Self { pub fn new(start_hash: &Sha256Hash, ms_per_tick: Option<u64>) -> Self {
use std::sync::mpsc::channel; use std::sync::mpsc::sync_channel;
let (sender, event_receiver) = channel(); let (sender, event_receiver) = sync_channel(4000);
let (entry_sender, receiver) = channel(); let (entry_sender, receiver) = sync_channel(4000);
let thread_hdl = create_logger(*start_hash, ms_per_tick, event_receiver, entry_sender); let thread_hdl = create_logger(*start_hash, ms_per_tick, event_receiver, entry_sender);
Historian { Historian {
sender, sender,