/** * 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.drill.exec.planner.physical; import java.io.IOException; import java.util.Iterator; import java.util.List; import org.apache.drill.exec.physical.base.PhysicalOperator; import org.apache.drill.exec.planner.common.DrillWriterRelBase; import org.apache.drill.exec.planner.logical.CreateTableEntry; import org.apache.drill.exec.planner.physical.visitor.PrelVisitor; import org.apache.drill.exec.record.BatchSchema.SelectionVectorMode; import org.eigenbase.rel.RelNode; import org.eigenbase.relopt.RelOptCluster; import org.eigenbase.relopt.RelTraitSet; public class WriterPrel extends DrillWriterRelBase implements Prel { public WriterPrel(RelOptCluster cluster, RelTraitSet traits, RelNode child, CreateTableEntry createTableEntry) { super(Prel.DRILL_PHYSICAL, cluster, traits, child, createTableEntry); setRowType(); } @Override public WriterPrel copy(RelTraitSet traitSet, List inputs) { return new WriterPrel(getCluster(), traitSet, sole(inputs), getCreateTableEntry()); } @Override public PhysicalOperator getPhysicalOperator(PhysicalPlanCreator creator) throws IOException { Prel child = (Prel) this.getChild(); PhysicalOperator g = getCreateTableEntry().getWriter(child.getPhysicalOperator(creator)); return creator.addMetadata(this, g); } @Override public Iterator iterator() { return PrelUtil.iter(getChild()); } @Override public T accept(PrelVisitor logicalVisitor, X value) throws E { return logicalVisitor.visitWriter(this, value); } @Override public SelectionVectorMode[] getSupportedEncodings() { return SelectionVectorMode.DEFAULT; } @Override public SelectionVectorMode getEncoding() { return SelectionVectorMode.NONE; } @Override public boolean needsFinalColumnReordering() { return true; } }