如何将只包含头（不包含行）的数据集写入hdfs位置（csv格式），以便在下载时包含头？

Question

我有一个数据集，它只包含头(id、名称、年龄）和0行。我想将它作为csv文件写入hdfs位置，使用

DataFrameWriter dataFrameWriter = dataset.write();
Map<String, String> csvOptions = new HashMap<>();
csvOptions.put("header", "true");
dataFrameWriter = dataFrameWriter.options(csvOptions);
dataFrameWriter.mode(SaveMode.Overwrite).csv(location);

1. _SUCCESS
2. tempFile.csv

Answer 1

如果查看代码，您会发现只有在至少有一行的情况下才会写入标头。

UnivocityGenerator.Scala

  def write(row: InternalRow): Unit = {
    if (printHeader) {
      gen.writeHeaders()
    }
    gen.writeRow(convertRow(row): _*)
    printHeader = false
  }