Spark SQL

一、Spark SQL概述

1.1 什么是Spark SQL

Spark SQL是Spark用来处理结构化数据的一个模块,它提供了2个编程抽象:DataFrame和DataSet,并且作为分布式SQL查询引擎的作用。
我们已经学习了Hive,它是将Hive SQL转换成MapReduce然后提交到集群上执行,大大简化了编写MapReduc的程序的复杂性,由于MapReduce这种计算模型执行效率比较慢。所有Spark SQL的应运而生,它是将Spark SQL转换成RDD,然后提交到集群执行,执行效率非常快!

1.2 什么是DataFrame

与RDD类似,DataFrame也是一个分布式数据容器。然而DataFrame更像传统数据库的二维表格,除了数据以外,还记录数据的结构信息,即schema。同时,与Hive类似,DataFrame也支持嵌套数据类型(struct、array和map)。从API易用性的角度上看,DataFrame API提供的是一套高层的关系操作,比函数式的RDD API要更加友好,门槛更低。

在这里插入图片描述
上图直观地体现了DataFrame和RDD的区别。左侧的RDD[Person]虽然以Person为类型参数,但Spark框架本身不了解Person类的内部结构。而右侧的DataFrame却提供了详细的结构信息,使得Spark SQL可以清楚地知道该数据集中包含哪些列,每列的名称和类型各是什么。DataFrame是为数据提供了Schema的视图。可以把它当做数据库中的一张表来对待,DataFrame也是懒执行的。性能上比RDD要高,主要原因:
优化的执行计划:查询计划通过Spark catalyst optimiser进行优化。
在这里插入图片描述

在这里插入图片描述

在这里插入图片描述
为了说明查询优化,我们来看上图展示的人口数据分析的示例。图中构造了两个DataFrame,将它们join之后又做了一次filter操作。如果原封不动地执行这个执行计划,最终的执行效率是不高的。因为join是一个代价较大的操作,也可能会产生一个较大的数据集。如果我们能将filter下推到 join下方,先对DataFrame进行过滤,再join过滤后的较小的结果集,便可以有效缩短执行时间。而Spark SQL的查询优化器正是这样做的。简而言之,逻辑查询计划优化就是一个利用基于关系代数的等价变换,将高成本的操作替换为低成本操作的过程。

1.3 什么是DataSet

1)是Dataframe API的一个扩展,是Spark最新的数据抽象。
2)用户友好的API风格,既具有类型安全检查也具有Dataframe的查询优化特性。
3)Dataset支持编解码器,当需要访问非堆上的数据时可以避免反序列化整个对象,提高了效率。
4)样例类被用来在Dataset中定义数据的结构信息,样例类中每个属性的名称直接映射到DataSet中的字段名称。
5) Dataframe是Dataset的特列,DataFrame=Dataset[Row] ,所以可以通过as方法将Dataframe转换为Dataset。Row是一个类型,跟Car、Person这些的类型一样,所有的表结构信息我都用Row来表示。
6)DataSet是强类型的。比如可以有Dataset[Car],Dataset[Person].
7)DataFrame只是知道字段,但是不知道字段的类型,所以在执行这些操作的时候是没办法在编译的时候检查是否类型失败的,比如你可以对一个String进行减法操作,在执行的时候才报错,而DataSet不仅仅知道字段,而且知道字段类型,所以有更严格的错误检查。就跟JSON对象和类对象之间的类比。

二、SparkSQL编程

2.1 SparkSession新的起始点

在老的版本中,SparkSQL提供两种SQL查询起始点:一个叫SQLContext,用于Spark自己提供的SQL查询;一个叫HiveContext,用于连接Hive的查询。
SparkSession是Spark最新的SQL查询起始点,实质上是SQLContext和HiveContext的组合,所以在SQLContext和HiveContext上可用的API在SparkSession上同样是可以使用的。SparkSession内部封装了sparkContext,所以计算实际上是由sparkContext完成的。

2.2 DataFrame

2.2.1 创建

在Spark SQL中SparkSession是创建DataFrame和执行SQL的入口,创建DataFrame有三种方式:通过Spark的数据源进行创建;从一个存在的RDD进行转换;还可以从Hive Table进行查询返回。
1)从Spark数据源进行创建
(1)查看Spark数据源进行创建的文件格式
在这里插入图片描述
2)读取json文件创建DataFrame
在这里插入图片描述

(3)展示结果

在这里插入图片描述
2)从RDD进行转换

3)从Hive Table进行查询返回

2.2.2 SQL风格语法

1)创建一个DataFrame
在这里插入图片描述

