91 lines
2.5 KiB
Rust
91 lines
2.5 KiB
Rust
|
// Copyright 2015-2017 Parity Technologies (UK) Ltd.
|
||
|
|
// This file is part of Parity.
|
||
|
|
|
||
|
|
// Parity is free software: you can redistribute it and/or modify
|
||
|
|
// it under the terms of the GNU General Public License as published by
|
||
|
|
// the Free Software Foundation, either version 3 of the License, or
|
||
|
|
// (at your option) any later version.
|
||
|
|
|
||
|
|
// Parity is distributed in the hope that it will be useful,
|
||
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
|
|
// GNU General Public License for more details.
|
||
|
|
|
||
|
|
// You should have received a copy of the GNU General Public License
|
||
|
|
// along with Parity. If not, see <http://www.gnu.org/licenses/>.
|
||
|
|
|
||
|
|
use key_server_cluster::{Error, NodeId};
|
||
|
|
use key_server_cluster::message::Message;
|
||
|
|
|
||
|
|
/// Cluster access for single encryption/decryption participant.
|
||
|
|
pub trait Cluster {
|
||
|
|
/// Broadcast message to all other nodes.
|
||
|
|
fn broadcast(&self, message: Message) -> Result<(), Error>;
|
||
|
|
/// Send message to given node.
|
||
|
|
fn send(&self, to: &NodeId, message: Message) -> Result<(), Error>;
|
||
|
|
/// Blacklist node, close connection and remove all pending messages.
|
||
|
|
fn blacklist(&self, node: &NodeId);
|
||
|
|
}
|
||
|
|
|
||
|
|
#[cfg(test)]
|
||
|
|
pub mod tests {
|
||
|
|
use std::collections::VecDeque;
|
||
|
|
use parking_lot::Mutex;
|
||
|
|
use key_server_cluster::{NodeId, Error};
|
||
|
|
use key_server_cluster::message::Message;
|
||
|
|
use key_server_cluster::cluster::Cluster;
|
||
|
|
|
||
|
|
#[derive(Debug)]
|
||
|
|
pub struct DummyCluster {
|
||
|
|
id: NodeId,
|
||
|
|
data: Mutex<DummyClusterData>,
|
||
|
|
}
|
||
|
|
|
||
|
|
#[derive(Debug, Default)]
|
||
|
|
struct DummyClusterData {
|
||
|
|
nodes: Vec<NodeId>,
|
||
|
|
messages: VecDeque<(NodeId, Message)>,
|
||
|
|
}
|
||
|
|
|
||
|
|
impl DummyCluster {
|
||
|
|
pub fn new(id: NodeId) -> Self {
|
||
|
|
DummyCluster {
|
||
|
|
id: id,
|
||
|
|
data: Mutex::new(DummyClusterData::default())
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn node(&self) -> NodeId {
|
||
|
|
self.id.clone()
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn add_node(&self, node: NodeId) {
|
||
|
|
self.data.lock().nodes.push(node);
|
||
|
|
}
|
||
|
|
|
||
|
|
pub fn take_message(&self) -> Option<(NodeId, Message)> {
|
||
|
|
self.data.lock().messages.pop_front()
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
impl Cluster for DummyCluster {
|
||
|
|
fn broadcast(&self, message: Message) -> Result<(), Error> {
|
||
|
|
let mut data = self.data.lock();
|
||
|
|
let all_nodes: Vec<_> = data.nodes.iter().cloned().filter(|n| n != &self.id).collect();
|
||
|
|
for node in all_nodes {
|
||
|
|
data.messages.push_back((node, message.clone()));
|
||
|
|
}
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
fn send(&self, to: &NodeId, message: Message) -> Result<(), Error> {
|
||
|
|
debug_assert!(&self.id != to);
|
||
|
|
self.data.lock().messages.push_back((to.clone(), message));
|
||
|
|
Ok(())
|
||
|
|
}
|
||
|
|
|
||
|
|
fn blacklist(&self, _node: &NodeId) {
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|