mirror of
https://github.com/facebook/rocksdb.git
synced 2024-11-27 02:44:18 +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
116 lines
4.6 KiB
C++
116 lines
4.6 KiB
C++
// Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
|
|
// Copyright (c) 2011 The LevelDB Authors. All rights reserved.
|
|
// Use of this source code is governed by a BSD-style license that can be
|
|
// found in the LICENSE file. See the AUTHORS file for names of contributors.
|
|
|
|
#include "utilities/merge_operators.h"
|
|
|
|
#include <memory>
|
|
|
|
#include "rocksdb/merge_operator.h"
|
|
#include "rocksdb/options.h"
|
|
#include "rocksdb/utilities/customizable_util.h"
|
|
#include "rocksdb/utilities/object_registry.h"
|
|
#include "utilities/merge_operators/bytesxor.h"
|
|
#include "utilities/merge_operators/max_operator.h"
|
|
#include "utilities/merge_operators/put_operator.h"
|
|
#include "utilities/merge_operators/sortlist.h"
|
|
#include "utilities/merge_operators/string_append/stringappend.h"
|
|
#include "utilities/merge_operators/string_append/stringappend2.h"
|
|
#include "utilities/merge_operators/uint64add.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
static int RegisterBuiltinMergeOperators(ObjectLibrary& library,
|
|
const std::string& /*arg*/) {
|
|
size_t num_types;
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(StringAppendOperator::kClassName())
|
|
.AnotherName(StringAppendOperator::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new StringAppendOperator(","));
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(StringAppendTESTOperator::kClassName())
|
|
.AnotherName(StringAppendTESTOperator::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new StringAppendTESTOperator(","));
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(SortList::kClassName())
|
|
.AnotherName(SortList::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new SortList());
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(BytesXOROperator::kClassName())
|
|
.AnotherName(BytesXOROperator::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new BytesXOROperator());
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(UInt64AddOperator::kClassName())
|
|
.AnotherName(UInt64AddOperator::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new UInt64AddOperator());
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(MaxOperator::kClassName())
|
|
.AnotherName(MaxOperator::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new MaxOperator());
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(PutOperatorV2::kClassName())
|
|
.AnotherName(PutOperatorV2::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new PutOperatorV2());
|
|
return guard->get();
|
|
});
|
|
library.AddFactory<MergeOperator>(
|
|
ObjectLibrary::PatternEntry(PutOperator::kNickName()),
|
|
[](const std::string& /*uri*/, std::unique_ptr<MergeOperator>* guard,
|
|
std::string* /*errmsg*/) {
|
|
guard->reset(new PutOperator());
|
|
return guard->get();
|
|
});
|
|
|
|
return static_cast<int>(library.GetFactoryCount(&num_types));
|
|
}
|
|
|
|
Status MergeOperator::CreateFromString(const ConfigOptions& config_options,
|
|
const std::string& value,
|
|
std::shared_ptr<MergeOperator>* result) {
|
|
static std::once_flag once;
|
|
std::call_once(once, [&]() {
|
|
RegisterBuiltinMergeOperators(*(ObjectLibrary::Default().get()), "");
|
|
});
|
|
return LoadSharedObject<MergeOperator>(config_options, value, result);
|
|
}
|
|
|
|
std::shared_ptr<MergeOperator> MergeOperators::CreateFromStringId(
|
|
const std::string& id) {
|
|
std::shared_ptr<MergeOperator> result;
|
|
Status s = MergeOperator::CreateFromString(ConfigOptions(), id, &result);
|
|
if (s.ok()) {
|
|
return result;
|
|
} else {
|
|
// Empty or unknown, just return nullptr
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
} // namespace ROCKSDB_NAMESPACE
|