asterix-bad/src/main/java/org/apache/asterix/bad/lang/BADRuleSetFactory.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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 org.apache.asterix.bad.lang;

import java.util.ArrayList;
import java.util.List;

import org.apache.asterix.bad.rules.InsertBrokerNotifierForChannelRule;
import org.apache.asterix.common.dataflow.ICcApplicationContext;
import org.apache.asterix.compiler.provider.DefaultRuleSetFactory;
import org.apache.asterix.compiler.provider.IRuleSetFactory;
import org.apache.asterix.optimizer.base.RuleCollections;
import org.apache.asterix.optimizer.rules.UnnestToDataScanRule;
import org.apache.hyracks.algebricks.common.exceptions.AlgebricksException;
import org.apache.hyracks.algebricks.common.utils.Pair;
import org.apache.hyracks.algebricks.compiler.rewriter.rulecontrollers.SequentialOnceRuleController;
import org.apache.hyracks.algebricks.core.rewriter.base.AbstractRuleController;
import org.apache.hyracks.algebricks.core.rewriter.base.IAlgebraicRewriteRule;

public class BADRuleSetFactory implements IRuleSetFactory {

    @Override
    public List<Pair<AbstractRuleController, List<IAlgebraicRewriteRule>>> getLogicalRewrites(
            ICcApplicationContext appCtx) throws AlgebricksException {
        List<Pair<AbstractRuleController, List<IAlgebraicRewriteRule>>> logicalRuleSet =
                DefaultRuleSetFactory.buildLogical(appCtx);

        List<IAlgebraicRewriteRule> normalizationCollection = RuleCollections.buildNormalizationRuleCollection(appCtx);
        List<IAlgebraicRewriteRule> alteredNormalizationCollection = new ArrayList<>();
        alteredNormalizationCollection.addAll(normalizationCollection);

        //Create a normalization collection that includes the broker rule
        for (int i = 0; i < alteredNormalizationCollection.size(); i++) {
            IAlgebraicRewriteRule rule = alteredNormalizationCollection.get(i);
            if (rule instanceof UnnestToDataScanRule) {
                alteredNormalizationCollection.add(i + 1, new InsertBrokerNotifierForChannelRule());
                break;
            }
        }

        //Find instances of the normalization collection and replace them with the new one
        SequentialOnceRuleController seqOnceCtrl = new SequentialOnceRuleController(true);
        for (int i = 0; i < logicalRuleSet.size(); i++) {
            List<IAlgebraicRewriteRule> collection = logicalRuleSet.get(i).second;
            if (collection.size() == normalizationCollection.size()) {
                boolean isNormalizationCollection = true;
                for (int j = 0; j < collection.size(); j++) {
                    //Make sure the set of rules is the same
                    if (!collection.get(j).getClass().equals(normalizationCollection.get(j).getClass())) {
                        isNormalizationCollection = false;
                        break;
                    }
                }
                if (isNormalizationCollection) {
                    //replace with the new collection
                    logicalRuleSet.set(i, new Pair<>(seqOnceCtrl, alteredNormalizationCollection));
                }
            }
        }
        return logicalRuleSet;
    }

    @Override
    public List<Pair<AbstractRuleController, List<IAlgebraicRewriteRule>>> getPhysicalRewrites(
            ICcApplicationContext appCtx) {
        return DefaultRuleSetFactory.buildPhysical(appCtx);
    }
}