5个不稳定版本
0.3.0-surreal.1 | 2024年6月24日 |
---|---|
0.2.0-surreal.2 | 2023年8月8日 |
0.2.0-surreal.1 | 2023年7月25日 |
0.1.0-surreal.2 | 2023年7月22日 |
0.1.0-surreal.1 | 2023年6月19日 |
#70 在 数据库接口 中排名
3,714 每月下载量
在 15 个crate中使用了(直接使用7个)
1.5MB
27K SLoC
TiKV客户端(Rust)
此crate提供了一个易于使用的TiKV客户端,TiKV是一个用Rust编写的分布式、事务型键值数据库。
此crate允许您连接到TiKV(>= v5.0.0
)集群,并使用事务型或原始(简单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
使用客户端crate的一般流程是创建一个原始或事务客户端对象(可配置),然后使用客户端对象发送命令,或者用它来创建事务对象。在后一种情况下,事务通过各种命令构建,然后提交(或回滚)。
示例
原始模式
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
结构体。使用某些选项,您可以自己接管协议的部分(例如重试失败的消息)。
最低级别的抽象是直接创建和发送gRPC消息到TiKV(和PD)节点。`tikv-client-store`和`tikv-client-pd`crate使这比直接使用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
:一对Key
和Value
。它提供方便的方法用于转换到和从其他类型。
BoundRange
:用于范围相关请求,如scan
。它实现了Rust范围的对From
,因此您可以将Rust键的范围传递给请求,例如,client.delete_range(vec![]..)
。
原始请求
请求 | 主要参数类型 | 结果类型 | 值得注意的行为 |
---|---|---|---|
put |
KvPair |
||
get |
Key |
Option<Value> |
|
delete |
Key |
||
delete_range |
BoundRange |
||
scan |
BoundRange |
Vec<KvPair> |
|
batch_put |
Iter<KvPair> |
||
batch_get |
Iter<Key> |
Vec<KvPair> |
跳过不存在的键;不保留顺序 |
batch_delete |
Iter<Key> |
||
batch_scan |
Iter<BoundRange> |
Vec<KvPair> |
请参阅 each_limit 参数行为的文档。保留范围顺序。 |
batch_scan_keys |
Iter<BoundRange> |
Vec<Key> |
请参阅 each_limit 参数行为的文档。保留范围顺序。 |
compare_and_swap |
Key + 2x Value |
(Option<Value>, bool) |
事务请求
请求 | 主要参数类型 | 结果类型 | 值得注意的行为 |
---|---|---|---|
put |
KvPair |
||
get |
Key |
Option<value> |
|
get_for_update |
Key |
Option<value> |
|
key_exists |
Key |
bool |
|
delete |
Key |
||
scan |
BoundRange |
Iter<KvPair> |
|
scan_keys |
BoundRange |
Iter<Key> |
|
batch_get |
Iter<Key> |
Iter<KvPair> |
跳过不存在的键;不保留顺序 |
batch_get_for_update |
Iter<Key> |
Iter<KvPair> |
跳过不存在的键;不保留顺序 |
lock_keys |
Iter<Key> |
||
send_heart_beat |
u64 (TTL) |
||
gc |
Timestamp |
bool |
如果 PD 中最新的安全点等于参数,则返回 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。
依赖项
~16–31MB
~570K SLoC