如何通过PySpark将带有struct列的数据框写入Elasticsearch

时间:2020-05-11 13:09:19

标签: elasticsearch pyspark pyspark-dataframes elasticsearch-hadoop elasticsearch-spark

我正在尝试将包含struct列的数据框写入Elasticsearch:

df1 = spark.createDataFrame([{"date": "2020.04.10","approach": "test", "outlier_score": 1, "a":"1","b":2},
                       {"date": "2020.04.10","approach": "test", "outlier_score": 0, "a":"2","b":1}],
                       )

df1 = df1.withColumn('details', to_json(struct(
   col('a'),
   col('b')
)))

df1.show(truncate=False)

df1.select('date','approach','outlier_score','details').write.format("org.elasticsearch.spark.sql").option('es.resource', 'outliers').save(mode="append")

结果为:

+---+--------+---+----------+-------------+---------------+
|a  |approach|b  |date      |outlier_score|details        |
+---+--------+---+----------+-------------+---------------+
|1  |test    |2  |2020.04.10|1            |{"a":"1","b":2}|
|2  |test    |1  |2020.04.10|0            |{"a":"2","b":1}|
+---+--------+---+----------+-------------+---------------+   

这确实有效,但是JSON被转义了,因此在Kibana中无法单击相应的 details 字段:

    {
  "_index": "outliers",
  "_type": "_doc",
  "_id": "NuDSA3IBhHa_VjuWENYR",
  "_version": 1,
  "_score": 0,
  "_source": {
    "date": "2020.04.10",
    "approach": "test",
    "outlier_score": 1,
    "details": "{\"a\":\"1\",\"b\":2}"
  },
  "highlight": {
    "date": [
      "@kibana-highlighted-field@2020.04.10@/kibana-highlighted-field@"
    ]
  }
}

我尝试提供 .option(“ es.input.json”,“ true”),但出现异常:

org.elasticsearch.hadoop.rest.EsHadoopRemoteException: mapper_parsing_exception: failed to parse;org.elasticsearch.hadoop.rest.EsHadoopRemoteException: not_x_content_exception: Compressor detection can only be called on some xcontent bytes or compressed xcontent bytes

相反,如果我尝试写数据而不转换为JSON,即从原始代码中删除 to_json(,则出现另一个异常:

org.elasticsearch.hadoop.rest.EsHadoopRemoteException: mapper_parsing_exception: failed to parse field [details] of type [text] in document with id 'TuDWA3IBhHa_VjuWFNmX'. Preview of field's value: '{a=2, b=1}';org.elasticsearch.hadoop.rest.EsHadoopRemoteException: illegal_state_exception: Can't get text on a START_OBJECT at 1:68
    {"index":{}}
{"date":"2020.04.10","approach":"test","outlier_score":0,"details":{"a":"2","b":1}}

所以问题是如何将带有嵌套JSON列的PySpark数据框写入Elasticsearch,以使JSON不会被转义?

1 个答案:

答案 0 :(得分:0)

在不转换为JSON的情况下写入数据(没有 to_json )实际上不会产生任何异常。问题在于,已经为转义的JSON字段自动创建了映射。

为了解决该异常,应删除或重​​新创建索引。之后,将自动为作为对象的详细信息字段创建映射。或者,也可以删除带有明细字段的所有记录,然后将该字段的映射更改为对象类型。