2)对DataFrame创建一个临时表
在这里插入图片描述

3)通过SQL语句实现查询全表
在这里插入图片描述

4)结果展示

在这里插入图片描述

**注意:**临时表是Session范围内的,Session退出后,表就失效了。如果想应用范围内有效,可以使用全局表。注意使用全局表时需要全路径访问,如:global_temp.people

5)对于DataFrame创建一个全局表
在这里插入图片描述
6)通过SQL语句实现查询全表
在这里插入图片描述

2.2.3 DSL风格语法

1)创建一个DateFrame
在这里插入图片描述
2)查看DataFrame的Schema信息
在这里插入图片描述

3)只查看”name”列数据
在这里插入图片描述

4)查看”name”列数据以及”age+1”数据
在这里插入图片描述
5)查看”age”大于”21”的数据
在这里插入图片描述
6)按照”age”分组,查看数据条数
在这里插入图片描述

2.2.4 RDD转换为DateFrame

**注意:**如果需要RDD与DF或者DS之间操作,那么都需要引入 import spark.implicits._ 【spark不是包名,而是sparkSession对象的名称】
前置条件:导入隐式转换并创建一个RDD
在这里插入图片描述
1)通过手动确定转换
在这里插入图片描述
2)通过反射确定(需要用到样例类)
(1)创建一个样例类
在这里插入图片描述
(2)根据样例类将RDD转换为DataFrame
在这里插入图片描述
3)通过编程的方式
(1)导入所需的类型
在这里插入图片描述
(2)创建Schema
在这里插入图片描述
(3)导入所需的类型
在这里插入图片描述
(4)根据给定的类型创建二元组RDD
在这里插入图片描述
(5)根据数据及给定的schema创建DataFrame
在这里插入图片描述

2.2.5 DateFrame转换为RDD

直接调用rdd即可
1)创建一个DataFrame
在这里插入图片描述
2)将DataFrame转换为RDD
在这里插入图片描述
3)打印RDD
在这里插入图片描述

2.3 DataSet

Dataset是具有强类型的数据集合,需要提供对应的类型信息。

2.3.1 创建

1)创建一个样例类
在这里插入图片描述
2)创建DataSet
在这里插入图片描述

2.3.2 RDD转换为DataSet

SparkSQL能够自动将包含有case类的RDD转换成DataFrame,case类定义了table的结构,case类属性通过反射变成了表的列名。
1)创建一个RDD
在这里插入图片描述
2)创建一个样例类
在这里插入图片描述
3)将RDD转化为DataSet
在这里插入图片描述

2.3.3 DataSet转换为RDD

调用rdd方法即可。
1)创建一个DataSet
在这里插入图片描述
2)将DataSet转换为RDD
在这里插入图片描述

2.4 DataFrame与DataSet的互操作

2.4.1. DataFrame转换为DataSet

1)创建一个DateFrame
在这里插入图片描述
2)创建一个样例类
在这里插入图片描述
3)将DateFrame转化为DataSet
在这里插入图片描述

2.4.2. DataSet转换为DataFrame

1)创建一个样例类
在这里插入图片描述
2)创建DataSet
在这里插入图片描述
3)将DataSet转化为DataFrame
在这里插入图片描述

2.4.3 DataSet与DataFrame之间转化

DataSet转DataFrame

这个很简单,因为只是把case class封装成Row
(1)导入隐式转换
import spark.implicits._
(2)转换
val testDF = testDS.toDF

DataFrame转DataSet

(1)导入隐式转换
import spark.implicits._
(2)创建样例类
case class Coltest(col1:String,col2:Int)extends Serializable //定义字段名和类型
(3)转换
val testDS = testDF.as[Coltest]
这种方法就是在给出每一列的类型后,使用as方法,转成Dataset,这在数据类型是DataFrame又需要针对各个字段处理时极为方便。在使用一些特殊的操作时,一定要加上 import spark.implicits._ 不然toDF、toDS无法使用。

2.5 RDD、DataFrame、DataSet

在这里插入图片描述
在SparkSQL中Spark为我们提供了两个新的抽象,分别是DataFrame和DataSet。他们和RDD有什么区别呢?首先从版本的产生上来看:
RDD (Spark1.0) —> Dataframe(Spark1.3) —> Dataset(Spark1.6)
如果同样的数据都给到这三个数据结构,他们分别计算之后,都会给出相同的结果。不同是的他们的执行效率和执行方式。
在后期的Spark版本中,DataSet会逐步取代RDD和DataFrame成为唯一的API接口。

