• Home
• tech.sourced
• gitbase-spark-connector
• 0.3.0
• source code
• UastChildren.scala

×

Project download

Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $

You can buy this project and download/modify it how often you want.

tech.sourced.gitbase.spark.udf.UastChildren.scala Maven / Gradle / Ivy

package tech.sourced.gitbase.spark.udf

import org.apache.spark.sql.expressions.UserDefinedFunction
import org.apache.spark.sql.functions.udf

object UastChildren extends CustomUDF {
  /** Name of the function. */
  override def name: String = "uast_children"

  /** Function to execute when this function is called. */
  override def function: UserDefinedFunction = udf(get _)

  def get(marshaledNodes: Array[Byte]): Option[Array[Byte]] = {
    val nodes = BblfshUtils.unmarshalNodes(marshaledNodes).getOrElse(Seq.empty)
    val children = nodes.flatMap(_.children)
    BblfshUtils.marshalNodes(children)
  }
}