use timely::dataflow::scopes::child::Iterative;
use timely::dataflow::Scope;
use timely::progress::Timestamp;
use differential_dataflow::lattice::Lattice;
use differential_dataflow::operators::{Join, Threshold};
use crate::binding::{AsBinding, Binding};
use crate::plan::{Dependencies, ImplContext, Implementable};
use crate::{CollectionRelation, Implemented, Relation, ShutdownHandle, Var, VariableMap};
#[derive(Hash, PartialEq, Eq, PartialOrd, Ord, Clone, Debug, Serialize, Deserialize)]
pub struct Antijoin<P1: Implementable, P2: Implementable> {
pub variables: Vec<Var>,
pub left_plan: Box<P1>,
pub right_plan: Box<P2>,
}
impl<P1: Implementable, P2: Implementable> Implementable for Antijoin<P1, P2> {
fn dependencies(&self) -> Dependencies {
Dependencies::merge(
self.left_plan.dependencies(),
self.right_plan.dependencies(),
)
}
fn into_bindings(&self) -> Vec<Binding> {
unimplemented!();
}
fn implement<'b, T, I, S>(
&self,
nested: &mut Iterative<'b, S, u64>,
local_arrangements: &VariableMap<Iterative<'b, S, u64>>,
context: &mut I,
) -> (Implemented<'b, S>, ShutdownHandle)
where
T: Timestamp + Lattice,
I: ImplContext<T>,
S: Scope<Timestamp = T>,
{
let mut shutdown_handle = ShutdownHandle::empty();
let left = {
let (left, shutdown) = self
.left_plan
.implement(nested, local_arrangements, context);
shutdown_handle.merge_with(shutdown);
left
};
let right = {
let (right, shutdown) = self
.right_plan
.implement(nested, local_arrangements, context);
shutdown_handle.merge_with(shutdown);
right
};
let variables = self
.variables
.iter()
.cloned()
.chain(
left.variables()
.drain(..)
.filter(|x| !self.variables.contains(x)),
)
.collect();
let right_projected = {
let (projected, shutdown) = right.projected(nested, context, &self.variables);
shutdown_handle.merge_with(shutdown);
projected
};
let left_arranged = {
let (arranged, shutdown) = left.tuples_by_variables(nested, context, &self.variables);
shutdown_handle.merge_with(shutdown);
arranged
};
let tuples = left_arranged
.distinct()
.antijoin(&right_projected.distinct())
.map(|(key, tuple)| key.iter().cloned().chain(tuple.iter().cloned()).collect());
let relation = CollectionRelation { variables, tuples };
(Implemented::Collection(relation), shutdown_handle)
}
}