001/*
002 * Copyright (c) 2016 Chris K Wensel <chris@wensel.net>. All Rights Reserved.
003 * Copyright (c) 2007-2017 Xplenty, Inc. All Rights Reserved.
004 *
005 * Project and contact information: http://www.cascading.org/
006 *
007 * This file is part of the Cascading project.
008 *
009 * Licensed under the Apache License, Version 2.0 (the "License");
010 * you may not use this file except in compliance with the License.
011 * You may obtain a copy of the License at
012 *
013 *     http://www.apache.org/licenses/LICENSE-2.0
014 *
015 * Unless required by applicable law or agreed to in writing, software
016 * distributed under the License is distributed on an "AS IS" BASIS,
017 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
018 * See the License for the specific language governing permissions and
019 * limitations under the License.
020 */
021
022package cascading.flow.tez.planner.rule.expressiongraph;
023
024import cascading.flow.planner.iso.expression.AnnotationExpression;
025import cascading.flow.planner.iso.expression.ElementCapture;
026import cascading.flow.planner.iso.expression.ElementExpression;
027import cascading.flow.planner.iso.expression.ExpressionGraph;
028import cascading.flow.planner.iso.expression.FlowElementExpression;
029import cascading.flow.planner.iso.expression.PathScopeExpression;
030import cascading.flow.planner.iso.expression.TypeExpression;
031import cascading.flow.planner.iso.finder.SearchOrder;
032import cascading.flow.planner.rule.elementexpression.BoundariesElementExpression;
033import cascading.flow.stream.graph.IORole;
034import cascading.pipe.Boundary;
035import cascading.pipe.Group;
036import cascading.pipe.HashJoin;
037import cascading.pipe.Merge;
038import cascading.tap.Tap;
039
040import static cascading.flow.planner.iso.expression.AndElementExpression.and;
041import static cascading.flow.planner.iso.expression.NotElementExpression.not;
042import static cascading.flow.planner.iso.expression.OrElementExpression.or;
043
044/**
045 *
046 */
047public class BottomUpConsecutiveBoundariesExpressionGraph extends ExpressionGraph
048  {
049  public BottomUpConsecutiveBoundariesExpressionGraph()
050    {
051    super( SearchOrder.ReverseTopological );
052
053    ElementExpression head = or(
054      new FlowElementExpression( Boundary.class ),
055      new FlowElementExpression( Tap.class ),
056      new FlowElementExpression( Group.class, TypeExpression.Topo.LinearOut ),
057      new FlowElementExpression( Merge.class, TypeExpression.Topo.LinearOut )
058    );
059
060    FlowElementExpression shared = new FlowElementExpression( ElementCapture.Secondary, HashJoin.class );
061
062    ElementExpression tail = or(
063      ElementCapture.Primary,
064      and(
065        new BoundariesElementExpression( TypeExpression.Topo.LinearIn ),
066        not( new AnnotationExpression( IORole.sink ) )
067      ),
068      new BoundariesElementExpression( TypeExpression.Topo.Splice )
069    );
070
071    this.arc(
072      head,
073
074      PathScopeExpression.ANY,
075
076      shared
077    );
078
079    this.arc(
080      shared,
081
082      PathScopeExpression.ANY,
083
084      tail
085    );
086    }
087  }