aboutsummaryrefslogtreecommitdiff
path: root/src/garage/repair/offline.rs
blob: f4edcf038e4372d94abddd659387b5dc509aa5d8 (plain) (blame)
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
use std::path::PathBuf;

use garage_util::config::*;
use garage_util::error::*;

use garage_model::garage::Garage;

use crate::cli::structs::*;
use crate::{fill_secrets, Secrets};

pub async fn offline_repair(
	config_file: PathBuf,
	secrets: Secrets,
	opt: OfflineRepairOpt,
) -> Result<(), Error> {
	if !opt.yes {
		return Err(Error::Message(
			"Please add the --yes flag to launch repair operation".into(),
		));
	}

	info!("Loading configuration...");
	let config = fill_secrets(read_config(config_file)?, secrets);

	info!("Initializing Garage main data store...");
	let garage = Garage::new(config)?;

	info!("Launching repair operation...");
	match opt.what {
		#[cfg(feature = "k2v")]
		OfflineRepairWhat::K2VItemCounters => {
			garage
				.k2v
				.counter_table
				.offline_recount_all(&garage.k2v.item_table)?;
		}
		OfflineRepairWhat::ObjectCounters => {
			garage
				.object_counter_table
				.offline_recount_all(&garage.object_table)?;
		}
	}

	info!("Repair operation finished, shutting down...");

	Ok(())
}