    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.ValueAssertion;
    027import cascading.pipe.Each;
    028import cascading.pipe.OperatorException;
    029import cascading.tuple.Fields;
    030import cascading.tuple.TupleEntry;
    031
    032/**
    033 *
    034 */
    035public class ValueAssertionEachStage extends EachStage
    036  {
    037  private ValueAssertion valueAssertion;
    038
    039  public ValueAssertionEachStage( 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    valueAssertion = each.getValueAssertion();
    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      valueAssertion.doAssert( flowProcess, operationCall );
    072
    073      next.receive( this, incomingEntry );
    074      }
    075    catch( CascadingException exception )
    076      {
    077      handleException( exception, argumentsEntry );
    078      }
    079    catch( Throwable throwable )
    080      {
    081      handleException( new OperatorException( each, "operator Each failed executing operation", throwable ), argumentsEntry );
    082      }
    083    }
    084  }