2.5.1 三者的共性

1、RDD、DataFrame、Dataset全都是spark平台下的分布式弹性数据集,为处理超大型数据提供便利
2、三者都有惰性机制,在进行创建、转换,如map方法时,不会立即执行,只有在遇到Action如foreach时,三者才会开始遍历运算。
3、三者都会根据spark的内存情况自动缓存运算,这样即使数据量很大,也不用担心会内存溢出。
4、三者都有partition的概念
5、三者有许多共同的函数,如filter,排序等
6、在对DataFrame和Dataset进行操作许多操作都需要这个包进行支持
import spark.implicits._
7、DataFrame和Dataset均可使用模式匹配获取各个字段的值和类型
DataFrame:

testDF.map{
      case Row(col1:String,col2:Int)=>
        println(col1);println(col2)
        col1
      case _=>
        ""
    }

Dataset:

case class Coltest(col1:String,col2:Int)extends Serializable //定义字段名和类型
    testDS.map{
      case Coltest(col1:String,col2:Int)=>
        println(col1);println(col2)
        col1
      case _=>
        ""
    }

2.5.2 三者的区别

  1. RDD:
    1)RDD一般和spark mlib同时使用
    2)RDD不支持sparksql操作
  2. DataFrame:
    1)与RDD和Dataset不同,DataFrame每一行的类型固定为Row,每一列的值没法直接访问,只有通过解析才能获取各个字段的值,如:
testDF.foreach{
  line =>
    val col1=line.getAs[String]("col1")
    val col2=line.getAs[String]("col2")
}

2)DataFrame与Dataset一般不与spark mlib同时使用
3)DataFrame与Dataset均支持sparksql的操作,比如select,groupby之类,还能注册临时表/视窗,进行sql语句操作,如:

dataDF.createOrReplaceTempView("tmp")
spark.sql("select  ROW,DATE from tmp where DATE is not null order by DATE").show(100,false)

4)DataFrame与Dataset支持一些特别方便的保存方式,比如保存成csv,可以带上表头,这样每一列的字段名一目了然
//保存

val saveoptions = Map("header" -> "true", "delimiter" -> "\t", "path" -> "hdfs://hadoop102:9000/test")
datawDF.write.format("com.atguigu.spark.csv").mode(SaveMode.Overwrite).options(saveoptions).save()
//读取
val options = Map("header" -> "true", "delimiter" -> "\t", "path" -> "hdfs://hadoop102:9000/test")
val datarDF= spark.read.options(options).format("com.atguigu.spark.csv").load()

利用这样的保存方式,可以方便的获得字段名和列的对应,而且分隔符(delimiter)可以自由指定。
3. Dataset:
1)Dataset和DataFrame拥有完全相同的成员函数,区别只是每一行的数据类型不同。
2)DataFrame也可以叫Dataset[Row],每一行的类型是Row,不解析,每一行究竟有哪些字段,各个字段又是什么类型都无从得知,只能用上面提到的getAS方法或者共性中的第七条提到的模式匹配拿出特定字段。而Dataset中,每一行是什么类型是不一定的,在自定义了case class之后可以很自由的获得每一行的信息

case class Coltest(col1:String,col2:Int)extends Serializable //定义字段名和类型
/**
 rdd
 ("a", 1)
 ("b", 1)
 ("a", 1)
**/
val test: Dataset[Coltest]=rdd.map{line=>
      Coltest(line._1,line._2)
    }.toDS
test.map{
      line=>
        println(line.col1)
        println(line.col2)
    }

可以看出,Dataset在需要访问列中的某个字段时是非常方便的,然而,如果要写一些适配性很强的函数时,如果使用Dataset,行的类型又不确定,可能是各种case class,无法实现适配,这时候用DataFrame即Dataset[Row]就能比较好的解决问题

2.6 IDEA创建SparkSQL程序

IDEA中程序的打包和运行方式都和SparkCore类似,Maven依赖中需要添加新的依赖项:

<dependency>
    <groupId>org.apache.spark</groupId>
    <artifactId>spark-sql_2.11</artifactId>
    <version>2.1.1</version>
</dependency>

2.6.1 创建SparkSession对象

方法一

程序如下:

