Merge pull request #145 from neithernut/fix-store-retrieve

Fix Store::retrieve()
This commit is contained in:
Matthias Beyer 2016-01-29 16:29:38 +01:00
commit 3b51328aa7
1 changed files with 11 additions and 9 deletions

View File

@ -142,14 +142,16 @@ impl Store {
/// Borrow a given Entry. When the `FileLockEntry` is either `update`d or
/// dropped, the new Entry is written to disk
pub fn retrieve<'a>(&'a self, id: StoreId) -> Result<FileLockEntry<'a>> {
let hsmap = self.entries.write();
if hsmap.is_err() {
return Err(StoreError::new(StoreErrorKind::LockPoisoned, None))
}
hsmap.unwrap().get_mut(&id)
.ok_or(StoreError::new(StoreErrorKind::IdNotFound, None))
.and_then(|store_entry| store_entry.get_entry())
.and_then(|entry| Ok(FileLockEntry::new(self, entry, id)))
self.entries
.write()
.map_err(|_| StoreError::new(StoreErrorKind::LockPoisoned, None))
.and_then(|mut es| {
let mut se = es.entry(id.clone()).or_insert_with(|| StoreEntry::new(id.clone()));
let entry = se.get_entry();
se.status = StoreEntryStatus::Borrowed;
entry
})
.map(|e| FileLockEntry::new(self, e, id))
}
/// Iterate over all StoreIds for one module name