com.yahoo.yqlplus.engine.internal.plan.streams.RecordsBuiltinsModule.java Source code

Introduction

Here is the source code for com.yahoo.yqlplus.engine.internal.plan.streams.RecordsBuiltinsModule.java

Source

/*
 * Copyright (c) 2016 Yahoo Inc.
 * Licensed under the terms of the Apache version 2.0 license.
 * See LICENSE file for terms.
 */

package com.yahoo.yqlplus.engine.internal.plan.streams;

import com.google.common.collect.Lists;
import com.yahoo.yqlplus.engine.internal.plan.ContextPlanner;
import com.yahoo.yqlplus.engine.internal.plan.DynamicExpressionEvaluator;
import com.yahoo.yqlplus.engine.internal.plan.ModuleType;
import com.yahoo.yqlplus.engine.internal.plan.ast.PhysicalExprOperator;
import com.yahoo.yqlplus.engine.internal.plan.ast.PhysicalProjectOperator;
import com.yahoo.yqlplus.language.logical.ExpressionOperator;
import com.yahoo.yqlplus.language.operator.OperatorNode;
import com.yahoo.yqlplus.language.parser.Location;
import com.yahoo.yqlplus.language.parser.ProgramCompileException;

import java.util.List;

public class RecordsBuiltinsModule implements ModuleType {
    @Override
    public OperatorNode<PhysicalExprOperator> call(Location location, ContextPlanner context, String name,
            List<OperatorNode<ExpressionOperator>> arguments) {
        return callInRowContext(location, context, name, arguments, null);
    }

    @Override
    public OperatorNode<PhysicalExprOperator> callInRowContext(Location location, ContextPlanner context,
            String name, List<OperatorNode<ExpressionOperator>> arguments, OperatorNode<PhysicalExprOperator> row) {
        DynamicExpressionEvaluator eval = new DynamicExpressionEvaluator(context, row);
        if ("merge".equals(name)) {
            List<OperatorNode<PhysicalExprOperator>> args = eval.applyAll(arguments);
            List<OperatorNode<PhysicalProjectOperator>> ops = Lists.newArrayListWithExpectedSize(args.size());
            for (OperatorNode<PhysicalExprOperator> arg : args) {
                ops.add(OperatorNode.create(arg.getLocation(), PhysicalProjectOperator.MERGE, arg));
            }
            return OperatorNode.create(PhysicalExprOperator.PROJECT, ops);
        } else if ("map".equals(name)) {
            List<OperatorNode<PhysicalExprOperator>> args = eval.applyAll(arguments);
            List<OperatorNode<PhysicalProjectOperator>> ops = Lists.newArrayListWithExpectedSize(args.size());
            for (OperatorNode<PhysicalExprOperator> arg : args) {
                ops.add(OperatorNode.create(arg.getLocation(), PhysicalProjectOperator.MERGE, arg));
            }
            OperatorNode<PhysicalExprOperator> projectNode = OperatorNode.create(PhysicalExprOperator.PROJECT, ops);
            projectNode.putAnnotation("project:type", "map");
            return projectNode;
        }
        throw new ProgramCompileException(location, "Unknown records function '%s'", name);
    }

    @Override
    public OperatorNode<PhysicalExprOperator> property(Location location, ContextPlanner context, String name) {
        return null;
    }

    @Override
    public StreamValue pipe(Location location, ContextPlanner context, String name, StreamValue input,
            List<OperatorNode<ExpressionOperator>> arguments) {
        return null;
    }
}