Files
abomonation
abomonation_derive
ansi_term
async_trait
atty
bincode
bitflags
byteorder
bytes
cfg_if
chrono
clap
dirs
dirs_sys
erdos
fixedbitset
fnv
futures
futures_channel
futures_core
futures_executor
futures_io
futures_macro
futures_sink
futures_task
futures_util
async_await
future
io
lock
sink
stream
task
indexmap
iovec
lazy_static
libc
log
memchr
mio
net2
num_cpus
num_integer
num_traits
petgraph
pin_project_lite
pin_utils
proc_macro2
proc_macro_hack
proc_macro_nested
quote
rand
rand_chacha
rand_core
rand_hc
rand_isaac
rand_jitter
rand_os
rand_pcg
rand_xorshift
serde
serde_derive
sha1
slab
slog
slog_term
strsim
syn
synstructure
term
textwrap
thread_local
time
tokio
future
io
loom
macros
net
park
runtime
stream
sync
task
time
util
tokio_macros
tokio_serde
tokio_serde_bincode
tokio_util
unicode_width
unicode_xid
uuid
vec_map
 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
use crate::dataflow::graph::{Channel, Graph, Vertex};

// Crate-wide visible submodules
pub(crate) mod endpoints_manager;

// Public exports
pub mod channel_manager;

/// Schedules a dataflow graph. Assigns operators to nodes and updates channels.
/// After running this method, there should be no unscheduled channels remaining.
pub(crate) fn schedule(graph: &Graph) -> Graph {
    let mut scheduled_graph = graph.clone();
    for stream in scheduled_graph.get_streams_ref_mut() {
        let source_node_id = match stream.get_source() {
            Vertex::Driver(node_id) => node_id,
            Vertex::Operator(operator_id) => graph.get_operator(operator_id).unwrap().node_id,
        };

        let mut channels = Vec::new();
        for channel in stream.get_channels() {
            channels.push(match channel {
                Channel::Unscheduled(cm) => {
                    let sink_node_id = match cm.sink {
                        Vertex::Driver(node_id) => node_id,
                        Vertex::Operator(operator_id) => {
                            graph.get_operator(operator_id).unwrap().node_id
                        }
                    };
                    if source_node_id == sink_node_id {
                        Channel::InterThread(cm)
                    } else {
                        Channel::InterNode(cm)
                    }
                }
                c => c,
            });
        }
        stream.set_channels(channels);
    }
    scheduled_graph
}