Spark SQL基本概念与基本用法

蔚落 2022-01-05 10:07 409阅读 0赞

# 1. Spark SQL概述 #

## 1.1 什么是Spark SQL ##

Spark SQL是Spark用来处理结构化数据的一个模块，它提供了两个编程抽象分别叫做DataFrame和DataSet，它们用于作为分布式SQL查询引擎。从下图可以查看RDD、DataFrames与DataSet的关系。

![1217276-20180512172718536-491676089.png][]

## 1.2 为什么要学习Spark SQL ##

Hive，它是将Hive SQL转换成MapReduce，然后提交到集群上执行的，大大简化了编写MapReduce程序的复杂性，而且MapReduce这种计算模型执行效率比较慢。类比Hive，Spark SQL，它时将Spark SQL转换成RDD，然后提交到集群上执行，执行效率非常快！

# 2. DataFrames #

## 2.1 什么是DataFrames ##

与RDD类似，DataFrame也是一个分布式数据容器。然而DataFrame更像传统数据库的二维表格，除了数据以外，还记录数据的结构信息，即schema。同时，与Hive类似，DataFrame也支持嵌套数据类型（struct、array和map）。从API易用性的角度上看，DataFrame API提供的是一套高层的关系操作，比函数式的RDD API要更加友好，门槛更低。由于与R和Pandas的DataFrame类似，Spark DataFrame很好地继承了传统单机数据分析的开发体验。

![1217276-20180512011416077-2063816032.png][]

## 2.2 创建DataFrames ##

在Spark SQL中SQLContext是创建DataFrames和执行SQL的入口，在spark中已经内置了一个sqlContext。

创建DataFrames的步骤：

1) 在本地创建一个文件，有三列，分别是id、name、age，用空格分隔，然后上传到hdfs上

hdfs dfs -put person.txt /

2) 在spark shell执行下面命令，读取数据，将每一行的数据使用列分隔符分割

val lineRDD = sc.textFile("hdfs://hadoop1:9000/person.txt").map(_.split(" "))

3) 定义case class(相当于表的schema)

case class Person(id:int, name:string, age:int)

4) 将RDD和case class关联

val personRDD = lineRDD.map(x => Person(x(0).toInt, x(1), x(2).toInt))

5) 将RDD转换成DataFrame

val personDF = personRDD.toDF

6) 对DataFrame进行处理

personDF.show

## 2.3 DataFrame常用操作 ##

### 2.3.1 DSL风格语法 ###

查看DataFrame中的内容

personDF.show

查看DataFrame部分列中的内容

personDF.select(personDF.col("name")).show
    personDF.select(col("name"),col("age")).show
    personDF.select("name").show

打印DataFrame的Scheme信息

personDF.printSchema

查询所有的name和age，并将age+1

personDF.select(col("id"),col("name"),col("age")+1).show
    personDF.select(personDF("id"), personDF("name"), personDF("age")+1).show

过滤age大于等于18的

personDF.filter(col("age") >= 18 ).show

按年龄进行分组并统计相同年龄的人数

personDF.groupBy("age").count().show

### 2.3.2 SQL风格语法 ###

如果想使用SQL风格的语法，需要将DataFrame注册成表

personDF.createOrReplaceTempView("t_person") //spark2.x以上写法
    personDF.registerTempTable("t_person")   //spark1.6以下写法

查询年龄最大的前两名

sqlContext.sql("select * from t_person order by age desc limit 2").show

显示表的Schema信息

sqlContext.sql("desc t_person").show

# 3. DataSet #

## 3.1 什么是DataSet ##

DataSet是从Spark 1.6开始引入的一个新的抽象。DataSet是特定域对象中的强类型集合，它可以使用函数或者相关操作并行地进行转换等操作。每个DataSet都有一个称为DataFrame的非类型化的视图，这个视图是行的数据集。为了有效地支持特定域对象，DataSet引入了Encoder（编码器）。例如，给出一个Person的类，有两个字段：name(string)和age(int)，通过一个encoder来告诉spark在运行的时候产生代码把Person对象转换成一个二进制结构。这种二进制结构通常有更低的内存占用，以及优化的数据处理效率（例如在一个柱状格式）。若要了解数据的内部二进制表示，请使用schema(表结构)函数。

