All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.hazelcast.jet.sql.impl.opt.logical.CalcDropLateItemsTransposeRule Maven / Gradle / Ivy

There is a newer version: 5.5.0
Show newest version
/*
 * Copyright 2021 Hazelcast Inc.
 *
 * Licensed under the Hazelcast Community License (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://hazelcast.com/hazelcast-community-license
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.hazelcast.jet.sql.impl.opt.logical;

import com.hazelcast.org.apache.calcite.plan.RelOptRule;
import com.hazelcast.org.apache.calcite.plan.RelOptRuleCall;
import com.hazelcast.org.apache.calcite.plan.RelRule;
import com.hazelcast.org.apache.calcite.rel.RelNode;
import com.hazelcast.org.apache.calcite.rel.core.Calc;
import com.hazelcast.org.apache.calcite.rel.rules.TransformationRule;
import com.hazelcast.org.apache.calcite.rex.RexInputRef;
import com.hazelcast.org.apache.calcite.rex.RexNode;
import org.immutables.value.Value;

import java.util.List;

import static com.hazelcast.jet.sql.impl.opt.Conventions.LOGICAL;
import static com.hazelcast.jet.sql.impl.opt.OptUtils.getTargetField;

/**
 * Logical rule that transposes {@link DropLateItemsLogicalRel} with {@link Calc}.
 * 

* Before: *

 * Calc[...]
 *   DropLateItemsRel[...]
 * 
* After: *
 * DropLateItemsRel
 *   Calc[...]
 * 
*/ @Value.Enclosing public class CalcDropLateItemsTransposeRule extends RelRule implements TransformationRule { @Value.Immutable public interface Config extends RelRule.Config { Config DEFAULT = ImmutableCalcDropLateItemsTransposeRule.Config.builder() .description(CalcDropLateItemsTransposeRule.class.getSimpleName()) .operandSupplier(b0 -> b0 .operand(Calc.class) .trait(LOGICAL) .inputs(b1 -> b1 .operand(DropLateItemsLogicalRel.class) .anyInputs())) .build(); @Override default RelOptRule toRule() { return new CalcDropLateItemsTransposeRule(this); } } public static final RelOptRule INSTANCE = new CalcDropLateItemsTransposeRule(Config.DEFAULT); protected CalcDropLateItemsTransposeRule(Config config) { super(config); } @Override public boolean matches(RelOptRuleCall call) { Calc calc = call.rel(0); DropLateItemsLogicalRel dropRel = call.rel(1); List rexNodes = calc.getProgram().expandList(calc.getProgram().getProjectList()); return rexNodes.stream() .filter(r -> r instanceof RexInputRef) .map(r -> (RexInputRef) r) .anyMatch(r -> r.getIndex() == dropRel.wmField()); } @Override public void onMatch(RelOptRuleCall call) { Calc calc = call.rel(0); DropLateItemsLogicalRel dropRel = call.rel(1); RelNode input = dropRel.getInput(); int newWmFieldIndex = getTargetField(calc.getProgram(), dropRel.wmField()); assert newWmFieldIndex >= 0; // we ensure this in `matches()` Calc newCalc = calc.copy(calc.getTraitSet(), input, calc.getProgram()); DropLateItemsLogicalRel newDropRel = dropRel.copy( dropRel.getTraitSet(), newCalc, newWmFieldIndex); call.transformTo(newDropRel); } }




© 2015 - 2024 Weber Informatics LLC | Privacy Policy