Rust

Databend 提供了一个用 Rust 编写的驱动（crates.io - databend-driver），它便于使用 Rust 编程语言开发应用程序，并与 Databend 建立连接。请注意，驱动目前不支持处理数组。

有关安装说明、示例和源代码，请参见 GitHub - databend-driver。

Databend Rust 驱动行为总结

下表总结了 Rust 驱动的主要行为和功能及其用途：

函数名称	描述
info	返回客户端的连接信息。
version	返回执行 SELECT VERSION() 语句的结果。
exec	执行一个 SQL 语句并返回受影响的行数。
query_iter	执行一个 SQL 查询并返回一个迭代器，用于逐行处理结果。
query_iter_ext	执行一个 SQL 查询并返回一个包含结果统计信息的迭代器。
query_row	执行一个 SQL 查询并返回单行结果。
get_presigned_url	基于操作和 Stage 参数生成 PRESIGN 语句，返回 HTTP 方法、头信息和 URL。
upload_to_stage	上传数据到 Stage。默认使用 PRESIGN UPLOAD 获得 URL，或者如果 PRESIGN 被禁用，则使用 v1/upload_to_stage API。
load_data	上传数据到内置 Stage（upload_to_stage）并执行插入/替换操作，使用 Stage Attachment。
load_file	重用 load_data 逻辑来上传文件并插入数据。
stream_load	读取数据作为 Vec，将其转换为 CSV，然后调用 load_data 方法。

教程 -1：使用 Rust 与 Databend 集成

在开始之前，请确保您已成功安装了本地 Databend。有关详细说明，请参见 本地和 Docker 部署。

步骤 1. 准备一个 SQL 用户账户

为了将您的程序连接到 Databend 并执行 SQL 操作，您必须在代码中提供具有适当权限的 SQL 用户账户。如果需要，请在 Databend 中创建一个，并确保 SQL 用户仅具有出于安全考虑所需的权限。

本教程使用名为 'user1'，密码为 'abc123' 的 SQL 用户作为示例。由于程序将向 Databend 写入数据，因此用户需要 ALL 权限。有关如何管理 SQL 用户及其权限的信息，请参见 用户与角色。

CREATE USER user1 IDENTIFIED BY 'abc123';
GRANT ALL on *.* TO user1;

步骤 2. 编写一个 Rust 程序

在这一步中，您将创建一个简单的 Rust 程序，该程序将与 Databend 通信。程序将涉及创建表、插入数据和执行数据查询等任务。

1

创建一个新项目

cargo new databend-demo --bin

Cargo.toml

[package]
name = "databend-demo"
version = "0.1.0"
edition = "2021"

# 有关更多键及其定义，请参见 https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
databend-driver = "0.7"
tokio = { version = "1", features = ["full"] }
tokio-stream = "0.1.12"

2

将以下代码复制并粘贴到文件 main.rs 中

备注

代码下方的 hostname 值必须与您的 Databend 查询服务的 HTTP 处理程序设置保持一致。

main.rs

use databend_driver::Client;
use tokio_stream::StreamExt;

#[tokio::main]
async fn main() {
    // 下面的代码以用户名 "user1" 的 SQL 用户和密码 "abc123" 为例连接到本地 Databend 实例。
    // 在保持相同格式的条件下，您可以随意使用自己的值。
    let dsn = "databend://user1:abc123@localhost:8000/default?sslmode=disable";
    let client = Client::new(dsn.to_string());
    let conn = client.get_conn().await.unwrap();

    let sql_db_create = "CREATE DATABASE IF NOT EXISTS book_db;";
    conn.exec(sql_db_create).await.unwrap();

    let sql_table_create = "CREATE TABLE book_db.books (
    title VARCHAR,
    author VARCHAR,
    date VARCHAR
);";

    conn.exec(sql_table_create).await.unwrap();
    let sql_insert = "INSERT INTO book_db.books VALUES ('mybook', 'author', '2022');";
    conn.exec(sql_insert).await.unwrap();

    let mut rows = conn.query_iter("SELECT * FROM book_db.books;").await.unwrap();
    while let Some(row) = rows.next().await {
        let (title, author, date): (String, String, String) = row.unwrap().try_into().unwrap();
        println!("{} {} {}", title, author, date);
    }

    let sql_table_drop = "DROP TABLE book_db.books;";
    conn.exec(sql_table_drop).await.unwrap();

    let sql_db_drop = "DROP DATABASE book_db;";
    conn.exec(sql_db_drop).await.unwrap();
}

3

运行程序。

cargo run

输出

mybook author 2022

教程 -2：使用 Rust 与 Databend Cloud 集成

在开始之前，请确保您已成功创建一个计算集群并获得了连接信息。有关如何操作，请参见 连接到计算集群。

步骤 1. 创建一个 Rust Crate

$ cargo new databend-sample --bin

步骤 2. 添加依赖项

使用以下代码编辑名为 Cargo.toml 的文件：

[package]
name = "databend-sample"
version = "0.1.0"
edition = "2021"

# 有关更多键及其定义，请参见 https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
chrono = "0.4"
databend-driver = "0.6"
tokio = { version = "1", features = ["full"] }
tokio-stream = "0.1"

步骤 3. 使用 databend-driver 连接

使用以下代码编辑名为 main.rs 的文件：

use databend_driver::Client;
use tokio_stream::StreamExt;

#[tokio::main]
async fn main() {
    let dsn = "databend://{USER}:{PASSWORD}@${HOST}:443/{DATABASE}?&warehouse={WAREHOUSE_NAME}";
    let client = Client::new(dsn.to_string());
    let conn = client.get_conn().await.unwrap();

    let sql_table_drop = "DROP TABLE IF EXISTS data;";
    conn.exec(sql_table_drop).await.unwrap();

    let sql_table_create = "CREATE TABLE IF NOT EXISTS data (
        i64 Int64,
        u64 UInt64,
        f64 Float64,
        s   String,
        s2  String,
        d   Date,
        t   DateTime)";

    conn.exec(sql_table_create).await.unwrap();
    let sql_insert = "INSERT INTO data VALUES ('1234', '2345', '3.1415', 'test', 'test2', '2021-01-01', '2021-01-01 00:00:00');";
    conn.exec(sql_insert).await.unwrap();

    let mut rows = conn.query_iter("SELECT * FROM data;").await.unwrap();
    当 let Some(row) = rows.next().await {
        let (col1, col2, col3, col4, col5, col6, col7): (
            i64,
            u64,
            f64,
            String,
            String,
            chrono::NaiveDate,
            chrono::NaiveDateTime,
        ) = row.unwrap().try_into().unwrap();
        println!(
            "{} {} {} {} {} {} {}",
            col1, col2, col3, col4, col5, col6, col7
        );
    }
}

提示

将代码中的 {USER}、{PASSWORD}、{HOST}、{WAREHOUSE_NAME} 和 {DATABASE} 替换为您的连接信息。关于如何 获取连接信息， 请参阅连接到计算集群。

第 4 步。使用 Cargo 运行示例

$ cargo run