dalbodeule/snap-admin
1
0
Fork 0
mirror of https://github.com/dalbodeule/snap-admin.git synced 2025-06-08 21:38:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

WIP XLSX export

Browse Source
This commit is contained in:
Francesco 2023-10-13 21:46:41 +02:00
parent 913f1bb05f
commit a8386fcd02
2 changed files with 108 additions and 38 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,12 +1,21 @@
package tech.ailef.dbadmin.external.controller; package tech.ailef.dbadmin.external.controller;
import java.io.ByteArrayOutputStream;
import java.io.IOException; import java.io.IOException;
import java.io.StringWriter; import java.io.StringWriter;
import java.util.ArrayList;
import java.util.List; import java.util.List;
import java.util.Set; import java.util.Set;
import java.util.stream.Collectors;
import org.apache.commons.csv.CSVFormat; import org.apache.commons.csv.CSVFormat;
import org.apache.commons.csv.CSVPrinter; import org.apache.commons.csv.CSVPrinter;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.Row;
import org.apache.poi.ss.usermodel.Sheet;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.beans.factory.annotation.Autowired; import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpHeaders; import org.springframework.http.HttpHeaders;
import org.springframework.http.ResponseEntity; import org.springframework.http.ResponseEntity;
@ -23,65 +32,121 @@ import tech.ailef.dbadmin.external.dbmapping.DbAdminRepository;
import tech.ailef.dbadmin.external.dbmapping.DbField; import tech.ailef.dbadmin.external.dbmapping.DbField;
import tech.ailef.dbadmin.external.dbmapping.DbObject; import tech.ailef.dbadmin.external.dbmapping.DbObject;
import tech.ailef.dbadmin.external.dbmapping.DbObjectSchema; import tech.ailef.dbadmin.external.dbmapping.DbObjectSchema;
import tech.ailef.dbadmin.external.dto.DataExportFormat;
import tech.ailef.dbadmin.external.dto.QueryFilter; import tech.ailef.dbadmin.external.dto.QueryFilter;
import tech.ailef.dbadmin.external.exceptions.DbAdminException; import tech.ailef.dbadmin.external.exceptions.DbAdminException;
import tech.ailef.dbadmin.external.misc.Utils; import tech.ailef.dbadmin.external.misc.Utils;
@Controller @Controller
@RequestMapping(value= {"/${dbadmin.baseUrl}/export", "/${dbadmin.baseUrl}/export/"}) @RequestMapping(value = { "/${dbadmin.baseUrl}/export", "/${dbadmin.baseUrl}/export/" })
public class DataExportController { public class DataExportController {
@Autowired @Autowired
private DbAdmin dbAdmin; private DbAdmin dbAdmin;
@Autowired @Autowired
private DbAdminRepository repository; private DbAdminRepository repository;
@GetMapping("/{className}") @GetMapping("/{className}")
@ResponseBody @ResponseBody
public ResponseEntity<byte[]> export(@PathVariable String className, public ResponseEntity<byte[]> export(@PathVariable String className, @RequestParam(required = false) String query,
@RequestParam(required=false) String query, @RequestParam(required = false) String format, @RequestParam MultiValueMap<String, String> otherParams) {
@RequestParam MultiValueMap<String, String> otherParams) {
if (format == null)
format = "CSV";
DataExportFormat exportFormat = null;
try {
exportFormat = DataExportFormat.valueOf(format.toUpperCase());
} catch (IllegalArgumentException e) {
throw new DbAdminException("Unsupported export format: " + format);
}
List<String> fieldToInclude = otherParams.getOrDefault("fields[]", new ArrayList<>());
DbObjectSchema schema = dbAdmin.findSchemaByClassName(className); DbObjectSchema schema = dbAdmin.findSchemaByClassName(className);
List<DbField> fields = schema.getSortedFields().stream().filter(f -> fieldToInclude.contains(f.getName()))
.collect(Collectors.toList());
Set<QueryFilter> queryFilters = Utils.computeFilters(schema, otherParams); Set<QueryFilter> queryFilters = Utils.computeFilters(schema, otherParams);
List<DbObject> results = repository.search(schema, query, queryFilters); List<DbObject> results = repository.search(schema, query, queryFilters);
String result = toCsv(results, schema.getSortedFields()); switch (exportFormat) {
return ResponseEntity.ok().header(HttpHeaders.CONTENT_DISPOSITION, case CSV:
"attachment; filename=\"export_" + schema.getJavaClass().getSimpleName() + ".csv\"") return ResponseEntity.ok()
.body(result.getBytes()); .header(HttpHeaders.CONTENT_DISPOSITION,
"attachment; filename=\"export_" + schema.getJavaClass().getSimpleName() + ".csv\"")
.body(toCsv(results, fields).getBytes());
case XLSX:
return ResponseEntity.ok()
.header(HttpHeaders.CONTENT_DISPOSITION,
"attachment; filename=\"export_" + schema.getJavaClass().getSimpleName() + ".xlsx\"")
.body(toXlsx(results, fields));
case JSON:
throw new DbAdminException("JSON TODO");
default:
throw new DbAdminException("Unable to detect export format");
}
} }
private byte[] toXlsx(List<DbObject> items, List<DbField> fields) {
Workbook workbook = new XSSFWorkbook();
Sheet sheet = workbook.createSheet("SchemaName");
int rowIndex = 0;
for (DbObject item : items) {
Row row = sheet.createRow(rowIndex++);
int cellIndex = 0;
for (DbField field : fields) {
Cell cell = row.createCell(cellIndex++);
if (field.isForeignKey()) {
DbObject linkedItem = item.traverse(field);
cell.setCellValue(linkedItem.getPrimaryKeyValue() + " (" + linkedItem.getDisplayName() + ")");
} else {
cell.setCellValue(item.get(field).getFormattedValue());
}
}
}
// lets write the excel data to file now
ByteArrayOutputStream fos = new ByteArrayOutputStream();
try {
workbook.write(fos);
fos.close();
} catch (IOException e) {
throw new DbAdminException("Error writing XLSX file");
}
return fos.toByteArray();
}
private String toCsv(List<DbObject> items, List<DbField> fields) { private String toCsv(List<DbObject> items, List<DbField> fields) {
if (items.isEmpty()) return ""; if (items.isEmpty())
return "";
StringWriter sw = new StringWriter(); StringWriter sw = new StringWriter();
String[] header = fields.stream().map(f -> f.getName()).toArray(String[]::new); String[] header = fields.stream().map(f -> f.getName()).toArray(String[]::new);
CSVFormat csvFormat = CSVFormat.DEFAULT.builder() CSVFormat csvFormat = CSVFormat.DEFAULT.builder().setHeader(header).build();
.setHeader(header)
.build(); try (final CSVPrinter printer = new CSVPrinter(sw, csvFormat)) {
for (DbObject item : items) {
try (final CSVPrinter printer = new CSVPrinter(sw, csvFormat)) { printer.printRecord(fields.stream().map(f -> {
for (DbObject item : items) { if (f.isForeignKey()) {
printer.printRecord(fields.stream().map(f -> { DbObject linkedItem = item.traverse(f);
if (f.isForeignKey()) { return linkedItem.getPrimaryKeyValue() + " (" + linkedItem.getDisplayName() + ")";
DbObject linkedItem = item.traverse(f); } else {
return linkedItem.getPrimaryKeyValue() + " (" + linkedItem.getDisplayName() + ")"; return item.get(f).getFormattedValue();
} else { }
return item.get(f).getFormattedValue(); }));
} }
}));
} return sw.toString();
} catch (IOException e) {
return sw.toString(); throw new DbAdminException(e);
} catch (IOException e) {
throw new DbAdminException(e);
} }
} }
} }

5
src/main/java/tech/ailef/dbadmin/external/dto/DataExportFormat.java vendored Normal file
View File

@ -0,0 +1,5 @@
package tech.ailef.dbadmin.external.dto;
public enum DataExportFormat {
CSV, XLSX, JSON;
}