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 com.google.common.collect.ImmutableSet;
  import org.apache.spark.sql.connector.catalog.SupportsRead;
  import org.apache.spark.sql.connector.catalog.Table;
  import org.apache.spark.sql.connector.catalog.TableCapability;
  import org.apache.spark.sql.connector.read.ScanBuilder;
  import org.apache.spark.sql.types.StructType;
  import org.apache.spark.sql.util.CaseInsensitiveStringMap;
  
  import java.util.Set;
  
  public class BallistaTable implements Table, SupportsRead {
  
    private TableMeta tableMeta;
  
    public BallistaTable(TableMeta tableMeta) {
      this.tableMeta = tableMeta;
    }
  
    @Override
    public String name() {
      return tableMeta.tableName;
    }
  
    @Override
    public StructType schema() {
      return tableMeta.schema;
    }
  
    @Override
    public Set<TableCapability> capabilities() {
      return ImmutableSet.of(TableCapability.BATCH_READ);
    }
  
    @Override
    public ScanBuilder newScanBuilder(CaseInsensitiveStringMap options) {
      return new BallistaScanBuilder(tableMeta);
    }
  }
  