ToplingDB 使用入门

本文原始链接：101 · topling/rockside Wiki

从 RocksDB 迁移到 ToplingDB，只需要修改 Open/Close 相关的代码，其它代码不用任何改动。

示例代码（省略错误处理，完整版在这里）

#include <topling/side_plugin_repo.h>
#include <rocksdb/db.h>
int main(int argc, char* argv[]) {
  using namespace rocksdb;
  SidePluginRepo repo;
  repo.ImportAutoFile(argv[1]); // argv[1] is json/yaml conf file
  DB* db = nullptr;
  repo.OpenDB(&db);
  repo.StartHttpServer();
  if (strcasecmp(argv[2], "set") == 0) {
    db->Put(WriteOptions(), argv[3], argv[4]);
  }
  else if (strcasecmp(argv[2], "get") == 0) {
    std::string val;
    db->Get(ReadOptions(), argv[3], &val);
    fprintf(stdout, "%s\n", val.c_str());
  }
  fprintf(stderr, "now visit the web(defined in json/yaml conf file)\n");
  fprintf(stderr, "press enter to exit\n");
  getchar(); // wait for enter
  repo.CloseAllDB();
  return 0;
}

1. Open

-----------------------------------

SidePluginRepo 是 SidePlugin 的核心 class，所有组件都被包含在其中，要使用 SidePlugin，就要先定一个 SidePluginRepo 对象：

SidePluginRepo repo;

-----------------------------------
有了 SidePluginRepo 对象，接下来就是导入配置参数，最便捷的方式是调用 ImportAutoFile，该函数根据文件后缀名自动识别 json 和 yaml 文件：

repo.ImportAutoFile(argv[1]); // argv[1] is json/yaml conf file

-----------------------------------

现在，我们可以打开 DB 了：

  DB* db = nullptr;
  repo.OpenDB(&db);

在 RocksDB 中，我们一般使用 DB::Open，少数情况下使用其它 Open 函数，但是在 SidePlugin 中，我们统一用 OpenDB，具体使用的是哪个 Open 函数，是在 json/yaml 中进行配置的：例如 Todis从结点。

在这个例子中我们使用的是默认的，只有一个 ColumnFamily 的 DB，但其实 OpenDB 是个重载的函数，也可以用来 Open 有多个 ColumnFamily 的 DB（后面修改的 db_bench 就是这样的例子）：

  DB_MultiCF* dbmcf = nullptr;
  repo.OpenDB(&dbmcf); // overload function name

2. 启动内嵌的 Http Web Server

理论上，内嵌的 Http Web Server 可以在 Open 中自动启动，但是，我们仍然决定让用户来显式启动：

repo.StartHttpServer();

主要有三个原因：

• 用户可以显式打开同一个 json/yaml 中定义的多个 DB，但内嵌的 Http 只有一个
• 以 RocksDB 的接口风格，错误处理使用返回值 Status，难以使用同一个 Status 来区分，DB Open 与 Http 的错误信息
• 即使 json/yaml 中定义了 http，用户也可能并不想启用内嵌的 Http Web Server

3. Close

需要显式 Close，就一行代码：

repo.CloseAllDB();

---

复杂一点的：db_bench 针对 ToplingDB 的修改

db_bench_tool.cc 是一个 8000 多行的文件，我们让它支持 ToplingDB，只新增了 47 行代码， 无任何其它修改，就拥有了 ToplingDB 的全部能力！以下是 diff 内容：

diff --git a/tools/db_bench_tool.cc b/tools/db_bench_tool.cc
index fcd3c157a..3a1f632a5 100644
--- a/tools/db_bench_tool.cc
+++ b/tools/db_bench_tool.cc
@@ -97,6 +97,8 @@
 #include <io.h>  // open/close
 #endif

+#include "sideplugin/rockside/src/topling/side_plugin_repo.h"
+
 using GFLAGS_NAMESPACE::ParseCommandLineFlags;
 using GFLAGS_NAMESPACE::RegisterFlagValidator;
 using GFLAGS_NAMESPACE::SetUsageMessage;
@@ -1052,6 +1054,7 @@ DEFINE_int32(trace_replay_threads, 1,
 DEFINE_bool(io_uring_enabled, true,
             "If true, enable the use of IO uring if the platform supports it");
 extern "C" bool RocksDbIOUringEnable() { return FLAGS_io_uring_enabled; }
+DEFINE_string(json, "", "json config file.");
 #endif  // ROCKSDB_LITE

 DEFINE_bool(adaptive_readahead, false,
@@ -3087,6 +3090,7 @@ class Benchmark {
   }

   void DeleteDBs() {
+    repo_.CloseAllDB(false);
     db_.DeleteDBs();
     for (const DBWithColumnFamilies& dbwcf : multi_dbs_) {
       delete dbwcf.db;
@@ -3104,6 +3108,11 @@ class Benchmark {
     }
   }

+  void exit(int code) {
+    this->~Benchmark();
+    ::exit(code);
+  }
+
   Slice AllocateKey(std::unique_ptr<const char[]>* key_guard) {
     char* data = new char[key_size_];
     const char* const_data = data;
@@ -3236,6 +3245,7 @@ class Benchmark {
       ErrorExit();
     }
     Open(&open_options_);
+    open_options_ = db_.db->GetOptions();
     PrintHeader(open_options_);
     std::stringstream benchmark_stream(FLAGS_benchmarks);
     std::string name;
@@ -4516,9 +4526,45 @@ class Benchmark {
     InitializeOptionsGeneral(opts);
   }

+  SidePluginRepo repo_;
   void OpenDb(Options options, const std::string& db_name,
       DBWithColumnFamilies* db) {
     uint64_t open_start = FLAGS_report_open_timing ? FLAGS_env->NowNanos() : 0;
+    if (!FLAGS_json.empty()) {
+      repo_.CloseAllDB(false);
+      repo_.CleanResetRepo();
+      DB_MultiCF* dbmcf = nullptr;
+      Status s = repo_.ImportAutoFile(FLAGS_json);
+      if (!s.ok()) {
+        fprintf(stderr, "ERROR: ImportAutoFile(%s): %s\n",
+                FLAGS_json.c_str(), s.ToString().c_str());
+        exit(1);
+      }
+      s = repo_.OpenDB(&dbmcf);
+      if (!s.ok()) {
+        fprintf(stderr, "ERROR: OpenDB(): Config File=%s: %s\n",
+                FLAGS_json.c_str(), s.ToString().c_str());
+        exit(1);
+      }
+      s = repo_.StartHttpServer();
+      if (!s.ok()) {
+        fprintf(stderr, "ERROR: StartHttpServer(): JsonFile=%s: %s\n",
+                FLAGS_json.c_str(), s.ToString().c_str());
+        exit(1);
+      }
+      db->cfh = dbmcf->cf_handles;
+      db->db = dbmcf->db;
+      if (auto tdb = dynamic_cast<OptimisticTransactionDB*>(dbmcf->db)) {
+        db->opt_txn_db = tdb;
+        db->db = tdb->GetBaseDB();
+      }
+      db->num_created = FLAGS_num_column_families;
+      db->num_hot = FLAGS_num_column_families;
+      DBOptions dbo = db->db->GetDBOptions();
+      dbstats = dbo.statistics;
+      FLAGS_db = db->db->GetName();
+      return;
+    }
     Status s;
     // Open with column families if necessary.
     if (FLAGS_num_column_families > 1) {