clickhouse-qol 0.1.2

Quality of Life tools for ClickHouse
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45

use clickhouse::Client;
use error::ClickhouseToolError;
use url::Url;
use urlencoding::decode;

pub mod error;
pub mod migrations;
pub mod worker;

pub async fn init_client(db_uri: &str) -> Result<Client, ClickhouseToolError> {
    let endpoint: Url = db_uri
        .parse()
        .map_err(|_| ClickhouseToolError::InvaidArgs("Invalid URL".to_string()))?;

    let url = format!(
        "{}://{}:{}",
        endpoint.scheme(),
        endpoint.host().expect("clickhouse must have host"),
        endpoint.port().expect("clickhouse must have port")
    );
    let user = endpoint.username();
    let password = decode(endpoint.password().unwrap_or(""))
        .map_err(|_| ClickhouseToolError::InvaidArgs("Invalid password".to_string()))?;
    let database = endpoint
        .path()
        .split('/')
        .nth(1)
        .expect("clickhouse ep must have database");

    let mut client = Client::default().with_url(&url);
    if !user.is_empty() {
        client = client.with_user(user);
    }

    if !password.is_empty() {
        client = client.with_password(password);
    }

    if !database.is_empty() {
        client = client.with_database(database);
    }

    client.query("SELECT 1").execute().await?;
    Ok(client)
}