Add a fold_ok
utility.
Add a utility that folds an iterator into a result and uses it to reduce boilerplate in the codebase.
This commit is contained in:
parent
3a2010c35b
commit
2c40b8734e
8 changed files with 93 additions and 83 deletions
|
@ -14,3 +14,6 @@ path = "../libimagstore"
|
||||||
[dependencies.libimagerror]
|
[dependencies.libimagerror]
|
||||||
path = "../libimagerror"
|
path = "../libimagerror"
|
||||||
|
|
||||||
|
[dependencies.libimagutil]
|
||||||
|
path = "../libimagutil"
|
||||||
|
|
||||||
|
|
|
@ -19,6 +19,7 @@ extern crate clap;
|
||||||
extern crate toml;
|
extern crate toml;
|
||||||
|
|
||||||
extern crate libimagstore;
|
extern crate libimagstore;
|
||||||
|
extern crate libimagutil;
|
||||||
#[macro_use] extern crate libimagerror;
|
#[macro_use] extern crate libimagerror;
|
||||||
|
|
||||||
pub mod cli;
|
pub mod cli;
|
||||||
|
|
|
@ -5,6 +5,7 @@ use lister::Lister;
|
||||||
use result::Result;
|
use result::Result;
|
||||||
|
|
||||||
use libimagstore::store::FileLockEntry;
|
use libimagstore::store::FileLockEntry;
|
||||||
|
use libimagutil::iter::fold_ok;
|
||||||
|
|
||||||
pub struct LineLister<'a> {
|
pub struct LineLister<'a> {
|
||||||
unknown_output: &'a str,
|
unknown_output: &'a str,
|
||||||
|
@ -26,13 +27,10 @@ impl<'a> Lister for LineLister<'a> {
|
||||||
use error::ListError as LE;
|
use error::ListError as LE;
|
||||||
use error::ListErrorKind as LEK;
|
use error::ListErrorKind as LEK;
|
||||||
|
|
||||||
entries.fold(Ok(()), |accu, entry| {
|
fold_ok(entries, |entry| {
|
||||||
accu.and_then(|_| {
|
write!(stdout(), "{:?}\n", entry.get_location().to_str().unwrap_or(self.unknown_output))
|
||||||
write!(stdout(), "{:?}\n",
|
|
||||||
entry.get_location().to_str().unwrap_or(self.unknown_output))
|
|
||||||
.map_err(|e| LE::new(LEK::FormatError, Some(Box::new(e))))
|
.map_err(|e| LE::new(LEK::FormatError, Some(Box::new(e))))
|
||||||
})
|
})
|
||||||
})
|
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -5,6 +5,7 @@ use lister::Lister;
|
||||||
use result::Result;
|
use result::Result;
|
||||||
|
|
||||||
use libimagstore::store::FileLockEntry;
|
use libimagstore::store::FileLockEntry;
|
||||||
|
use libimagutil::iter::fold_ok;
|
||||||
|
|
||||||
pub struct PathLister {
|
pub struct PathLister {
|
||||||
absolute: bool,
|
absolute: bool,
|
||||||
|
@ -26,8 +27,8 @@ impl Lister for PathLister {
|
||||||
use error::ListError as LE;
|
use error::ListError as LE;
|
||||||
use error::ListErrorKind as LEK;
|
use error::ListErrorKind as LEK;
|
||||||
|
|
||||||
entries.fold(Ok(()), |accu, entry| {
|
fold_ok(entries, |entry| {
|
||||||
accu.and_then(|_| Ok(entry.get_location().clone()))
|
Ok(entry.get_location().clone())
|
||||||
.and_then(|pb| {
|
.and_then(|pb| {
|
||||||
if self.absolute {
|
if self.absolute {
|
||||||
pb.canonicalize().map_err(|e| LE::new(LEK::FormatError, Some(Box::new(e))))
|
pb.canonicalize().map_err(|e| LE::new(LEK::FormatError, Some(Box::new(e))))
|
||||||
|
|
|
@ -1,4 +1,5 @@
|
||||||
use libimagerror::trace::trace_error;
|
use libimagerror::trace::trace_error;
|
||||||
|
use libimagutil::iter::fold_ok;
|
||||||
|
|
||||||
use store::FileLockEntry;
|
use store::FileLockEntry;
|
||||||
use storeid::StoreId;
|
use storeid::StoreId;
|
||||||
|
@ -45,10 +46,7 @@ impl StoreIdAccessor for Aspect {
|
||||||
return Err(HE::new(HEK::AccessTypeViolation, None));
|
return Err(HE::new(HEK::AccessTypeViolation, None));
|
||||||
}
|
}
|
||||||
|
|
||||||
accessors
|
fold_ok(accessors.iter(), |accessor| {
|
||||||
.iter()
|
|
||||||
.fold(Ok(()), |acc, accessor| {
|
|
||||||
acc.and_then(|_| {
|
|
||||||
let res = match accessor {
|
let res = match accessor {
|
||||||
&HDA::StoreIdAccess(accessor) => accessor.access(id),
|
&HDA::StoreIdAccess(accessor) => accessor.access(id),
|
||||||
_ => unreachable!(),
|
_ => unreachable!(),
|
||||||
|
@ -67,7 +65,6 @@ impl StoreIdAccessor for Aspect {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
})
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -91,8 +88,7 @@ impl MutableHookDataAccessor for Aspect {
|
||||||
// More sophisticated version would check whether there are _chunks_ of
|
// More sophisticated version would check whether there are _chunks_ of
|
||||||
// NonMutableAccess accessors and execute these chunks in parallel. We do not have
|
// NonMutableAccess accessors and execute these chunks in parallel. We do not have
|
||||||
// performance concerns yet, so this is okay.
|
// performance concerns yet, so this is okay.
|
||||||
accessors.iter().fold(Ok(()), |acc, accessor| {
|
fold_ok(accessors.iter(), |accessor| {
|
||||||
acc.and_then(|_| {
|
|
||||||
let res = match accessor {
|
let res = match accessor {
|
||||||
&HDA::MutableAccess(ref accessor) => accessor.access_mut(fle),
|
&HDA::MutableAccess(ref accessor) => accessor.access_mut(fle),
|
||||||
&HDA::NonMutableAccess(ref accessor) => accessor.access(fle),
|
&HDA::NonMutableAccess(ref accessor) => accessor.access(fle),
|
||||||
|
@ -112,7 +108,6 @@ impl MutableHookDataAccessor for Aspect {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
})
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -123,10 +118,7 @@ impl NonMutableHookDataAccessor for Aspect {
|
||||||
return Err(HE::new(HEK::AccessTypeViolation, None));
|
return Err(HE::new(HEK::AccessTypeViolation, None));
|
||||||
}
|
}
|
||||||
|
|
||||||
accessors
|
fold_ok(accessors.iter(), |accessor| {
|
||||||
.iter()
|
|
||||||
.fold(Ok(()), |acc, accessor| {
|
|
||||||
acc.and_then(|_| {
|
|
||||||
let res = match accessor {
|
let res = match accessor {
|
||||||
&HDA::NonMutableAccess(accessor) => accessor.access(fle),
|
&HDA::NonMutableAccess(accessor) => accessor.access(fle),
|
||||||
_ => unreachable!(),
|
_ => unreachable!(),
|
||||||
|
@ -145,7 +137,6 @@ impl NonMutableHookDataAccessor for Aspect {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
})
|
})
|
||||||
})
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
|
@ -37,6 +37,7 @@ use hook::position::HookPosition;
|
||||||
use hook::Hook;
|
use hook::Hook;
|
||||||
|
|
||||||
use libimagerror::into::IntoError;
|
use libimagerror::into::IntoError;
|
||||||
|
use libimagutil::iter::fold_ok;
|
||||||
|
|
||||||
use self::glob_store_iter::*;
|
use self::glob_store_iter::*;
|
||||||
|
|
||||||
|
@ -692,13 +693,16 @@ impl Store {
|
||||||
id: &StoreId)
|
id: &StoreId)
|
||||||
-> HookResult<()>
|
-> HookResult<()>
|
||||||
{
|
{
|
||||||
|
fold_ok(
|
||||||
match aspects.lock() {
|
match aspects.lock() {
|
||||||
Err(_) => return Err(HookErrorKind::HookExecutionError.into()),
|
Err(_) => return Err(HookErrorKind::HookExecutionError.into()),
|
||||||
Ok(g) => g
|
Ok(g) => g
|
||||||
}.iter().fold(Ok(()), |acc, aspect| {
|
}.iter(),
|
||||||
|
|aspect| {
|
||||||
debug!("[Aspect][exec]: {:?}", aspect);
|
debug!("[Aspect][exec]: {:?}", aspect);
|
||||||
acc.and_then(|_| (aspect as &StoreIdAccessor).access(id))
|
(aspect as &StoreIdAccessor).access(id)
|
||||||
}).map_err(Box::new).map_err(|e| HookErrorKind::HookExecutionError.into_error_with_cause(e))
|
}).map_err(Box::new)
|
||||||
|
.map_err(|e| HookErrorKind::HookExecutionError.into_error_with_cause(e))
|
||||||
}
|
}
|
||||||
|
|
||||||
fn execute_hooks_for_mut_file(&self,
|
fn execute_hooks_for_mut_file(&self,
|
||||||
|
@ -706,13 +710,16 @@ impl Store {
|
||||||
fle: &mut FileLockEntry)
|
fle: &mut FileLockEntry)
|
||||||
-> HookResult<()>
|
-> HookResult<()>
|
||||||
{
|
{
|
||||||
|
fold_ok(
|
||||||
match aspects.lock() {
|
match aspects.lock() {
|
||||||
Err(_) => return Err(HookErrorKind::HookExecutionError.into()),
|
Err(_) => return Err(HookErrorKind::HookExecutionError.into()),
|
||||||
Ok(g) => g
|
Ok(g) => g
|
||||||
}.iter().fold(Ok(()), |acc, aspect| {
|
}.iter(),
|
||||||
|
|aspect| {
|
||||||
debug!("[Aspect][exec]: {:?}", aspect);
|
debug!("[Aspect][exec]: {:?}", aspect);
|
||||||
acc.and_then(|_| aspect.access_mut(fle))
|
aspect.access_mut(fle)
|
||||||
}).map_err(Box::new).map_err(|e| HookErrorKind::HookExecutionError.into_error_with_cause(e))
|
}).map_err(Box::new)
|
||||||
|
.map_err(|e| HookErrorKind::HookExecutionError.into_error_with_cause(e))
|
||||||
}
|
}
|
||||||
|
|
||||||
}
|
}
|
||||||
|
|
8
libimagutil/src/iter.rs
Normal file
8
libimagutil/src/iter.rs
Normal file
|
@ -0,0 +1,8 @@
|
||||||
|
/// Processes `iter` returning the last successful result or the first error.
|
||||||
|
pub fn fold_ok<X, I, R, E, F>(iter: I, mut func: F) -> Result<R, E>
|
||||||
|
where I: Iterator<Item = X>,
|
||||||
|
R: Default,
|
||||||
|
F: FnMut(X) -> Result<R, E>
|
||||||
|
{
|
||||||
|
iter.fold(Ok(R::default()), |acc, item| acc.and_then(|_| func(item)))
|
||||||
|
}
|
|
@ -18,5 +18,6 @@
|
||||||
extern crate regex;
|
extern crate regex;
|
||||||
|
|
||||||
pub mod ismatch;
|
pub mod ismatch;
|
||||||
|
pub mod iter;
|
||||||
pub mod key_value_split;
|
pub mod key_value_split;
|
||||||
pub mod variants;
|
pub mod variants;
|
||||||
|
|
Loading…
Reference in a new issue