dalbodeule/snap-admin
1
0
Fork 0
mirror of https://github.com/dalbodeule/snap-admin.git synced 2025-12-15 04:41:58 +09:00
Code Issues Packages Projects Releases Wiki Activity

WIP CSV export

Browse Source
This commit is contained in:
Francesco
2023-10-13 17:57:05 +02:00
parent 0556f8c041
commit ada0f60b61
2 changed files with 37 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/main/java/tech/ailef/dbadmin/external/controller/DataExportController.java vendored
View File

@@ -48,13 +48,11 @@ public class DataExportController {
Set<QueryFilter> queryFilters = Utils.computeFilters(schema, otherParams);
System.out.println("QF = " + queryFilters);
List<DbObject> results = repository.search(schema, query, queryFilters);
String result = toCsv(results, schema.getSortedFields());
return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION,
"attachment; filename=\"export_" + schema.getClass().getSimpleName() + ".csv\"")
"attachment; filename=\"export_" + schema.getJavaClass().getSimpleName() + ".csv\"")
.body(result.getBytes());
}
@@ -63,14 +61,21 @@ public class DataExportController {
StringWriter sw = new StringWriter();
String[] header = fields.stream().map(f -> f.getName()).toArray(String[]::new);
CSVFormat csvFormat = CSVFormat.DEFAULT.builder()
// .setHeader(HEADERS)
.setHeader(header)
.build();
try (final CSVPrinter printer = new CSVPrinter(sw, csvFormat)) {
for (DbObject item : items) {
printer.printRecord(fields.stream().map(f -> {
return item.get(f).getFormattedValue();
if (f.isForeignKey()) {
DbObject linkedItem = item.traverse(f);
return linkedItem.getPrimaryKeyValue() + " (" + linkedItem.getDisplayName() + ")";
} else {
return item.get(f).getFormattedValue();
}
}));
}