001/*
002 * Copyright (c) 2007-2016 Concurrent, Inc. All Rights Reserved.
003 *
004 * Project and contact information: http://www.cascading.org/
005 *
006 * This file is part of the Cascading project.
007 *
008 * Licensed under the Apache License, Version 2.0 (the "License");
009 * you may not use this file except in compliance with the License.
010 * You may obtain a copy of the License at
011 *
012 *     http://www.apache.org/licenses/LICENSE-2.0
013 *
014 * Unless required by applicable law or agreed to in writing, software
015 * distributed under the License is distributed on an "AS IS" BASIS,
016 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
017 * See the License for the specific language governing permissions and
018 * limitations under the License.
019 */
020
021package cascading.flow.stream.element;
022
023import cascading.CascadingException;
024import cascading.flow.FlowProcess;
025import cascading.flow.stream.duct.Duct;
026import cascading.operation.Filter;
027import cascading.pipe.Each;
028import cascading.pipe.OperatorException;
029import cascading.tuple.Fields;
030import cascading.tuple.TupleEntry;
031
032/**
033 *
034 */
035public class FilterEachStage extends EachStage
036  {
037  private Filter filter;
038
039  public FilterEachStage( FlowProcess flowProcess, Each each )
040    {
041    super( flowProcess, each );
042    }
043
044  @Override
045  protected Fields getIncomingPassThroughFields()
046    {
047    return incomingScopes.get( 0 ).getIncomingFunctionPassThroughFields();
048    }
049
050  @Override
051  protected Fields getIncomingArgumentsFields()
052    {
053    return incomingScopes.get( 0 ).getIncomingFunctionArgumentFields();
054    }
055
056  @Override
057  public void initialize()
058    {
059    super.initialize();
060
061    filter = each.getFilter();
062    }
063
064  @Override
065  public void receive( Duct previous, TupleEntry incomingEntry )
066    {
067    argumentsEntry.setTuple( argumentsBuilder.makeResult( incomingEntry.getTuple(), null ) );
068
069    try
070      {
071      if( filter.isRemove( flowProcess, operationCall ) )
072        return;
073
074      next.receive( this, incomingEntry );
075      }
076    catch( CascadingException exception )
077      {
078      handleException( exception, argumentsEntry );
079      }
080    catch( Throwable throwable )
081      {
082      handleException( new OperatorException( each, "operator Each failed executing operation", throwable ), argumentsEntry );
083      }
084    }
085  }