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
66 lines
2.4 KiB
C++
66 lines
2.4 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 <mutex>
|
|
|
|
#include "rocksdb/convenience.h"
|
|
#include "rocksdb/table.h"
|
|
#include "rocksdb/utilities/customizable_util.h"
|
|
#include "rocksdb/utilities/object_registry.h"
|
|
#include "table/block_based/block_based_table_factory.h"
|
|
#include "table/cuckoo/cuckoo_table_factory.h"
|
|
#include "table/plain/plain_table_factory.h"
|
|
|
|
namespace ROCKSDB_NAMESPACE {
|
|
|
|
static void RegisterTableFactories(const std::string& /*arg*/) {
|
|
#ifndef ROCKSDB_LITE
|
|
static std::once_flag loaded;
|
|
std::call_once(loaded, []() {
|
|
auto library = ObjectLibrary::Default();
|
|
library->Register<TableFactory>(
|
|
TableFactory::kBlockBasedTableName(),
|
|
[](const std::string& /*uri*/, std::unique_ptr<TableFactory>* guard,
|
|
std::string* /* errmsg */) {
|
|
guard->reset(new BlockBasedTableFactory());
|
|
return guard->get();
|
|
});
|
|
library->Register<TableFactory>(
|
|
TableFactory::kPlainTableName(),
|
|
[](const std::string& /*uri*/, std::unique_ptr<TableFactory>* guard,
|
|
std::string* /* errmsg */) {
|
|
guard->reset(new PlainTableFactory());
|
|
return guard->get();
|
|
});
|
|
library->Register<TableFactory>(
|
|
TableFactory::kCuckooTableName(),
|
|
[](const std::string& /*uri*/, std::unique_ptr<TableFactory>* guard,
|
|
std::string* /* errmsg */) {
|
|
guard->reset(new CuckooTableFactory());
|
|
return guard->get();
|
|
});
|
|
});
|
|
#endif // ROCKSDB_LITE
|
|
}
|
|
|
|
static bool LoadFactory(const std::string& name,
|
|
std::shared_ptr<TableFactory>* factory) {
|
|
if (name == TableFactory::kBlockBasedTableName()) {
|
|
factory->reset(new BlockBasedTableFactory());
|
|
return true;
|
|
} else {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
Status TableFactory::CreateFromString(const ConfigOptions& config_options,
|
|
const std::string& value,
|
|
std::shared_ptr<TableFactory>* factory) {
|
|
RegisterTableFactories("");
|
|
return LoadSharedObject<TableFactory>(config_options, value, LoadFactory,
|
|
factory);
|
|
}
|
|
} // namespace ROCKSDB_NAMESPACE
|