chm 0.1.17

Migrations CLI tool 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
46
47
48
49
50
51
52
53
54
55
56

use clap::Args;
use serde::{Deserialize, Serialize};

use crate::{
    errors::CLIError,
    operators::{
        clickhouse_operators::{
            apply_migrations, create_migrations_table,
            get_clickhouse_client_and_ping, get_migrations_from_clickhouse,
        },
        migrations_operators::{get_migrations_from_dir, MigrationOnDisk},
    },
};

#[derive(Args, Clone, Deserialize, Serialize)]
pub struct SetupArgs {
    /// Clickhouse URL
    #[arg(env = "CLICKHOUSE_URL", default_value = None)]
    pub url: Option<String>,
    /// Clickhouse User
    #[arg(env = "CLICKHOUSE_USER", default_value = None)]
    pub user: Option<String>,
    /// Clickhouse Password
    #[arg(env = "CLICKHOUSE_PASSWORD", default_value = None)]
    pub password: Option<String>,
    /// Clickhouse Database
    #[arg(env = "CLICKHOUSE_DB", default_value = None)]
    pub database: Option<String>,
}

pub async fn run_pending_migrations(config: SetupArgs) -> Result<(), CLIError> {
    let client = get_clickhouse_client_and_ping(config).await?;

    create_migrations_table(client.clone()).await?;

    let local_migrations = get_migrations_from_dir().await?;

    let applied_migrations = get_migrations_from_clickhouse(client.clone()).await?;

    let local_migrations_not_in_db: Vec<MigrationOnDisk> = local_migrations
        .iter()
        .filter_map(|m| {
            if applied_migrations
                .iter()
                .any(|applied_migration| applied_migration.version == m.version)
            {
                return None;
            }
            Some(m.clone())
        })
        .collect();

    apply_migrations(client.clone(), local_migrations_not_in_db).await?;

    Ok(())
}