mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-28 15:33:54 +00:00
b6640c3117
Summary: The primary purpose of the FactoryFunc was to support LITE mode where the ObjectRegistry was not available. With the removal of LITE mode, the function was no longer required. Note that the MergeOperator had some private classes defined in header files. To gain access to their constructors (and name methods), the class definitions were moved into header files. Pull Request resolved: https://github.com/facebook/rocksdb/pull/11203 Reviewed By: cbi42 Differential Revision: D43160255 Pulled By: pdillinger fbshipit-source-id: f3a465fd5d1a7049b73ecf31e4b8c3762f6dae6c
84 lines
2.9 KiB
C++
84 lines
2.9 KiB
C++
// Copyright (c) 2011-present, Facebook, Inc. All rights reserved.
|
|
// 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).
|
|
//
|
|
|
|
#include "rocksdb/sst_partitioner.h"
|
|
|
|
#include <algorithm>
|
|
|
|
#include "rocksdb/utilities/customizable_util.h"
|
|
#include "rocksdb/utilities/object_registry.h"
|
|
#include "rocksdb/utilities/options_type.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
static std::unordered_map<std::string, OptionTypeInfo>
|
|
sst_fixed_prefix_type_info = {
|
|
{"length",
|
|
{0, OptionType::kSizeT, OptionVerificationType::kNormal,
|
|
OptionTypeFlags::kNone}},
|
|
};
|
|
|
|
SstPartitionerFixedPrefixFactory::SstPartitionerFixedPrefixFactory(size_t len)
|
|
: len_(len) {
|
|
RegisterOptions("Length", &len_, &sst_fixed_prefix_type_info);
|
|
}
|
|
|
|
PartitionerResult SstPartitionerFixedPrefix::ShouldPartition(
|
|
const PartitionerRequest& request) {
|
|
Slice last_key_fixed(*request.prev_user_key);
|
|
if (last_key_fixed.size() > len_) {
|
|
last_key_fixed.size_ = len_;
|
|
}
|
|
Slice current_key_fixed(*request.current_user_key);
|
|
if (current_key_fixed.size() > len_) {
|
|
current_key_fixed.size_ = len_;
|
|
}
|
|
return last_key_fixed.compare(current_key_fixed) != 0 ? kRequired
|
|
: kNotRequired;
|
|
}
|
|
|
|
bool SstPartitionerFixedPrefix::CanDoTrivialMove(
|
|
const Slice& smallest_user_key, const Slice& largest_user_key) {
|
|
return ShouldPartition(PartitionerRequest(smallest_user_key, largest_user_key,
|
|
0)) == kNotRequired;
|
|
}
|
|
|
|
std::unique_ptr<SstPartitioner>
|
|
SstPartitionerFixedPrefixFactory::CreatePartitioner(
|
|
const SstPartitioner::Context& /* context */) const {
|
|
return std::unique_ptr<SstPartitioner>(new SstPartitionerFixedPrefix(len_));
|
|
}
|
|
|
|
std::shared_ptr<SstPartitionerFactory> NewSstPartitionerFixedPrefixFactory(
|
|
size_t prefix_len) {
|
|
return std::make_shared<SstPartitionerFixedPrefixFactory>(prefix_len);
|
|
}
|
|
|
|
namespace {
|
|
static int RegisterSstPartitionerFactories(ObjectLibrary& library,
|
|
const std::string& /*arg*/) {
|
|
library.AddFactory<SstPartitionerFactory>(
|
|
SstPartitionerFixedPrefixFactory::kClassName(),
|
|
[](const std::string& /*uri*/,
|
|
std::unique_ptr<SstPartitionerFactory>* guard,
|
|
std::string* /* errmsg */) {
|
|
guard->reset(new SstPartitionerFixedPrefixFactory(0));
|
|
return guard->get();
|
|
});
|
|
return 1;
|
|
}
|
|
} // namespace
|
|
|
|
Status SstPartitionerFactory::CreateFromString(
|
|
const ConfigOptions& options, const std::string& value,
|
|
std::shared_ptr<SstPartitionerFactory>* result) {
|
|
static std::once_flag once;
|
|
std::call_once(once, [&]() {
|
|
RegisterSstPartitionerFactories(*(ObjectLibrary::Default().get()), "");
|
|
});
|
|
return LoadSharedObject<SstPartitionerFactory>(options, value, result);
|
|
}
|
|
} // namespace ROCKSDB_NAMESPACE
|