7fd68b7c39
Summary: Pull Request resolved: https://github.com/facebook/rocksdb/pull/8638 Reviewed By: zhichao-cao Differential Revision: D31024729 Pulled By: mrambacher fbshipit-source-id: 954c04ccab0b8dee64050a27aadf78ed119106c0
91 lines
3.1 KiB
C++
91 lines
3.1 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 = {
|
|
#ifndef ROCKSDB_LITE
|
|
{"length",
|
|
{0, OptionType::kSizeT, OptionVerificationType::kNormal,
|
|
OptionTypeFlags::kNone}},
|
|
#endif // ROCKSDB_LITE
|
|
};
|
|
|
|
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);
|
|
}
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
namespace {
|
|
static int RegisterSstPartitionerFactories(ObjectLibrary& library,
|
|
const std::string& /*arg*/) {
|
|
library.Register<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
|
|
#endif // ROCKSDB_LITE
|
|
|
|
Status SstPartitionerFactory::CreateFromString(
|
|
const ConfigOptions& options, const std::string& value,
|
|
std::shared_ptr<SstPartitionerFactory>* result) {
|
|
#ifndef ROCKSDB_LITE
|
|
static std::once_flag once;
|
|
std::call_once(once, [&]() {
|
|
RegisterSstPartitionerFactories(*(ObjectLibrary::Default().get()), "");
|
|
});
|
|
#endif // ROCKSDB_LITE
|
|
return LoadSharedObject<SstPartitionerFactory>(options, value, nullptr,
|
|
result);
|
|
}
|
|
} // namespace ROCKSDB_NAMESPACE
|