Function criterion_benchmark

datafusion/functions/benches/initcap.rs:69–206

(c: &mut Criterion)

}

fn criterion_benchmark(c: &mut Criterion) {
    let initcap = unicode::initcap();
    let config_options = Arc::new(ConfigOptions::default());

    // Array benchmarks: vary both row count and string length
    for size in [1024, 4096, 8192] {
        for str_len in [16, 128] {
            let mut group =
                c.benchmark_group(format!("initcap size={size} str_len={str_len}"));
            group.sampling_mode(SamplingMode::Flat);
            group.sample_size(10);
            group.measurement_time(Duration::from_secs(10));

            // Utf8
            let array_args = create_args::<i32>(size, str_len, false);
            let array_arg_fields = vec![Field::new("arg_0", DataType::Utf8, true).into()];

            group.bench_function("array_utf8", |b| {
                b.iter(|| {
                    black_box(initcap.invoke_with_args(ScalarFunctionArgs {
                        args: array_args.clone(),
                        arg_fields: array_arg_fields.clone(),
                        number_rows: size,
                        return_field: Field::new("f", DataType::Utf8, true).into(),
                        config_options: Arc::clone(&config_options),
                    }))
                })
            });

            // Utf8View
            let array_view_args = create_args::<i32>(size, str_len, true);
            let array_view_arg_fields =
                vec![Field::new("arg_0", DataType::Utf8View, true).into()];

            group.bench_function("array_utf8view", |b| {
                b.iter(|| {
                    black_box(initcap.invoke_with_args(ScalarFunctionArgs {
                        args: array_view_args.clone(),
                        arg_fields: array_view_arg_fields.clone(),
                        number_rows: size,
                        return_field: Field::new("f", DataType::Utf8View, true).into(),
                        config_options: Arc::clone(&config_options),
                    }))
                })
            });

            group.finish();
        }
    }

    // Unicode array benchmarks
    for size in [1024, 4096, 8192] {
        let mut group = c.benchmark_group(format!("initcap unicode size={size}"));
        group.sampling_mode(SamplingMode::Flat);
        group.sample_size(10);
        group.measurement_time(Duration::from_secs(10));

        let unicode_args = create_unicode_utf8_args(size);