Improved example (#1131)

jorgecarleitao · Jun 30, 2022 · c52b3e7 · c52b3e7
1 parent 9b095ea
commit c52b3e7
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/examples/cow.rs b/examples/cow.rs
@@ -19,4 +19,20 @@ fn main() {
 
     // confirm that it gives the right result :)
     assert_eq!(array, &PrimitiveArray::from_vec(vec![10i32, 20]));
+
+    // alternatively, you can use `get_mut_values`. Unwrap works because it is single owned
+    let values = array.get_mut_values().unwrap();
+    values[0] = 0;
+
+    assert_eq!(array, &PrimitiveArray::from_vec(vec![0, 20]));
+
+    // you can also modify the validity:
+    array.set_validity(Some([true, false].into()));
+    array.apply_validity(|bitmap| {
+        let mut mut_bitmap = bitmap.into_mut().right().unwrap();
+        mut_bitmap.set(1, true);
+        mut_bitmap.into()
+    });
+
+    assert_eq!(array, &PrimitiveArray::from_vec(vec![0, 20]));
 }