vortex_array/array/
visitor.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
use vortex_buffer::Buffer;
use vortex_error::VortexResult;

use crate::validity::Validity;
use crate::ArrayData;

pub trait AcceptArrayVisitor {
    fn accept(&self, visitor: &mut dyn ArrayVisitor) -> VortexResult<()>;
}

// TODO(ngates): maybe we make this more like the inverse of TryFromParts?
pub trait ArrayVisitor {
    /// Visit a child of this array.
    fn visit_child(&mut self, _name: &str, _array: &ArrayData) -> VortexResult<()> {
        Ok(())
    }

    /// Utility for visiting Array validity.
    fn visit_validity(&mut self, validity: &Validity) -> VortexResult<()> {
        if let Some(v) = validity.as_array() {
            self.visit_child("validity", v)
        } else {
            Ok(())
        }
    }

    fn visit_buffer(&mut self, _buffer: &Buffer) -> VortexResult<()> {
        Ok(())
    }
}