在DataSet上的操作，分为transformations和actions。transformations会产生新的DataSet，而actions则是触发计算并产生结果。transformations包括：map，filter，select和aggregate等操作。而actions包括：count，show或把数据写入到文件系统中。

RDD也是可以并行化的操作，DataSet和RDD主要的区别是：DataSet是特定域的对象集合；然而RDD是任何对象的集合。DataSet的API总是强类型的；而且可以利用这些模式进行优化，然而RDD却不行。

DataFrame是特殊的DataSet，它在编译时不会对模式进行检测。

## 3.2 创建DataSet ##

1) 直接通过Seq创建

case class Data(a: Int， b: String)
    val ds = Seq(Data(1, "one"), Data(2, "two")).toDS()
    ds.collect()
    ds.show()

2) 通过sqlContext创建

case class Person(name: String, zip: Long)
    val df = sqlContext.read.json(sc.parallelize("""{"zip": 94709, "name": "Michael"}""" :: Nil))
    df.as[Person].collect()
    df.as[Person].show()

## 3.3 DataSet使用示例 ##

1) DataSet的WordCount

import org.apache.spark.sql.functions._
    #创建DataSet
    val ds = sqlContext.read.text("hdfs://node-1.itcast.cn:9000/wc").as[String]
    val result = ds.flatMap(_.split(" "))
                          .filter(_ != "")
                          .toDF()
                          .groupBy($"value")
                          .agg(count("*") as "numOccurances")
                          .orderBy($"numOccurances" desc)
                          .show()
    
    
    val wordCount = ds.flatMap(_.split(" ")).filter(_ != "").groupBy(_.toLowerCase()).count().show()

2) 对hdfs上的json文件进行分析

样例schools.json：

{"name": "UC Berkeley", "yearFounded": 1868, "numStudents": 37581}
    {"name": "MIT", "yearFounded": 1860, "numStudents": 11318}

把schools.json上传到hdfs上：

hdfs dfs -put schools.json /

编写代码：

#定义case class
    case class University(name: String, numStudents: Long, yearFounded: Long)
    
    #创建DataSet
    val schools = sqlContext.read.json("hdfs://hadoop1:9000/schools.json").as[University]
    
    #操作DataSet
    schools.map(sc => s"${sc.name} is ${2015 - sc.yearFounded} years old").show

3) 对person.json进行分析

把json转DataFrame

#JSON -> DataFrame
    val df = sqlContext.read.json("hdfs://hadoop1:9000/person.json")

使用DataFrame操作

df.where($"age" >= 20).show
    df.where(col("age") >= 20).show
    df.printSchema

DataFrame转DataSet

#DataFrame -> Dataset
    case class Person(age: Long, name: String)
    
    val ds = df.as[Person]
    ds.filter(_.age >= 20).show

DataSet转DataFrame

# Dataset -> DataFrame
    val df2 = ds.toDF

DataFrame和DataSet操作对比

import org.apache.spark.sql.types._
    
    df.where($"age" > 0)
       .groupBy((($"age" / 10) cast IntegerType) * 10 as "decade")
       .agg(count("*"))
       .orderBy($"decade")
       .show 
      
    ds.filter(_.age > 0)
        .groupBy(p => (p.age / 10) * 10)
        .agg(count("name"))
        .toDF()
        .withColumnRenamed("value", "decade")
        .orderBy("decade")
        .show

4) 对student.json进行分析

把json转DataFrame

val df = sqlContext.read.json("hdfs://hadoop1:9000/student.json")

DataFrame转DataSet

case class Student(name: String, age: Long, major: String)
    val studentDS = df.as[Student]
    studentDS.select($"name".as[String], $"age".as[Long]).filter(_._2 > 19).collect()

DataSet根据major分组求和

studentDS.groupBy(_.major).count().collect()

DataSet根据major分组聚合

import org.apache.spark.sql.functions._
    
    studentDS.groupBy(_.major).agg(avg($"age").as[Double]).collect()

创建Major的分类

case class Major(shortName: String, fullName: String)
    val majors = Seq(Major("CS", "Computer Science"), Major("Math", "Mathematics")).toDS()