import org.apache.spark.SparkConf
import org.apache.spark.sql.{DataFrame, Dataset, SparkSession}

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

        //创建SparkSession对象的三种方法
        //方法一
        val sparkConf: SparkConf = new SparkConf().setMaster("local[*]").setAppName("sc")
        val spark: SparkSession = SparkSession.builder().config(sparkConf).getOrCreate()
        import spark.implicits._ //必须导入这项,否则dataframe和dataset之间转换会报错.spark是SparkSession对象的名字
        //创建dataframe
        val df: DataFrame = spark.read.json("in/json.json")
        //dataframe转换为dataset
        val ds: Dataset[Student] = df.as[Student]
        //展示数据
        ds.show()
        //关闭
        spark.stop()
    }
}
方法二
import org.apache.spark.sql.SparkSession

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

        //创建SparkSession对象的三种方法
        //方法二在Hive中存取数据
       // val sparkConf: SparkConf = new SparkConf().setMaster("local[*]").setAppName("sc")
        val spark: SparkSession = SparkSession.builder
            .appName("sc") // 设置APP的名称
            .config(" spark.sql.warehouse.dir","指定hive元数据库的位置") //指定Hive元数据库的位置
            .enableHiveSupport() //支持Hive
            .getOrCreate() //创建SparkSession //必须导入这项,否则dataframe和dataset之间转换会报错
    }
}
方法三
import org.apache.spark.sql.SparkSession

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

        //创建SparkSession对象的三种方法
        //方法三
       val spark: SparkSession = SparkSession
            .builder()
            .appName("sc")
            .master("local[*]")
            .getOrCreate()
    }
}

2.6.2 IDEA实现RDD、DataFrame、DataSet之间的转换

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.sql.{DataFrame, Dataset, SparkSession}

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

        val conf: SparkConf = new SparkConf().setAppName("sc").setMaster("local[*]")

        // 创建SparkContext对象
        val sc: SparkContext = SparkContext.getOrCreate(conf)

        //创建SparkSession对象
        val spark: SparkSession = SparkSession.builder().config(conf).getOrCreate()

        import spark.implicits._

        //创建RDD 数据格式 zhangsan 12
        val rdd: RDD[String] = sc.textFile("in/student.txt")

        //转换为RDD[(String, String)]类型RDD
        val rdd2: RDD[(String, Int)] = rdd.map(
            x => {
                val para = x.split(" ")
                (para(0), para(1).toInt)
            })

        //创建DataFrame
        val df: DataFrame = spark.read.json("in/json.json")

        //RDD转换为DF
        val rddToDF: DataFrame = rdd2.toDF("name","age")

        //RDD直接转换为DS
        val ds2: Dataset[(String, Int)] = rdd2.toDS()
        ds2.show()
        //RDD转换为DS,使用样例类作为辅助
        val ds3: Dataset[Student] = rdd2.map(x=>Student(x._1,x._2)).toDS()

        //DF转换为DS,使用样例类作为辅助
        val ds: Dataset[Student] = df.as[Student]

        //DS转换为DF
        val df3: DataFrame = ds.toDF()
        df3.show()

        //DF转换为RDD
        val rdd2: RDD[Row] = df.rdd

        //DS转换为RDD
        val rdd3: RDD[Student] = ds.rdd
        
        spark.stop()
    }
}

//样例类,用于将dataframe映射为dataset
case class Student(name:String,age:Long)

2.7 用户自定义函数

在Shell窗口中可以通过spark.udf功能用户可以自定义函数。

2.7.1 用户自定义UDF函数

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

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

        val conf: SparkConf = new SparkConf().setAppName("sc").setMaster("local[*]")

        // 创建SparkContext对象
        val sc: SparkContext = SparkContext.getOrCreate(conf)

        //创建SparkSession对象
        val spark: SparkSession = SparkSession.builder().config(conf).getOrCreate()

        //创建DataFrame
        val df: DataFrame = spark.read.json("in/json.json")

        //注册udf
        spark.udf.register("appName", (x:String) =>"name: "+x)

        //创建临时表
        df.createOrReplaceTempView("student")

        //使用udf
        spark.sql("select appName(name), age from student").show()
        df.show()

    }
}

2.7.2 用户自定义聚合函数

强类型的Dataset和弱类型的DataFrame都提供了相关的聚合函数, 如 count(),countDistinct(),avg(),max(),min()。除此之外,用户可以设定自己的自定义聚合函数。
弱类型用户自定义聚合函数:通过继承UserDefinedAggregateFunction来实现用户自定义聚合函数。下面展示一个求平均年龄的自定义聚合函数。

import org.apache.spark.sql.expressions.{MutableAggregationBuffer, UserDefinedAggregateFunction}
import org.apache.spark.sql.types._
import org.apache.spark.sql.{DataFrame, Row, SparkSession}

object MyAverage extends UserDefinedAggregateFunction {

