Make things work.

tests/tolstoy.rs

@@ -45,7 +45,8 @@ impl TxCountingReceiver {
 }
 
 impl TxReceiver for TxCountingReceiver {
-    fn tx(&mut self, tx_id: Entid, d: &mut DatomsIterator) -> Result<()> {
+    fn tx<T>(&mut self, tx_id: Entid, d: &mut T) -> Result<()>
+        where T: Iterator<Item=TxPart> {
         self.tx_count = self.tx_count + 1;
         Ok(())
     }
@@ -72,10 +73,11 @@ impl TestingReceiver {
 }
 
 impl TxReceiver for TestingReceiver {
-    fn tx(&mut self, tx_id: Entid, d: &mut DatomsIterator) -> Result<()> {
+    fn tx<T>(&mut self, tx_id: Entid, d: &mut T) -> Result<()>
+        where T: Iterator<Item=TxPart> {
         let datoms = self.txes.entry(tx_id).or_insert(vec![]);
         for datom in d {
-            datoms.push(datom?);
+            datoms.push(datom);
         }
         Ok(())
     }

tolstoy/src/tx_processor.rs

@@ -43,37 +43,37 @@ pub struct Tx {
 }
 
 pub trait TxReceiver {
-    fn tx(&mut self, tx_id: Entid, d: &mut DatomsIterator) -> Result<()>;
+    fn tx<T>(&mut self, tx_id: Entid, d: &mut T) -> Result<()>
+        where T: Iterator<Item=TxPart>;
     fn done(&mut self) -> Result<()>;
 }
 
 pub struct Processor {}
 
-type DemRows<'s> = Peekable<rusqlite::AndThenRows<'s, Result<TxPart>>>;
+pub struct DatomsIterator<'conn, 't, T>
-
+where T: Sized + Iterator<Item=Result<TxPart>> + 't {
-pub struct DatomsIterator<'conn> {
     at_first: bool,
     at_last: bool,
     first: &'conn TxPart,
-    rows: DemRows<'conn>,
+    rows: &'t mut Peekable<T>,
 }
 
-impl<'conn> DatomsIterator<'conn> {
+impl<'conn, 't, T> DatomsIterator<'conn, 't, T>
-    fn new<I>(first: &'conn TxPart, rows: &'conn mut DemRows<'conn>) -> DatomsIterator<'conn>
+where T: Sized + Iterator<Item=Result<TxPart>> + 't {
-    where
+    fn new(first: &'conn TxPart, rows: &'t mut Peekable<T>) -> DatomsIterator<'conn, 't, T>
-        I: Iterator<Item=Result<TxPart>>
     {
         DatomsIterator {
             at_first: true,
             at_last: false,
             first: first,
-            rows: Box::new(rows),
+            rows: rows,
         }
     }
 }
 
-impl<'conn> Iterator for DatomsIterator<'conn> {
+impl<'conn, 't, T> Iterator for DatomsIterator<'conn, 't, T>
-    type Item = Result<TxPart>;
+where T: Sized + Iterator<Item=Result<TxPart>> + 't {
+    type Item = TxPart;
 
     fn next(&mut self) -> Option<Self::Item> {
         if self.at_last {
@@ -82,7 +82,7 @@ impl<'conn> Iterator for DatomsIterator<'conn> {
 
         if self.at_first {
             self.at_first = false;
-            return Some(Ok(self.first.clone()));
+            return Some(self.first.clone());
         }
 
         // Look ahead to see if we're about to cross into
@@ -90,9 +90,7 @@ impl<'conn> Iterator for DatomsIterator<'conn> {
         {
             let next_option = self.rows.peek();
             match next_option {
-                Some(next_result) => {
+                Some(&Ok(ref next)) => {
-                    match *next_result {
-                        Ok(ref next) => {
                     if next.tx != self.first.tx {
                         self.at_last = true;
                         return None;
@@ -100,41 +98,32 @@ impl<'conn> Iterator for DatomsIterator<'conn> {
                 },
                 _ => ()
             }
-                },
-                _ => ()
-            }
         }
 
         // We're in the correct partition.
-        if let Some(row) = self.rows.next() {
+        if let Some(result) = self.rows.next() {
-            let datom = match row {
+            match result {
-                Ok(r) => r,
+                Err(_) => None,
-                Err(e) => {
+                Ok(datom) => {
-                    self.at_last = true;
+                    Some(TxPart {
-                    return Some(Err(e));
-                }
-            };
-
-            if datom.a == entids::DB_TX_INSTANT && datom.tx == datom.e {
-                self.at_last = true;
-            }
-
-            return Some(Ok(TxPart {
                         e: datom.e,
                         a: datom.a,
                         v: datom.v.clone(),
                         tx: datom.tx,
                         added: datom.added,
-            }));
+                    })
+                },
+            }
         } else {
             self.at_last = true;
-            return None;
+            None
         }
     }
 }
 
 impl Processor {
-    pub fn process(sqlite: &rusqlite::Connection, receiver: &mut TxReceiver) -> Result<()> {
+    pub fn process<R>(sqlite: &rusqlite::Connection, receiver: &mut R) -> Result<()>
+    where R: TxReceiver {
         let mut stmt = sqlite.prepare(
             "SELECT e, a, v, tx, added, value_type_tag FROM transactions ORDER BY tx"
         )?;