文盘Rust -- rust 连接云上数仓 starwift

作者：京东云 贾世闻

最近想看看 rust 如何集成 clickhouse，又犯了好吃懒做的心理（不想自己建环境），刚好京东云发布了兼容ck 的云原生数仓 Starwfit，于是搞了个实例折腾一番。

Starwfit 是京东云自主研发的新一代云原生数据仓库,通过存算分离降低了存储成本，同时兼具性能和扩展弹性。其写入和查询速度可达到传统数据仓库的数倍，为用户提供实时数据分析能力。广泛应用于流量分析、精准营销、用户画像、广告实时竞价、BI报表分析、日志分析、促销选品、物联网等业务场景。

言归正传，看看 rust 如何与 starwift 打交道。

创建集群

• 登录控制台，直接选配创建即可

• 创建用户

• 开启白名单 为了保证starwift能够被外部资源访问需要开启白名单，默认情况下新创建的实例只有该实例所在vpc下才能访问

为了方便，实验环境直接编辑 default 分组，编辑 ip 列表为 0.0.0.0/0，对所有ip开放。在实际生产环境需要根据实际情况配置白名单避免安全问题。

• 开启公网连接

  为了实验方便，开启公网访问域名便于开发和测试

mac客户端安装

• 安装客户端

  macos 客户端安装

  1 curl -O 'https://builds.clickhouse.com/master/macos/clickhouse' && chmod a  +x ./clickhouse
  

  其他客户端安装 参见 clickhosue 官方文档

• 测试连通性

  1clickhouse client --host service-terrabase-9s29mdlsb7.  terrabase-9s29mdlsb7-hb-public.jvessel2.jdcloud.com \
  2--port 9000 \
  3--user sample \
  4--password xxxxxx
  

  1clickhouse:) show databases
  

rust 连接 starwift

starwift 支持 tcp 和 http 两种协议连接server。下面我们分别给出两种方式的示例代码

• tcp 连接

  • 依赖 crate

    1# clickhouse tcp
    2tokio = { version = "1.21.2", features = ["full"] }
    3clickhouse-rs = { git = "https://github.com/suharev7/clickhouse-rs", features = ["default"]}
    

  • 示例代码

    1use clickhouse_rs::Pool;
    2
    3#[tokio::main]
    4async fn main() {
    5    let database_url="tcp://username:password@service-terrabase-9s29mdlsb7.terrabase-9s2mdsb-hb-public.jvessel2.jdcloud.com:9000?compression=lz4".to_string();
    6    let pool = Pool::new(database_url);
    7
    8    let mut client = pool.get_handle().await.unwrap();
    9    let sql = "show databases;";
    10    let r = client.query(sql).fetch_all().await;
    11    println!("result is: {:?}", r);
    12}
    13
    

• http 连接

  • 依赖 crate

    1# clickhouse http
    2clickhouse = {git = "https://github.com/loyd/clickhouse.rs", features = ["test-util"]}
    

  • 示例代码

    1use clickhouse::Client;
    2use clickhouse::Row;
    3use serde::{Deserialize, Serialize};
    4
    5#[derive(Debug, Row, Serialize, Deserialize)]
    6struct Database {
    7    name: String,
    8}
    9
    10#[tokio::main]
    11async fn main() {
    12    let client = Client::default()
    13    .with_url("https://service-terrabase-9s29mdlsb7.terrabase-9sdlb7-hb-public.jvessel2.jdcloud.com:8123")
    14    .with_user("username")
    15    .with_password("password");
    16
    17    let sql = "SHOW databases";
    18    let r = client.query(sql).fetch_all::<Database>().await;
    19    println!("result is: {:?}", r);
    20}
    

测试过程中，遇到了依赖冲突问题，clickhouse-rs 和 clickhouse.rs 同时依赖了 clickhouse-rs-cityhash-sys 但依赖方式不同，导致编译不通过，下一期详细聊聊踩坑及爬坑过程。 再次感谢京东云数据库团队提供的 starwift 测试实例。 下期见。