CSVFormat 在标题中添加引号

问题描述

我正在使用 apache.commons.csv 创建一个 CSV 文件。生成文件时，我想使用 .withHeader() 方法解析标头。我解析的值应该有像“FILENAME”这样的引号。为此，我想使用枚举类。但是，我不太确定在名称周围添加引号是如何工作的。我使用该代码对其进行了测试：

    private enum Header{
        FILENAME("\"FILENAME\""),TEST("\"Test\"");
    
        private final String string;
    
        Header(String string){
            this.string = string;
        }
    }

    public void addHeader(File outFile) throws IOException{
            CSVPrinter printer = new CSVPrinter(outFile,CSVFormat.DEFAULT.withHeader(Header.class));
    }

然而，当创建 outfile 时，它只包含没有引号的标题。有谁知道如何解决它？我知道我使用字符串数组作为标题，但我想避免这种情况，因为我有很多值，字符串会变得很长。

解决方法

是因为withHeader(Header.class)的实现方式，它使用Enums的.name()方法来获取枚举值作为字符串，它获取实际枚举值作为字符串：FILENAME而不是{ {1}}。要在 csv 中获取双引号，您可以使用这样的字符串数组：

"\"FILENAME\""

根据 George Nechifor 的建议，我们可以使用 CSVFormat#withHeader(String...) 以获得更好的灵活性。为了避免重复和硬编码，我们当然应该使用枚举中的值，并进行简单的映射。

Arrays.stream(Header.values()).map(Header::getCsvHeader).toArray(String[]::new)

并且会导致
"""FILENAME""","""Test"""
一个可运行的示例如下（使用 apache commons-csv 1.8）：

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;
import java.util.Arrays;

public class CsvWithEnumHeader {
    public static void main(String[] args) throws IOException {
        addHeader(new File("C://temp/demo.csv"));
    }

    public static void addHeader(File outFile) throws IOException {
        // apache common csv 1.8 cannot compile with new CSVPrinter(outFile,...)
        try (CSVPrinter printer = new CSVPrinter(new FileWriter(outFile),CSVFormat.DEFAULT.withHeader(Arrays.stream(Header.values()).map(Header::getCsvHeader).toArray(String[]::new)))) {
        }
    }
}

public enum Header {
    FILENAME("\"FILENAME\""),TEST("\"Test\"");

    private final String csvHeader;

    Header(String csvHeader) {
        this.csvHeader = csvHeader;
    }

    public String getCsvHeader() {
        return csvHeader;
    }
}

从评论中更新

似乎OP想要的是标题被引用而不是包含双引号的标题。我们可能会使用 withQuoteMode(QuoteMode.ALL) 并删除 \" 中的 getCsvHeader。

CSVFormat.DEFAULT.withQuoteMode(QuoteMode.ALL).withHeader(Arrays.stream(Header.values()).map(Header::getCsvHeader).toArray(String[]::new)))

结果将是
"FILENAME","Test"
请注意，这也会引用 csv 中的记录。如果我们只想引用header，请参考CSVPrinter remove quotes only from header，虽然是逆向的，但是技巧还是可以的。

apache-commons-csv java java