如何以CSV格式编写窗口聚合?

时间:2019-01-09 14:22:51

标签: apache-spark apache-spark-sql spark-structured-streaming

我正在开发一个Spark结构化流应用程序,该应用程序可以流csv文件并将其与静态数据结合在一起。加入后我已经做了一些汇总。

将查询结果以CSV格式写入HDFS时,出现以下错误:

19/01/09 14:00:30 ERROR MicroBatchExecution: Query [id = 830ca987-b55a-4c03-aa13-f71bc57e47ad, runId = 87cdb029-0022-4f1c-b55e-c2443c9f058a] terminated with error java.lang.UnsupportedOperationException: CSV data source does not support struct<start:timestamp,end:timestamp> data type.
    at org.apache.spark.sql.execution.datasources.csv.CSVUtils$.org$apache$spark$sql$execution$datasources$csv$CSVUtils$$verifyType$1(CSVUtils.scala:127)
    at org.apache.spark.sql.execution.datasources.csv.CSVUtils$$anonfun$verifySchema$1.apply(CSVUtils.scala:131)
    at org.apache.spark.sql.execution.datasources.csv.CSVUtils$$anonfun$verifySchema$1.apply(CSVUtils.scala:131)

可能是根本原因?

这是我代码的相关部分:

val spark = SparkSession
  .builder
  .enableHiveSupport()
  .config("hive.exec.dynamic.partition", "true")
  .config("hive.exec.dynamic.partition.mode", "nonstrict")
  .config("spark.sql.streaming.checkpointLocation", "/user/sas/sparkCheckpoint")
  .getOrCreate

...

val df_agg_without_time = sqlResultjoin
  .withWatermark("event_time", "10 seconds")
  .groupBy(
    window($"event_time", "10 seconds", "5 seconds"),
    $"section",
    $"timestamp")
  .agg(sum($"total") as "total")

...

finalTable_repo
  .writeStream
  .outputMode("append")
  .partitionBy("xml_data_dt")
  .format("csv")
  .trigger(Trigger.ProcessingTime("2 seconds"))
  .option("path", "hdfs://op/apps/hive/warehouse/area.db/finalTable_repo")
  .start

1 个答案:

答案 0 :(得分:1)

进行聚合的行.groupBy(window($"event_time", "10 seconds", "5 seconds"), $"section", $"timestamp")创建了CSV数据源不支持的struct<start:timestamp,end:timestamp>数据类型。

df_agg_without_time.printSchema,您会看到该列。

一种解决方案就是将其转换为其他一些更简单的类型(可能使用selectwithColumn)或仅将其转换为select(即不包括在以下数据框中)。 / p>


以下是一个示例批处理(非流式)结构化查询,该查询显示了流式结构化查询使用的架构(创建df_agg_without_time时)。

val q = spark
  .range(4)
  .withColumn("t", current_timestamp)
  .groupBy(window($"t", "10 seconds"))
  .count
scala> q.printSchema
root
 |-- window: struct (nullable = false)
 |    |-- start: timestamp (nullable = true)
 |    |-- end: timestamp (nullable = true)
 |-- count: long (nullable = false)

对于样本流查询,您可以使用费率数据源。

val q = spark
  .readStream
  .format("rate")
  .load
  .groupBy(window($"timestamp", "10 seconds"))
  .count
scala> q.printSchema
root
 |-- window: struct (nullable = false)
 |    |-- start: timestamp (nullable = true)
 |    |-- end: timestamp (nullable = true)
 |-- count: long (nullable = false)