    //创建SparkSession对象
    val spark:SparkSession = SparkSession.builder().master("local[*]").appName("myAverage").getOrCreate()

    // 聚合函数输入参数的数据类型
    override def inputSchema: StructType = StructType(StructField("inputColumn", LongType) :: Nil)

    //聚合缓冲区中值的数据类型
    override def bufferSchema: StructType = {
        StructType(StructField("sum", LongType) :: StructField("count", LongType) :: Nil)
    }

    // 返回值的数据类型
    override def dataType: DataType = DoubleType

    //对于相同的输入是否一直返回相同的输出
    override def deterministic: Boolean = true

    //初始化
    override def initialize(buffer: MutableAggregationBuffer): Unit = {
        //年龄初始值
        buffer(0) = 0L

        //初始人数
        buffer(1) = 0L
    }

    //相同Execute之间的数据合并
    override def update(buffer: MutableAggregationBuffer, input: Row): Unit = {
        if (!input.isNullAt(0)) {
            buffer(0) = buffer.getLong(0) + input.getLong(0)
            buffer(1) = buffer.getLong(1) + 1
        }
    }

    // 不同Execute间的数据合并
    override def merge(buffer1: MutableAggregationBuffer, buffer2: Row): Unit = {
        buffer1(0) = buffer1.getLong(0) + buffer2.getLong(0)
        buffer1(1) = buffer1.getLong(1) + buffer2.getLong(1)
    }

    //计算最终结果
    override def evaluate(buffer: Row): Double = {
        buffer.getLong(0).toDouble / buffer.getLong(1)
    }

    // 注册函数
    spark.udf.register("myAverage", MyAverage)

    //创建DataFrame
    val df:DataFrame = spark.read.json("in/json.json")

    // 创建临时表
    df.createOrReplaceGlobalTempView("student")

    // 通过自定义聚合函数计算平均年龄
    val df2: DataFrame = spark.sql("select myAverage(age) as average_age from global_temp.student")
    
    def main(args: Array[String]): Unit = {
        df2.show()
    }
}

强类型用户自定义聚合函数:通过继承Aggregator来实现强类型自定义聚合函数,同样是求平均年龄

import org.apache.spark.sql.expressions.Aggregator
import org.apache.spark.sql.{Dataset, Encoder, Encoders, SparkSession, TypedColumn}


//既然是强类型,就使用case类
case class Student(name: String, age: Long)

case class Average(var sum: Long, var count: Long)

object MyAverage02 extends Aggregator[Student, Average, Double] {

    // 创建SparkSession对象
    val spark: SparkSession = SparkSession.builder().master("local[*]").appName("MyAverage2").getOrCreate()

    //定义一个数据结构,保存总年龄和总人数,初始化都为0
    override def zero: Average = Average(0L, 0L)

    //将缓存中的数据聚合
    override def reduce(buffer: Average, student: Student): Average = {
        buffer.sum += student.age
        buffer.count += 1
        buffer
    }

    // 聚合不同Execute的结果
    override def merge(b1: Average, b2: Average): Average = {
        b1.sum += b2.sum
        b1.count += b2.count
        b1
    }

    // 计算输出
    override def finish(reduction: Average): Double = reduction.sum.toDouble / reduction.count

    //设置值类型的编码器,转换成case类
    //Encoders.product是进行Scala元组和case类转换的编码器
    override def bufferEncoder: Encoder[Average] = Encoders.product

    //设定最终输出值的编码器
    override def outputEncoder: Encoder[Double] = Encoders.scalaDouble

    import spark.implicits._

    // 创建DataSe
    val ds: Dataset[Student] = spark.read.json("in/json.json").as[Student]

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

        //给聚合行数处理的列一个名字
        val averageAge: TypedColumn[Student, Double] = MyAverage02.toColumn.name("average_age")
        
        val ds2: Dataset[Double] = ds.select(averageAge)
        ds2.show()
    }

}

三、SparkSQL数据源

3.1 通用加载/保存方法

3.2 JSON文件

3.3 Parquet文件

3.4 从JDBC中加载数据

Spark SQL可以通过JDBC从关系型数据库中读取数据的方式创建DataFrame,通过对DataFrame一系列的计算后,还可以将数据再写回关系型数据库中。
注意:
(1)在IDEA中需要将相关的数据库驱动放(mysql-connector-java-5.1.37-bin.jar)导入依赖包中,即,在POM.xml中添加

    <dependency>
        <groupId>mysql</groupId>
        <artifactId>mysql-connector-java</artifactId>
        <version>8.0.11</version>
    </dependency>

