use crate::{
atomic_write_batch,
block::{Transaction, Transition},
cow_to_cloned,
cow_to_copied,
process::{Execution, Fee},
snark::Proof,
store::{
helpers::{memory_map::MemoryMap, Map, MapRead},
TransitionMemory,
TransitionStorage,
TransitionStore,
},
};
use console::network::prelude::*;
use anyhow::Result;
use core::marker::PhantomData;
use std::borrow::Cow;
pub trait ExecutionStorage<N: Network>: Clone + Send + Sync {
type IDMap: for<'a> Map<'a, N::TransactionID, (Vec<N::TransitionID>, Option<N::TransitionID>)>;
type ReverseIDMap: for<'a> Map<'a, N::TransitionID, N::TransactionID>;
type TransitionStorage: TransitionStorage<N>;
type InclusionMap: for<'a> Map<'a, N::TransactionID, (N::StateRoot, Option<Proof<N>>)>;
type FeeMap: for<'a> Map<'a, N::TransactionID, (N::StateRoot, Option<Proof<N>>)>;
fn open(transition_store: TransitionStore<N, Self::TransitionStorage>) -> Result<Self>;
fn id_map(&self) -> &Self::IDMap;
fn reverse_id_map(&self) -> &Self::ReverseIDMap;
fn transition_store(&self) -> &TransitionStore<N, Self::TransitionStorage>;
fn inclusion_map(&self) -> &Self::InclusionMap;
fn fee_map(&self) -> &Self::FeeMap;
fn dev(&self) -> Option<u16> {
self.transition_store().dev()
}
fn start_atomic(&self) {
self.id_map().start_atomic();
self.reverse_id_map().start_atomic();
self.transition_store().start_atomic();
self.inclusion_map().start_atomic();
self.fee_map().start_atomic();
}
fn is_atomic_in_progress(&self) -> bool {
self.id_map().is_atomic_in_progress()
|| self.reverse_id_map().is_atomic_in_progress()
|| self.transition_store().is_atomic_in_progress()
|| self.inclusion_map().is_atomic_in_progress()
|| self.fee_map().is_atomic_in_progress()
}
fn abort_atomic(&self) {
self.id_map().abort_atomic();
self.reverse_id_map().abort_atomic();
self.transition_store().abort_atomic();
self.inclusion_map().abort_atomic();
self.fee_map().abort_atomic();
}
fn finish_atomic(&self) -> Result<()> {
self.id_map().finish_atomic()?;
self.reverse_id_map().finish_atomic()?;
self.transition_store().finish_atomic()?;
self.inclusion_map().finish_atomic()?;
self.fee_map().finish_atomic()
}
fn insert(&self, transaction: &Transaction<N>) -> Result<()> {
let (transaction_id, execution, optional_additional_fee) = match transaction {
Transaction::Deploy(..) => {
bail!("Attempted to insert non-execution transaction into execution storage.")
}
Transaction::Execute(transaction_id, execution, optional_additional_fee) => {
(transaction_id, execution, optional_additional_fee)
}
};
let transitions = execution.transitions();
let transition_ids = execution.transitions().map(Transition::id).copied().collect();
let global_state_root = execution.global_state_root();
let inclusion_proof = execution.inclusion_proof().cloned();
let optional_additional_fee_id = optional_additional_fee.as_ref().map(|additional_fee| *additional_fee.id());
atomic_write_batch!(self, {
self.id_map().insert(*transaction_id, (transition_ids, optional_additional_fee_id))?;
for transition in transitions {
self.reverse_id_map().insert(*transition.id(), *transaction_id)?;
self.transition_store().insert(transition)?;
}
self.inclusion_map().insert(*transaction_id, (global_state_root, inclusion_proof))?;
if let Some(additional_fee) = optional_additional_fee {
self.reverse_id_map().insert(*additional_fee.transition_id(), *transaction_id)?;
self.transition_store().insert(additional_fee)?;
self.fee_map().insert(
*transaction_id,
(additional_fee.global_state_root(), additional_fee.inclusion_proof().cloned()),
)?;
}
Ok(())
});
Ok(())
}
fn remove(&self, transaction_id: &N::TransactionID) -> Result<()> {
let (transition_ids, optional_additional_fee_id) = match self.id_map().get(transaction_id)? {
Some(ids) => cow_to_cloned!(ids),
None => bail!("Failed to get the transition IDs for the transaction '{transaction_id}'"),
};
atomic_write_batch!(self, {
self.id_map().remove(transaction_id)?;
for transition_id in transition_ids {
self.reverse_id_map().remove(&transition_id)?;
self.transition_store().remove(&transition_id)?;
}
self.inclusion_map().remove(transaction_id)?;
if let Some(additional_fee_id) = optional_additional_fee_id {
self.reverse_id_map().remove(&additional_fee_id)?;
self.transition_store().remove(&additional_fee_id)?;
self.fee_map().remove(transaction_id)?;
}
Ok(())
});
Ok(())
}
fn find_transaction_id(&self, transition_id: &N::TransitionID) -> Result<Option<N::TransactionID>> {
match self.reverse_id_map().get(transition_id)? {
Some(transaction_id) => Ok(Some(cow_to_copied!(transaction_id))),
None => Ok(None),
}
}
fn get_execution(&self, transaction_id: &N::TransactionID) -> Result<Option<Execution<N>>> {
let (transition_ids, _) = match self.id_map().get(transaction_id)? {
Some(ids) => cow_to_cloned!(ids),
None => return Ok(None),
};
let (global_state_root, inclusion_proof) = match self.inclusion_map().get(transaction_id)? {
Some(inclusion) => cow_to_cloned!(inclusion),
None => bail!("Failed to get the inclusion proof for the transaction '{transaction_id}'"),
};
let mut transitions = Vec::new();
for transition_id in &transition_ids {
match self.transition_store().get_transition(transition_id)? {
Some(transition) => transitions.push(transition),
None => bail!("Failed to get transition '{transition_id}' for transaction '{transaction_id}'"),
};
}
Ok(Some(Execution::from(transitions.into_iter(), global_state_root, inclusion_proof)?))
}
fn get_transaction(&self, transaction_id: &N::TransactionID) -> Result<Option<Transaction<N>>> {
let (transition_ids, optional_additional_fee_id) = match self.id_map().get(transaction_id)? {
Some(ids) => cow_to_cloned!(ids),
None => return Ok(None),
};
let (global_state_root, inclusion_proof) = match self.inclusion_map().get(transaction_id)? {
Some(inclusion) => cow_to_cloned!(inclusion),
None => bail!("Failed to get the inclusion proof for the transaction '{transaction_id}'"),
};
let mut transitions = Vec::new();
for transition_id in &transition_ids {
match self.transition_store().get_transition(transition_id)? {
Some(transition) => transitions.push(transition),
None => bail!("Failed to get transition '{transition_id}' for transaction '{transaction_id}'"),
};
}
let execution = Execution::from(transitions.into_iter(), global_state_root, inclusion_proof)?;
let transaction = match optional_additional_fee_id {
Some(additional_fee_id) => {
let additional_fee_transition = match self.transition_store().get_transition(&additional_fee_id)? {
Some(additional_fee_transition) => additional_fee_transition,
None => bail!("Failed to get the additional fee for transaction '{transaction_id}'"),
};
let (global_state_root, inclusion_proof) = match self.fee_map().get(&additional_fee_id)? {
Some(fee) => cow_to_cloned!(fee),
None => bail!("Failed to get the additional fee for transaction '{transaction_id}'"),
};
Transaction::from_execution(
execution,
Some(Fee::from(additional_fee_transition, global_state_root, inclusion_proof)),
)?
}
None => Transaction::from_execution(execution, None)?,
};
match *transaction_id == transaction.id() {
true => Ok(Some(transaction)),
false => bail!("Mismatching transaction ID for transaction '{transaction_id}'"),
}
}
}
#[derive(Clone)]
#[allow(clippy::type_complexity)]
pub struct ExecutionMemory<N: Network> {
id_map: MemoryMap<N::TransactionID, (Vec<N::TransitionID>, Option<N::TransitionID>)>,
reverse_id_map: MemoryMap<N::TransitionID, N::TransactionID>,
transition_store: TransitionStore<N, TransitionMemory<N>>,
inclusion_map: MemoryMap<N::TransactionID, (N::StateRoot, Option<Proof<N>>)>,
fee_map: MemoryMap<N::TransactionID, (N::StateRoot, Option<Proof<N>>)>,
}
#[rustfmt::skip]
impl<N: Network> ExecutionStorage<N> for ExecutionMemory<N> {
type IDMap = MemoryMap<N::TransactionID, (Vec<N::TransitionID>, Option<N::TransitionID>)>;
type ReverseIDMap = MemoryMap<N::TransitionID, N::TransactionID>;
type TransitionStorage = TransitionMemory<N>;
type InclusionMap = MemoryMap<N::TransactionID, (N::StateRoot, Option<Proof<N>>)>;
type FeeMap = MemoryMap<N::TransactionID, (N::StateRoot, Option<Proof<N>>)>;
fn open(transition_store: TransitionStore<N, Self::TransitionStorage>) -> Result<Self> {
Ok(Self {
id_map: MemoryMap::default(),
reverse_id_map: MemoryMap::default(),
transition_store,
inclusion_map: MemoryMap::default(),
fee_map: MemoryMap::default(),
})
}
fn id_map(&self) -> &Self::IDMap {
&self.id_map
}
fn reverse_id_map(&self) -> &Self::ReverseIDMap {
&self.reverse_id_map
}
fn transition_store(&self) -> &TransitionStore<N, Self::TransitionStorage> {
&self.transition_store
}
fn inclusion_map(&self) -> &Self::InclusionMap {
&self.inclusion_map
}
fn fee_map(&self) -> &Self::FeeMap {
&self.fee_map
}
}
#[derive(Clone)]
pub struct ExecutionStore<N: Network, E: ExecutionStorage<N>> {
storage: E,
_phantom: PhantomData<N>,
}
impl<N: Network, E: ExecutionStorage<N>> ExecutionStore<N, E> {
pub fn open(transition_store: TransitionStore<N, E::TransitionStorage>) -> Result<Self> {
let storage = E::open(transition_store)?;
Ok(Self { storage, _phantom: PhantomData })
}
pub fn from(storage: E) -> Self {
Self { storage, _phantom: PhantomData }
}
pub fn insert(&self, transaction: &Transaction<N>) -> Result<()> {
self.storage.insert(transaction)
}
pub fn remove(&self, transaction_id: &N::TransactionID) -> Result<()> {
self.storage.remove(transaction_id)
}
pub fn transition_store(&self) -> &TransitionStore<N, E::TransitionStorage> {
self.storage.transition_store()
}
pub fn start_atomic(&self) {
self.storage.start_atomic();
}
pub fn is_atomic_in_progress(&self) -> bool {
self.storage.is_atomic_in_progress()
}
pub fn abort_atomic(&self) {
self.storage.abort_atomic();
}
pub fn finish_atomic(&self) -> Result<()> {
self.storage.finish_atomic()
}
pub fn dev(&self) -> Option<u16> {
self.storage.dev()
}
}
impl<N: Network, E: ExecutionStorage<N>> ExecutionStore<N, E> {
pub fn get_transaction(&self, transaction_id: &N::TransactionID) -> Result<Option<Transaction<N>>> {
self.storage.get_transaction(transaction_id)
}
pub fn get_execution(&self, transaction_id: &N::TransactionID) -> Result<Option<Execution<N>>> {
self.storage.get_execution(transaction_id)
}
}
impl<N: Network, E: ExecutionStorage<N>> ExecutionStore<N, E> {
pub fn find_transaction_id(&self, transition_id: &N::TransitionID) -> Result<Option<N::TransactionID>> {
self.storage.find_transaction_id(transition_id)
}
}
impl<N: Network, E: ExecutionStorage<N>> ExecutionStore<N, E> {
pub fn execution_ids(&self) -> impl '_ + Iterator<Item = Cow<'_, N::TransactionID>> {
self.storage.id_map().keys()
}
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_insert_get_remove() {
let rng = &mut TestRng::default();
let transaction = crate::vm::test_helpers::sample_execution_transaction(rng);
let transaction_id = transaction.id();
let transition_store = TransitionStore::open(None).unwrap();
let execution_store = ExecutionMemory::open(transition_store).unwrap();
let candidate = execution_store.get_transaction(&transaction_id).unwrap();
assert_eq!(None, candidate);
execution_store.insert(&transaction).unwrap();
let candidate = execution_store.get_transaction(&transaction_id).unwrap();
assert_eq!(Some(transaction), candidate);
execution_store.remove(&transaction_id).unwrap();
let candidate = execution_store.get_transaction(&transaction_id).unwrap();
assert_eq!(None, candidate);
}
#[test]
fn test_find_transaction_id() {
let rng = &mut TestRng::default();
let transaction = crate::vm::test_helpers::sample_execution_transaction(rng);
let transaction_id = transaction.id();
let transition_ids = match transaction {
Transaction::Execute(_, ref execution, _) => {
execution.transitions().map(|transition| *transition.id()).collect::<Vec<_>>()
}
_ => panic!("Incorrect transaction type"),
};
let transition_store = TransitionStore::open(None).unwrap();
let execution_store = ExecutionMemory::open(transition_store).unwrap();
let candidate = execution_store.get_transaction(&transaction_id).unwrap();
assert_eq!(None, candidate);
for transition_id in transition_ids {
let candidate = execution_store.find_transaction_id(&transition_id).unwrap();
assert_eq!(None, candidate);
execution_store.insert(&transaction).unwrap();
let candidate = execution_store.find_transaction_id(&transition_id).unwrap();
assert_eq!(Some(transaction_id), candidate);
execution_store.remove(&transaction_id).unwrap();
let candidate = execution_store.find_transaction_id(&transition_id).unwrap();
assert_eq!(None, candidate);
}
}
}