1个不稳定版本
| 0.1.0 | 2023年2月15日 |
|---|
#718 在 WebAssembly
504 每月下载量
31KB
554 行
Rust辅助库,用于Scylla UDFs
此crate允许编写纯Rust函数,这些函数可以用作Scylla UDFs。
注意:此crate官方支持且可用。然而,UDFs仍然是ScyllaDB的一个实验性特性,该crate尚未广泛应用,因此它仍然处于beta版本,其API可能发生变化。我们感谢错误报告和pull请求!
用法
先决条件
要在Scylla中使用此辅助库,您需要
- Rust标准库
wasm32-wasi- 可以在rustup安装中使用以下命令添加:
rustup target add wasm32-wasi - 对于非rustup配置,您可以尝试遵循以下步骤:https://wasm.rust-lang.net.cn/docs/wasm-pack/prerequisites/non-rustup-setups.html
- 也可作为rpm提供:
rust-std-static-wasm32-wasi
- 可以在rustup安装中使用以下命令添加:
wasm2wat解析器- 在许多发行版的
wabt软件包中提供
- 在许多发行版的
编译
我们推荐使用cargo配置。
- 从库包开始
cargo new --lib
- 将以下行添加到Cargo.toml中,以设置crate类型为cdylib
[lib]
crate-type = ["cdylib"]
- 实现您的包,使用
scylla_udf::export_udf宏导出Scylla UDFs。 - 使用wasm32-wasi目标构建包
cargo build --target=wasm32-wasi
- 找到编译后的
.wasm二进制文件。假设它是target/wasm32-wasi/debug/abc.wasm。 - (可选)使用
wasm-opt -O3 target/wasm32-wasi-debug/abc.wasm优化二进制代码(可以与使用cargo build --release配置文件) - 将二进制代码翻译为
wat
wasm2wat target/wasm32-wasi/debug/abc.wasm > target/wasm32/wasi/debug/abc.wat
CQL 语句
生成的 target/wasm32/wasi/debug/abc.wat 代码现在可以直接在 CREATE FUNCTION 语句中使用。生成的代码可能包含 ' 字符,因此可能需要先将它们替换为 '',以便在 CQL 字符串中使用。
例如,如果你有一个使用逗号连接单词列表的 Rust UDF,你可以使用以下语句创建 Scylla UDF
CREATE FUNCTION commas(string list<text>) CALLED ON NULL INPUT RETURNS text AS ' (module ...) '
CQL 类型映射
使用 #[export_udf] 注释的函数中使用的参数和返回值类型必须映射到 Scylla 中使用的 CREATE FUNCTION 语句中的 CQL 类型,如下表所示。
如果 Scylla 函数是用与 Rust 函数中使用的类型不匹配的类型创建的,则调用 UDF 将失败或产生任意结果。
本地类型
| CQL 类型 | Rust 类型 |
|---|---|
| ASCII | String |
| BIGINT | i64 |
| BLOB | Vec<u8> |
| BOOLEAN | bool |
| COUNTER | scylla_udf::Counter |
| DATE | chrono::NaiveDate |
| DECIMAL | bigdecimal::Decimal |
| DOUBLE | f64 |
| DURATION | scylla_udf::CqlDuration |
| FLOAT | f32 |
| INET | std::net::IpAddr |
| INT | i32 |
| SMALLINT | i16 |
| TEXT | String |
| TIME | scylla_udf::Time |
| TIMESTAMP | scylla_udf::Timestamp |
| TIMEUUID | uuid::Uuid |
| TINYINT | i8 |
| UUID | uuid::Uuid |
| VARCHAR | String |
| VARINT | num_bigint::BigInt |
集合
如果 CQL 类型 T 映射到 Rust 类型 RustT,则可以使用它作为集合参数
| CQL 类型 | Rust 类型 |
|---|---|
| LIST<T> | Vec<RustT> |
| MAP<T> | std::collections::BTreeMap<RustT>, std::collections::HashMap<RustT> |
| SET<T> | Vec<RustT>, std::collections::BTreeSet<RustT>, std::collections::HashSet<RustT> |
元组
如果 CQL 类型 T1、T2、... 映射到 Rust 类型 RustT1、RustT2、...,则可以在元组中使用它们
| CQL 类型 | Rust 类型 |
|---|---|
| TUPLE<T1, T2, ...> | (RustT1, RustT2, ...) |
空值
如果映射到类型 RustT 的 CQL 值 T 可能是空值(在非 RETURNS NULL ON NULL INPUT UDFs)中,则 Rust 函数中使用的类型应为 Option<RustT>。
贡献
通常情况下,尽量遵循https://github.com/scylladb/scylla-rust-driver/blob/main/CONTRIBUTING.md中的相同规则。
测试
这个crate旨在编译成wasm32-wasi目标并在WASM运行时中运行。使用WASM特定代码的测试很可能在以不同方式执行(特别是使用简单的cargo test命令)时不会成功。
例如,如果您已安装并配置了wasmtime运行时,并且它位于PATH中,您可以使用以下命令运行测试
CARGO_TARGET_WASM32_WASI_RUNNER="wasmtime --allow-unknown-exports" cargo test --target=wasm32-wasi
依赖
~7.5MB
~141K SLoC