1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13

use crate::prelude::Arc;
use arrow::array::{Array, ArrayRef, UInt32Array, Utf8Array};
use arrow::buffer::Buffer;
use arrow::datatypes::DataType;

pub(crate) fn string_lengths(array: &Utf8Array<i64>) -> ArrayRef {
    let values = array.offsets().windows(2).map(|x| (x[1] - x[0]) as u32);

    let values = Buffer::from_trusted_len_iter(values);

    let array = UInt32Array::from_data(DataType::UInt32, values, array.validity().clone());
    Arc::new(array)
}