scala spark sql 若该条记录第一次 进入数仓 dwd 层则 dwd_insert_time、dwd_modify_time 均存当前操作时间, 并进行数据类型转换。若该数据在进入 dwd 层时发生了合并修改,则 dwd_insert_time 时间不变,dwd_modify_time 存当前操作时间,其余列存 最新的值,必须用sql语法
时间: 2023-06-13 18:06:59 浏览: 178
可以使用 Scala Spark SQL 中的 `merge` 语法来实现该需求,具体实现如下:
```scala
import org.apache.spark.sql.functions._
import spark.implicits._
// 假设输入数据集为 inputDF,包含字段 id、name、age、dwd_insert_time、dwd_modify_time
val inputDF = Seq(
(1, "Alice", 20, "2021-01-01 10:00:00", "2021-01-01 10:00:00"),
(2, "Bob", 25, "2021-01-01 10:00:00", "2021-01-01 10:00:00")
).toDF("id", "name", "age", "dwd_insert_time", "dwd_modify_time")
// 假设输出数据集为 outputDF,包含字段 id、name、age、dwd_insert_time、dwd_modify_time
val outputDF = Seq(
(1, "Alice", 20, "2021-01-01 10:00:00", "2021-01-02 11:00:00"), // 修改 dwd_modify_time
(3, "Charlie", 30, "2021-01-02 12:00:00", "2021-01-02 12:00:00") // 新增记录
).toDF("id", "name", "age", "dwd_insert_time", "dwd_modify_time")
// 定义合并逻辑
val mergedDF = inputDF.as("input")
.merge(outputDF.as("output"), "id")
.whenMatched(
// 如果存在匹配的 id,则表示数据被修改过,更新 dwd_modify_time 和其他列的值
updateExprs = Map(
"name" -> $"output.name",
"age" -> $"output.age",
"dwd_modify_time" -> current_timestamp()
)
)
.whenNotMatched(
// 如果不存在匹配的 id,则表示是新的记录,插入当前时间作为 dwd_insert_time 和其他列的值
insertExprs = Map(
"name" -> $"output.name",
"age" -> $"output.age",
"dwd_insert_time" -> current_timestamp(),
"dwd_modify_time" -> current_timestamp()
)
)
.as("merged")
// 选择需要的列并进行数据类型转换
val resultDF = mergedDF.select(
$"merged.id",
$"merged.name",
$"merged.age".cast("int"),
$"merged.dwd_insert_time".cast("timestamp"),
$"merged.dwd_modify_time".cast("timestamp")
)
// 输出结果
resultDF.show()
```
以上代码中,我们使用 `merge` 语法将输入数据集和输出数据集合并,当存在匹配的 id 时更新 dwd_modify_time 和其他列的值,当不存在匹配的 id 时插入当前时间作为 dwd_insert_time 和其他列的值。最后选择需要的列并进行数据类型转换,输出结果。
阅读全文