public CsvConfiguration<T> build() { CsvConfiguration<T> config = new CsvConfiguration<>(clazz); config.entityInfo = entityInfo; config.charset = charset; config.typeSerializer = typeSerializer; config.header = header; config.fieldSeparator = fieldSeparator; config.textDelimiter = textDelimiter; config.alwaysDelimitText = alwaysDelimitText; config.lineDelimiter = lineDelimiter; return config; }
public void writeAll(OutputStream output, Stream<T> stream) throws IOException { BufferedWriter writer = new BufferedWriter(new OutputStreamWriter(output, config.getCharset())); if (config.isHeader()) { writeHeader(writer, columns); } Iterator<T> iterator = stream.iterator(); while (iterator.hasNext()) { add(writer, iterator.next()); } writer.flush(); }
private EntityInfo getEntityInfo(CsvConfiguration<T> config) { EntityInfo entityInfo = config.getEntityInfo(); if (entityInfo != null) { return entityInfo; } return new JpaEntityInspector(config.getClazz()).inspect(); }
public CsvEscaper(CsvConfiguration<?> configuration) { this.delimiterChar = configuration.getTextDelimiter(); this.separatorChar = configuration.getFieldSeparator(); this.alwaysDelimit = configuration.isAlwaysDelimitText(); if (this.alwaysDelimit) { this.baseState = NEEDS_BUT_DOESNT_HAVE; } }
private void writeHeader(BufferedWriter writer, List<ColumnInfo> columns) throws IOException { for (int i = 0; i < columns.size(); i++) { ColumnInfo info = columns.get(i); writer.write(csvEscaper.escapeAndDelimite(info.getColumnName())); if (i < columns.size() - 1) { writer.write(config.getFieldSeparator()); } } writer.write(config.getLineDelimiter()); }
private void add(BufferedWriter writer, T entity) throws IOException { for (int i = 0; i < columns.size(); i++) { ColumnInfo column = columns.get(i); Object value = column.getAccessor().apply(entity); if (value != null) { String valueStr = typeSerializer.toString(value, column.getFieldType()); String escapedValue = csvEscaper.escapeAndDelimite(valueStr); writer.write(escapedValue); } else if (config.isAlwaysDelimitText()) { writer.write(emptyText); } if (i < columns.size() - 1) { writer.write(config.getFieldSeparator()); } } writer.write(config.getLineDelimiter()); }
public CsvWriter(CsvConfiguration<T> config) { this.config = config; this.typeSerializer = config.getTypeSerializer(); this.columns = getEntityInfo(config).getColumns(); this.csvEscaper = new CsvEscaper(config); this.emptyText = Character.toString(config.getTextDelimiter()) + config.getTextDelimiter(); }
public CsvWriter(Class<T> clazz) { this(new CsvConfiguration<>(clazz)); }
@Test public void emptyCollectionOnlyGeneratesHeader() throws IOException { CsvConfiguration<SomeEntity> config = new CsvConfiguration<>(createBuilderForSomeEntity().build()); String result = writeCsvToString(config, Collections.emptyList()); assertEquals("name,age" + ls, result); }
@Test public void columnsMantainsDeclaredOrderInClass() throws IOException { CsvConfiguration<AnnotatedEntity> config = new CsvConfiguration<>(AnnotatedEntity.class); Address address = new Address("221B Baker Street", "London"); String result = writeCsvToString(config, new AnnotatedEntity("Sherlock", 63, address)); assertEquals("title,duration,street,city" + ls + "Sherlock,63,221B Baker Street,London" + ls, result); }
@Test public void elementIsAfterHeader() throws IOException { CsvConfiguration<SomeEntity> config = new CsvConfiguration<>(createBuilderForSomeEntity().build()); String result = writeCsvToString(config, new SomeEntity("John", 10)); assertEquals("name,age" + ls + "John,10" + ls, result); }
@Test public void columnsMantainsDeclaredOrder() throws IOException { EntityInfoBuilder<SomeEntity> entityBuilder = new EntityInfoBuilder<>(SomeEntity.class); entityBuilder.addColumn("age", FieldTypeEnum.INT, SomeEntity::getAge); entityBuilder.addColumn("name", FieldTypeEnum.STRING, SomeEntity::getName); CsvConfiguration<SomeEntity> config = new CsvConfiguration<>(entityBuilder.build()); String result = writeCsvToString(config, new SomeEntity("John", 10)); assertEquals("age,name" + ls + "10,John" + ls, result); }
@Test public void canCreateSyntheticColumns() throws IOException { EntityInfoBuilder<SomeEntity> entityBuilder = new EntityInfoBuilder<>(SomeEntity.class); entityBuilder.addFields("name", "age"); entityBuilder.addColumn("adult", FieldTypeEnum.BOOLEAN, e -> e.getAge() >= 18); CsvConfiguration<SomeEntity> config = new CsvConfiguration<>(entityBuilder.build()); String result = writeCsvToString(config, new SomeEntity("John", 10), new SomeEntity("Amanda", 34)); assertEquals("name,age,adult" + ls + "John,10,false" + ls + "Amanda,34,true" + ls, result); }