(2) 在Shell中需要在Spark安装目录的子目录jar中放入数据库驱动(mysql-connector-java-5.1.37-bin.jar)

3.4.1 在Shell中

(1)启动spark-shell

bin/spark-shell // Spark的安装目录下

(2)从Mysql数据库加载数据方式一
在这里插入图片描述
命令解释

val jdbcDF = spark.read
.format("jdbc") // 通过jdbc方式
.option("url", "jdbc:mysql://hadoop:3306/test") //表数据库url(这是在linux虚拟机上安装的MYSQL)
.option("dbtable", "users") //表名
.option("user", "root") // 登录数据库用户名
.option("password", "assiduity") //登录数据库密码
.load()

数据展示
在这里插入图片描述
(3)从Mysql数据库加载数据方式二
在这里插入图片描述
命令解释

val connectionProperties = new java.util.Properties() // 创建属性对象

connectionProperties.put("user", "root") // 登录数据库用户名
connectionProperties.put("password", "assiduity") //登录数据库密码

val jdbcDF2 = spark.read.jdbc("jdbc:mysql://hadoop:3306/test", "student", connectionProperties) //加载 数据库test 的student表的数据
3.4.2 通过IDEA

(1) 方式一

import org.apache.spark.sql.{DataFrame, SparkSession}

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

        // 创建SparkSession对象
        val spark: SparkSession = SparkSession.builder().master("local[*]").appName("jdbc").getOrCreate()

        //
        val df: DataFrame = spark.read
            .format("jdbc") //数据来源格式
            .option("url", "jdbc:mysql://localhost:3306/test?rewriteBatchedStatements=true&useSSL=false&serverTimezone=GMT%2B8&allowPublicKeyRetrieval=true")
            .option("driver", "com.mysql.cj.jdbc.Driver") //驱动,shell中不需要,idea中不加该配置会报错Exception in thread "main" java.sql.SQLException: No suitable driver
            .option("dbtable", "student") //数据库表名
            .option("user", "root") //数据库用户名
            .option("password", "assiduity") //数据库登录密码
            .load()
        df.show()

    }
}

(2) 方式二

import java.util.Properties

import org.apache.spark.sql.{DataFrame, SparkSession}

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

        // 创建SparkSession对象
        val spark: SparkSession = SparkSession.builder().master("local[*]").appName("jdbc").getOrCreate()
        
        // 创建Proterties对象
        val connectionProperties: Properties = new Properties()
        
        // 登录数据库的用户名
        connectionProperties.put("user","root")
        // 登录数据库的密码
        connectionProperties.put("password","assiduity")

        // 参数分别为:数据库url 表名  连接数据库的用户名,密码配置信息
        val df: DataFrame = spark.read.jdbc("jdbc:mysql://localhost:3306/test?rewriteBatchedStatements=true&useSSL=false&serverTimezone=GMT%2B8&allowPublicKeyRetrieval=true", "users", connectionProperties)

        df.show()
    }

3.5 向JDBC中写入数据

(1)方式一

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.sql.{DataFrame, SparkSession}

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

        //向JDBC中写入数据方式一

        //创建SparkContext对象
        val sparkConf: SparkConf = new SparkConf().setMaster("local[*]").setAppName("ToJDBC")

        //创建SparkContext对象
        val sc: SparkContext = SparkContext.getOrCreate(sparkConf)

        //创建SparkSession对象
        val spark: SparkSession = SparkSession.builder().config(sparkConf).getOrCreate()

        //创建RDD对象
        val rdd: RDD[String] = sc.textFile("in/users.txt") //5 zhangsan101 19

        import spark.implicits._

        //创建DataFrame
        val df: DataFrame = rdd.map(cols => {
            val col = cols.split(" ")
            (col(0).toInt, col(1), col(2).toLong)
        }).toDF("id", "name", "age")

        df.write
            .format("jdbc")
            .mode("append") // 1.追加模式 还有 2."error"(default) 如果文件存在,则报错 3."overwrite" 覆写 4."ignore" 数据存在,则忽略
            .option("url", "jdbc:mysql://localhost:3306/test?rewriteBatchedStatements=true&useSSL=false&serverTimezone=GMT%2B8&allowPublicKeyRetrieval=true")
            .option("dbtable", "student")
            .option("user", "root")
            .option("password", "assiduity")
            .save()
    }
}

(2)方式二

import java.util.Properties

