89f66d4484
Summary: Useful for allowing new classes to create and manage Customizable objects without using internal APIs. Pull Request resolved: https://github.com/facebook/rocksdb/pull/8301 Reviewed By: zhichao-cao Differential Revision: D29428303 Pulled By: mrambacher fbshipit-source-id: 3d33d5197cc8379fe35b54d3d169f91f0dfe7a47
108 lines
3.6 KiB
C++
108 lines
3.6 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/customizable.h"
|
|
|
|
#include "options/configurable_helper.h"
|
|
#include "options/options_helper.h"
|
|
#include "rocksdb/convenience.h"
|
|
#include "rocksdb/status.h"
|
|
#include "util/string_util.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
std::string Customizable::GetOptionName(const std::string& long_name) const {
|
|
const std::string& name = Name();
|
|
size_t name_len = name.size();
|
|
if (long_name.size() > name_len + 1 &&
|
|
long_name.compare(0, name_len, name) == 0 &&
|
|
long_name.at(name_len) == '.') {
|
|
return long_name.substr(name_len + 1);
|
|
} else {
|
|
return Configurable::GetOptionName(long_name);
|
|
}
|
|
}
|
|
|
|
#ifndef ROCKSDB_LITE
|
|
Status Customizable::GetOption(const ConfigOptions& config_options,
|
|
const std::string& opt_name,
|
|
std::string* value) const {
|
|
if (opt_name == ConfigurableHelper::kIdPropName) {
|
|
*value = GetId();
|
|
return Status::OK();
|
|
} else {
|
|
return Configurable::GetOption(config_options, opt_name, value);
|
|
}
|
|
}
|
|
|
|
std::string Customizable::SerializeOptions(const ConfigOptions& config_options,
|
|
const std::string& prefix) const {
|
|
std::string result;
|
|
std::string parent;
|
|
if (!config_options.IsShallow()) {
|
|
parent = Configurable::SerializeOptions(config_options, "");
|
|
}
|
|
if (parent.empty()) {
|
|
result = GetId();
|
|
} else {
|
|
result.append(prefix + ConfigurableHelper::kIdPropName + "=" + GetId() +
|
|
config_options.delimiter);
|
|
result.append(parent);
|
|
}
|
|
return result;
|
|
}
|
|
|
|
#endif // ROCKSDB_LITE
|
|
|
|
bool Customizable::AreEquivalent(const ConfigOptions& config_options,
|
|
const Configurable* other,
|
|
std::string* mismatch) const {
|
|
if (config_options.sanity_level > ConfigOptions::kSanityLevelNone &&
|
|
this != other) {
|
|
const Customizable* custom = reinterpret_cast<const Customizable*>(other);
|
|
if (GetId() != custom->GetId()) {
|
|
*mismatch = ConfigurableHelper::kIdPropName;
|
|
return false;
|
|
} else if (config_options.sanity_level >
|
|
ConfigOptions::kSanityLevelLooselyCompatible) {
|
|
bool matches =
|
|
Configurable::AreEquivalent(config_options, other, mismatch);
|
|
return matches;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
|
|
Status Customizable::GetOptionsMap(
|
|
const ConfigOptions& config_options, const Customizable* customizable,
|
|
const std::string& value, std::string* id,
|
|
std::unordered_map<std::string, std::string>* props) {
|
|
if (customizable != nullptr) {
|
|
Status status = ConfigurableHelper::GetOptionsMap(
|
|
value, customizable->GetId(), id, props);
|
|
#ifdef ROCKSDB_LITE
|
|
(void)config_options;
|
|
#else
|
|
if (status.ok() && customizable->IsInstanceOf(*id)) {
|
|
// The new ID and the old ID match, so the objects are the same type.
|
|
// Try to get the existing options, ignoring any errors
|
|
ConfigOptions embedded = config_options;
|
|
embedded.delimiter = ";";
|
|
std::string curr_opts;
|
|
if (customizable->GetOptionString(embedded, &curr_opts).ok()) {
|
|
std::unordered_map<std::string, std::string> curr_props;
|
|
if (StringToMap(curr_opts, &curr_props).ok()) {
|
|
props->insert(curr_props.begin(), curr_props.end());
|
|
}
|
|
}
|
|
}
|
|
#endif // ROCKSDB_LITE
|
|
return status;
|
|
} else {
|
|
return ConfigurableHelper::GetOptionsMap(value, "", id, props);
|
|
}
|
|
}
|
|
} // namespace ROCKSDB_NAMESPACE
|