cnidarium/
write_batch.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
use std::sync::Arc;

// HashMap is okay here because we don't care about ordering of substore roots.
use std::collections::HashMap;

use crate::{
    cache::Cache,
    store::{multistore, substore::SubstoreConfig},
    RootHash,
};

/// A staged write batch that can be committed to RocksDB.
///
/// This allows for write batches to be prepared and committed at a later time.
pub struct StagedWriteBatch {
    /// The write batch to commit to RocksDB.
    pub(crate) write_batch: rocksdb::WriteBatch,
    /// The new version of the chain state.
    pub(crate) version: jmt::Version,
    /// The new versions of each substore.
    pub(crate) multistore_versions: multistore::MultistoreCache,
    /// The root hash of the chain state corresponding to this set of changes.
    pub(crate) root_hash: RootHash,
    /// The configs, root hashes, and new versions of each substore
    /// that was updated in this batch.
    #[allow(clippy::disallowed_types)]
    pub(crate) substore_roots: HashMap<Arc<SubstoreConfig>, (RootHash, u64)>,
    /// Whether or not to perform a migration.
    pub(crate) perform_migration: bool,
    /// A lightweight copy of the changeset, this is useful to provide
    /// a stream of changes to subscribers.
    pub(crate) changes: Arc<Cache>,
}

impl StagedWriteBatch {
    /// Returns the new version of the chain state corresponding to this set of changes.
    pub fn version(&self) -> jmt::Version {
        self.version
    }

    /// Returns the root hash of the jmt corresponding to this set of changes.
    pub fn root_hash(&self) -> &RootHash {
        &self.root_hash
    }

    /// Returns the version of a substore in this batch, if it exists
    /// and `None` otherwise.
    pub fn substore_version(&self, prefix: &str) -> Option<jmt::Version> {
        let Some(substore_config) = self
            .multistore_versions
            .config
            .find_substore(prefix.as_bytes())
        else {
            return None;
        };

        self.multistore_versions.get_version(&substore_config)
    }
}