libimagcounter: Rewrite error handling

This commit is contained in:
Matthias Beyer 2017-09-03 16:09:55 +02:00
parent c9dbf76395
commit 31dc0eebc2
3 changed files with 23 additions and 22 deletions

View File

@ -38,7 +38,7 @@ use module_path::ModuleEntryPath;
use result::Result; use result::Result;
use error::CounterError as CE; use error::CounterError as CE;
use error::CounterErrorKind as CEK; use error::CounterErrorKind as CEK;
use error::error::MapErrInto; use error::ResultExt;
pub type CounterName = String; pub type CounterName = String;
@ -71,8 +71,8 @@ impl<'a> Counter<'a> {
let fle = { let fle = {
let id = try!(ModuleEntryPath::new(name.clone()) let id = try!(ModuleEntryPath::new(name.clone())
.into_storeid() .into_storeid()
.map_err_into(CEK::StoreWriteError)); .chain_err(|| CEK::StoreWriteError));
let mut lockentry = try!(store.create(id).map_err_into(CEK::StoreWriteError)); let mut lockentry = try!(store.create(id).chain_err(|| CEK::StoreWriteError));
{ {
let entry = lockentry.deref_mut(); let entry = lockentry.deref_mut();
@ -116,26 +116,26 @@ impl<'a> Counter<'a> {
pub fn inc(&mut self) -> Result<()> { pub fn inc(&mut self) -> Result<()> {
let header = self.fle.deref_mut().get_header_mut(); let header = self.fle.deref_mut().get_header_mut();
let query = String::from("counter.value"); let query = String::from("counter.value");
match try!(header.read(&query).map_err_into(CEK::StoreReadError)) { match try!(header.read(&query).chain_err(|| CEK::StoreReadError)) {
Some(&Value::Integer(i)) => { Some(&Value::Integer(i)) => {
header.set(&query, Value::Integer(i + 1)) header.set(&query, Value::Integer(i + 1))
.map_err_into(CEK::StoreWriteError) .chain_err(|| CEK::StoreWriteError)
.map(|_| ()) .map(|_| ())
}, },
_ => Err(CE::new(CEK::StoreReadError, None)), _ => Err(CE::from_kind(CEK::StoreReadError)),
} }
} }
pub fn dec(&mut self) -> Result<()> { pub fn dec(&mut self) -> Result<()> {
let header = self.fle.deref_mut().get_header_mut(); let header = self.fle.deref_mut().get_header_mut();
let query = String::from("counter.value"); let query = String::from("counter.value");
match try!(header.read(&query).map_err_into(CEK::StoreReadError)) { match try!(header.read(&query).chain_err(|| CEK::StoreReadError)) {
Some(&Value::Integer(i)) => { Some(&Value::Integer(i)) => {
header.set(&query, Value::Integer(i - 1)) header.set(&query, Value::Integer(i - 1))
.map_err_into(CEK::StoreWriteError) .chain_err(|| CEK::StoreWriteError)
.map(|_| ()) .map(|_| ())
}, },
_ => Err(CE::new(CEK::StoreReadError, None)), _ => Err(CE::from_kind(CEK::StoreReadError)),
} }
} }
@ -148,7 +148,7 @@ impl<'a> Counter<'a> {
.deref_mut() .deref_mut()
.get_header_mut() .get_header_mut()
.set(&String::from("counter.value"), Value::Integer(v)) .set(&String::from("counter.value"), Value::Integer(v))
.map_err_into(CEK::StoreWriteError) .chain_err(|| CEK::StoreWriteError)
.map(|_| ()) .map(|_| ())
} }
@ -187,7 +187,7 @@ impl<'a> Counter<'a> {
self.fle self.fle
.get_header() .get_header()
.read(&String::from(name)) .read(&String::from(name))
.map_err_into(CEK::StoreWriteError) .chain_err(|| CEK::StoreWriteError)
.and_then(f) .and_then(f)
} }
@ -195,7 +195,7 @@ impl<'a> Counter<'a> {
debug!("Loading counter: '{}'", name); debug!("Loading counter: '{}'", name);
let id = try!(ModuleEntryPath::new(name) let id = try!(ModuleEntryPath::new(name)
.into_storeid() .into_storeid()
.map_err_into(CEK::StoreWriteError)); .chain_err(|| CEK::StoreWriteError));
Counter::from_storeid(store, id) Counter::from_storeid(store, id)
} }
@ -203,14 +203,14 @@ impl<'a> Counter<'a> {
debug!("Deleting counter: '{}'", name); debug!("Deleting counter: '{}'", name);
let id = try!(ModuleEntryPath::new(name) let id = try!(ModuleEntryPath::new(name)
.into_storeid() .into_storeid()
.map_err_into(CEK::StoreWriteError)); .chain_err(|| CEK::StoreWriteError));
store.delete(id).map_err_into(CEK::StoreWriteError) store.delete(id).chain_err(|| CEK::StoreWriteError)
} }
pub fn all_counters(store: &Store) -> Result<CounterIterator> { pub fn all_counters(store: &Store) -> Result<CounterIterator> {
store.retrieve_for_module("counter") store.retrieve_for_module("counter")
.map(|iter| CounterIterator::new(store, iter)) .map(|iter| CounterIterator::new(store, iter))
.map_err_into(CEK::StoreReadError) .chain_err(|| CEK::StoreReadError)
} }
} }
@ -224,11 +224,11 @@ impl<'a> FromStoreId for Counter<'a> {
fn from_storeid(store: &Store, id: StoreId) -> Result<Counter> { fn from_storeid(store: &Store, id: StoreId) -> Result<Counter> {
debug!("Loading counter from storeid: '{:?}'", id); debug!("Loading counter from storeid: '{:?}'", id);
match store.retrieve(id) { match store.retrieve(id) {
Err(e) => Err(CE::new(CEK::StoreReadError, Some(Box::new(e)))), Err(e) => Err(e).chain_err(|| CEK::StoreReadError),
Ok(c) => { Ok(c) => {
let mut counter = Counter { fle: c, unit: None }; let mut counter = Counter { fle: c, unit: None };
counter.read_unit() counter.read_unit()
.map_err_into(CEK::StoreReadError) .chain_err(|| CEK::StoreReadError)
.and_then(|u| { .and_then(|u| {
counter.unit = u; counter.unit = u;
Ok(counter) Ok(counter)

View File

@ -17,6 +17,10 @@
// Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA // Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
// //
use std::error::Error;
use libimagerror::into::IntoError;
error_chain! { error_chain! {
types { types {
CounterError, CounterErrorKind, ResultExt, Result; CounterError, CounterErrorKind, ResultExt, Result;
@ -51,9 +55,6 @@ error_chain! {
} }
} }
pub use self::error::CounterError;
pub use self::error::CounterErrorKind;
impl IntoError for CounterErrorKind { impl IntoError for CounterErrorKind {
type Target = CounterError; type Target = CounterError;
@ -61,7 +62,7 @@ impl IntoError for CounterErrorKind {
CounterError::from_kind(self) CounterError::from_kind(self)
} }
fn into_error_with_cause(self, cause: Box<Error>) -> Self::Target { fn into_error_with_cause(self, _: Box<Error>) -> Self::Target {
CounterError::from_kind(self) CounterError::from_kind(self)
} }
} }

View File

@ -41,7 +41,7 @@ extern crate toml_query;
#[macro_use] extern crate error_chain; #[macro_use] extern crate error_chain;
#[macro_use] extern crate libimagstore; #[macro_use] extern crate libimagstore;
#[macro_use] extern crate libimagerror; extern crate libimagerror;
module_entry_path_mod!("counter"); module_entry_path_mod!("counter");