-
Notifications
You must be signed in to change notification settings - Fork 271
/
mod.rs
105 lines (94 loc) · 2.99 KB
/
mod.rs
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
100
101
102
103
104
105
//! Progress tracking mechanisms to support notification in timely dataflow
use serde::{Deserialize, Serialize};
pub use self::operate::Operate;
pub use self::subgraph::{Subgraph, SubgraphBuilder};
pub use self::timestamp::{Timestamp, PathSummary};
pub use self::change_batch::ChangeBatch;
pub use self::frontier::Antichain;
pub mod change_batch;
pub mod frontier;
pub mod timestamp;
pub mod operate;
pub mod broadcast;
pub mod reachability;
pub mod subgraph;
/// A timely dataflow location.
#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Serialize, Deserialize)]
pub struct Location {
/// A scope-local operator identifier.
pub node: usize,
/// An operator port identifier.`
pub port: Port,
}
impl Location {
/// Creates a new target location (operator input or scope output).
pub fn new_target(node: usize, port: usize) -> Location {
Location { node, port: Port::Target(port) }
}
/// Creates a new source location (operator output or scope input).
pub fn new_source(node: usize, port: usize) -> Location {
Location { node, port: Port::Source(port) }
}
/// If the location is a target.
pub fn is_target(&self) -> bool { matches!(self.port, Port::Target(_)) }
/// If the location is a source.
pub fn is_source(&self) -> bool { matches!(self.port, Port::Source(_)) }
}
impl From<Target> for Location {
fn from(target: Target) -> Self {
Location {
node: target.node,
port: Port::Target(target.port),
}
}
}
impl From<Source> for Location {
fn from(source: Source) -> Self {
Location {
node: source.node,
port: Port::Source(source.port),
}
}
}
/// An operator port.
#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Hash, Debug, Serialize, Deserialize)]
pub enum Port {
/// An operator input.
Target(usize),
/// An operator output.
Source(usize),
}
/// Names a source of a data stream.
///
/// A source of data is either a child output, or an input from a parent.
/// Conventionally, `index` zero is used for parent input.
#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Debug)]
pub struct Source {
/// Index of the source operator.
pub node: usize,
/// Number of the output port from the operator.
pub port: usize,
}
impl Source {
/// Creates a new source from node and port identifiers.
pub fn new(node: usize, port: usize) -> Self {
Self { node, port }
}
}
/// Names a target of a data stream.
///
/// A target of data is either a child input, or an output to a parent.
/// Conventionally, `index` zero is used for parent output.
#[derive(Copy, Clone, Eq, PartialEq, Ord, PartialOrd, Debug)]
pub struct Target {
/// Index of the target operator.
pub node: usize,
/// Number of the input port to the operator.
pub port: usize,
}
impl Target {
/// Creates a new target from node and port identifiers.
pub fn new(node: usize, port: usize) -> Self {
Self { node, port }
}
}