Function gen_string_array

datafusion/functions/benches/contains.rs:31–53

Generate a StringArray/StringViewArray with random ASCII strings

(
    n_rows: usize,
    str_len: usize,
    is_string_view: bool,
)

/// Generate a StringArray/StringViewArray with random ASCII strings
fn gen_string_array(
    n_rows: usize,
    str_len: usize,
    is_string_view: bool,
) -> ColumnarValue {
    let mut rng = StdRng::seed_from_u64(42);
    let strings: Vec<Option<String>> = (0..n_rows)
        .map(|_| {
            let s: String = (&mut rng)
                .sample_iter(&Alphanumeric)
                .take(str_len)
                .map(char::from)
                .collect();
            Some(s)
        })
        .collect();

    if is_string_view {
        ColumnarValue::Array(Arc::new(StringViewArray::from(strings)))
    } else {
        ColumnarValue::Array(Arc::new(StringArray::from(strings)))
    }
}

/// Generate a scalar search string
fn gen_scalar_search(search_str: &str, is_string_view: bool) -> ColumnarValue {