spark运行时加载hive，hdfs配置文件

 

以前我用spark连接hive都是把hive的配置文件放在spark的conf目录，后来有个项目在运行时才能确定要连接哪个hive源，我就找了个能在运行时加载配置文件的方法，代码如下：

import org.apache.hadoop.conf.Configuration
import org.apache.hadoop.fs.Path
import org.apache.spark.sql.SparkSession
import scala.collection.JavaConverters._

object ReadHive {

  def main(args: Array[String]): Unit = {
    val sparkBuilder = SparkSession
      .builder
      .master("local")
      .appName("Spk Pi")
    val conf = new Configuration()
    // 这里的文件地址可以换成从数据库里查询
    val core = new Path("C:\\Users\\shadow\\Desktop\\core-site.xml")
    val hdfs = new Path("C:\\Users\\shadow\\Desktop\\hdfs-site.xml")
    val hive = new Path("C:\\Users\\shadow\\Desktop\\hive-site.xml")
    conf.addResource(core)
    conf.addResource(hdfs)
    conf.addResource(hive)
    for (c <- conf.iterator().asScala){
      sparkBuilder.config(c.getKey, c.getValue)
    }
    val spark = sparkBuilder.enableHiveSupport().getOrCreate()
    spark.sql("select * from default.wt_test1").show()
  }

}