当前位置:Gxlcms > 数据库问题 > Spark(九) -- SparkSQL API编程

Spark(九) -- SparkSQL API编程

时间:2021-07-01 10:21:17 帮助过:30人阅读

TextFile{ def main(args:Array[String]){ } }

SparkSQL编程模型:

第一步:
需要一个SQLContext对象,该对象是SparkSQL操作的入口
而构建一个SQLContext对象需要一个SparkContext

第二步:
构建好入口对象之后,要引入隐式转换的方法,作用是将读取到的各种文件转换成DataFrame,DataFrame是SparkSQL上进行统一操作的数据类型

第三步:
根据数据的格式,构建一个样例类。作用是提供将读取到的各种各样的数据类型隐式转换成一个统一的数据格式,方便编程

第四步:
使用SQLContext对象读取文件,并将其转换成DataFrame

第五步:
对数据进行相关操作。
1.DataFrame自带的操作方式。DataFrame提供了很多操作数据的方法,如where,select等

2.DSL方式。DSL其实使用的也是DataFrame提供的方法,但是在操作属性时可以方便的使用’ + 属性名的方式进行操作

3.将数据注册成表,通过SQL语句操作

  1. <code class=" hljs go">object TextFile{
  2. def main(args:Array[String]){
  3. <span class="hljs-comment">//第一步</span>
  4. <span class="hljs-comment">//构建SparkContext对象,主要要使用new调用构造方法,否则就变成使用样例类的Apply方法了</span>
  5. val sc = <span class="hljs-built_in">new</span> SparkContext()
  6. <span class="hljs-comment">//构建SQLContext对象</span>
  7. val sqlContext = <span class="hljs-built_in">new</span> SQLContext(sc)
  8. <span class="hljs-comment">//第二步</span>
  9. <span class="hljs-keyword">import</span> sqlContext.implicits._
  10. <span class="hljs-comment">//第三步</span>
  11. <span class="hljs-keyword">case</span> Person(name:String,age:Int)
  12. <span class="hljs-comment">//第四步,textFile从指定路径读取文件如果是集群模式要写hdfs文件地址;通过两个map操作将读取到的文件转换成Person类的对象,每一行对应一个Person对象;toDF将其转换成DataFrame</span>
  13. val people = sc.textFile(<span class="hljs-string">"文件路径"</span>).<span class="hljs-keyword">map</span>(_.split(<span class="hljs-string">","</span>)).<span class="hljs-keyword">map</span>{<span class="hljs-keyword">case</span> (name,age) => Person(name,age.toInt)}.toDF()
  14. <span class="hljs-comment">//第五步</span>
  15. <span class="hljs-comment">//DataFrame方法</span>
  16. <span class="hljs-built_in">println</span>(<span class="hljs-string">"------------------------DataFrame------------------------------------"</span>)
  17. <span class="hljs-comment">//赛选出age>10的记录,然后只选择name属性,show方法将其输出</span>
  18. people.where(people(<span class="hljs-string">"age"</span>) ><span class="hljs-number"> 10</span>).<span class="hljs-keyword">select</span>(people(<span class="hljs-string">"name"</span>)).show()
  19. <span class="hljs-comment">//DSL</span>
  20. <span class="hljs-built_in">println</span>(<span class="hljs-string">"---------------------------DSL---------------------------------"</span>)
  21. people.where(<span class="hljs-string">‘age > 10).select(‘</span>name).show()
  22. <span class="hljs-comment">//SQL</span>
  23. <span class="hljs-built_in">println</span>(<span class="hljs-string">"-----------------------------SQL-------------------------------"</span>)
  24. <span class="hljs-comment">//将people注册成people表</span>
  25. people.registerTempTable(<span class="hljs-string">"people"</span>)
  26. <span class="hljs-comment">//使用sqlContext的sql方法来写SQL语句</span>
  27. <span class="hljs-comment">//查询返回的是RDD,所以对其进行collect操作,之后循环打印</span>
  28. sqlContext.sql(<span class="hljs-string">"select name from people where age > 10"</span>).collect.foreach(<span class="hljs-built_in">println</span>)
  29. <span class="hljs-comment">//保存为parquet文件,之后的parquet演示会用到</span>
  30. people.saveAsParquet(<span class="hljs-string">"保存的路径"</span>)
  31. }
  32. }</code>

