Как напечатать только заголовок в строку в Jackson CSVMapper - PullRequest
Новая
       52

Как напечатать только заголовок в строку в Jackson CSVMapper

0 голосов
/ 27 апреля 2019

Я использую Jackson CSVMapper для чтения разделенного табуляцией текстового файла в список объектов Java и записи его обратно в другой разделенный табуляцией текстовый файл. Мой класс выглядит так: 

@Getter
@Setter
@ToString
@NoArgsConstructor
@JsonPropertyOrder({
    "Id",
    "Name",
    "Address 1",
    "Address 2"})
public class InputSchema {
    @JsonProperty("Id")private String id;
    @JsonProperty("Name") private String name;
    @JsonProperty("Address 1") private String address1;
    @JsonProperty("Address 2") private String address2;
}

Код парсера: 

public List<InputSchema> parseAddresses(String fileContent) throws IOException {
    List<InputSchema> inputAddresses = new ArrayList<>();
    InputStream inputStream = new ByteArrayInputStream(fileContent.getBytes());
    MappingIterator<InputSchema> iterator = getInputSchemaMappingIterator(inputStream);

        while (iterator.hasNextValue()) {
            InputSchema inputAddress = iterator.nextValue();
            log.info(inputAddress.toString());
            inputAddresses.add(inputAddress);
        }

    return inputAddresses;
}

private MappingIterator<InputSchema> getInputSchemaMappingIterator(InputStream inputStream) throws IOException {
    CsvMapper mapper = new CsvMapper();
    mapper.configure(CsvParser.Feature.FAIL_ON_MISSING_COLUMNS, true);
    mapper.configure(CsvParser.Feature.TRIM_SPACES, true);
    CsvSchema schema = mapper.schemaFor(InputSchema.class)
                        .withColumnSeparator('\t')
                        .withHeader()
                        .withStrictHeaders(true);

    ObjectReader reader = mapper.readerFor(InputSchema.class).with(schema);
    MappingIterator<InputSchema> iterator = reader.readValues(inputStream);

    return iterator;
}

Код писателя: 

private String toTxt(List<InputSchema> inputAddresses) throws JsonProcessingException {
    CsvMapper mapper = new CsvMapper();
    CsvSchema schema = mapper.schemaFor(InputSchema.class)
        .withColumnSeparator('\t')
        .withLineSeparator("\n")
        .withHeader()
        .withoutQuoteChar();

    return mapper.writer(schema).writeValueAsString(inputAddresses);
}

Я хочу извлечь только заголовок независимо от данных. Таким образом, 

String header = extractHeader();

должен вернуть 

"Id\tName\tAddress 1\tAddress 2"

Как я могу это сделать?

Спасибо!

1 Ответ

0 голосов
/ 03 мая 2019

Сработало следующее: 

    public static String displayHeader() {
    CsvMapper mapper = new CsvMapper();

    JavaType javaType = mapper.getTypeFactory().constructType(InputSchema.class);
    BeanDescription beanDescription = mapper.getSerializationConfig().introspect(javaType);
    List<BeanPropertyDefinition> properties = beanDescription.findProperties();

    List<String> columnNames = properties.stream()
        .map(e -> e.getName())
        .collect(Collectors.toList());
    String header = String.join("\t", columnNames);

    return header;
}

ссылка: @ cassiomolin 's ответ

...