mirror of https://github.com/facebook/rocksdb.git
52 lines
1.4 KiB
C++
52 lines
1.4 KiB
C++
// Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
// This source code is licensed under both the GPLv2 (found in the
|
|
// COPYING file in the root directory) and Apache 2.0 License
|
|
// (found in the LICENSE.Apache file in the root directory).
|
|
|
|
#ifdef GFLAGS
|
|
|
|
#include "db_stress_tool/db_stress_wide_merge_operator.h"
|
|
|
|
#include "db_stress_tool/db_stress_common.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
bool DBStressWideMergeOperator::FullMergeV3(
|
|
const MergeOperationInputV3& merge_in,
|
|
MergeOperationOutputV3* merge_out) const {
|
|
assert(!merge_in.operand_list.empty());
|
|
assert(merge_out);
|
|
|
|
const Slice& latest = merge_in.operand_list.back();
|
|
|
|
if (latest.size() < sizeof(uint32_t)) {
|
|
return false;
|
|
}
|
|
|
|
const uint32_t value_base = GetValueBase(latest);
|
|
|
|
if (FLAGS_use_put_entity_one_in == 0 ||
|
|
(value_base % FLAGS_use_put_entity_one_in) != 0) {
|
|
merge_out->new_value = latest;
|
|
return true;
|
|
}
|
|
|
|
const auto columns = GenerateWideColumns(value_base, latest);
|
|
|
|
merge_out->new_value = MergeOperationOutputV3::NewColumns();
|
|
auto& new_columns =
|
|
std::get<MergeOperationOutputV3::NewColumns>(merge_out->new_value);
|
|
new_columns.reserve(columns.size());
|
|
|
|
for (const auto& column : columns) {
|
|
new_columns.emplace_back(column.name().ToString(),
|
|
column.value().ToString());
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|
|
|
|
#endif // GFLAGS
|