myst-client 0.1.2

A client library for the Myst network
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40

use std::collections::{HashMap, HashSet};

use serde::{Deserialize, Serialize};

use super::NodeId;

#[derive(Clone, Default, Debug, Serialize, Deserialize)]
pub struct NetworkState {
    pub graph: HashMap<NodeId, HashSet<NodeId>>,
}

impl NetworkState {
    pub fn new() -> Self {
        Self {
            graph: HashMap::new(),
        }
    }

    pub fn update(&mut self, new_state: NetworkState) {
        for (node, new_neighbors) in new_state.graph.into_iter() {
            let entry = self.graph.entry(node).or_default();
            entry.extend(new_neighbors);
        }
    }

    pub fn neighbors(&self, node_id: &NodeId) -> HashSet<NodeId> {
        self.graph.get(node_id).cloned().unwrap_or_default()
    }

    pub fn insert_adjacencies(&mut self, node: NodeId, neighbors: HashSet<NodeId>) {
        self.graph
            .entry(node.clone())
            .or_default()
            .extend(neighbors.clone());

        for neighbor in neighbors {
            self.graph.entry(neighbor).or_default().insert(node.clone());
        }
    }
}