当从文件的第一行读取 cvsMapper 使用的 csvSchema 时，有什么方法可以检索它？

Question

我设置了 CsvSchema 和 CsvMapper 以读取第一行中列名的 CSV 文件。

我需要能够对数据进行一些工作，更改特定字段，然后在保留列顺序的同时将数据写回。 我假设 CsvMapper 具有从第一行创建的内部模式，因为该地图具有所有正确的 key-value 对。 那么，有什么方法可以使用CsvMapper 的writer 函数来使用该模式来排序输出数据？

CsvSchema schema = CsvSchema.emptySchema().withHeader();
CsvMapper mapper = new CsvMapper();
MappingIterator<Map<String, String>> it = this.mapper.readerFor(Map.class)
            .with(this.schema)
            .readValues(stream);

现在，我在第一列数据上收到了一个 CsvMappingException 表示“无法识别的列”，所以看起来它没有使用架构。

Answer 1

您可以使用从输入文件中读取的第一行中的架构构建SequenceWriter。 Jackson 在后面使用 LinkedHashMap，因此，订单保持不变。假设您的 CSV 输入包含：

C1,name,type
I1,John,T1
I2,Adam,T2

您可以读取、更新和写入控制台，如下例所示：

import com.fasterxml.jackson.databind.MappingIterator;
import com.fasterxml.jackson.databind.ObjectWriter;
import com.fasterxml.jackson.databind.SequenceWriter;
import com.fasterxml.jackson.dataformat.csv.CsvMapper;
import com.fasterxml.jackson.dataformat.csv.CsvSchema;

import java.io.Closeable;
import java.io.File;
import java.io.IOException;
import java.util.Map;

public class CsvApp {

    private CsvMapper csvMapper = new CsvMapper();

    private void handle() throws Exception {
        File csvFile = new File("./resource/test.csv").getAbsoluteFile();

        CsvSchema schema = CsvSchema.emptySchema().withHeader();
        MappingIterator<Map<String, String>> iterator = csvMapper.readerFor(Map.class).with(schema).readValues(csvFile);

        SequenceWriter writer = null;
        while (iterator.hasNext()) {
            final Map<String, String> row = iterator.next();
            if (writer == null) {
                writer = createWriter(row).writeValues(System.out);
            }

            // do something ...
            for (Map.Entry<String, String> item : row.entrySet()) {
                row.replace(item.getKey(), item.getValue() + "U");
            }

            // write back
            writer.write(row);
        }

        close(writer);
    }

    private ObjectWriter createWriter(Map<String, String> row) {
        CsvSchema.Builder writeSchema = new CsvSchema.Builder();
        writeSchema.setUseHeader(true);
        row.keySet().forEach(writeSchema::addColumn);

        return csvMapper.writerFor(Map.class).with(writeSchema.build());
    }

    private void close(Closeable closeable) throws IOException {
        if (closeable != null) {
            closeable.close();
        }
    }

    public static void main(String[] args) throws Exception {
        new CsvApp().handle();
    }
}

上面的代码打印：

C1,name,type
I1U,JohnU,T1U
I2U,AdamU,T2U

另见：

1. jackson-dataformats-text - Overview