55 lines
2.1 KiB
Rust
55 lines
2.1 KiB
Rust
// Licensed to the Apache Software Foundation (ASF) under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing,
|
|
// software distributed under the License is distributed on an
|
|
// "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
// KIND, either express or implied. See the License for the
|
|
// specific language governing permissions and limitations
|
|
// under the License.
|
|
|
|
use arrow_array::builder::{Decimal128Builder, Decimal256Builder};
|
|
use arrow_buffer::i256;
|
|
use criterion::*;
|
|
use std::hint;
|
|
|
|
fn criterion_benchmark(c: &mut Criterion) {
|
|
let len = 8192;
|
|
let mut builder_128 = Decimal128Builder::with_capacity(len);
|
|
let mut builder_256 = Decimal256Builder::with_capacity(len);
|
|
for i in 0..len {
|
|
if i % 10 == 0 {
|
|
builder_128.append_value(i128::MAX);
|
|
builder_256.append_value(i256::from_i128(i128::MAX));
|
|
} else {
|
|
builder_128.append_value(i as i128);
|
|
builder_256.append_value(i256::from_i128(i as i128));
|
|
}
|
|
}
|
|
let array_128 = builder_128.finish();
|
|
let array_256 = builder_256.finish();
|
|
|
|
c.bench_function("validate_decimal_precision_128", |b| {
|
|
b.iter(|| hint::black_box(array_128.validate_decimal_precision(8)));
|
|
});
|
|
c.bench_function("null_if_overflow_precision_128", |b| {
|
|
b.iter(|| hint::black_box(array_128.null_if_overflow_precision(8)));
|
|
});
|
|
c.bench_function("validate_decimal_precision_256", |b| {
|
|
b.iter(|| hint::black_box(array_256.validate_decimal_precision(8)));
|
|
});
|
|
c.bench_function("null_if_overflow_precision_256", |b| {
|
|
b.iter(|| hint::black_box(array_256.null_if_overflow_precision(8)));
|
|
});
|
|
}
|
|
|
|
criterion_group!(benches, criterion_benchmark);
|
|
criterion_main!(benches);
|