Line data Source code
1 : use anyhow::*;
2 : use clap::{value_parser, Arg, ArgMatches, Command};
3 : use std::{path::PathBuf, str::FromStr};
4 : use wal_craft::*;
5 :
6 0 : fn main() -> Result<()> {
7 0 : env_logger::Builder::from_env(env_logger::Env::default().default_filter_or("wal_craft=info"))
8 0 : .init();
9 0 : let arg_matches = cli().get_matches();
10 0 :
11 0 : let wal_craft = |arg_matches: &ArgMatches, client| {
12 0 : let (intermediate_lsns, end_of_wal_lsn) = match arg_matches
13 0 : .get_one::<String>("type")
14 0 : .map(|s| s.as_str())
15 0 : .context("'type' is required")?
16 : {
17 0 : Simple::NAME => Simple::craft(client)?,
18 0 : LastWalRecordXlogSwitch::NAME => LastWalRecordXlogSwitch::craft(client)?,
19 0 : LastWalRecordXlogSwitchEndsOnPageBoundary::NAME => {
20 0 : LastWalRecordXlogSwitchEndsOnPageBoundary::craft(client)?
21 : }
22 0 : WalRecordCrossingSegmentFollowedBySmallOne::NAME => {
23 0 : WalRecordCrossingSegmentFollowedBySmallOne::craft(client)?
24 : }
25 0 : LastWalRecordCrossingSegment::NAME => LastWalRecordCrossingSegment::craft(client)?,
26 0 : a => panic!("Unknown --type argument: {a}"),
27 : };
28 0 : for lsn in intermediate_lsns {
29 0 : println!("intermediate_lsn = {lsn}");
30 0 : }
31 0 : println!("end_of_wal = {end_of_wal_lsn}");
32 0 : Ok(())
33 0 : };
34 :
35 0 : match arg_matches.subcommand() {
36 0 : None => panic!("No subcommand provided"),
37 0 : Some(("print-postgres-config", _)) => {
38 0 : for cfg in REQUIRED_POSTGRES_CONFIG.iter() {
39 0 : println!("{cfg}");
40 0 : }
41 0 : Ok(())
42 : }
43 :
44 0 : Some(("with-initdb", arg_matches)) => {
45 0 : let cfg = Conf {
46 0 : pg_version: *arg_matches
47 0 : .get_one::<u32>("pg-version")
48 0 : .context("'pg-version' is required")?,
49 0 : pg_distrib_dir: arg_matches
50 0 : .get_one::<PathBuf>("pg-distrib-dir")
51 0 : .context("'pg-distrib-dir' is required")?
52 0 : .to_owned(),
53 0 : datadir: arg_matches
54 0 : .get_one::<PathBuf>("datadir")
55 0 : .context("'datadir' is required")?
56 0 : .to_owned(),
57 0 : };
58 0 : cfg.initdb()?;
59 0 : let srv = cfg.start_server()?;
60 0 : wal_craft(arg_matches, &mut srv.connect_with_timeout()?)?;
61 0 : srv.kill();
62 0 : Ok(())
63 : }
64 0 : Some(("in-existing", arg_matches)) => wal_craft(
65 0 : arg_matches,
66 0 : &mut postgres::Config::from_str(
67 0 : arg_matches
68 0 : .get_one::<String>("connection")
69 0 : .context("'connection' is required")?,
70 : )
71 0 : .context(
72 0 : "'connection' argument value could not be parsed as a postgres connection string",
73 0 : )?
74 0 : .connect(postgres::NoTls)?,
75 : ),
76 0 : Some(_) => panic!("Unknown subcommand"),
77 : }
78 0 : }
79 :
80 2 : fn cli() -> Command {
81 2 : let type_arg = &Arg::new("type")
82 2 : .help("Type of WAL to craft")
83 2 : .value_parser([
84 2 : Simple::NAME,
85 2 : LastWalRecordXlogSwitch::NAME,
86 2 : LastWalRecordXlogSwitchEndsOnPageBoundary::NAME,
87 2 : WalRecordCrossingSegmentFollowedBySmallOne::NAME,
88 2 : LastWalRecordCrossingSegment::NAME,
89 2 : ])
90 2 : .required(true);
91 2 :
92 2 : Command::new("Postgres WAL crafter")
93 2 : .about("Crafts Postgres databases with specific WAL properties")
94 2 : .subcommand(
95 2 : Command::new("print-postgres-config")
96 2 : .about("Print the configuration required for PostgreSQL server before running this script")
97 2 : )
98 2 : .subcommand(
99 2 : Command::new("with-initdb")
100 2 : .about("Craft WAL in a new data directory first initialized with initdb")
101 2 : .arg(type_arg)
102 2 : .arg(
103 2 : Arg::new("datadir")
104 2 : .help("Data directory for the Postgres server")
105 2 : .value_parser(value_parser!(PathBuf))
106 2 : .required(true)
107 2 : )
108 2 : .arg(
109 2 : Arg::new("pg-distrib-dir")
110 2 : .long("pg-distrib-dir")
111 2 : .value_parser(value_parser!(PathBuf))
112 2 : .help("Directory with Postgres distributions (bin and lib directories, e.g. pg_install containing subpath `v14/bin/postgresql`)")
113 2 : .default_value("/usr/local")
114 2 : )
115 2 : .arg(
116 2 : Arg::new("pg-version")
117 2 : .long("pg-version")
118 2 : .help("Postgres version to use for the initial tenant")
119 2 : .value_parser(value_parser!(u32))
120 2 : .required(true)
121 2 :
122 2 : )
123 2 : )
124 2 : .subcommand(
125 2 : Command::new("in-existing")
126 2 : .about("Craft WAL at an existing recently created Postgres database. Note that server may append new WAL entries on shutdown.")
127 2 : .arg(type_arg)
128 2 : .arg(
129 2 : Arg::new("connection")
130 2 : .help("Connection string to the Postgres database to populate")
131 2 : .required(true)
132 2 : )
133 2 : )
134 2 : }
135 :
136 2 : #[test]
137 2 : fn verify_cli() {
138 2 : cli().debug_assert();
139 2 : }
|
