forked from Deuxfleurs/garage
Small refactorings
This commit is contained in:
parent
9f8b3b5a18
commit
2bea76ce16
7 changed files with 22 additions and 21 deletions
|
@ -221,7 +221,7 @@ async fn put_block(garage: Arc<Garage>, hash: Hash, data: Vec<u8>) -> Result<(),
|
|||
&who[..],
|
||||
&Message::PutBlock(PutBlockMessage { hash, data }),
|
||||
(garage.system.config.data_replication_factor + 1) / 2,
|
||||
DEFAULT_TIMEOUT,
|
||||
BLOCK_RW_TIMEOUT,
|
||||
)
|
||||
.await?;
|
||||
Ok(())
|
||||
|
@ -368,7 +368,7 @@ async fn get_block(garage: Arc<Garage>, hash: &Hash) -> Result<Vec<u8>, Error> {
|
|||
&who[..],
|
||||
&Message::GetBlock(hash.clone()),
|
||||
1,
|
||||
DEFAULT_TIMEOUT,
|
||||
BLOCK_RW_TIMEOUT,
|
||||
)
|
||||
.await?;
|
||||
|
||||
|
|
|
@ -40,7 +40,7 @@ pub struct BlockRefTable {
|
|||
}
|
||||
|
||||
#[async_trait]
|
||||
impl TableFormat for BlockRefTable {
|
||||
impl TableSchema for BlockRefTable {
|
||||
type P = Hash;
|
||||
type S = UUID;
|
||||
type E = BlockRef;
|
||||
|
|
|
@ -92,7 +92,7 @@ pub struct ObjectTable {
|
|||
}
|
||||
|
||||
#[async_trait]
|
||||
impl TableFormat for ObjectTable {
|
||||
impl TableSchema for ObjectTable {
|
||||
type P = String;
|
||||
type S = String;
|
||||
type E = Object;
|
||||
|
|
|
@ -4,7 +4,8 @@ use std::time::Duration;
|
|||
|
||||
use crate::data::*;
|
||||
|
||||
pub const DEFAULT_TIMEOUT: Duration = Duration::from_secs(30);
|
||||
pub const DEFAULT_TIMEOUT: Duration = Duration::from_secs(10);
|
||||
pub const BLOCK_RW_TIMEOUT: Duration = Duration::from_secs(42);
|
||||
|
||||
#[derive(Debug, Serialize, Deserialize)]
|
||||
pub enum Message {
|
||||
|
|
|
@ -84,7 +84,7 @@ impl Garage {
|
|||
timeout: DEFAULT_TIMEOUT,
|
||||
};
|
||||
|
||||
let block_ref_table = Arc::new(Table::new(
|
||||
let block_ref_table = Table::new(
|
||||
BlockRefTable {
|
||||
background: background.clone(),
|
||||
block_manager: block_manager.clone(),
|
||||
|
@ -93,8 +93,8 @@ impl Garage {
|
|||
&db,
|
||||
"block_ref".to_string(),
|
||||
data_rep_param.clone(),
|
||||
));
|
||||
let version_table = Arc::new(Table::new(
|
||||
);
|
||||
let version_table = Table::new(
|
||||
VersionTable {
|
||||
background: background.clone(),
|
||||
block_ref_table: block_ref_table.clone(),
|
||||
|
@ -103,8 +103,8 @@ impl Garage {
|
|||
&db,
|
||||
"version".to_string(),
|
||||
meta_rep_param.clone(),
|
||||
));
|
||||
let object_table = Arc::new(Table::new(
|
||||
);
|
||||
let object_table = Table::new(
|
||||
ObjectTable {
|
||||
background: background.clone(),
|
||||
version_table: version_table.clone(),
|
||||
|
@ -113,7 +113,7 @@ impl Garage {
|
|||
&db,
|
||||
"object".to_string(),
|
||||
meta_rep_param.clone(),
|
||||
));
|
||||
);
|
||||
|
||||
let mut garage = Self {
|
||||
db,
|
||||
|
|
18
src/table.rs
18
src/table.rs
|
@ -13,7 +13,7 @@ use crate::membership::System;
|
|||
use crate::proto::*;
|
||||
use crate::rpc_client::*;
|
||||
|
||||
pub struct Table<F: TableFormat> {
|
||||
pub struct Table<F: TableSchema> {
|
||||
pub instance: F,
|
||||
|
||||
pub name: String,
|
||||
|
@ -38,12 +38,12 @@ pub trait TableRpcHandler {
|
|||
async fn handle(&self, rpc: &[u8]) -> Result<Vec<u8>, Error>;
|
||||
}
|
||||
|
||||
struct TableRpcHandlerAdapter<F: TableFormat> {
|
||||
struct TableRpcHandlerAdapter<F: TableSchema> {
|
||||
table: Arc<Table<F>>,
|
||||
}
|
||||
|
||||
#[async_trait]
|
||||
impl<F: TableFormat + 'static> TableRpcHandler for TableRpcHandlerAdapter<F> {
|
||||
impl<F: TableSchema + 'static> TableRpcHandler for TableRpcHandlerAdapter<F> {
|
||||
async fn handle(&self, rpc: &[u8]) -> Result<Vec<u8>, Error> {
|
||||
let msg = rmp_serde::decode::from_read_ref::<_, TableRPC<F>>(rpc)?;
|
||||
let rep = self.table.handle(msg).await?;
|
||||
|
@ -52,7 +52,7 @@ impl<F: TableFormat + 'static> TableRpcHandler for TableRpcHandlerAdapter<F> {
|
|||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
pub enum TableRPC<F: TableFormat> {
|
||||
pub enum TableRPC<F: TableSchema> {
|
||||
Ok,
|
||||
|
||||
ReadEntry(F::P, F::S),
|
||||
|
@ -115,7 +115,7 @@ impl SortKey for Hash {
|
|||
}
|
||||
|
||||
#[async_trait]
|
||||
pub trait TableFormat: Send + Sync {
|
||||
pub trait TableSchema: Send + Sync {
|
||||
type P: PartitionKey + Clone + PartialEq + Serialize + for<'de> Deserialize<'de> + Send + Sync;
|
||||
type S: SortKey + Clone + Serialize + for<'de> Deserialize<'de> + Send + Sync;
|
||||
type E: Entry<Self::P, Self::S>;
|
||||
|
@ -123,23 +123,23 @@ pub trait TableFormat: Send + Sync {
|
|||
async fn updated(&self, old: Option<Self::E>, new: Self::E);
|
||||
}
|
||||
|
||||
impl<F: TableFormat + 'static> Table<F> {
|
||||
impl<F: TableSchema + 'static> Table<F> {
|
||||
pub fn new(
|
||||
instance: F,
|
||||
system: Arc<System>,
|
||||
db: &sled::Db,
|
||||
name: String,
|
||||
param: TableReplicationParams,
|
||||
) -> Self {
|
||||
) -> Arc<Self> {
|
||||
let store = db.open_tree(&name).expect("Unable to open DB tree");
|
||||
Self {
|
||||
Arc::new(Self {
|
||||
instance,
|
||||
name,
|
||||
system,
|
||||
store,
|
||||
partitions: Vec::new(),
|
||||
param,
|
||||
}
|
||||
})
|
||||
}
|
||||
|
||||
pub fn rpc_handler(self: Arc<Self>) -> Box<dyn TableRpcHandler + Send + Sync> {
|
||||
|
|
|
@ -58,7 +58,7 @@ pub struct VersionTable {
|
|||
}
|
||||
|
||||
#[async_trait]
|
||||
impl TableFormat for VersionTable {
|
||||
impl TableSchema for VersionTable {
|
||||
type P = Hash;
|
||||
type S = EmptySortKey;
|
||||
type E = Version;
|
||||
|
|
Loading…
Reference in a new issue