feat: add get_rows

insolvent-capital · May 6, 2024 · b3eecc7 · b3eecc7
1 parent 121d52c
commit b3eecc7
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/src/worktable.rs b/src/worktable.rs
@@ -26,7 +26,7 @@ pub enum Value {
 impl Eq for Value {}
 
 impl PartialOrd for Value {
-    fn partial_cmp(&self, other: &Self) -> Option<std::cmp::Ordering> {
+    fn partial_cmp(&self, other: &Self) -> Option<Ordering> {
         Some(self.cmp(other))
     }
 }
@@ -346,14 +346,14 @@ impl WorkTable {
                 .collect(),
         )
     }
-    pub fn find_row_index(&self, mut check: impl FnMut(&RowView) -> bool) -> Option<usize> {
+    pub fn find_row_index(&self, mut check: impl FnMut(RowView) -> bool) -> Option<usize> {
         for i in 0..self.shape().0 {
             let row = RowView {
                 index: i,
                 column_names: &self.columns,
                 columns: &self.column_values,
             };
-            if check(&row) {
+            if check(row) {
                 return Some(i);
             }
         }
@@ -451,6 +451,18 @@ impl WorkTable {
             *col = new_columns;
         }
     }
+    pub fn get_rows(&self) -> impl Iterator<Item = RowView> {
+        let len = self.shape().0;
+        (0..len)
+            .map(|index| {
+                RowView {
+                    index,
+                    column_names: &self.columns,
+                    columns: &self.column_values,
+                }
+            })
+
+    }
 }
 pub type SyncWorkTable = Arc<RwLock<WorkTable>>;