Add len() to InMemoryWriteableCursor (#564)

* Add len() to InMemoryWriteableCursor * Add is_empty to make clippy happy
apache · Jul 20, 2021 · 6c82bdc · 6c82bdc
1 parent 03269aa
commit 6c82bdc
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/parquet/src/util/cursor.rs b/parquet/src/util/cursor.rs
@@ -154,6 +154,18 @@ impl InMemoryWriteableCursor {
         let inner = self.buffer.lock().unwrap();
         inner.get_ref().to_vec()
     }
+
+    /// Returns a length of the underlying buffer
+    pub fn len(&self) -> usize {
+        let inner = self.buffer.lock().unwrap();
+        inner.get_ref().len()
+    }
+
+    /// Returns true if the underlying buffer contains no elements
+    pub fn is_empty(&self) -> bool {
+        let inner = self.buffer.lock().unwrap();
+        inner.get_ref().is_empty()
+    }
 }
 
 impl TryClone for InMemoryWriteableCursor {