garage/src/membership.rs

86 lines
1.8 KiB
Rust

use std::sync::Arc;
use std::collections::HashMap;
use std::time::Duration;
use std::net::SocketAddr;
use hyper::client::Client;
use tokio::sync::RwLock;
use crate::Config;
use crate::error::Error;
use crate::data::*;
use crate::proto::*;
use crate::rpc::*;
const PING_INTERVAL: Duration = Duration::from_secs(10);
const PING_TIMEOUT: Duration = Duration::from_secs(2);
const MAX_FAILED_PINGS: usize = 3;
pub struct System {
pub config: Config,
pub id: UUID,
pub rpc_client: Client<hyper::client::HttpConnector, hyper::Body>,
pub members: RwLock<Members>,
}
pub struct Members {
pub present: Vec<UUID>,
pub status: HashMap<UUID, NodeStatus>,
pub config: HashMap<UUID, NodeConfig>,
pub config_version: u64,
}
pub struct NodeStatus {
pub addr: SocketAddr,
remaining_ping_attempts: usize,
}
pub struct NodeConfig {
pub n_tokens: u32,
}
impl System {
pub fn new(config: Config, id: UUID) -> Self {
System{
config,
id,
rpc_client: Client::new(),
members: RwLock::new(Members{
present: Vec::new(),
status: HashMap::new(),
config: HashMap::new(),
config_version: 0,
}),
}
}
pub async fn broadcast(&self) -> Vec<UUID> {
self.members.read().await.present.clone()
}
}
pub async fn bootstrap(system: Arc<System>) {
rpc_call_many_addr(system.clone(),
&system.config.bootstrap_peers,
&Message::Ping(PingMessage{
id: system.id,
rpc_port: system.config.rpc_port,
present_hash: [0u8; 32],
config_version: 0,
}),
None,
PING_TIMEOUT).await;
unimplemented!() //TODO
}
pub async fn handle_ping(sys: Arc<System>, from: &SocketAddr, ping: &PingMessage) -> Result<Message, Error> {
unimplemented!() //TODO
}
pub async fn handle_advertise_node(sys: Arc<System>, ping: &AdvertiseNodeMessage) -> Result<Message, Error> {
unimplemented!() //TODO
}