gnosis/lighthouse
1
0
Fork 0
mirror of https://github.com/sigp/lighthouse.git synced 2026-03-10 12:11:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

Modify runtime to allow memory or disk db

Browse Source 
DiskDB is not working yet, but we'll get there!
This commit is contained in:
Paul Hauner
2019-03-31 18:57:48 +11:00
parent 9a0ebac687
commit 08b1808745
6 changed files with 107 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
beacon_node/client/src/client_config.rs
View File

@@ -119,6 +119,12 @@ impl ClientConfig {
}
}
match args.value_of("db") {
Some("rocks") => config.db_type = DBType::RocksDB,
Some("memory") => config.db_type = DBType::Memory,
_ => unreachable!(), // clap prevents this.
};
Ok(config)
}
}

20
beacon_node/client/src/client_types.rs
View File

@@ -34,9 +34,9 @@ impl ClientTypes for StandardClientType {
}
}
pub struct TestingClientType;
pub struct MemoryDBTestingClientType;
impl ClientTypes for TestingClientType {
impl ClientTypes for MemoryDBTestingClientType {
type DB = MemoryDB;
type SlotClock = SystemTimeSlotClock;
type ForkChoice = BitwiseLMDGhost<MemoryDB>;
@@ -44,6 +44,20 @@ impl ClientTypes for TestingClientType {
fn initialise_beacon_chain(
config: &ClientConfig,
) -> Arc<BeaconChain<Self::DB, Self::SlotClock, Self::ForkChoice>> {
initialise::initialise_test_beacon_chain(&config.spec, None)
initialise::initialise_test_beacon_chain_with_memory_db(&config.spec, None)
}
}
pub struct DiskDBTestingClientType;
impl ClientTypes for DiskDBTestingClientType {
type DB = DiskDB;
type SlotClock = SystemTimeSlotClock;
type ForkChoice = BitwiseLMDGhost<DiskDB>;
fn initialise_beacon_chain(
config: &ClientConfig,
) -> Arc<BeaconChain<Self::DB, Self::SlotClock, Self::ForkChoice>> {
initialise::initialise_test_beacon_chain_with_disk_db(&config.spec, Some(&config.db_name))
}
}