从spark/scala项目代码中资源文件夹中的sql文件读取查询

Question

我在 IntelliJ 中的文件夹结构如下 src-->主-->资源-->sql-->samplequery.sql 我在文件夹 src--> main-->scala-->workflow 中有 scala 对象文件samplequeryexecute

在samplequery.sql文件中，我提到了一个简单的查询，例如 从表名中选择 * 在samplequeryexecute scala对象文件中，只想从samplequery.sql获取查询并执行它。假设所有其他必要的 Spark 属性，我已经拥有了。

请仅建议执行此操作的代码。

我还没有尝试过任何东西。

Answer 1

类路径资源

的函数

import scala.io.Source

def readResourceAsStream(resourceName: String): Iterator[String] = {
  val resourceStream = getClass.getResourceAsStream(resourceName)
  if (resourceStream == null) {
    throw new IllegalArgumentException(s"Resource not found: $resourceName")
  }
  Source.fromInputStream(resourceStream).getLines()
}

用途：

val lines = readResourceAsStream("/sql/samplequery.sql")
lines.foreach(println)