Rename DB traits

This commit is contained in:
Paul Hauner 2019-05-01 11:59:18 +10:00
parent 85266f8db0
commit 157d4900aa
No known key found for this signature in database
GPG Key ID: D362883A9218FCC6
2 changed files with 13 additions and 24 deletions

View File

@ -9,20 +9,20 @@ pub use errors::Error;
pub use types::*;
pub type DBValue = Vec<u8>;
pub trait StoreDB: Sync + Send + Sized {
fn put(&self, key: &Hash256, item: &impl DBRecord) -> Result<(), Error> {
pub trait Store: Sync + Send + Sized {
fn put(&self, key: &Hash256, item: &impl StorableItem) -> Result<(), Error> {
item.db_put(self, key)
}
fn get<I: DBRecord>(&self, key: &Hash256) -> Result<Option<I>, Error> {
fn get<I: StorableItem>(&self, key: &Hash256) -> Result<Option<I>, Error> {
I::db_get(self, key)
}
fn exists<I: DBRecord>(&self, key: &Hash256) -> Result<bool, Error> {
fn exists<I: StorableItem>(&self, key: &Hash256) -> Result<bool, Error> {
I::db_exists(self, key)
}
fn delete<I: DBRecord>(&self, key: &Hash256) -> Result<(), Error> {
fn delete<I: StorableItem>(&self, key: &Hash256) -> Result<(), Error> {
I::db_delete(self, key)
}
@ -35,17 +35,6 @@ pub trait StoreDB: Sync + Send + Sized {
fn key_delete(&self, col: &str, key: &[u8]) -> Result<(), Error>;
}
pub trait DBStore {
fn db_column(&self) -> DBColumn;
}
/// Currently available database options
#[derive(Debug, Clone)]
pub enum DBType {
Memory,
RocksDB,
}
pub enum DBColumn {
Block,
State,
@ -63,10 +52,10 @@ impl<'a> Into<&'a str> for DBColumn {
}
}
pub trait DBRecord: DBEncode + DBDecode {
pub trait StorableItem: DBEncode + DBDecode {
fn db_column() -> DBColumn;
fn db_put(&self, store: &impl StoreDB, key: &Hash256) -> Result<(), Error> {
fn db_put(&self, store: &impl Store, key: &Hash256) -> Result<(), Error> {
let column = Self::db_column().into();
let key = key.as_bytes();
@ -75,7 +64,7 @@ pub trait DBRecord: DBEncode + DBDecode {
.map_err(|e| e.into())
}
fn db_get(store: &impl StoreDB, key: &Hash256) -> Result<Option<Self>, Error> {
fn db_get(store: &impl Store, key: &Hash256) -> Result<Option<Self>, Error> {
let column = Self::db_column().into();
let key = key.as_bytes();
@ -88,14 +77,14 @@ pub trait DBRecord: DBEncode + DBDecode {
}
}
fn db_exists(store: &impl StoreDB, key: &Hash256) -> Result<bool, Error> {
fn db_exists(store: &impl Store, key: &Hash256) -> Result<bool, Error> {
let column = Self::db_column().into();
let key = key.as_bytes();
store.key_exists(column, key)
}
fn db_delete(store: &impl StoreDB, key: &Hash256) -> Result<(), Error> {
fn db_delete(store: &impl Store, key: &Hash256) -> Result<(), Error> {
let column = Self::db_column().into();
let key = key.as_bytes();
@ -116,7 +105,7 @@ mod tests {
b: u64,
}
impl DBRecord for StorableThing {
impl StorableItem for StorableThing {
fn db_column() -> DBColumn {
DBColumn::Block
}

View File

@ -1,4 +1,4 @@
use super::{DBValue, Error, StoreDB};
use super::{DBValue, Error, Store};
use parking_lot::RwLock;
use std::collections::HashMap;
@ -22,7 +22,7 @@ impl MemoryDB {
}
}
impl StoreDB for MemoryDB {
impl Store for MemoryDB {
/// Get the value of some key from the database. Returns `None` if the key does not exist.
fn get_bytes(&self, col: &str, key: &[u8]) -> Result<Option<DBValue>, Error> {
let column_key = MemoryDB::get_key_for_col(col, key);