import org.apache.spark.rdd.RDD
import org.apache.spark.{SparkConf, SparkContext}
import org.apache.spark.sql.{DataFrame, SparkSession}

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

        //向JDBC中写入数据方式一

        //创建SparkContext对象
        val sparkConf: SparkConf = new SparkConf().setMaster("local[*]").setAppName("ToJDBC")

        //创建SparkContext对象
        val sc: SparkContext = SparkContext.getOrCreate(sparkConf)

        //创建SparkSession对象
        val spark: SparkSession = SparkSession.builder().config(sparkConf).getOrCreate()

        //创建RDD对象
        val rdd: RDD[String] = sc.textFile("in/users.txt") //5 zhangsan101 19

        import spark.implicits._

        //创建DataFrame
        val df: DataFrame = rdd.map(cols => {
            val col = cols.split(" ")
            (col(0).toInt, col(1), col(2).toLong)
        }).toDF("id", "name", "age")

        val connectionProperties = new Properties()
        connectionProperties.put("user", "root")
        connectionProperties.put("password", "assiduity")

        df.write
                .mode("append")
            .jdbc("jdbc:mysql://localhost:3306/test?rewriteBatchedStatements=true&useSSL=false&serverTimezone=GMT%2B8&allowPublicKeyRetrieval=true","student",connectionProperties)
    }
}

3.6 Hive数据库

Apache Hive是Hadoop上的SQL引擎,Spark SQL编译时可以包含Hive支持,也可以不包含。包含Hive支持的Spark SQL可以支持Hive表访问、UDF(用户自定义函数)以及 Hive 查询语言(HiveQL/HQL)等。需要强调的一点是,如果要在Spark SQL中包含Hive的库,并不需要事先安装Hive。一般来说,最好还是在编译Spark SQL时引入Hive支持,这样就可以使用这些特性了。如果你下载的是二进制版本的 Spark,它应该已经在编译时添加了 Hive 支持。
若要把Spark SQL连接到一个部署好的Hive上,你必须把hive-site.xml复制到 Spark的配置文件目录中($SPARK_HOME/conf)。即使没有部署好Hive,Spark SQL也可以运行。 需要注意的是,如果你没有部署好Hive,Spark SQL会在当前的工作目录中创建出自己的Hive 元数据仓库,叫作 metastore_db。此外,如果你尝试使用 HiveQL 中的 CREATE TABLE (并非 CREATE EXTERNAL TABLE)语句来创建表,这些表会被放在你默认的文件系统中的 /user/hive/warehouse 目录中(如果你的 classpath 中有配好的 hdfs-site.xml,默认的文件系统就是 HDFS,否则就是本地文件系统)。

3.5.1 内嵌Hive应用

如果要使用内嵌的Hive,什么都不用做,直接用就可以了。
可以通过添加参数初次指定数据仓库地址:–conf spark.sql.warehouse.dir=hdfs://hadoop102/spark-wearhouse
注意:
如果你使用的是内部的Hive,在Spark2.0之后,spark.sql.warehouse.dir用于指定数据仓库的地址,如果你需要是用HDFS作为路径,那么需要将core-site.xml和hdfs-site.xml 加入到Spark conf目录,否则只会创建master节点上的warehouse目录,查询时会出现文件找不到的问题,这是需要使用HDFS,则需要将metastore删除,重启集群。

3.5.2 外部Hive应用

如果想连接外部已经部署好的Hive,需要通过以下几个步骤。

  1. 将Hive中的hive-site.xml拷贝或者软连接到Spark安装目录下的conf目录下。
  2. 打开spark shell,注意带上访问Hive元数据库的JDBC客户端
    $ bin/spark-shell --jars mysql-connector-java-5.1.27-bin.jar