把studentDS和majors求join

val joined = studentDS.joinWith(majors, $"major" === $"shortName")
    
    joined.map(s => (s._1.name, s._2.fullName)).show()
    
    joined.explain()

# 4. 以编程方式执行Spark SQL查询 #

## 4.1 编写Spark SQL查询程序 ##

前面我们学习了如何在Spark Shell中使用SQL完成查询，现在我们来实现在自定义的程序中编写Spark SQL查询程序。首先在maven项目的pom.xml中添加Spark SQL的依赖

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

### 4.1.1 通过反射推断Schema ###

import org.apache.spark.sql.{SQLContext, SparkSession}
    import org.apache.spark.{SparkConf, SparkContext}
    
    /**
      * @author y15079
      * @create 2018-03-11 14:23
      * @desc
      **/
    object SQLDemo {
    
      def main(args: Array[String]): Unit = {
        //本地运行
        val conf = new SparkConf().setAppName("SQLDemo").setMaster("local[2]")
        //val conf = new SparkConf().setAppName("SQLDemo") 要打包到spark集群上运行则不需要后面的setMaster("local[2]")
    
        //SQLContext要依赖SparkContext
        val sc = new SparkContext(conf)
        //创建SQLContext spark1.6.1以下的写法
        //val sqlContext = new SQLContext(sc)
    
        //spark2.0 以上的写法
        val sqlContext = SparkSession.builder().config(conf).getOrCreate()
    
        //提交到spark集群上运行，需要设置用户，否则无权限执行，本地运行则无需
        //System.setProperty("user.name", "bigdata")
    
        //集群hdfs路径  hdfs://node-1.itcast.cn:9000/person.txt
        //下面由于是本地运行，所以采用本地路径
        //将RDD和case class关联
        val personRdd = sc.textFile("D:\\IDEA\\HelloSpark\\src\\main\\files\\day4\\person.txt").map({line =>
          val fields = line.split(",")
          Person(fields(0).toLong, fields(1), fields(2).toInt)
        })
    
        //导入隐式转换，如果不导入无法将RDD转换成DataFrame
        //将RDD转换成DataFrame
        import sqlContext.implicits._
        val personDf = personRdd.toDF()
    
        //采用SQL编写风格 注册表
        //personDf.registerTempTable("person") spark 1.6.1以下的写法
        personDf.createOrReplaceTempView("person")
    
        sqlContext.sql("select * from person where age >= 20 order by age desc limit 2").show()
      }
    }
    //case class一定要放到外面
    case class Person(id:Long, name:String, age:Int)

### 4.1.2 通过StructType直接指定Schema ###

import org.apache.spark.{SparkConf, SparkContext}
    import org.apache.spark.sql.{Row, SQLContext, SparkSession}
    import org.apache.spark.sql.types.{IntegerType, StringType, StructField, StructType}
    
    /**
      * @author y15079
      * @create 2018-05-12 2:12
      * @desc
      **/
    object SQLSchemaDemo {
      def main(args: Array[String]): Unit = {
        //本地运行
        val conf = new SparkConf().setAppName("SQLSchemaDemo").setMaster("local[2]")
        //val conf = new SparkConf().setAppName("SQLDemo") 要打包到spark集群上运行则不需要后面的setMaster("local[2]")
        //SQLContext要依赖SparkContext
        val sc = new SparkContext(conf)
        //创建SQLContext spark1.6.1以下的写法
        //val sqlContext = new SQLContext(sc)
    
        //spark2.0 以上的写法
        val sqlContext = SparkSession.builder().config(conf).getOrCreate()
    
        //提交到spark集群上运行，需要设置用户，否则无权限执行，本地运行则无需
        //System.setProperty("user.name", "bigdata")
    
        //集群hdfs路径  hdfs://node-1.itcast.cn:9000/person.txt
        //下面由于是本地运行，所以采用本地路径
        //从指定的地址创建RDD
        val personRDD = sc.textFile("D:\\IDEA\\HelloSpark\\src\\main\\files\\day4\\person.txt").map(_.split(","))
        //通过StructType直接指定每个字段的schema
        val schema = StructType(
          List(
            StructField("id", IntegerType, true),
            StructField("name", StringType, true),
            StructField("age", IntegerType, true)
          )
        )
    
        //将RDD映射到rowRDD
        val rowRDD = personRDD.map(p=>Row(p(0).toInt, p(1).trim, p(2).toInt))
        //将schema信息应用到rowRDD上
        val personDataFrame = sqlContext.createDataFrame(rowRDD, schema)
    
        //注册表
        //personDataFrame.registerTempTable("person") spark 1.6.1以下的写法
        personDataFrame.createOrReplaceTempView("person")
        //执行SQL
        val df = sqlContext.sql("select * from person where age >= 20 order by age desc limit 2").show()
    
        sc.stop()
      }
    }

