content: // Copyright 2020 Andy Grove
  //
  // Licensed 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 io.andygrove.kquery.spark.datasource;
  
  import org.apache.spark.sql.catalyst.InternalRow;
  import org.apache.spark.sql.connector.read.InputPartition;
  import org.apache.spark.sql.connector.read.PartitionReader;
  import org.apache.spark.sql.connector.read.PartitionReaderFactory;
  import org.apache.spark.sql.vectorized.ColumnarBatch;
  
  public class BallistaPartitionReaderFactory implements PartitionReaderFactory {
  
    private final TableMeta tableMeta;
  
    public BallistaPartitionReaderFactory(TableMeta tableMeta) {
      this.tableMeta = tableMeta;
    }
  
    @Override
    public PartitionReader<InternalRow> createReader(InputPartition partition) {
      throw new UnsupportedOperationException();
    }
  
    @Override
    public PartitionReader<ColumnarBatch> createColumnarReader(InputPartition partition) {
      return new BallistaColumnarReader(tableMeta);
    }
  
    @Override
    public boolean supportColumnarReads(InputPartition partition) {
      return true;
    }
  }
  