Spark自定义函数UDF UDAF UDTF编写

一、Spark自定义函数UDF

import org.apache.spark.sql.{Row, SparkSession}
import org.apache.spark.{SparkConf, SparkContext}

object UDFDemo {
//创建样例类
  case class Hobbies(name: String, hobbies: String)

  def main(args: Array[String]): Unit = {
//建立连接
    val conf = new SparkConf().setAppName("Function").setMaster("local[*]")
    val spark = SparkSession.builder().config(conf).getOrCreate()
    val sc = spark.sparkContext

//需要手动导入一个隐式转换,否则RDD无法转换成DF
    import spark.implicits._
//导入文件
    val rdd = sc.textFile("in/hobbies.txt")
    //根据文件情况进行分割,然后分组,转换为DataFrame
    val hobbyDF = rdd.map(_.split(" "))
      .map(x => Hobbies(x(0), x(1))).toDF()

    //注册为临时表
    hobbyDF.createOrReplaceTempView("hobby")
    //注册自定义函数
    spark.udf.register("hobby_num", (x: String) => x.split(",").size )
    //.show(false)字段不隐藏
    spark.sql("select name,hobbies,hobby_num(hobbies) as hobbyNum from hobby").show(false)

  }

}


运行结果如下

+-----+----------------------+--------+
|name |hobbies               |hobbyNum|
+-----+----------------------+--------+
|alice|jogging,Coding,cooking|3       |
|lina |travel,dance          |2       |
+-----+----------------------+--------+


文档 hobbies.txt内容

alice jogging,Coding,cooking
lina travel,dance

### 二、Spark自定义函数UDAF

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值