Rewrite StoreIdIterator extensions
This patch reimplements the iterator extensions. As we iterate (in StoreIdIterator) over Result<StoreId> now anyways, we don't need the extensions for Result iterators anymore. This patch rewrites the extensions to be more simple in every way and generic over the error type in the iterator. All the errors have to do is implement From<StoreError>, which is what they do when linking the generated error types with error_chain to the libimagstore error types.
This commit is contained in:
parent
6ee3f4a3f6
commit
d08c27e623
2 changed files with 51 additions and 127 deletions
|
@ -17,111 +17,7 @@
|
||||||
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
||||||
//
|
//
|
||||||
|
|
||||||
macro_rules! mk_iterator {
|
|
||||||
{
|
|
||||||
modname = $modname:ident,
|
|
||||||
itername = $itername:ident,
|
|
||||||
iteryield = $yield:ty,
|
|
||||||
extname = $extname:ident,
|
|
||||||
extfnname = $extfnname:ident,
|
|
||||||
fun = $fun:expr
|
|
||||||
} => {
|
|
||||||
use storeid::StoreId;
|
|
||||||
#[allow(unused_imports)]
|
|
||||||
use store::FileLockEntry;
|
|
||||||
use store::Store;
|
|
||||||
use error::Result;
|
|
||||||
|
|
||||||
pub struct $itername<'a>(Box<Iterator<Item = Result<StoreId>> + 'a>, &'a Store);
|
|
||||||
|
|
||||||
impl<'a> $itername<'a> {
|
|
||||||
pub fn new(inner: Box<Iterator<Item = Result<StoreId>> + 'a>, store: &'a Store) -> Self {
|
|
||||||
$itername(inner, store)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a> Iterator for $itername<'a> {
|
|
||||||
type Item = Result<$yield>;
|
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
|
||||||
self.0.next().map(|id| $fun(id?, self.1))
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub trait $extname<'a> {
|
|
||||||
fn $extfnname(self, store: &'a Store) -> $itername<'a>;
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, I> $extname<'a> for I
|
|
||||||
where I: Iterator<Item = Result<StoreId>> + 'a
|
|
||||||
{
|
|
||||||
fn $extfnname(self, store: &'a Store) -> $itername<'a> {
|
|
||||||
$itername(Box::new(self), store)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
use error::StoreError;
|
|
||||||
|
|
||||||
pub enum ExtensionError<E> {
|
|
||||||
Forwarded(E),
|
|
||||||
StoreError(StoreError)
|
|
||||||
}
|
|
||||||
|
|
||||||
macro_rules! mk_iterator_mod {
|
macro_rules! mk_iterator_mod {
|
||||||
{
|
|
||||||
modname = $modname:ident,
|
|
||||||
itername = $itername:ident,
|
|
||||||
iteryield = $yield:ty,
|
|
||||||
extname = $extname:ident,
|
|
||||||
extfnname = $extfnname:ident,
|
|
||||||
fun = $fun:expr,
|
|
||||||
resultitername = $resultitername:ident,
|
|
||||||
resultextname = $resultextname:ident
|
|
||||||
} => {
|
|
||||||
pub mod $modname {
|
|
||||||
mk_iterator! {
|
|
||||||
modname = $modname,
|
|
||||||
itername = $itername,
|
|
||||||
iteryield = $yield,
|
|
||||||
extname = $extname,
|
|
||||||
extfnname = $extfnname,
|
|
||||||
fun = $fun
|
|
||||||
}
|
|
||||||
|
|
||||||
use std::result::Result as RResult;
|
|
||||||
|
|
||||||
pub struct $resultitername<'a, I>(I, &'a Store);
|
|
||||||
|
|
||||||
impl<'a, I, E> Iterator for $resultitername<'a, I>
|
|
||||||
where I: Iterator<Item = RResult<StoreId, E>>
|
|
||||||
{
|
|
||||||
type Item = RResult<$yield, $crate::iter::ExtensionError<E>>;
|
|
||||||
|
|
||||||
fn next(&mut self) -> Option<Self::Item> {
|
|
||||||
match self.0.next() {
|
|
||||||
Some(Ok(sid)) => Some($fun(sid, self.1).map_err($crate::iter::ExtensionError::StoreError)),
|
|
||||||
Some(Err(e)) => Some(Err($crate::iter::ExtensionError::Forwarded(e))),
|
|
||||||
None => None,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
pub trait $resultextname<'a> : Iterator {
|
|
||||||
fn $extfnname(self, store: &'a Store) -> $resultitername<'a, Self>
|
|
||||||
where Self: Sized
|
|
||||||
{
|
|
||||||
$resultitername(self, store)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl<'a, I> $resultextname<'a> for I
|
|
||||||
where I: Iterator
|
|
||||||
{ /* empty */ }
|
|
||||||
}
|
|
||||||
};
|
|
||||||
|
|
||||||
{
|
{
|
||||||
modname = $modname:ident,
|
modname = $modname:ident,
|
||||||
itername = $itername:ident,
|
itername = $itername:ident,
|
||||||
|
@ -131,13 +27,47 @@ macro_rules! mk_iterator_mod {
|
||||||
fun = $fun:expr
|
fun = $fun:expr
|
||||||
} => {
|
} => {
|
||||||
pub mod $modname {
|
pub mod $modname {
|
||||||
mk_iterator! {
|
use storeid::StoreId;
|
||||||
modname = $modname,
|
#[allow(unused_imports)]
|
||||||
itername = $itername,
|
use store::FileLockEntry;
|
||||||
iteryield = $yield,
|
use store::Store;
|
||||||
extname = $extname,
|
use error::StoreError;
|
||||||
extfnname = $extfnname,
|
use std::result::Result as RResult;
|
||||||
fun = $fun
|
|
||||||
|
pub struct $itername<'a, E>(Box<Iterator<Item = RResult<StoreId, E>> + 'a>, &'a Store)
|
||||||
|
where E: From<StoreError>;
|
||||||
|
|
||||||
|
impl<'a, E> $itername<'a, E>
|
||||||
|
where E: From<StoreError>
|
||||||
|
{
|
||||||
|
pub fn new(inner: Box<Iterator<Item = RResult<StoreId, E>> + 'a>, store: &'a Store) -> Self {
|
||||||
|
$itername(inner, store)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, E> Iterator for $itername<'a, E>
|
||||||
|
where E: From<StoreError>
|
||||||
|
{
|
||||||
|
type Item = RResult<$yield, E>;
|
||||||
|
|
||||||
|
fn next(&mut self) -> Option<Self::Item> {
|
||||||
|
self.0.next().map(|id| $fun(id?, self.1).map_err(E::from))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
pub trait $extname<'a, E>
|
||||||
|
where E: From<StoreError>
|
||||||
|
{
|
||||||
|
fn $extfnname(self, store: &'a Store) -> $itername<'a, E>;
|
||||||
|
}
|
||||||
|
|
||||||
|
impl<'a, I, E> $extname<'a, E> for I
|
||||||
|
where I: Iterator<Item = RResult<StoreId, E>> + 'a,
|
||||||
|
E: From<StoreError>
|
||||||
|
{
|
||||||
|
fn $extfnname(self, store: &'a Store) -> $itername<'a, E> {
|
||||||
|
$itername(Box::new(self), store)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
@ -149,9 +79,7 @@ mk_iterator_mod! {
|
||||||
iteryield = FileLockEntry<'a>,
|
iteryield = FileLockEntry<'a>,
|
||||||
extname = StoreIdCreateIteratorExtension,
|
extname = StoreIdCreateIteratorExtension,
|
||||||
extfnname = into_create_iter,
|
extfnname = into_create_iter,
|
||||||
fun = |id: StoreId, store: &'a Store| store.create(id),
|
fun = |id: StoreId, store: &'a Store| store.create(id)
|
||||||
resultitername = StoreCreateResultIterator,
|
|
||||||
resultextname = StoreIdCreateResultIteratorExtension
|
|
||||||
}
|
}
|
||||||
|
|
||||||
mk_iterator_mod! {
|
mk_iterator_mod! {
|
||||||
|
@ -160,9 +88,7 @@ mk_iterator_mod! {
|
||||||
iteryield = (),
|
iteryield = (),
|
||||||
extname = StoreIdDeleteIteratorExtension,
|
extname = StoreIdDeleteIteratorExtension,
|
||||||
extfnname = into_delete_iter,
|
extfnname = into_delete_iter,
|
||||||
fun = |id: StoreId, store: &'a Store| store.delete(id),
|
fun = |id: StoreId, store: &'a Store| store.delete(id)
|
||||||
resultitername = StoreDeleteResultIterator,
|
|
||||||
resultextname = StoreIdDeleteResultIteratorExtension
|
|
||||||
}
|
}
|
||||||
|
|
||||||
mk_iterator_mod! {
|
mk_iterator_mod! {
|
||||||
|
@ -171,9 +97,7 @@ mk_iterator_mod! {
|
||||||
iteryield = Option<FileLockEntry<'a>>,
|
iteryield = Option<FileLockEntry<'a>>,
|
||||||
extname = StoreIdGetIteratorExtension,
|
extname = StoreIdGetIteratorExtension,
|
||||||
extfnname = into_get_iter,
|
extfnname = into_get_iter,
|
||||||
fun = |id: StoreId, store: &'a Store| store.get(id),
|
fun = |id: StoreId, store: &'a Store| store.get(id)
|
||||||
resultitername = StoreGetResultIterator,
|
|
||||||
resultextname = StoreIdGetResultIteratorExtension
|
|
||||||
}
|
}
|
||||||
|
|
||||||
mk_iterator_mod! {
|
mk_iterator_mod! {
|
||||||
|
@ -182,9 +106,7 @@ mk_iterator_mod! {
|
||||||
iteryield = FileLockEntry<'a>,
|
iteryield = FileLockEntry<'a>,
|
||||||
extname = StoreIdRetrieveIteratorExtension,
|
extname = StoreIdRetrieveIteratorExtension,
|
||||||
extfnname = into_retrieve_iter,
|
extfnname = into_retrieve_iter,
|
||||||
fun = |id: StoreId, store: &'a Store| store.retrieve(id),
|
fun = |id: StoreId, store: &'a Store| store.retrieve(id)
|
||||||
resultitername = StoreRetrieveResultIterator,
|
|
||||||
resultextname = StoreIdRetrieveResultIteratorExtension
|
|
||||||
}
|
}
|
||||||
|
|
||||||
#[cfg(test)]
|
#[cfg(test)]
|
||||||
|
|
|
@ -287,6 +287,8 @@ impl<'a> Iterator for StoreIdIteratorWithStore<'a> {
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
use error::StoreError;
|
||||||
|
|
||||||
impl<'a> StoreIdIteratorWithStore<'a> {
|
impl<'a> StoreIdIteratorWithStore<'a> {
|
||||||
|
|
||||||
pub fn new(iter: Box<Iterator<Item = Result<StoreId>>>, store: &'a Store) -> Self {
|
pub fn new(iter: Box<Iterator<Item = Result<StoreId>>>, store: &'a Store) -> Self {
|
||||||
|
@ -300,7 +302,7 @@ impl<'a> StoreIdIteratorWithStore<'a> {
|
||||||
/// Transform the iterator into a StoreCreateIterator
|
/// Transform the iterator into a StoreCreateIterator
|
||||||
///
|
///
|
||||||
/// This immitates the API from `libimagstore::iter`.
|
/// This immitates the API from `libimagstore::iter`.
|
||||||
pub fn into_create_iter(self) -> StoreCreateIterator<'a> {
|
pub fn into_create_iter(self) -> StoreCreateIterator<'a, StoreError> {
|
||||||
StoreCreateIterator::new(Box::new(self.0), self.1)
|
StoreCreateIterator::new(Box::new(self.0), self.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -308,7 +310,7 @@ impl<'a> StoreIdIteratorWithStore<'a> {
|
||||||
///
|
///
|
||||||
///
|
///
|
||||||
/// This immitates the API from `libimagstore::iter`.
|
/// This immitates the API from `libimagstore::iter`.
|
||||||
pub fn into_delete_iter(self) -> StoreDeleteIterator<'a> {
|
pub fn into_delete_iter(self) -> StoreDeleteIterator<'a, StoreError> {
|
||||||
StoreDeleteIterator::new(Box::new(self.0), self.1)
|
StoreDeleteIterator::new(Box::new(self.0), self.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -316,7 +318,7 @@ impl<'a> StoreIdIteratorWithStore<'a> {
|
||||||
///
|
///
|
||||||
///
|
///
|
||||||
/// This immitates the API from `libimagstore::iter`.
|
/// This immitates the API from `libimagstore::iter`.
|
||||||
pub fn into_get_iter(self) -> StoreGetIterator<'a> {
|
pub fn into_get_iter(self) -> StoreGetIterator<'a, StoreError> {
|
||||||
StoreGetIterator::new(Box::new(self.0), self.1)
|
StoreGetIterator::new(Box::new(self.0), self.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -324,7 +326,7 @@ impl<'a> StoreIdIteratorWithStore<'a> {
|
||||||
///
|
///
|
||||||
///
|
///
|
||||||
/// This immitates the API from `libimagstore::iter`.
|
/// This immitates the API from `libimagstore::iter`.
|
||||||
pub fn into_retrieve_iter(self) -> StoreRetrieveIterator<'a> {
|
pub fn into_retrieve_iter(self) -> StoreRetrieveIterator<'a, StoreError> {
|
||||||
StoreRetrieveIterator::new(Box::new(self.0), self.1)
|
StoreRetrieveIterator::new(Box::new(self.0), self.1)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|
Loading…
Reference in a new issue