mirror/lighthouse-pulse
1
0
Fork 0
mirror of https://gitlab.com/pulsechaincom/lighthouse-pulse.git synced 2025-01-03 17:54:28 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update db crate for new BeaconStateTypes

Browse Source
This commit is contained in:
Paul Hauner 2019-05-08 17:06:39 +10:00
parent 6ea538938b
commit ae09a00090
No known key found for this signature in database
GPG Key ID: D362883A9218FCC6
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
beacon_node/db/src/stores/beacon_state_store.rs
View File

@ -2,7 +2,7 @@ use super::STATES_DB_COLUMN as DB_COLUMN;
use super::{ClientDB, DBError}; use super::{ClientDB, DBError};
use ssz::decode; use ssz::decode;
use std::sync::Arc; use std::sync::Arc;
use types::{BeaconState, Hash256}; use types::{BeaconState, BeaconStateTypes, FoundationBeaconState, Hash256};
pub struct BeaconStateStore<T> pub struct BeaconStateStore<T>
where where
@ -19,11 +19,14 @@ impl<T: ClientDB> BeaconStateStore<T> {
Self { db } Self { db }
} }
pub fn get_deserialized(&self, hash: &Hash256) -> Result<Option<BeaconState>, DBError> { pub fn get_deserialized<B: BeaconStateTypes>(
&self,
hash: &Hash256,
) -> Result<Option<BeaconState<B>>, DBError> {
match self.get(&hash)? { match self.get(&hash)? {
None => Ok(None), None => Ok(None),
Some(ssz) => { Some(ssz) => {
let state = decode::<BeaconState>(&ssz).map_err(|_| DBError { let state = decode::<BeaconState<B>>(&ssz).map_err(|_| DBError {
message: "Bad State SSZ.".to_string(), message: "Bad State SSZ.".to_string(),
})?; })?;
Ok(Some(state)) Ok(Some(state))
@ -50,7 +53,7 @@ mod tests {
let store = BeaconStateStore::new(db.clone()); let store = BeaconStateStore::new(db.clone());
let mut rng = XorShiftRng::from_seed([42; 16]); let mut rng = XorShiftRng::from_seed([42; 16]);
let state = BeaconState::random_for_test(&mut rng); let state: FoundationBeaconState = BeaconState::random_for_test(&mut rng);
let state_root = state.canonical_root(); let state_root = state.canonical_root();
store.put(&state_root, &ssz_encode(&state)).unwrap(); store.put(&state_root, &ssz_encode(&state)).unwrap();