[xh@hadoop bin]$ ./spark-shell 
2021-02-27 05:02:22 WARN  NativeCodeLoader:62 - Unable to load native-hadoop library for your platform... using builtin-java classes where applicable
Setting default log level to "WARN".
To adjust logging level use sc.setLogLevel(newLevel). For SparkR, use setLogLevel(newLevel).
Spark context Web UI available at http://hadoop:4040
Spark context available as 'sc' (master = local[*], app id = local-1614430977756).
Spark session available as 'spark'.
Welcome to
      ____              __
     / __/__  ___ _____/ /__
    _\ \/ _ \/ _ `/ __/  '_/
   /___/ .__/\_,_/_/ /_/\_\   version 2.3.4
      /_/
         
Using Scala version 2.11.8 (Java HotSpot(TM) 64-Bit Server VM, Java 1.8.0_212)
Type in expressions to have them evaluated.
Type :help for more information.

scala> spark.sql("show tables").show
2021-02-27 05:10:49 WARN  ObjectStore:568 - Failed to get database global_temp, returning NoSuchObjectException
+--------+-----------------+-----------+
|database|        tableName|isTemporary|
+--------+-----------------+-----------+
| default|     product_info|      false|
| default|        user_info|      false|
| default|user_visit_action|      false|
+--------+-----------------+-----------+


scala> spark.sql("select * from default.product_info limit 20")
res1: org.apache.spark.sql.DataFrame = [product_id: bigint, product_name: string ... 1 more field]

scala> spark.sql("select * from default.product_info limit 20").show
+----------+------------+--------------------+                                  
|product_id|product_name|         extend_info|
+----------+------------+--------------------+
|        75|   product75|{"product_status"...|
|        76|   product76|{"product_status"...|
|        77|   product77|{"product_status"...|
|        78|   product78|{"product_status"...|
|        79|   product79|{"product_status"...|
|        80|   product80|{"product_status"...|
|        81|   product81|{"product_status"...|
|        82|   product82|{"product_status"...|
|        83|   product83|{"product_status"...|
|        84|   product84|{"product_status"...|
|        85|   product85|{"product_status"...|
|        86|   product86|{"product_status"...|
|        87|   product87|{"product_status"...|
|        88|   product88|{"product_status"...|
|        89|   product89|{"product_status"...|
|        90|   product90|{"product_status"...|
|        91|   product91|{"product_status"...|
|        92|   product92|{"product_status"...|
|        93|   product93|{"product_status"...|
|        94|   product94|{"product_status"...|
+----------+------------+--------------------+


scala> spark.sql("desc default.product_info").show
+------------+---------+-------+
|    col_name|data_type|comment|
+------------+---------+-------+
|  product_id|   bigint|   null|
|product_name|   string|   null|
| extend_info|   string|   null|
+------------+---------+-------+


scala> spark.sql("desc default.product_info; select * from default.product_info limit 20;").show
org.apache.spark.sql.catalyst.parser.ParseException:
mismatched input ';' expecting <EOF>(line 1, pos 25)

== SQL ==
desc default.product_info; select * from default.product_info limit 20;
-------------------------^^^

  at org.apache.spark.sql.catalyst.parser.ParseException.withCommand(ParseDriver.scala:239)
  at org.apache.spark.sql.catalyst.parser.AbstractSqlParser.parse(ParseDriver.scala:115)
  at org.apache.spark.sql.execution.SparkSqlParser.parse(SparkSqlParser.scala:48)
  at org.apache.spark.sql.catalyst.parser.AbstractSqlParser.parsePlan(ParseDriver.scala:69)
  at org.apache.spark.sql.SparkSession.sql(SparkSession.scala:642)
  ... 49 elided

scala> spark.sql("desc default.user_info").show
+------------+---------+-------+
|    col_name|data_type|comment|
+------------+---------+-------+
|     user_id|   bigint|   null|
|    username|   string|   null|
|        name|   string|   null|
|         age|      int|   null|
|professional|   string|   null|
|        city|   string|   null|
|         sex|   string|   null|
+------------+---------+-------+

3.5.3 运行Spark SQL CLI

Spark SQL CLI可以很方便的在本地运行Hive元数据服务以及从命令行执行查询任务。在Spark目录下执行如下命令启动Spark SQL CLI:

./bin/spark-sql

在这里插入图片描述
果然通过spark-sql执行sql语句计算hive表数据,速度极快!!!
在Spark SQL CLI界面就像在Hive sql beeline界面一样,hive的sql可以在此自如执行。

3.5.4 代码中使用Hive

(1)添加依赖:

<!-- https://mvnrepository.com/artifact/org.apache.spark/spark-hive -->
<dependency>
    <groupId>org.apache.spark</groupId>
    <artifactId>spark-hive_2.11</artifactId>
    <version>2.1.1</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.apache.hive/hive-exec -->
<dependency>
    <groupId>org.apache.hive</groupId>
    <artifactId>hive-exec</artifactId>
    <version>1.2.1</version>
</dependency>

(2)创建SparkSession时需要添加hive支持(红色部分)
Spark内置Hive

val warehouseLocation: String = new File("spark-warehouse").getAbsolutePath

val spark = SparkSession
.builder()
.appName("Spark Hive Example")
.config("spark.sql.warehouse.dir", warehouseLocation)
.enableHiveSupport()
.getOrCreate()

外部Hive

val spark = SparkSession
.builder()
.appName("Spark Hive Example")
.enableHiveSupport()
.getOrCreate()

若使用的是外部Hive,还需要将hive-site.xml添加到ClassPath下。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值