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 : #[derive(Clone)]
15 : pub struct NodeOs {
16 : world: Arc<World>,
17 : internal: Arc<Node>,
18 : }
19 :
20 : impl NodeOs {
21 147537 : pub fn new(world: Arc<World>, internal: Arc<Node>) -> NodeOs {
22 147537 : NodeOs { world, internal }
23 147537 : }
24 :
25 : /// Get the node id.
26 76252 : pub fn id(&self) -> NodeId {
27 76252 : self.internal.id
28 76252 : }
29 :
30 : /// Opens a bidirectional connection with the other node. Always successful.
31 267272 : pub fn open_tcp(&self, dst: NodeId) -> TCP {
32 267272 : self.world.open_tcp(dst)
33 267272 : }
34 :
35 : /// Returns a channel to receive node events (socket Accept and internal messages).
36 150477 : pub fn node_events(&self) -> Chan<NodeEvent> {
37 150477 : self.internal.node_events()
38 150477 : }
39 :
40 : /// Get current time.
41 2472324 : pub fn now(&self) -> u64 {
42 2472324 : self.world.now()
43 2472324 : }
44 :
45 : /// Generate a random number in range [0, max).
46 263324 : pub fn random(&self, max: u64) -> u64 {
47 263324 : self.internal.rng.lock().gen_range(0..max)
48 263324 : }
49 :
50 : /// Append a new event to the world event log.
51 212835 : pub fn log_event(&self, data: String) {
52 212835 : self.internal.log_event(data)
53 212835 : }
54 : }
|