/*
* Copyright (c) 2007-2014 Concurrent, Inc. All Rights Reserved.
*
* Project and contact information: http://www.cascading.org/
*
* This file is part of the Cascading project.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package cascading.lingual.optiq;
import java.lang.reflect.Constructor;
import java.util.List;
import cascading.lingual.optiq.enumerable.CascadingFlowRunnerEnumerable;
import cascading.lingual.optiq.enumerable.CascadingPlannerEnumerable;
import cascading.lingual.optiq.enumerable.CascadingValueInsertEnumerable;
import cascading.lingual.optiq.meta.Branch;
import cascading.lingual.optiq.meta.FlowHolder;
import cascading.lingual.optiq.meta.ValuesHolder;
import cascading.lingual.platform.PlannerPlatformBroker;
import net.hydromatic.linq4j.expressions.BlockBuilder;
import net.hydromatic.linq4j.expressions.BlockStatement;
import net.hydromatic.linq4j.expressions.Expressions;
import net.hydromatic.optiq.rules.java.EnumerableConvention;
import net.hydromatic.optiq.rules.java.EnumerableRel;
import net.hydromatic.optiq.rules.java.EnumerableRelImplementor;
import net.hydromatic.optiq.rules.java.JavaRowFormat;
import net.hydromatic.optiq.rules.java.PhysType;
import net.hydromatic.optiq.rules.java.PhysTypeImpl;
import org.eigenbase.rel.RelNode;
import org.eigenbase.rel.SingleRel;
import org.eigenbase.relopt.RelOptCluster;
import org.eigenbase.relopt.RelOptCost;
import org.eigenbase.relopt.RelOptPlanner;
import org.eigenbase.relopt.RelTraitSet;
import org.eigenbase.relopt.volcano.VolcanoPlanner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
/**
*
*/
class CascadingEnumerableRel extends SingleRel implements EnumerableRel
{
private static final Logger LOG = LoggerFactory.getLogger( CascadingEnumerableRel.class );
public CascadingEnumerableRel( RelOptCluster cluster, RelTraitSet traitSet, RelNode input )
{
super( cluster, traitSet, input );
if( getConvention() != EnumerableConvention.INSTANCE )
throw new IllegalStateException( "unsupported convention " + getConvention() );
}
@Override
public RelOptCost computeSelfCost( RelOptPlanner planner )
{
return super.computeSelfCost( planner ).multiplyBy( .1 );
}
@Override
public RelNode copy( RelTraitSet traitSet, List<RelNode> inputs )
{
return new CascadingEnumerableRel( getCluster(), traitSet, sole( inputs ) );
}
@Override
public Result implement( EnumerableRelImplementor implementor, Prefer pref )
{
LOG.debug( "implementing enumerable" );
CascadingRelNode input = (CascadingRelNode) getChild();
Branch branch = input.visitChild( new Stack() );
VolcanoPlanner planner = (VolcanoPlanner) getCluster().getPlanner();
PhysType physType = PhysTypeImpl.of( implementor.getTypeFactory(), input.getRowType(), JavaRowFormat.ARRAY );
BlockStatement block;
if( branch.tuples != null )
block = handleInsert( branch, planner );
else if( branch.platformBroker instanceof PlannerPlatformBroker ) // used by the SQLPlanner
block = handlePrepare( branch, physType, planner );
else
block = handleFlow( branch, physType, planner );
return implementor.result( physType, block );
}
private BlockStatement handleInsert( Branch branch, VolcanoPlanner planner )
{
ValuesHolder holder = new ValuesHolder( branch, planner );
long ordinal = CascadingValueInsertEnumerable.addHolder( holder );
Constructor<CascadingValueInsertEnumerable> constructor = getConstructorFor( CascadingValueInsertEnumerable.class );
return new BlockBuilder().append( Expressions.new_( constructor, Expressions.constant( ordinal ) ) ).toBlock();
}
private BlockStatement handleFlow( Branch branch, PhysType physType, VolcanoPlanner planner )
{
FlowHolder flowHolder = new FlowHolder( physType, branch, planner );
long ordinal = CascadingFlowRunnerEnumerable.addHolder( flowHolder );
Constructor<CascadingFlowRunnerEnumerable> constructor = getConstructorFor( CascadingFlowRunnerEnumerable.class );
return new BlockBuilder().append( Expressions.new_( constructor, Expressions.constant( ordinal ) ) ).toBlock();
}
private BlockStatement handlePrepare( Branch branch, PhysType physType, VolcanoPlanner planner )
{
FlowHolder flowHolder = new FlowHolder( physType, branch, planner );
long ordinal = CascadingPlannerEnumerable.addHolder( flowHolder );
Constructor<CascadingPlannerEnumerable> constructor = getConstructorFor( CascadingPlannerEnumerable.class );
return new BlockBuilder().append( Expressions.new_( constructor, Expressions.constant( ordinal ) ) ).toBlock();
}
static <T> Constructor<T> getConstructorFor( Class<T> type )
{
Constructor<T> constructor;
try
{
constructor = type.getConstructor( long.class );
}
catch( NoSuchMethodException exception )
{
throw new RuntimeException( exception );
}
return constructor;
}
}