Как читать сжатые файлы avro (.gz) в spark?

#python #apache-spark #pyspark #gzip

#python #apache-spark #pyspark #gzip

Вопрос:

Я пытаюсь прочитать файл avro gzip (с расширением.gz) с помощью spark, но я получаю сообщение об ошибке ниже. Из документации я вижу, что spark должен иметь возможность читать файлы .gz без каких-либо дополнительных преобразований (возможно, для csv / текстовых файлов).

Я попытался выполнить приведенную ниже команду, но она выдает ошибку:

 df= spark.read.format("com.databricks.spark.avro").load("/user/data/test1.avro.gz")
 

Ошибка:

 Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "/usr/hdp/2.6.1.0-129/spark2/python/pyspark/sql/readwriter.py", line 149, in load
    return self._df(self._jreader.load(path))
  File "/usr/hdp/2.6.1.0-129/spark2/python/lib/py4j-0.10.4-src.zip/py4j/java_gateway.py", line 1133, in __call__
  File "/usr/hdp/2.6.1.0-129/spark2/python/pyspark/sql/utils.py", line 63, in deco
    return f(*a, **kw)
  File "/usr/hdp/2.6.1.0-129/spark2/python/lib/py4j-0.10.4-src.zip/py4j/protocol.py", line 319, in get_return_value
py4j.protocol.Py4JJavaError: An error occurred while calling o72.load.
: java.io.IOException: Not an Avro data file
        at org.apache.avro.file.DataFileReader.openReader(DataFileReader.java:63)
        at com.databricks.spark.avro.DefaultSource$anonfun$5.apply(DefaultSource.scala:80)
        at com.databricks.spark.avro.DefaultSource$anonfun$5.apply(DefaultSource.scala:77)
        at scala.Option.getOrElse(Option.scala:121)
        at com.databricks.spark.avro.DefaultSource.inferSchema(DefaultSource.scala:77)
        at org.apache.spark.sql.execution.datasources.DataSource$anonfun$7.apply(DataSource.scala:184)
        at org.apache.spark.sql.execution.datasources.DataSource$anonfun$7.apply(DataSource.scala:184)
        at scala.Option.orElse(Option.scala:289)
        at org.apache.spark.sql.execution.datasources.DataSource.org$apache$spark$sql$execution$datasources$DataSource$getOrInferFileFormatSchema(DataSource.scala:183)
        at org.apache.spark.sql.execution.datasources.DataSource.resolveRelation(DataSource.scala:387)
        at org.apache.spark.sql.DataFrameReader.load(DataFrameReader.scala:152)
        at org.apache.spark.sql.DataFrameReader.load(DataFrameReader.scala:135)
        at sun.reflect.NativeMethodAccessorImpl.invoke0(Native Method)
        at sun.reflect.NativeMethodAccessorImpl.invoke(NativeMethodAccessorImpl.java:62)
        at sun.reflect.DelegatingMethodAccessorImpl.invoke(DelegatingMethodAccessorImpl.java:43)
        at java.lang.reflect.Method.invoke(Method.java:498)
        at py4j.reflection.MethodInvoker.invoke(MethodInvoker.java:244)
        at py4j.reflection.ReflectionEngine.invoke(ReflectionEngine.java:357)
        at py4j.Gateway.invoke(Gateway.java:280)
        at py4j.commands.AbstractCommand.invokeMethod(AbstractCommand.java:132)
        at py4j.commands.CallCommand.execute(CallCommand.java:79)
        at py4j.GatewayConnection.run(GatewayConnection.java:214)
        at java.lang.Thread.run(Thread.java:748)
 

Ответ №1:

Сжатие в файле avro выполняется путем раздельного сжатия отдельных блоков данных, сам файл avro не сжимается (docs). Сжатие ORC и Parquet работает аналогичным образом, именно так эти форматы могут быть разделяемыми.

Другими словами, вы не можете запустить gzip несжатый файл .avro и прочитать его напрямую, как вы можете с обычными текстовыми файлами.

Сжатие происходит при записи файла avro, в spark это контролируется либо spark.sql.avro.compression.codec настройкой SparkConf, либо параметром сжатия в writer (docs).

Комментарии:

1. не могли бы вы добавить соответствующие ссылки на документацию spark / avro к вашему ответу. Спасибо!

2. обновлено @smaug. В соответствующей части спецификации Avro говорится, что «Объекты хранятся в блоках, которые могут быть сжаты».