Extract createdAt and updatedAt from v3 dump

This commit is contained in:
Francis Murillo 2023-01-13 22:45:45 +08:00
parent e27bb8ab3e
commit a97281af08
No known key found for this signature in database
GPG Key ID: 2D816FFCF0F44A70
2 changed files with 67 additions and 15 deletions

View File

@ -114,6 +114,10 @@ impl V3Reader {
V3IndexReader::new( V3IndexReader::new(
index.uid.clone(), index.uid.clone(),
&self.dump.path().join("indexes").join(index.uuid.to_string()), &self.dump.path().join("indexes").join(index.uuid.to_string()),
&index,
BufReader::new(
File::open(self.dump.path().join("updates").join("data.jsonl")).unwrap(),
),
) )
})) }))
} }
@ -147,6 +151,7 @@ impl V3Reader {
} }
} }
#[derive(Debug)]
pub struct V3IndexReader { pub struct V3IndexReader {
metadata: IndexMetadata, metadata: IndexMetadata,
settings: Settings<Checked>, settings: Settings<Checked>,
@ -155,16 +160,43 @@ pub struct V3IndexReader {
} }
impl V3IndexReader { impl V3IndexReader {
pub fn new(name: String, path: &Path) -> Result<Self> { pub fn new(
name: String,
path: &Path,
index_uuid: &IndexUuid,
tasks: BufReader<File>,
) -> Result<Self> {
let meta = File::open(path.join("meta.json"))?; let meta = File::open(path.join("meta.json"))?;
let meta: DumpMeta = serde_json::from_reader(meta)?; let meta: DumpMeta = serde_json::from_reader(meta)?;
let mut created_entry = (u64::MAX, None);
let mut updated_entry = (u64::MIN, None);
for line in tasks.lines() {
let task: Task = serde_json::from_str(&line?)?;
if &task.uuid == &index_uuid.uuid {
let update_id = task.update.id();
if update_id <= created_entry.0 {
created_entry.0 = update_id;
created_entry.1 = task.update.created_at();
}
if update_id >= updated_entry.0 {
updated_entry.0 = update_id;
updated_entry.1 = task.update.processed_at();
}
}
}
let current_time = OffsetDateTime::now_utc();
let metadata = IndexMetadata { let metadata = IndexMetadata {
uid: name, uid: name,
primary_key: meta.primary_key, primary_key: meta.primary_key,
// FIXME: Iterate over the whole task queue to find the creation and last update date. created_at: created_entry.1.unwrap_or(current_time),
created_at: OffsetDateTime::now_utc(), updated_at: updated_entry.1.unwrap_or(current_time),
updated_at: OffsetDateTime::now_utc(),
}; };
let ret = V3IndexReader { let ret = V3IndexReader {
@ -263,12 +295,12 @@ pub(crate) mod test {
assert!(indexes.is_empty()); assert!(indexes.is_empty());
// products // products
insta::assert_json_snapshot!(products.metadata(), { ".createdAt" => "[now]", ".updatedAt" => "[now]" }, @r###" insta::assert_json_snapshot!(products.metadata(), @r###"
{ {
"uid": "products", "uid": "products",
"primaryKey": "sku", "primaryKey": "sku",
"createdAt": "[now]", "createdAt": "2022-10-07T11:38:54.734594617Z",
"updatedAt": "[now]" "updatedAt": "2022-10-07T11:38:55.963185778Z"
} }
"###); "###);
@ -278,12 +310,12 @@ pub(crate) mod test {
meili_snap::snapshot_hash!(format!("{:#?}", documents), @"548284a84de510f71e88e6cdea495cf5"); meili_snap::snapshot_hash!(format!("{:#?}", documents), @"548284a84de510f71e88e6cdea495cf5");
// movies // movies
insta::assert_json_snapshot!(movies.metadata(), { ".createdAt" => "[now]", ".updatedAt" => "[now]" }, @r###" insta::assert_json_snapshot!(movies.metadata(), @r###"
{ {
"uid": "movies", "uid": "movies",
"primaryKey": "id", "primaryKey": "id",
"createdAt": "[now]", "createdAt": "2022-10-07T11:38:54.004402239Z",
"updatedAt": "[now]" "updatedAt": "2022-10-07T11:39:04.188852537Z"
} }
"###); "###);
@ -293,11 +325,11 @@ pub(crate) mod test {
meili_snap::snapshot_hash!(format!("{:#?}", documents), @"d153b5a81d8b3cdcbe1dec270b574022"); meili_snap::snapshot_hash!(format!("{:#?}", documents), @"d153b5a81d8b3cdcbe1dec270b574022");
// movies2 // movies2
insta::assert_json_snapshot!(movies2.metadata(), { ".createdAt" => "[now]", ".updatedAt" => "[now]" }, @r###" insta::assert_json_snapshot!(movies2.metadata(), { ".updatedAt" => "[now]" }, @r###"
{ {
"uid": "movies_2", "uid": "movies_2",
"primaryKey": null, "primaryKey": null,
"createdAt": "[now]", "createdAt": "2022-10-07T11:39:03.703667164Z",
"updatedAt": "[now]" "updatedAt": "[now]"
} }
"###); "###);
@ -308,12 +340,12 @@ pub(crate) mod test {
meili_snap::snapshot_hash!(format!("{:#?}", documents), @"d751713988987e9331980363e24189ce"); meili_snap::snapshot_hash!(format!("{:#?}", documents), @"d751713988987e9331980363e24189ce");
// spells // spells
insta::assert_json_snapshot!(spells.metadata(), { ".createdAt" => "[now]", ".updatedAt" => "[now]" }, @r###" insta::assert_json_snapshot!(spells.metadata(), @r###"
{ {
"uid": "dnd_spells", "uid": "dnd_spells",
"primaryKey": "index", "primaryKey": "index",
"createdAt": "[now]", "createdAt": "2022-10-07T11:38:56.263041061Z",
"updatedAt": "[now]" "updatedAt": "2022-10-07T11:38:56.521004328Z"
} }
"###); "###);

View File

@ -74,6 +74,26 @@ impl UpdateStatus {
_ => None, _ => None,
} }
} }
pub fn created_at(&self) -> Option<OffsetDateTime> {
match self {
UpdateStatus::Processing(u) => Some(u.from.enqueued_at.clone()),
UpdateStatus::Enqueued(u) => Some(u.enqueued_at.clone()),
UpdateStatus::Processed(u) => Some(u.from.from.enqueued_at.clone()),
UpdateStatus::Aborted(u) => Some(u.from.enqueued_at.clone()),
UpdateStatus::Failed(u) => Some(u.from.from.enqueued_at.clone()),
}
}
pub fn processed_at(&self) -> Option<OffsetDateTime> {
match self {
UpdateStatus::Processing(u) => Some(u.started_processing_at.clone()),
UpdateStatus::Enqueued(_) => None,
UpdateStatus::Processed(u) => Some(u.processed_at.clone()),
UpdateStatus::Aborted(_) => None,
UpdateStatus::Failed(u) => Some(u.failed_at.clone()),
}
}
} }
#[derive(Debug, Deserialize, Clone)] #[derive(Debug, Deserialize, Clone)]