DataFrame转自定义JavaBean

本人在项目开发的阶段遇到了一个业务场景：需要将 DataFrame 的数据转成一个自定义的 JavaBean 对象，然后传给 Java 应用层。

执行的主函数代码：

package com.hanlian.spark.sql

import org.apache.spark.sql.SparkSession

object DataFrameToJavaBean {

  def main(args: Array[String]): Unit = {

    val sparkSession = SparkSession
      .builder()
      .appName("DataFrameToJavaBean")
      .master("local[*]")
      .enableHiveSupport()
      .getOrCreate()

    val hic_product = sparkSession.sql("select * from hive_table")

    val dataIterator = hic_product
      .collectAsList()
      .iterator()

    while (dataIterator.hasNext) {

      val nextData = dataIterator.next()

      val newBean = new NewBean

      newBean.setID(nextData.getAs[String]("id"))

      println(newBean.getID)

    }

  }

}

JavaBean代码：

package com.hailian.javabean;

public class NewBean {

    private String ID;

    public String getID() {
        return ID;
    }

    public void setID(String ID) {
        this.ID = ID;
    }
}