Line data    Source code

       1              : use std::sync::Arc;
       2              : 
       3              : use rand::Rng;
       4              : 
       5              : use crate::proto::NodeEvent;
       6              : 
       7              : use super::{
       8              :     chan::Chan,
       9              :     network::TCP,
      10              :     world::{Node, NodeId, World},
      11              : };
      12              : 
      13              : /// Abstraction with all functions (aka syscalls) available to the node.
      14        73354 : #[derive(Clone)]
      15              : pub struct NodeOs {
      16              :     world: Arc<World>,
      17              :     internal: Arc<Node>,
      18              : }
      19              : 
      20              : impl NodeOs {
      21       151537 :     pub fn new(world: Arc<World>, internal: Arc<Node>) -> NodeOs {
      22       151537 :         NodeOs { world, internal }
      23       151537 :     }
      24              : 
      25              :     /// Get the node id.
      26        78207 :     pub fn id(&self) -> NodeId {
      27        78207 :         self.internal.id
      28        78207 :     }
      29              : 
      30              :     /// Opens a bidirectional connection with the other node. Always successful.
      31       274100 :     pub fn open_tcp(&self, dst: NodeId) -> TCP {
      32       274100 :         self.world.open_tcp(dst)
      33       274100 :     }
      34              : 
      35              :     /// Returns a channel to receive node events (socket Accept and internal messages).
      36       154478 :     pub fn node_events(&self) -> Chan<NodeEvent> {
      37       154478 :         self.internal.node_events()
      38       154478 :     }
      39              : 
      40              :     /// Get current time.
      41      2534073 :     pub fn now(&self) -> u64 {
      42      2534073 :         self.world.now()
      43      2534073 :     }
      44              : 
      45              :     /// Generate a random number in range [0, max).
      46       270013 :     pub fn random(&self, max: u64) -> u64 {
      47       270013 :         self.internal.rng.lock().gen_range(0..max)
      48       270013 :     }
      49              : 
      50              :     /// Append a new event to the world event log.
      51       218977 :     pub fn log_event(&self, data: String) {
      52       218977 :         self.internal.log_event(data)
      53       218977 :     }
      54              : }