use crate::backtrace::Backtrace;
use crate::transactor::Transactor;
use fuel_tx::{Receipt, Transaction};
mod storage;
pub use storage::MemoryStorage;
#[derive(Debug, Default)]
pub struct MemoryClient<'a> {
transactor: Transactor<'a, MemoryStorage>,
}
impl<'a> AsRef<MemoryStorage> for MemoryClient<'a> {
fn as_ref(&self) -> &MemoryStorage {
self.transactor.as_ref()
}
}
impl<'a> AsMut<MemoryStorage> for MemoryClient<'a> {
fn as_mut(&mut self) -> &mut MemoryStorage {
self.transactor.as_mut()
}
}
impl<'a> MemoryClient<'a> {
pub fn new(storage: MemoryStorage) -> Self {
Self {
transactor: Transactor::new(storage),
}
}
pub fn from_txtor(transactor: Transactor<'a, MemoryStorage>) -> Self {
Self { transactor }
}
pub fn backtrace(&self) -> Option<Backtrace> {
self.transactor.backtrace()
}
pub fn receipts(&self) -> Option<&[Receipt]> {
self.transactor.receipts()
}
pub fn transact(&mut self, tx: Transaction) -> &[Receipt] {
self.transactor.transact(tx);
let state = self
.transactor
.result()
.expect("MemoryStorage implements `Infallible` as storage error. This means panic should be unreachable.");
if state.should_revert() {
self.transactor.as_mut().revert();
} else {
self.transactor.as_mut().commit();
}
self.transactor
.receipts()
.expect("The transaction was provided to the transactor.")
}
pub fn persist(&mut self) {
self.as_mut().persist();
}
}
impl<'a> From<MemoryStorage> for MemoryClient<'a> {
fn from(s: MemoryStorage) -> Self {
Self::new(s)
}
}