1个不稳定版本

0.1.0 2023年2月15日

#718WebAssembly

Download history 122/week @ 2024-03-13 57/week @ 2024-03-20 97/week @ 2024-03-27 56/week @ 2024-04-03 33/week @ 2024-04-10 106/week @ 2024-04-17 98/week @ 2024-04-24 86/week @ 2024-05-01 231/week @ 2024-05-08 292/week @ 2024-05-15 299/week @ 2024-05-22 193/week @ 2024-05-29 139/week @ 2024-06-05 124/week @ 2024-06-12 102/week @ 2024-06-19 95/week @ 2024-06-26

504 每月下载量

MIT/Apache

31KB
554

Rust辅助库,用于Scylla UDFs

此crate允许编写纯Rust函数,这些函数可以用作Scylla UDFs。

注意:此crate官方支持且可用。然而,UDFs仍然是ScyllaDB的一个实验性特性,该crate尚未广泛应用,因此它仍然处于beta版本,其API可能发生变化。我们感谢错误报告和pull请求!

用法

先决条件

要在Scylla中使用此辅助库,您需要

编译

我们推荐使用cargo配置。

  1. 从库包开始
cargo new --lib
  1. 将以下行添加到Cargo.toml中,以设置crate类型为cdylib
[lib]
crate-type = ["cdylib"]
  1. 实现您的包,使用scylla_udf::export_udf宏导出Scylla UDFs。
  2. 使用wasm32-wasi目标构建包
cargo build --target=wasm32-wasi
  1. 找到编译后的.wasm二进制文件。假设它是target/wasm32-wasi/debug/abc.wasm
  2. (可选)使用 wasm-opt -O3 target/wasm32-wasi-debug/abc.wasm 优化二进制代码(可以与使用 cargo build --release 配置文件)
  3. 将二进制代码翻译为 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 类型 T1T2、... 映射到 Rust 类型 RustT1RustT2、...,则可以在元组中使用它们

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