# 5. 数据源 #

## 5.1 JDBC ##

Spark SQL可以通过JDBC从关系型数据库中读取数据的方式创建DataFrame，通过对DataFrame一系列的计算后，还可以将数据再写回关系型数据库中。

### 5.1.1 从Mysql中加载数据库(Spark Shell 方式) ###

1) 启动Spark Shell，必须指定mysql连接驱动jar包

spark-shell --master spark://hadoop1:7077 --jars mysql-connector-java-5.1.35-bin.jar --driver-class-path mysql-connector-java-5.1.35-bin.jar

2) 从mysql中加载数据

val jdbcDF = sqlContext.read.format("jdbc").option(
         Map("url"->"jdbc:mysql://hadoop1:3306/bigdata",
                "driver"->"com.mysql.jdbc.Driver", 
                "dbtable"->"person", 
                "user"->"root",
                "password"->"123456")
         ).load()

3) 执行查询

jdbcDF.show()

### 5.1.2 将数据写入到MySQL中(打jar包方式) ###

1) 编写Spark SQL程序

import java.util.Properties
    import org.apache.spark.sql.{Row, SQLContext, SparkSession}
    import org.apache.spark.sql.types.{IntegerType, StringType, StructField, StructType}
    import org.apache.spark.{SparkConf, SparkContext}
    
    /**
      * @author y15079
      * @create 2018-05-12 2:50
      * @desc
      **/
    object JdbcDFDemo {
      def main(args: Array[String]): Unit = {
        val conf = new SparkConf().setAppName("MysqlDemo").setMaster("local[2]")
        val sc = new SparkContext(conf)
    
        //创建SQLContext spark1.6.1以下的写法
        //val sqlContext = new SQLContext(sc)
    
        //spark2.0 以上的写法
        val sqlContext = SparkSession.builder().config(conf).getOrCreate()
    
        //通过并行化创建RDD
        val personRDD = sc.parallelize(Array("1 tom 5", "2 jerry 3", "3 kitty 6")).map(_.split(" "))
        //通过StructType直接指定每个字段的schema
        val schema = StructType(
          List(
            StructField("id", IntegerType, true),
            StructField("name", StringType, true),
            StructField("age", IntegerType, true)
          )
        )
    
        //将RDD映射到rowRDD
        val rowRDD = personRDD.map(p=>Row(p(0).toInt, p(1).trim, p(2).toInt))
        //将schema信息应用到rowRDD上
        val personDataFrame = sqlContext.createDataFrame(rowRDD, schema)
        //创建Properties存储数据库相关属性
        val prop = new Properties()
        prop.put("user", "root")
        prop.put("password", "123456")
        //将数据追加到数据库
        personDataFrame.write.mode("append").jdbc("jdbc:mysql://localhost:3306/bigdata","bigdata.person", prop)
    
        sc.stop()
      }
    }

2) 用maven-shade-plugin插件将程序打包

3) 将jar包提交到spark集群

spark-submit 
    --class cn.itcast.spark.sql.jdbcDF 
    --master spark://hadoop1:7077 
    --jars mysql-connector-java-5.1.35-bin.jar 
    --driver-class-path mysql-connector-java-5.1.35-bin.jar 
    /root/demo.jar

[1217276-20180512172718536-491676089.png]: /images/20211229/641255ac374c4fa5ac8e4508250a95c8.png
[1217276-20180512011416077-2063816032.png]: /images/20211229/9cb783040372410c947e9a6be657e8e6.png