parquet格式文件测试:

  1. <code class=" hljs go">val sc = <span class="hljs-built_in">new</span> SparkContext()
  2. val sql = <span class="hljs-built_in">new</span> SQLContext(sc)
  3. <span class="hljs-keyword">import</span> sql.implicits._
  4. val parquet = sql.parquetFile(args<span class="hljs-number">(0</span>))
  5. <span class="hljs-built_in">println</span>(<span class="hljs-string">"------------------------DataFrame------------------------------------"</span>)
  6. <span class="hljs-built_in">println</span>(parquet.where(parquet(<span class="hljs-string">"age"</span>) ><span class="hljs-number"> 10</span>).<span class="hljs-keyword">select</span>(parquet(<span class="hljs-string">"name"</span>)).show())
  7. <span class="hljs-built_in">println</span>(<span class="hljs-string">"---------------------------DSL---------------------------------"</span>)
  8. <span class="hljs-built_in">println</span>(parquet.where(<span class="hljs-string">‘age > 10).select(‘</span>name).show())
  9. <span class="hljs-built_in">println</span>(<span class="hljs-string">"-----------------------------SQL-------------------------------"</span>)
  10. parquet.registerTempTable(<span class="hljs-string">"parquet"</span>)
  11. sql.sql(<span class="hljs-string">"select name from parquet where age > 10"</span>).<span class="hljs-keyword">map</span>(p => <span class="hljs-string">"name:"</span> + p<span class="hljs-number">(0</span>)).collect().foreach(<span class="hljs-built_in">println</span>)</code>

Json格式测试:

  1. <code class=" hljs go">val sc = <span class="hljs-built_in">new</span> SparkContext()
  2. val sql = <span class="hljs-built_in">new</span> SQLContext(sc)
  3. <span class="hljs-keyword">import</span> sql.implicits._
  4. val json = sql.jsonFile(args<span class="hljs-number">(0</span>))
  5. <span class="hljs-built_in">println</span>(<span class="hljs-string">"------------------------DataFrame------------------------------------"</span>)
  6. <span class="hljs-built_in">println</span>(json.where(json(<span class="hljs-string">"age"</span>) ><span class="hljs-number"> 10</span>).<span class="hljs-keyword">select</span>(json(<span class="hljs-string">"name"</span>)).show())
  7. <span class="hljs-built_in">println</span>(<span class="hljs-string">"---------------------------DSL---------------------------------"</span>)
  8. <span class="hljs-built_in">println</span>(json.where(<span class="hljs-string">‘age > 10).select(‘</span>name).show())
  9. <span class="hljs-built_in">println</span>(<span class="hljs-string">"-----------------------------SQL-------------------------------"</span>)
  10. json.registerTempTable(<span class="hljs-string">"json"</span>)
  11. sql.sql(<span class="hljs-string">"select name from json where age > 10"</span>).<span class="hljs-keyword">map</span>(p => <span class="hljs-string">"name:"</span> + p<span class="hljs-number">(0</span>)).collect().foreach(<span class="hljs-built_in">println</span>)</code>

可以看到上面的代码几乎和读取文本文件的一模一样,只不顾sc在读取文件的时候使用了parquetFile/jsonFile方法,而之后的操作是一摸一样的
由于parquet和json数据读取进来就是一个可操作的格式并且会自动转换成DataFrame,所以省去了case class的定义步骤和toDF的操作

以上为SparkSQL API的简单使用

Spark(九) -- SparkSQL API编程

标签:sparksql

人气教程排行