RocksDB入门指南：从零开始使用高性能键值存储引擎

原创于 2025-06-01 09:01:14 发布 · 438 阅读

CC 4.0 BY-SA版权

RocksDB入门指南：从零开始使用高性能键值存储引擎

rocksdb RocksDB 是一个嵌入式的、持久的键值存储库，由 Facebook 开发，基于 LevelDB。* 提供高性能的键值存储；支持快照；支持事务；支持自定义合并操作。* 特点：高性能；支持多种编程语言；支持多种操作系统；支持压缩。项目地址: https://gitcode.com/gh_mirrors/ro/rocksdb

什么是RocksDB？

RocksDB是一个由Facebook数据库工程团队维护的高性能持久化键值存储库。它基于Google的LevelDB构建，但进行了大量优化和改进。RocksDB特别适合需要快速存储和检索大量数据的应用场景，如数据库存储引擎、缓存系统等。

核心特性

RocksDB具有以下几个显著特点：

键值存储：支持任意字节数组作为键和值
有序存储：键按照用户指定的比较函数排序
高性能：针对SSD存储进行了优化
持久化：所有数据都持久保存在磁盘上

安装与配置

在开始使用RocksDB前，需要确保系统已安装必要的依赖项。RocksDB支持多种构建方式，可以通过包管理器或从源代码构建。

基本操作指南

1. 打开数据库

RocksDB数据库对应文件系统中的一个目录，所有数据库内容都存储在该目录下。以下是打开数据库的基本示例：

#include <cassert>
#include "rocksdb/db.h"

rocksdb::DB* db;
rocksdb::Options options;
options.create_if_missing = true;  // 如果数据库不存在则创建
rocksdb::Status status = rocksdb::DB::Open(options, "/tmp/testdb", &db);
assert(status.ok());

如果希望在数据库已存在时报错，可以设置：

options.error_if_exists = true;

2. 状态处理

RocksDB中大多数可能出错的操作都会返回rocksdb::Status对象，用于检查操作是否成功：

rocksdb::Status s = ...;
if (!s.ok()) {
    std::cerr << "操作失败: " << s.ToString() << std::endl;
    // 错误处理逻辑
}

3. 关闭数据库

使用完数据库后，只需删除数据库对象即可关闭：

delete db;  // 关闭并清理数据库资源

4. 基本读写操作

RocksDB提供了Put、Delete和Get方法来修改和查询数据库：

// 写入数据
rocksdb::Status s = db->Put(rocksdb::WriteOptions(), "key1", "value1");
assert(s.ok());

// 读取数据
std::string value;
s = db->Get(rocksdb::ReadOptions(), "key1", &value);
assert(s.ok());
std::cout << "key1的值是: " << value << std::endl;

// 删除数据
s = db->Delete(rocksdb::WriteOptions(), "key1");
assert(s.ok());

5. 数据迁移示例

以下代码展示了如何将键值从一个键迁移到另一个键：

std::string value;
rocksdb::Status s = db->Get(rocksdb::ReadOptions(), "source_key", &value);
if (s.ok()) {
    s = db->Put(rocksdb::WriteOptions(), "dest_key", value);
}
if (s.ok()) {
    s = db->Delete(rocksdb::WriteOptions(), "source_key");
}
if (!s.ok()) {
    std::cerr << "数据迁移失败: " << s.ToString() << std::endl;
}