mirror of
https://github.com/meilisearch/MeiliSearch
synced 2024-11-22 21:04:27 +01:00
Merge #3537
3537: fix a bug where the filestore could try to parse its own tmp file and fail (main) r=irevoire a=curquiza Co-authored-by: Tamo <tamo@meilisearch.com>
This commit is contained in:
commit
442c1e36de
@ -116,10 +116,20 @@ impl FileStore {
|
|||||||
|
|
||||||
/// List the Uuids of the files in the FileStore
|
/// List the Uuids of the files in the FileStore
|
||||||
pub fn all_uuids(&self) -> Result<impl Iterator<Item = Result<Uuid>>> {
|
pub fn all_uuids(&self) -> Result<impl Iterator<Item = Result<Uuid>>> {
|
||||||
Ok(self.path.read_dir()?.map(|entry| {
|
Ok(self.path.read_dir()?.filter_map(|entry| {
|
||||||
Ok(Uuid::from_str(
|
let file_name = match entry {
|
||||||
entry?.file_name().to_str().ok_or(Error::CouldNotParseFileNameAsUtf8)?,
|
Ok(entry) => entry.file_name(),
|
||||||
)?)
|
Err(e) => return Some(Err(e.into())),
|
||||||
|
};
|
||||||
|
let file_name = match file_name.to_str() {
|
||||||
|
Some(file_name) => file_name,
|
||||||
|
None => return Some(Err(Error::CouldNotParseFileNameAsUtf8)),
|
||||||
|
};
|
||||||
|
if file_name.starts_with('.') {
|
||||||
|
None
|
||||||
|
} else {
|
||||||
|
Some(Uuid::from_str(file_name).map_err(|e| e.into()))
|
||||||
|
}
|
||||||
}))
|
}))
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
@ -135,3 +145,34 @@ impl File {
|
|||||||
Ok(())
|
Ok(())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
#[cfg(test)]
|
||||||
|
mod test {
|
||||||
|
use std::io::Write;
|
||||||
|
|
||||||
|
use tempfile::TempDir;
|
||||||
|
|
||||||
|
use super::*;
|
||||||
|
|
||||||
|
#[test]
|
||||||
|
fn all_uuids() {
|
||||||
|
let dir = TempDir::new().unwrap();
|
||||||
|
let fs = FileStore::new(dir.path()).unwrap();
|
||||||
|
let (uuid, mut file) = fs.new_update().unwrap();
|
||||||
|
file.write_all(b"Hello world").unwrap();
|
||||||
|
file.persist().unwrap();
|
||||||
|
let all_uuids = fs.all_uuids().unwrap().collect::<Result<Vec<_>>>().unwrap();
|
||||||
|
assert_eq!(all_uuids, vec![uuid]);
|
||||||
|
|
||||||
|
let (uuid2, file) = fs.new_update().unwrap();
|
||||||
|
let all_uuids = fs.all_uuids().unwrap().collect::<Result<Vec<_>>>().unwrap();
|
||||||
|
assert_eq!(all_uuids, vec![uuid]);
|
||||||
|
|
||||||
|
file.persist().unwrap();
|
||||||
|
let mut all_uuids = fs.all_uuids().unwrap().collect::<Result<Vec<_>>>().unwrap();
|
||||||
|
all_uuids.sort();
|
||||||
|
let mut expected = vec![uuid, uuid2];
|
||||||
|
expected.sort();
|
||||||
|
assert_eq!(all_uuids, expected);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user