|
|
|
@ -6,6 +6,7 @@ use std::time::{Duration, Instant};
|
|
|
|
|
use tokio::sync::watch;
|
|
|
|
|
|
|
|
|
|
use garage_util::background::*;
|
|
|
|
|
use garage_util::data::*;
|
|
|
|
|
use garage_util::error::{Error, OkOrMessage};
|
|
|
|
|
use garage_util::persister::PersisterShared;
|
|
|
|
|
use garage_util::time::*;
|
|
|
|
@ -165,6 +166,7 @@ impl Worker for LifecycleWorker {
|
|
|
|
|
let object = self.garage.object_table.data.decode_entry(&object_bytes)?;
|
|
|
|
|
process_object(
|
|
|
|
|
&self.garage,
|
|
|
|
|
*date,
|
|
|
|
|
object,
|
|
|
|
|
objects_expired,
|
|
|
|
|
mpu_aborted,
|
|
|
|
@ -184,7 +186,7 @@ impl Worker for LifecycleWorker {
|
|
|
|
|
match &self.state {
|
|
|
|
|
State::Completed(d) => {
|
|
|
|
|
let now = now_msec();
|
|
|
|
|
let next_start = midnight_ts(d.succ());
|
|
|
|
|
let next_start = midnight_ts(d.succ_opt().expect("no next day"));
|
|
|
|
|
if now < next_start {
|
|
|
|
|
tokio::time::sleep_until(
|
|
|
|
|
(Instant::now() + Duration::from_millis(next_start - now)).into(),
|
|
|
|
@ -208,6 +210,7 @@ impl Worker for LifecycleWorker {
|
|
|
|
|
|
|
|
|
|
async fn process_object(
|
|
|
|
|
garage: &Arc<Garage>,
|
|
|
|
|
now_date: NaiveDate,
|
|
|
|
|
object: Object,
|
|
|
|
|
objects_expired: &mut usize,
|
|
|
|
|
mpu_aborted: &mut usize,
|
|
|
|
@ -229,24 +232,113 @@ async fn process_object(
|
|
|
|
|
.unwrap_or_default();
|
|
|
|
|
|
|
|
|
|
for rule in lifecycle_policy.iter() {
|
|
|
|
|
todo!()
|
|
|
|
|
if let Some(pfx) = &rule.filter.prefix {
|
|
|
|
|
if !object.key.starts_with(pfx) {
|
|
|
|
|
continue;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if let Some(expire) = &rule.expiration {
|
|
|
|
|
if let Some(current_version) = object.versions().iter().rev().find(|v| v.is_data()) {
|
|
|
|
|
let version_date = next_date(current_version.timestamp);
|
|
|
|
|
|
|
|
|
|
let current_version_data = match ¤t_version.state {
|
|
|
|
|
ObjectVersionState::Complete(c) => c,
|
|
|
|
|
_ => unreachable!(),
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
let size_match = check_size_filter(current_version_data, &rule.filter);
|
|
|
|
|
let date_match = match expire {
|
|
|
|
|
LifecycleExpiration::AfterDays(n_days) => {
|
|
|
|
|
(now_date - version_date) >= chrono::Duration::days(*n_days as i64)
|
|
|
|
|
}
|
|
|
|
|
LifecycleExpiration::AtDate(exp_date) => now_date >= *exp_date,
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
if size_match && date_match {
|
|
|
|
|
// Delete expired version
|
|
|
|
|
let deleted_object = Object::new(
|
|
|
|
|
object.bucket_id,
|
|
|
|
|
object.key.clone(),
|
|
|
|
|
vec![ObjectVersion {
|
|
|
|
|
uuid: gen_uuid(),
|
|
|
|
|
timestamp: std::cmp::max(now_msec(), current_version.timestamp + 1),
|
|
|
|
|
state: ObjectVersionState::Complete(ObjectVersionData::DeleteMarker),
|
|
|
|
|
}],
|
|
|
|
|
);
|
|
|
|
|
garage.object_table.insert(&deleted_object).await?;
|
|
|
|
|
*objects_expired += 1;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
if let Some(abort_mpu_days) = &rule.abort_incomplete_mpu_days {
|
|
|
|
|
let aborted_versions = object
|
|
|
|
|
.versions()
|
|
|
|
|
.iter()
|
|
|
|
|
.filter_map(|v| {
|
|
|
|
|
let version_date = next_date(v.timestamp);
|
|
|
|
|
match &v.state {
|
|
|
|
|
ObjectVersionState::Uploading { .. }
|
|
|
|
|
if (now_date - version_date)
|
|
|
|
|
>= chrono::Duration::days(*abort_mpu_days as i64) =>
|
|
|
|
|
{
|
|
|
|
|
Some(ObjectVersion {
|
|
|
|
|
state: ObjectVersionState::Aborted,
|
|
|
|
|
..*v
|
|
|
|
|
})
|
|
|
|
|
}
|
|
|
|
|
_ => None,
|
|
|
|
|
}
|
|
|
|
|
})
|
|
|
|
|
.collect::<Vec<_>>();
|
|
|
|
|
if !aborted_versions.is_empty() {
|
|
|
|
|
// Insert aborted mpu info
|
|
|
|
|
let n_aborted = aborted_versions.len();
|
|
|
|
|
let aborted_object =
|
|
|
|
|
Object::new(object.bucket_id, object.key.clone(), aborted_versions);
|
|
|
|
|
garage.object_table.insert(&aborted_object).await?;
|
|
|
|
|
*mpu_aborted += n_aborted;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
*last_bucket = Some(bucket);
|
|
|
|
|
Ok(())
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn check_size_filter(version_data: &ObjectVersionData, filter: &LifecycleFilter) -> bool {
|
|
|
|
|
let size = match version_data {
|
|
|
|
|
ObjectVersionData::Inline(meta, _) | ObjectVersionData::FirstBlock(meta, _) => meta.size,
|
|
|
|
|
_ => unreachable!(),
|
|
|
|
|
};
|
|
|
|
|
if let Some(size_gt) = filter.size_gt {
|
|
|
|
|
if !(size > size_gt) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
if let Some(size_lt) = filter.size_lt {
|
|
|
|
|
if !(size < size_lt) {
|
|
|
|
|
return false;
|
|
|
|
|
}
|
|
|
|
|
}
|
|
|
|
|
return true;
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn midnight_ts(date: NaiveDate) -> u64 {
|
|
|
|
|
date.and_hms(0, 0, 0).timestamp_millis() as u64
|
|
|
|
|
date.and_hms_opt(0, 0, 0)
|
|
|
|
|
.expect("midnight does not exist")
|
|
|
|
|
.timestamp_millis() as u64
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn next_date(ts: u64) -> NaiveDate {
|
|
|
|
|
NaiveDateTime::from_timestamp_millis(ts as i64)
|
|
|
|
|
.expect("bad timestamp")
|
|
|
|
|
.date()
|
|
|
|
|
.succ()
|
|
|
|
|
.succ_opt()
|
|
|
|
|
.expect("no next day")
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
fn today() -> NaiveDate {
|
|
|
|
|
Utc::today().naive_utc()
|
|
|
|
|
Utc::now().naive_utc().date()
|
|
|
|
|
}
|
|
|
|
|