dscale 0.4.1

A fast & deterministic simulation framework for benchmarking and testing distributed systems
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
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99

mod bandwidth;
mod latency;

use std::sync::Arc;

pub use bandwidth::BandwidthConfig;
pub(crate) use bandwidth::BandwidthQueue;
pub(crate) use latency::LatencyQueue;
use log::debug;

use crate::GLOBAL_POOL;
use crate::MessagePtr;
use crate::Rank;
use crate::actors::SimulationActor;
use crate::destination::Destination;
use crate::event::Event;
use crate::global::clock;
use crate::jiffy::Jiffies;
use crate::random::Randomizer;
use crate::random::Seed;
use crate::step::Step;
use crate::step::TimedStep;
use crate::topology::Topology;

pub(crate) struct NetworkActor {
    bandwidth_queue: BandwidthQueue,
    topology: Arc<Topology>,
}

impl NetworkActor {
    fn submit_single_message(
        &mut self,
        message: MessagePtr,
        source: Rank,
        destination: Destination,
    ) {
        let targets = match destination {
            Destination::BroadcastWithinPool(pool_name) => self.topology.list_pool(pool_name),
            Destination::Target(rank) => &[rank],
        };

        debug!("Submitting steps P{source} -> P{targets:?}");
        let base_time = clock::global_now() + Jiffies(1);
        for &target in targets {
            let timed_step = TimedStep {
                invocation_time: base_time,
                step: Step::NetworkStep {
                    source: source,
                    target: target,
                    message: message.clone(),
                },
            };
            self.bandwidth_queue.push(timed_step);
        }
    }
}

impl NetworkActor {
    pub(crate) fn new(
        seed: Seed,
        bandwidth_type: BandwidthConfig,
        topology: Arc<Topology>,
    ) -> Self {
        Self {
            bandwidth_queue: BandwidthQueue::new(
                bandwidth_type,
                topology.list_pool(GLOBAL_POOL).len(),
                LatencyQueue::new(Randomizer::new(seed), topology.clone()),
            ),
            topology,
        }
    }
}

impl SimulationActor for NetworkActor {
    fn next_step(&mut self) -> Step {
        self.bandwidth_queue
            .pop()
            .expect("Should not be empty")
            .step
    }

    fn peek_next_step(&self) -> Option<Jiffies> {
        self.bandwidth_queue.peek_closest()
    }

    fn submit(&mut self, event: Event) {
        match event {
            Event::NetworkEvent {
                source,
                destination,
                message,
            } => {
                self.submit_single_message(message, source, destination);
            }
            _ => unreachable!(),
        }
    }
}