Scala TreeNode类代码示例

OStack程序员社区-中国程序员成长平台 › 门户 › 编程› Scala›Scala教程

原作者: [db:作者] 来自: [db:来源] 收藏邀请

本文整理汇总了Scala中org.apache.spark.sql.catalyst.trees.TreeNode类的典型用法代码示例。如果您正苦于以下问题：Scala TreeNode类的具体用法？Scala TreeNode怎么用？Scala TreeNode使用的例子？那么恭喜您, 这里精选的类代码示例或许可以为您提供帮助。

在下文中一共展示了TreeNode类的1个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于我们的系统推荐出更棒的Scala代码示例。

示例1: CExpressions

//设置package包名称以及导入依赖的类
package org.sparkline.spark.udfs.contains

import org.apache.spark.sql.catalyst.trees.TreeNode
import org.sparkline.spark.udfs.BitmapUtils

object CExpressions {

  abstract class CExpression extends TreeNode[CExpression] {
    override def children: Seq[CExpression] = ???
    def apply(row : Array[Array[Byte]]) : Boolean

    def logicalOp(op : String, r : CExpression) : CExpression = op.toUpperCase match {
      case "AND" => And(this, r)
      case "OR" => Or(this, r)
    }

    def not(r : CExpression) : CExpression = Not(this)
  }


  case class BitmapContains(bIndex : Int, ids : Array[Int]) extends CExpression {
    override def children: Seq[CExpression] = Seq()
    override  def apply(row : Array[Array[Byte]]) : Boolean = {
      val b = row(bIndex)
      BitmapUtils.contains(b, ids)
    }

    override def asCode: String = {
      val idsStr = ids.mkString("new int[", ",", "]")
      s"new BitmapContains($bIndex, $idsStr)"
    }
  }

  case class And(l : CExpression, r : CExpression) extends CExpression {
    override def children: Seq[CExpression] = Seq(l,r)
    override  def apply(row : Array[Array[Byte]]) : Boolean = {
      l(row) && r(row)
    }
  }

  case class Or(l : CExpression, r : CExpression) extends CExpression {
    override def children: Seq[CExpression] = Seq(l,r)
    override  def apply(row : Array[Array[Byte]]) : Boolean = {
      l(row) || r(row)
    }
  }

  case class Not(c : CExpression) extends CExpression {
    override def children: Seq[CExpression] = Seq(c)
    override  def apply(row : Array[Array[Byte]]) : Boolean = {
      ! c(row)
    }
  }

}

开发者ID:SparklineData，项目名称:spark-functions，代码行数:56，代码来源:CExpressions.scala

注：本文中的org.apache.spark.sql.catalyst.trees.TreeNode类示例整理自Github/MSDocs等源码及文档管理平台，相关代码片段筛选自各路编程大神贡献的开源项目，源码版权归原作者所有，传播和使用请参考对应项目的License；未经允许，请勿转载。