|
| 1 | +#include <DataTypes/DataTypeNullable.h> |
| 2 | +#include <DataTypes/DataTypesNumber.h> |
| 3 | +#include <Columns/ColumnNullable.h> |
| 4 | +#include <Functions/FunctionHelpers.h> |
| 5 | +#include <Storages/StorageGenerateRandom.h> |
| 6 | +#include <benchmark/benchmark.h> |
| 7 | + |
| 8 | +using namespace DB; |
| 9 | + |
| 10 | +static void BM_WrapInNullable1(benchmark::State & state) |
| 11 | +{ |
| 12 | + pcg64 rng; |
| 13 | + UInt64 limit = DEFAULT_BLOCK_SIZE; |
| 14 | + UInt64 max_array_length = 10; |
| 15 | + UInt64 max_string_length = 10; |
| 16 | + DataTypePtr nullable_float64 = std::make_shared<DataTypeNullable>(std::make_shared<DataTypeFloat64>()); |
| 17 | + |
| 18 | + ColumnPtr arg_column = fillColumnWithRandomData(nullable_float64, limit, max_array_length, max_string_length, rng, nullptr); |
| 19 | + ColumnsWithTypeAndName args{{arg_column, nullable_float64, "arg"}}; |
| 20 | + |
| 21 | + ColumnPtr denull_src_column |
| 22 | + = fillColumnWithRandomData(removeNullable(nullable_float64), limit, max_array_length, max_string_length, rng, nullptr); |
| 23 | + |
| 24 | + DataTypePtr uint8_type = std::make_shared<DataTypeUInt8>(); |
| 25 | + for (auto _ : state) |
| 26 | + { |
| 27 | + state.PauseTiming(); |
| 28 | + ColumnPtr null_map_column = fillColumnWithRandomData(uint8_type, limit, max_array_length, max_string_length, rng, nullptr); |
| 29 | + ColumnPtr src_column = ColumnNullable::create(denull_src_column, std::move(null_map_column)); |
| 30 | + state.ResumeTiming(); |
| 31 | + |
| 32 | + auto result = wrapInNullable(std::move(src_column), args, nullable_float64, limit); |
| 33 | + benchmark::DoNotOptimize(result); |
| 34 | + } |
| 35 | +} |
| 36 | + |
| 37 | +static void BM_WrapInNullable2(benchmark::State & state) |
| 38 | +{ |
| 39 | + pcg64 rng; |
| 40 | + UInt64 limit = DEFAULT_BLOCK_SIZE; |
| 41 | + UInt64 max_array_length = 10; |
| 42 | + UInt64 max_string_length = 10; |
| 43 | + |
| 44 | + DataTypePtr nullable_float64 = std::make_shared<DataTypeNullable>(std::make_shared<DataTypeFloat64>()); |
| 45 | + ColumnPtr src_column = fillColumnWithRandomData(nullable_float64, limit, max_array_length, max_string_length, rng, nullptr); |
| 46 | + |
| 47 | + DataTypePtr uint8_type = std::make_shared<DataTypeUInt8>(); |
| 48 | + for (auto _ : state) |
| 49 | + { |
| 50 | + state.PauseTiming(); |
| 51 | + ColumnPtr null_map_column = fillColumnWithRandomData(uint8_type, limit, max_array_length, max_string_length, rng, nullptr); |
| 52 | + state.ResumeTiming(); |
| 53 | + |
| 54 | + auto result = wrapInNullable(src_column, std::move(null_map_column)); |
| 55 | + benchmark::DoNotOptimize(result); |
| 56 | + } |
| 57 | +} |
| 58 | + |
| 59 | +BENCHMARK(BM_WrapInNullable1); |
| 60 | +BENCHMARK(BM_WrapInNullable2); |
0 commit comments