[ 
https://issues.apache.org/jira/browse/FLINK-3848?page=com.atlassian.jira.plugin.system.issuetabpanels:comment-tabpanel&focusedCommentId=15725767#comment-15725767
 ] 

ASF GitHub Bot commented on FLINK-3848:
---------------------------------------

Github user fhueske commented on a diff in the pull request:

    https://github.com/apache/flink/pull/2810#discussion_r91059867
  
    --- Diff: 
flink-libraries/flink-table/src/main/scala/org/apache/flink/api/table/plan/rules/datastream/StreamProjectableTableSourceScanRule.scala
 ---
    @@ -0,0 +1,86 @@
    +/*
    + * 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.flink.api.table.plan.rules.datastream
    +
    +import org.apache.calcite.plan.RelOptRule._
    +import org.apache.calcite.plan.{RelOptRule, RelOptRuleCall, RelTraitSet}
    +import org.apache.calcite.rel.core.TableScan
    +import org.apache.calcite.rel.logical.{LogicalProject, LogicalTableScan}
    +import org.apache.calcite.rex.RexInputRef
    +import 
org.apache.flink.api.table.plan.nodes.datastream.{DataStreamConvention, 
StreamProjectableTableSourceScan, StreamTableSourceScan}
    +import org.apache.flink.api.table.plan.schema.TableSourceTable
    +import org.apache.flink.api.table.sources.{ProjectableTableSource, 
StreamTableSource}
    +
    +import scala.collection.JavaConverters._
    +
    +/** Rule to convert a [[LogicalTableScan]] with [[LogicalProject]]
    +  * into a [[StreamTableSourceScan]].
    +  */
    +class StreamProjectableTableSourceScanRule
    +  extends RelOptRule(
    +    operand(classOf[LogicalProject], operand(classOf[TableScan], none())),
    +    "StreamProjectableTableSourceScanRule") {
    +
    +  /** Rule must only match if TableScan targets a [[StreamTableSource]],
    +    * LogicalProject targets a [[ProjectableTableSource]] and all operands 
are [[RexInputRef]]
    +    */
    +  override def matches(call: RelOptRuleCall): Boolean = {
    +    val project = call.rel(0).asInstanceOf[LogicalProject]
    +    val scan: TableScan = call.rel(1).asInstanceOf[TableScan]
    +    val dataSetTable = scan.getTable.unwrap(classOf[TableSourceTable])
    +    dataSetTable match {
    +      case tst: TableSourceTable =>
    +        tst.tableSource match {
    +          case s: StreamTableSource[_] =>
    +            s match {
    +              case p: ProjectableTableSource[_] =>
    +                
project.getProjects.asScala.forall(_.isInstanceOf[RexInputRef])
    +              case _ =>
    +                false
    +            }
    +          case _ =>
    +            false
    +        }
    +      case _ =>
    +        false
    +    }
    +  }
    +
    +  override def onMatch(call: RelOptRuleCall): Unit = {
    --- End diff --
    
    The translation should be similar as in `BatchProjectableTableScanRule`.


> Add ProjectableTableSource interface and translation rule
> ---------------------------------------------------------
>
>                 Key: FLINK-3848
>                 URL: https://issues.apache.org/jira/browse/FLINK-3848
>             Project: Flink
>          Issue Type: New Feature
>          Components: Table API & SQL
>            Reporter: Fabian Hueske
>            Assignee: Anton Solovev
>
> Add a {{ProjectableTableSource}} interface for {{TableSource}} implementation 
> that support projection push-down.
> The interface could look as follows
> {code}
> def trait ProjectableTableSource {
>   def setProjection(fields: Array[String]): Unit
> }
> {code}
> In addition we need Calcite rules to push a projection into a TableScan that 
> refers to a {{ProjectableTableSource}}. We might need to tweak the cost model 
> as well to push the optimizer in the right direction.
> Moreover, the {{CsvTableSource}} could be extended to implement 
> {{ProjectableTableSource}}.



--
This message was sent by Atlassian JIRA
(v6.3.4#6332)

Reply via email to