Spark联合失败,嵌套的JSON数据帧

时间:2017-03-01 11:24:35

标签: scala apache-spark union spark-dataframe

我有以下两个JSON文件:

{
    "name" : "Agent1",
    "age" : "32",
    "details" : [{
            "d1" : 1,
            "d2" : 2
        }
    ]
}

{
    "name" : "Agent2",
    "age" : "42",
    "details" : []
}

我用spark阅读它们:

val jsonDf1 = spark.read.json(pathToJson1)
val jsonDf2 = spark.read.json(pathToJson2)

使用以下模式创建两个数据帧:

root
 |-- age: string (nullable = true)
 |-- details: array (nullable = true)
 |    |-- element: struct (containsNull = true)
 |    |    |-- d1: long (nullable = true)
 |    |    |-- d2: long (nullable = true)
 |-- name: string (nullable = true)

root
|-- age: string (nullable = true)
|-- details: array (nullable = true)
|    |-- element: string (containsNull = true)
|-- name: string (nullable = true)

当我尝试使用这两个数据帧执行联合时,我收到此错误:

jsonDf1.union(jsonDf2)


org.apache.spark.sql.AnalysisException: unresolved operator 'Union;;
'Union
:- LogicalRDD [age#0, details#1, name#2]
+- LogicalRDD [age#7, details#8, name#9]

我该如何解决这个问题?我有时会在JSON文件中得到空数组,火花作业会加载,但它仍然必须统一它们,这不应该是一个问题,因为Json文件的模式是相同的。

2 个答案:

答案 0 :(得分:4)

如果你试图联合2个数据帧,你会得到这个:

error:org.apache.spark.sql.AnalysisException: Union can only be performed on tables with the compatible column types. ArrayType(StringType,true) <> ArrayType(StructType(StructField(d1,StringType,true), StructField(d2,StringType,true)),true) at the second column of the second table

Json文件同时到达

要解决此问题,如果您可以同时阅读JSON,我建议:

val jsonDf1 = spark.read.json("json1.json", "json2.json")

这将给出这个架构:

jsonDf1.printSchema
 |-- age: string (nullable = true)
 |-- details: array (nullable = true)
 |    |-- element: struct (containsNull = true)
 |    |    |-- d1: long (nullable = true)
 |    |    |-- d2: long (nullable = true)
 |-- name: string (nullable = true)

数据输出

jsonDf1.show(10,truncate = false)
+---+-------+------+
|age|details|name  |
+---+-------+------+
|32 |[[1,2]]|Agent1|
|42 |null   |Agent2|
+---+-------+------+

Json文件在不同时间到达

如果您的json在不同时间到达,作为默认解决方案,我建议您阅读带有完整数组的模板JSON对象,这将使您的数据帧具有可能的空数组,对任何联合都有效。然后,在输出结果之前,您将使用过滤器删除此假JSON:

val df = spark.read.json("jsonWithMaybeAnEmptyArray.json", 
"TemplateFakeJsonWithAFullArray.json")

df.filter($"name" !== "FakeAgent").show(1)

请注意:已打开Jira卡以提高合并SQL数据类型的功能:https://issues.apache.org/jira/browse/SPARK-19536,这种操作应该可以在下一个Spark版本中使用。

答案 1 :(得分:2)

polomarcus的回答让我得到了这个解决方案: 我无法一次读取所有文件,因为我得到了一个文件列表作为输入,并且spark没有接收路径列表的API,但显然使用Scala它可以这样做:

val files = List("path1", "path2", "path3")
val dataframe = spark.read.json(files: _*)

这样我就得到了一个包含所有三个文件的数据框。