rust-pgdatadiff 0.1.8

Rust library for comparing two PostgreSQL databases
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108

// Path: examples/example_diff.rs
extern crate anyhow;
extern crate clap;
extern crate rust_pgdatadiff;

use anyhow::Result;
use clap::{Parser, Subcommand};
use rust_pgdatadiff::diff::diff_ops::Differ;
use rust_pgdatadiff::diff::diff_payload::DiffPayload;

#[derive(Parser)]
#[command(author, version, about, long_about = None)]
#[command(propagate_version = true)]
struct Cli {
    #[command(subcommand)]
    command: Commands,
}

#[derive(Subcommand)]
enum Commands {
    #[command(about = "Print the version")]
    Version,
    Diff {
        /// postgres://postgres:postgres@localhost:5438/example
        first_db: String,
        /// postgres://postgres:postgres@localhost:5439/example
        second_db: String,
        /// Only compare data, exclude sequences
        #[arg(long, default_value_t = false, required = false)]
        only_tables: bool,
        /// Only compare sequences, exclude data
        #[arg(long, default_value_t = false, required = false)]
        only_sequences: bool,
        /// Do a quick test based on counts alone
        #[arg(long, default_value_t = false, required = false)]
        only_count: bool,
        /// The chunk size when comparing data
        #[arg(long, default_value_t = 10000, required = false)]
        chunk_size: i64,
        /// The start position for the comparison
        #[arg(long, default_value_t = 0, required = false)]
        start_position: i64,
        /// Max connections for Postgres pool
        #[arg(long, default_value_t = 100, required = false)]
        max_connections: i64,
        /// Tables included in the comparison
        #[arg(short, long, value_delimiter = ',', num_args = 0.., required = false, conflicts_with = "exclude_tables")]
        include_tables: Vec<String>,
        /// Tables excluded from the comparison
        #[arg(short, long, value_delimiter = ',', num_args = 0.., required = false, conflicts_with = "include_tables")]
        exclude_tables: Vec<String>,
        /// Schema name
        #[arg(long, default_value = "public", required = false)]
        schema_name: String,
        /// Accept invalid TLS certificates for the first database
        #[arg(long, default_value_t = false, required = false)]
        accept_invalid_certs_first_db: bool,
        /// Accept invalid TLS certificates for the second database
        #[arg(long, default_value_t = false, required = false)]
        accept_invalid_certs_second_db: bool,
    },
}

#[tokio::main]
async fn main() -> Result<()> {
    tracing_subscriber::fmt::init();

    let cli = Cli::parse();
    match &cli.command {
        Commands::Version => {
            println!("Version: {}", env!("CARGO_PKG_VERSION"));
            Ok(())
        }
        Commands::Diff {
            first_db,
            second_db,
            only_tables,
            only_sequences,
            only_count,
            chunk_size,
            start_position,
            max_connections,
            include_tables,
            exclude_tables,
            schema_name,
            accept_invalid_certs_first_db,
            accept_invalid_certs_second_db,
        } => {
            let payload = DiffPayload::builder()
                .first_db(first_db.clone())
                .second_db(second_db.clone())
                .only_tables(*only_tables)
                .only_sequences(*only_sequences)
                .only_count(*only_count)
                .chunk_size(*chunk_size)
                .start_position(*start_position)
                .max_connections(*max_connections)
                .include_tables(include_tables.to_vec())
                .exclude_tables(exclude_tables.to_vec())
                .schema_name(schema_name.clone())
                .accept_invalid_certs_first_db(*accept_invalid_certs_first_db)
                .accept_invalid_certs_second_db(*accept_invalid_certs_second_db)
                .build();
            let _ = Differ::diff_dbs(payload).await;
            Ok(())
        }
    }
}