2 个版本
0.1.0-surreal.2 | 2023 年 7 月 22 日 |
---|---|
0.1.0-surreal.1 | 2023 年 6 月 19 日 |
#2502 在 数据库接口
92 每月下载量
95KB
665 行
TiKV 客户端 (Rust)
此 crate 为 TiKV(>= v5.0.0
)集群提供一个易于使用的客户端,TiKV 是用 Rust 编写的分布式、事务型键值数据库。
此 crate 允许您连接到 TiKV 集群,并使用事务型或原始(简单 get/put 风格,不带事务一致性保证)API 访问和更新您的数据。
TiKV Rust 客户端是一个开源(Apache 2)项目,由 TiKV 作者维护。我们欢迎贡献,下面有更多信息。
请注意,当前版本不适用于生产使用 - API 尚未稳定,crate 还未在实际使用中得到充分测试。
入门
TiKV 客户端是一个 Rust 库(crate)。要在项目中使用此 crate,请将以下依赖项添加到您的 Cargo.toml
[dependencies]
tikv-client = "0.3"
先决条件
rust
>=1.56.1
,需要hashbrown-v0.12.1
使用客户端库的一般流程是创建一个原始客户端对象或事务客户端对象(可以配置),然后使用客户端对象发送命令,或者用它来创建事务对象。在后一种情况下,事务通过各种命令构建,然后提交(或回滚)。
示例
原始模式
use tikv_client::RawClient;
let client = RawClient::new(vec!["127.0.0.1:2379"], None).await?;
client.put("key".to_owned(), "value".to_owned()).await?;
let value = client.get("key".to_owned()).await?;
事务模式
use tikv_client::TransactionClient;
let txn_client = TransactionClient::new(vec!["127.0.0.1:2379"], None).await?;
let mut txn = txn_client.begin_optimistic().await?;
txn.put("key".to_owned(), "value".to_owned()).await?;
let value = txn.get("key".to_owned()).await?;
txn.commit().await?;
由于 TiKV 客户端提供异步 API,您需要使用异步运行时(我们目前仅支持 Tokio)。有关完整示例,请参阅 getting-started.md。
API 概述
TiKV Rust 客户端支持多个抽象级别。使用客户端的最便捷方式是通过 RawClient
和 TransactionClient
。这提供了一种非常高级的 API,主要抽象了存储的分布式特性,并为所有协议提供了合理的默认值。此接口可以进行配置,主要是在创建客户端或事务对象时通过 Config
和 TransactionOptions
结构体进行。使用一些选项,您可以自己接管部分协议(例如重试失败的消息)。
最低级别的抽象是直接向 TiKV(和 PD)节点发送和接收 gRPC 消息。 tikv-client-store
和 tikv-client-pd
库使这比直接使用 protobuf 定义和 gRPC 库更容易,但提供了相同级别的控制。
在这些抽象级别之间,您可以向 TiKV 集群发送和接收单个消息,同时利用库代码执行常见操作,例如将数据解析到区域并从而到集群中的节点,或重试失败的消息。这对于测试 TiKV 集群或某些高级用例很有用。有关此 API,请参阅 client_rust::request
模块,以及 client_rust::raw::lowering
和 client_rust::transaction::lowering
以创建请求对象的便利方法。
本文件的其余部分仅描述 RawClient
/TransactionClient
API。
重要提示:不建议也不支持在同一个数据库上同时使用原始和事务 API。
类型
Key
:存储中的键。 String
和 Vec<u8>
实现 Into<Key>
,因此您可以直接将它们传递给客户端函数。
Value
:存储中的值;只是 Vec<u8>
的别名。
KvPair
:一个键值对的组合。它提供了方便的方法来转换到和从其他类型。
BoundRange
:用于范围相关的请求,如 scan
。它实现了 Rust 范围的 From
,因此您可以将 Rust 的键范围传递给请求,例如 client.delete_range(vec![]..)
。
原始请求
请求 | 主要参数类型 | 结果类型 | 值得注意的行为 |
---|---|---|---|
put |
KvPair |
||
get |
Key |
选项<值> |
|
删除 |
Key |
||
删除范围 |
范围边界 |
||
扫描 |
范围边界 |
Vec<KvPair> |
|
批量插入 |
迭代器<KvPair> |
||
批量获取 |
迭代器<Key> |
Vec<KvPair> |
跳过不存在的键;不保留顺序 |
批量删除 |
迭代器<Key> |
||
批量扫描 |
迭代器<范围边界> |
Vec<KvPair> |
请参阅关于 each_limit 参数行为的文档。保留范围顺序。 |
批量扫描键 |
迭代器<范围边界> |
Vec<Key> |
请参阅关于 each_limit 参数行为的文档。保留范围顺序。 |
比较并交换 |
Key + 2x Value |
(选项<值>, 布尔值) |
事务请求
请求 | 主要参数类型 | 结果类型 | 值得注意的行为 |
---|---|---|---|
put |
KvPair |
||
get |
Key |
选项<值> |
|
获取更新 |
Key |
选项<值> |
|
键存在 |
Key |
布尔值 |
|
删除 |
Key |
||
扫描 |
范围边界 |
迭代器<KvPair> |
|
扫描键 |
范围边界 |
迭代器<Key> |
|
批量获取 |
迭代器<Key> |
迭代器<KvPair> |
跳过不存在的键;不保留顺序 |
批量获取更新 |
迭代器<Key> |
迭代器<KvPair> |
跳过不存在的键;不保留顺序 |
锁定键 |
迭代器<Key> |
||
发送心跳 |
u64 (TTL) |
||
垃圾回收 |
时间戳 |
布尔值 |
如果PD中最近的safepoint等于参数,则返回true |
开发和贡献
我们欢迎您的贡献!贡献代码很棒,我们也感谢您提交问题以识别错误并提供反馈,添加测试或示例,以及改进文档。
构建和测试
我们使用标准的Cargo工作流程,例如,使用cargo build
进行构建和使用cargo test/nextest
运行单元测试。您需要使用夜间Rust工具链来构建和运行测试。可以使用nextest来加速ut,首先安装nextest。
cargo install cargo-nextest --locked
运行集成测试或手动使用TiKV集群测试客户端稍微复杂一些。最简单的方法是使用TiUp (>= 1.5)在您的本地机器上初始化集群
tiup playground nightly --mode tikv-slim
然后如果您想运行集成测试
PD_ADDRS="127.0.0.1:2379" cargo test --package tikv-client --test integration_tests --features integration-tests
创建PR
我们使用标准的GitHub PR工作流程。我们对每个PR运行CI,并要求所有PR在没有警告的情况下构建(包括clippy和Rustfmt警告),通过测试,有DCO签名(在提交时使用-s
,DCO机器人将引导您完成第一个PR的DCO协议),并且至少有一个审查。如果这些对您来说很困难,不要担心,在PR上提问。
要本地运行类似于CI的测试,我们建议您在提交PR之前运行cargo clippy
、cargo test/nextest run
和cargo fmt
。有关运行集成测试,请参阅上面,但您可能不需要在最初的几个PR上担心这一点。
请遵循PingCAP的Rust风格指南。所有代码PR应包括新的测试或测试用例。
寻求帮助
如果您需要帮助,无论是寻找要工作的事情,还是遇到任何技术问题,最快的方式是通过internals.tidb.io,这是TiDB开发者的论坛。
您也可以在Slack上提问。我们监视tikv-wg slack上的#client-rust频道。
您可以直接在GitHub issues或PR上提问;如果您没有收到回复,您应该ping @ekexium或@andylokandy。
依赖关系
~41MB
~797K SLoC