Contents

package org.embulk.filter.row;

import org.embulk.filter.row.RowFilterPlugin.PluginTask;
import org.embulk.filter.row.condition.Condition;

import org.embulk.filter.row.condition.ConditionConfig;
import org.embulk.filter.row.condition.ConditionFactory;
import org.embulk.spi.Column;
import org.embulk.spi.PageReader;
import org.embulk.spi.Schema;

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

abstract class AbstractGuardColumnVisitor
{
    PluginTask task;
    Schema inputSchema;
    Schema outputSchema;
    PageReader pageReader;

    AbstractGuardColumnVisitor(PluginTask task, Schema inputSchema, Schema outputSchema, PageReader pageReader)
    {
        this.task = task;
        this.inputSchema = inputSchema;
        this.outputSchema = outputSchema;
        this.pageReader = pageReader;
    }

    static HashMap<String, List<Condition>> buildConditionMap(PluginTask task, Schema outputSchema)
    {
        HashMap<String, List<Condition>> conditionMap = new HashMap<>();
        for (Column column : outputSchema.getColumns()) {
            String columnName = column.getName();
            conditionMap.put(columnName, new ArrayList<Condition>());
        }

        for (ConditionConfig conditionConfig : task.getConditions().get()) {
            String columnName = conditionConfig.getColumn();
            for (Column column : outputSchema.getColumns()) {
                if (columnName.equals(column.getName())) {
                    ConditionFactory factory = new ConditionFactory(column, conditionConfig);
                    Condition condition = factory.createCondition();
                    conditionMap.get(columnName).add(condition);
                    break;
                }
            }
        }
        return conditionMap;
    }

    abstract public boolean visitColumns(Schema inputSchema);
}

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
embulk-filter-row-0.5.1	src/main/java/org/embulk/filter/row/AbstractGuardColumnVisitor.java
embulk-filter-row-0.5.0	src/main/java/org/embulk/filter/row/AbstractGuardColumnVisitor.java