use failure::*;
-use crate::api::schema::*;
-use crate::api::router::*;
+use crate::tools;
+use crate::api_schema::*;
+use crate::api_schema::router::*;
//use crate::server::rest::*;
use serde_json::{json, Value};
+use std::collections::{HashSet, HashMap};
+use chrono::{DateTime, Datelike, Local};
+use std::path::PathBuf;
+use std::sync::Arc;
//use hyper::StatusCode;
//use hyper::rt::{Future, Stream};
use crate::config::datastore;
-use crate::backup::datastore::*;
+use crate::backup::*;
+use crate::server::WorkerTask;
-mod catar;
+mod pxar;
+mod upload;
-// this is just a test for mutability/mutex handling - will remove later
-fn start_garbage_collection(param: Value, _info: &ApiMethod) -> Result<Value, Error> {
+fn group_backups(backup_list: Vec<BackupInfo>) -> HashMap<String, Vec<BackupInfo>> {
+
+ let mut group_hash = HashMap::new();
+
+ for info in backup_list {
+ let group_id = info.backup_dir.group().group_path().to_str().unwrap().to_owned();
+ let time_list = group_hash.entry(group_id).or_insert(vec![]);
+ time_list.push(info);
+ }
+
+ group_hash
+}
+
+fn mark_selections<F: Fn(DateTime<Local>, &BackupInfo) -> String> (
+ mark: &mut HashSet<PathBuf>,
+ list: &Vec<BackupInfo>,
+ keep: usize,
+ select_id: F,
+){
+ let mut hash = HashSet::new();
+ for info in list {
+ let local_time = info.backup_dir.backup_time().with_timezone(&Local);
+ if hash.len() >= keep as usize { break; }
+ let backup_id = info.backup_dir.relative_path();
+ let sel_id: String = select_id(local_time, &info);
+ if !hash.contains(&sel_id) {
+ hash.insert(sel_id);
+ //println!(" KEEP ID {} {}", backup_id, local_time.format("%c"));
+ mark.insert(backup_id);
+ }
+ }
+}
+
+fn list_groups(
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
let store = param["store"].as_str().unwrap();
let datastore = DataStore::lookup_datastore(store)?;
- println!("Starting garbage collection on store {}", store);
+ let backup_list = datastore.list_backups()?;
+
+ let group_hash = group_backups(backup_list);
+
+ let mut groups = vec![];
+
+ for (_group_id, mut list) in group_hash {
+
+ BackupInfo::sort_list(&mut list, false);
+
+ let info = &list[0];
+ let group = info.backup_dir.group();
+
+ groups.push(json!({
+ "backup-type": group.backup_type(),
+ "backup-id": group.backup_id(),
+ "last-backup": info.backup_dir.backup_time().timestamp(),
+ "backup-count": list.len() as u64,
+ "files": info.files,
+ }));
+ }
+
+ Ok(json!(groups))
+}
+
+fn list_snapshot_files (
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
+
+ let store = tools::required_string_param(¶m, "store")?;
+ let backup_type = tools::required_string_param(¶m, "backup-type")?;
+ let backup_id = tools::required_string_param(¶m, "backup-id")?;
+ let backup_time = tools::required_integer_param(¶m, "backup-time")?;
+
+ let snapshot = BackupDir::new(backup_type, backup_id, backup_time);
+
+ let datastore = DataStore::lookup_datastore(store)?;
+
+ let files = datastore.list_files(&snapshot)?;
+
+ Ok(json!(files))
+}
- datastore.garbage_collection()?;
+fn delete_snapshots (
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
+
+ let store = tools::required_string_param(¶m, "store")?;
+ let backup_type = tools::required_string_param(¶m, "backup-type")?;
+ let backup_id = tools::required_string_param(¶m, "backup-id")?;
+ let backup_time = tools::required_integer_param(¶m, "backup-time")?;
+
+ let snapshot = BackupDir::new(backup_type, backup_id, backup_time);
+
+ let datastore = DataStore::lookup_datastore(store)?;
+
+ datastore.remove_backup_dir(&snapshot)?;
+
+ Ok(Value::Null)
+}
+
+fn list_snapshots (
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
+
+ let store = tools::required_string_param(¶m, "store")?;
+ let backup_type = tools::required_string_param(¶m, "backup-type")?;
+ let backup_id = tools::required_string_param(¶m, "backup-id")?;
+
+ let group = BackupGroup::new(backup_type, backup_id);
+
+ let datastore = DataStore::lookup_datastore(store)?;
+
+ let backup_list = datastore.list_backups()?;
+
+ let mut group_hash = group_backups(backup_list);
+
+ let group_id = group.group_path().to_str().unwrap().to_owned();
+
+ let group_snapshots = match group_hash.get_mut(&group_id) {
+ Some(data) => {
+ // new backups first
+ BackupInfo::sort_list(data, false);
+ data
+ }
+ None => bail!("Backup group '{}' does not exists.", group_id),
+ };
+
+ let mut snapshots = vec![];
+
+ for info in group_snapshots {
+
+ let group = info.backup_dir.group();
+
+ snapshots.push(json!({
+ "backup-type": group.backup_type(),
+ "backup-id": group.backup_id(),
+ "backup-time": info.backup_dir.backup_time().timestamp(),
+ "files": info.files,
+ }));
+ }
+
+ Ok(json!(snapshots))
+}
+
+fn prune(
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
+
+ let store = param["store"].as_str().unwrap();
+
+ let datastore = DataStore::lookup_datastore(store)?;
+
+ println!("Starting prune on store {}", store);
+
+ let backup_list = datastore.list_backups()?;
+
+ let group_hash = group_backups(backup_list);
+
+ for (_group_id, mut list) in group_hash {
+
+ let mut mark = HashSet::new();
+
+ BackupInfo::sort_list(&mut list, false);
+
+ if let Some(keep_last) = param["keep-last"].as_u64() {
+ list.iter().take(keep_last as usize).for_each(|info| {
+ mark.insert(info.backup_dir.relative_path());
+ });
+ }
+
+ if let Some(keep_daily) = param["keep-daily"].as_u64() {
+ mark_selections(&mut mark, &list, keep_daily as usize, |local_time, _info| {
+ format!("{}/{}/{}", local_time.year(), local_time.month(), local_time.day())
+ });
+ }
+
+ if let Some(keep_weekly) = param["keep-weekly"].as_u64() {
+ mark_selections(&mut mark, &list, keep_weekly as usize, |local_time, _info| {
+ format!("{}/{}", local_time.year(), local_time.iso_week().week())
+ });
+ }
+
+ if let Some(keep_monthly) = param["keep-monthly"].as_u64() {
+ mark_selections(&mut mark, &list, keep_monthly as usize, |local_time, _info| {
+ format!("{}/{}", local_time.year(), local_time.month())
+ });
+ }
+
+ if let Some(keep_yearly) = param["keep-yearly"].as_u64() {
+ mark_selections(&mut mark, &list, keep_yearly as usize, |local_time, _info| {
+ format!("{}/{}", local_time.year(), local_time.year())
+ });
+ }
+
+ let mut remove_list: Vec<BackupInfo> = list.into_iter()
+ .filter(|info| !mark.contains(&info.backup_dir.relative_path())).collect();
+
+ BackupInfo::sort_list(&mut remove_list, true);
+
+ for info in remove_list {
+ datastore.remove_backup_dir(&info.backup_dir)?;
+ }
+ }
Ok(json!(null))
}
+pub fn add_common_prune_prameters(schema: ObjectSchema) -> ObjectSchema {
+
+ schema
+ .optional(
+ "keep-last",
+ IntegerSchema::new("Number of backups to keep.")
+ .minimum(1)
+ )
+ .optional(
+ "keep-daily",
+ IntegerSchema::new("Number of daily backups to keep.")
+ .minimum(1)
+ )
+ .optional(
+ "keep-weekly",
+ IntegerSchema::new("Number of weekly backups to keep.")
+ .minimum(1)
+ )
+ .optional(
+ "keep-monthly",
+ IntegerSchema::new("Number of monthly backups to keep.")
+ .minimum(1)
+ )
+ .optional(
+ "keep-yearly",
+ IntegerSchema::new("Number of yearly backups to keep.")
+ .minimum(1)
+ )
+}
+
+fn api_method_prune() -> ApiMethod {
+ ApiMethod::new(
+ prune,
+ add_common_prune_prameters(
+ ObjectSchema::new("Prune the datastore.")
+ .required(
+ "store",
+ StringSchema::new("Datastore name.")
+ )
+ )
+ )
+}
+
+fn start_garbage_collection(
+ param: Value,
+ _info: &ApiMethod,
+ rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
+
+ let store = param["store"].as_str().unwrap().to_string();
+
+ let datastore = DataStore::lookup_datastore(&store)?;
+
+ println!("Starting garbage collection on store {}", store);
+
+ let to_stdout = if rpcenv.env_type() == RpcEnvironmentType::CLI { true } else { false };
+
+ let upid_str = WorkerTask::new_thread(
+ "garbage_collection", Some(store.clone()), "root@pam", to_stdout, move |worker|
+ {
+ worker.log(format!("starting garbage collection on store {}", store));
+ datastore.garbage_collection()
+ })?;
+
+ Ok(json!(upid_str))
+}
+
pub fn api_method_start_garbage_collection() -> ApiMethod {
ApiMethod::new(
start_garbage_collection,
)
}
-fn garbage_collection_status(param: Value, _info: &ApiMethod) -> Result<Value, Error> {
+fn garbage_collection_status(
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
let store = param["store"].as_str().unwrap();
)
}
-fn get_backup_list(param: Value, _info: &ApiMethod) -> Result<Value, Error> {
+fn get_backup_list(
+ param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
- let config = datastore::config()?;
+ //let config = datastore::config()?;
let store = param["store"].as_str().unwrap();
for info in datastore.list_backups()? {
list.push(json!({
- "backup_type": info.backup_type,
- "backup_id": info.backup_id,
- "backup_time": info.backup_time.timestamp(),
+ "backup-type": info.backup_dir.group().backup_type(),
+ "backup-id": info.backup_dir.group().backup_id(),
+ "backup-time": info.backup_dir.backup_time().timestamp(),
+ "files": info.files,
}));
}
Ok(result)
}
-fn get_datastore_list(_param: Value, _info: &ApiMethod) -> Result<Value, Error> {
+fn get_datastore_list(
+ _param: Value,
+ _info: &ApiMethod,
+ _rpcenv: &mut RpcEnvironment,
+) -> Result<Value, Error> {
let config = datastore::config()?;
pub fn router() -> Router {
+ let store_schema: Arc<Schema> = Arc::new(
+ StringSchema::new("Datastore name.").into()
+ );
+
let datastore_info = Router::new()
.get(ApiMethod::new(
- |_,_| Ok(json!([
+ |_,_,_| Ok(json!([
{"subdir": "backups" },
- {"subdir": "catar" },
- {"subdir": "status"},
- {"subdir": "gc" }
- ])),
+ {"subdir": "pxar" },
+ {"subdir": "gc" },
+ {"subdir": "groups" },
+ {"subdir": "snapshots" },
+ {"subdir": "status" },
+ {"subdir": "prune" },
+ ])),
ObjectSchema::new("Directory index.")
- .required("store", StringSchema::new("Datastore name.")))
+ .required("store", store_schema.clone()))
)
.subdir(
"backups",
.get(ApiMethod::new(
get_backup_list,
ObjectSchema::new("List backups.")
- .required("store", StringSchema::new("Datastore name.")))))
+ .required("store", store_schema.clone()))))
.subdir(
- "catar",
+ "pxar",
Router::new()
- .download(catar::api_method_download_catar())
- .upload(catar::api_method_upload_catar()))
+ .download(pxar::api_method_download_pxar())
+ .upload(pxar::api_method_upload_pxar()))
+ .subdir(
+ "test-upload",
+ Router::new()
+ .upgrade(upload::api_method_upgrade_upload()))
.subdir(
"gc",
Router::new()
.get(api_method_garbage_collection_status())
- .post(api_method_start_garbage_collection()));
+ .post(api_method_start_garbage_collection()))
+ .subdir(
+ "files",
+ Router::new()
+ .get(
+ ApiMethod::new(
+ list_snapshot_files,
+ ObjectSchema::new("List snapshot files.")
+ .required("store", store_schema.clone())
+ .required("backup-type", StringSchema::new("Backup type."))
+ .required("backup-id", StringSchema::new("Backup ID."))
+ .required("backup-time", IntegerSchema::new("Backup time (Unix epoch.)")
+ .minimum(1547797308))
+ )
+ )
+ )
+ .subdir(
+ "groups",
+ Router::new()
+ .get(ApiMethod::new(
+ list_groups,
+ ObjectSchema::new("List backup groups.")
+ .required("store", store_schema.clone()))))
+ .subdir(
+ "snapshots",
+ Router::new()
+ .get(
+ ApiMethod::new(
+ list_snapshots,
+ ObjectSchema::new("List backup groups.")
+ .required("store", store_schema.clone())
+ .required("backup-type", StringSchema::new("Backup type."))
+ .required("backup-id", StringSchema::new("Backup ID."))
+ )
+ )
+ .delete(
+ ApiMethod::new(
+ delete_snapshots,
+ ObjectSchema::new("Delete backup snapshot.")
+ .required("store", store_schema.clone())
+ .required("backup-type", StringSchema::new("Backup type."))
+ .required("backup-id", StringSchema::new("Backup ID."))
+ .required("backup-time", IntegerSchema::new("Backup time (Unix epoch.)")
+ .minimum(1547797308))
+ )
+ )
+ )
+ .subdir(
+ "prune",
+ Router::new()
+ .post(api_method_prune()));