casper-node 0.1.0

The Casper blockchain node
Documentation 
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

use std::{
    collections::{btree_map::Entry, BTreeMap},
    fmt::Debug,
    sync::RwLock,
};

use super::{BlockHeightStore, Result};

/// In-memory version of a store.
#[derive(Debug)]
pub(super) struct InMemBlockHeightStore<H> {
    inner: RwLock<BTreeMap<u64, H>>,
}

impl<H> InMemBlockHeightStore<H> {
    pub(crate) fn new() -> Self {
        InMemBlockHeightStore {
            inner: RwLock::new(BTreeMap::new()),
        }
    }
}

impl<H: Send + Sync + Clone> BlockHeightStore<H> for InMemBlockHeightStore<H> {
    fn put(&self, height: u64, block_hash: H) -> Result<bool> {
        if let Entry::Vacant(entry) = self.inner.write().expect("should lock").entry(height) {
            entry.insert(block_hash);
            return Ok(true);
        }
        Ok(false)
    }

    fn get(&self, height: u64) -> Result<Option<H>> {
        Ok(self
            .inner
            .read()
            .expect("should lock")
            .get(&height)
            .cloned())
    }

    fn highest(&self) -> Result<Option<H>> {
        Ok(self
            .inner
            .read()
            .expect("should lock")
            .values()
            .rev()
            .next()
            .cloned())
    }
}