Question

我有一个CsvSchema和CsvMapper设置为读取第一行中具有列名的CSV文件。

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

CsvMapper

现在，我在数据的第一列上获得了CsvSchema schema = CsvSchema.emptySchema().withHeader(); CsvMapper mapper = new CsvMapper(); MappingIterator<Map<String, String>> it = this.mapper.readerFor(Map.class) .with(this.schema) .readValues(stream);的“无法识别的列”，因此看起来好像没有在使用架构。

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

另请参阅：

jackson-dataformats-text - Overview

从文件的第一行读取cvsMapper所使用的csvSchema时，有什么方法吗？

1 个答案: