| | |
| | | import org.apache.poi.hssf.usermodel.HSSFWorkbook; |
| | | import org.apache.poi.ss.usermodel.*; |
| | | import org.apache.poi.xssf.usermodel.XSSFWorkbook; |
| | | import org.apache.poi.ss.util.CellRangeAddress; |
| | | |
| | | import javax.servlet.http.HttpServletResponse; |
| | | import jakarta.servlet.http.HttpServletResponse; |
| | | import java.io.IOException; |
| | | import java.lang.reflect.Field; |
| | | import java.lang.reflect.Method; |
| | | import java.lang.reflect.Modifier; |
| | | import java.net.URLEncoder; |
| | | import java.text.SimpleDateFormat; |
| | |
| | | return workbook; |
| | | } |
| | | |
| | | public static Workbook create(List<?> list, List<ColumnMeta> columns) { |
| | | return create(list, columns, null); |
| | | } |
| | | |
| | | public static Workbook create(List<?> list, List<ColumnMeta> columns, ReportMeta reportMeta) { |
| | | XSSFWorkbook workbook = new XSSFWorkbook(); |
| | | Sheet sheet = workbook.createSheet("export"); |
| | | List<ColumnMeta> safeColumns = columns == null ? Collections.emptyList() : columns; |
| | | int sheetColumnCount = safeColumns.size() + 1; |
| | | configureA4PrintLayout(sheet); |
| | | |
| | | CellStyle titleStyle = createTitleStyle(workbook); |
| | | CellStyle metaLabelStyle = createMetaLabelStyle(workbook); |
| | | CellStyle metaValueStyle = createMetaValueStyle(workbook); |
| | | CellStyle headerStyle = createHeaderStyle(workbook); |
| | | CellStyle bodyStyle = createBodyStyle(workbook); |
| | | CellStyle serialStyle = createCenteredBodyStyle(workbook); |
| | | |
| | | int rowIndex = 0; |
| | | if (reportMeta != null) { |
| | | Row titleRow = sheet.createRow(rowIndex++); |
| | | titleRow.setHeightInPoints(28); |
| | | Cell titleCell = titleRow.createCell(0); |
| | | titleCell.setCellValue(StringUtils.defaultIfBlank(reportMeta.getTitle(), "报表")); |
| | | titleCell.setCellStyle(titleStyle); |
| | | sheet.addMergedRegion(new CellRangeAddress(0, 0, 0, Math.max(0, sheetColumnCount - 1))); |
| | | |
| | | Row metaRow = sheet.createRow(rowIndex++); |
| | | int metaCol = 0; |
| | | metaCol = writeMetaPair(metaRow, metaCol, "报表日期", reportMeta.getReportDate(), metaLabelStyle, metaValueStyle); |
| | | writeMetaPair(metaRow, metaCol, "打印人", reportMeta.getPrintedBy(), metaLabelStyle, metaValueStyle); |
| | | |
| | | rowIndex++; |
| | | } |
| | | |
| | | int headerRowIndex = rowIndex; |
| | | Row header = sheet.createRow(rowIndex++); |
| | | Cell serialHeaderCell = header.createCell(0); |
| | | serialHeaderCell.setCellValue("序号"); |
| | | serialHeaderCell.setCellStyle(headerStyle); |
| | | |
| | | for (int i = 0; i < safeColumns.size(); i++) { |
| | | ColumnMeta column = safeColumns.get(i); |
| | | Cell headerCell = header.createCell(i + 1); |
| | | headerCell.setCellValue( |
| | | StringUtils.isBlank(column.getLabel()) ? column.getSource() : column.getLabel() |
| | | ); |
| | | headerCell.setCellStyle(headerStyle); |
| | | } |
| | | |
| | | if (list != null) { |
| | | int serialNo = 1; |
| | | for (Object rowObj : list) { |
| | | Row row = sheet.createRow(rowIndex++); |
| | | Cell serialCell = row.createCell(0); |
| | | serialCell.setCellValue(String.format("%03d", serialNo++)); |
| | | serialCell.setCellStyle(serialStyle); |
| | | for (int i = 0; i < safeColumns.size(); i++) { |
| | | Object value = getColumnValue(rowObj, safeColumns.get(i).getSource()); |
| | | Cell cell = row.createCell(i + 1); |
| | | cell.setCellStyle(bodyStyle); |
| | | if (value != null) { |
| | | if (value instanceof Date) { |
| | | SimpleDateFormat sdf = new SimpleDateFormat("yyyy-MM-dd HH:mm:ss"); |
| | | cell.setCellValue(sdf.format((Date) value)); |
| | | } else { |
| | | cell.setCellValue(value.toString()); |
| | | } |
| | | } |
| | | } |
| | | } |
| | | } |
| | | |
| | | for (int i = 0; i <= safeColumns.size(); i++) { |
| | | sheet.autoSizeColumn(i); |
| | | sheet.setColumnWidth(i, Math.min(sheet.getColumnWidth(i) + 1024, 12000)); |
| | | } |
| | | |
| | | sheet.setRepeatingRows(CellRangeAddress.valueOf((headerRowIndex + 1) + ":" + (headerRowIndex + 1))); |
| | | |
| | | return workbook; |
| | | } |
| | | |
| | | private static void configureA4PrintLayout(Sheet sheet) { |
| | | sheet.setAutobreaks(true); |
| | | sheet.setFitToPage(true); |
| | | sheet.setHorizontallyCenter(true); |
| | | sheet.setDisplayGridlines(false); |
| | | |
| | | PrintSetup printSetup = sheet.getPrintSetup(); |
| | | printSetup.setPaperSize(PrintSetup.A4_PAPERSIZE); |
| | | printSetup.setLandscape(true); |
| | | printSetup.setFitWidth((short) 1); |
| | | printSetup.setFitHeight((short) 0); |
| | | |
| | | sheet.setMargin(Sheet.LeftMargin, 0.3); |
| | | sheet.setMargin(Sheet.RightMargin, 0.3); |
| | | sheet.setMargin(Sheet.TopMargin, 0.4); |
| | | sheet.setMargin(Sheet.BottomMargin, 0.4); |
| | | } |
| | | |
| | | private static int writeMetaPair(Row row, int startCol, String label, String value, CellStyle labelStyle, CellStyle valueStyle) { |
| | | Cell labelCell = row.createCell(startCol); |
| | | labelCell.setCellValue(label + ":"); |
| | | labelCell.setCellStyle(labelStyle); |
| | | |
| | | Cell valueCell = row.createCell(startCol + 1); |
| | | valueCell.setCellValue(StringUtils.defaultString(value)); |
| | | valueCell.setCellStyle(valueStyle); |
| | | return startCol + 2; |
| | | } |
| | | |
| | | private static CellStyle createTitleStyle(Workbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.CENTER); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | style.setBorderBottom(BorderStyle.THICK); |
| | | Font font = workbook.createFont(); |
| | | font.setBold(true); |
| | | font.setFontHeightInPoints((short) 16); |
| | | style.setFont(font); |
| | | return style; |
| | | } |
| | | |
| | | private static CellStyle createMetaLabelStyle(Workbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.LEFT); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | style.setBorderBottom(BorderStyle.THIN); |
| | | Font font = workbook.createFont(); |
| | | font.setBold(true); |
| | | style.setFont(font); |
| | | return style; |
| | | } |
| | | |
| | | private static CellStyle createMetaValueStyle(Workbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.LEFT); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | style.setBorderBottom(BorderStyle.THIN); |
| | | return style; |
| | | } |
| | | |
| | | private static CellStyle createHeaderStyle(Workbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.CENTER); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | style.setFillForegroundColor(IndexedColors.GREY_25_PERCENT.getIndex()); |
| | | style.setFillPattern(FillPatternType.SOLID_FOREGROUND); |
| | | Font font = workbook.createFont(); |
| | | font.setBold(true); |
| | | style.setFont(font); |
| | | return style; |
| | | } |
| | | |
| | | private static CellStyle createBodyStyle(Workbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.LEFT); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | return style; |
| | | } |
| | | |
| | | private static CellStyle createCenteredBodyStyle(Workbook workbook) { |
| | | CellStyle style = workbook.createCellStyle(); |
| | | style.setAlignment(HorizontalAlignment.CENTER); |
| | | style.setVerticalAlignment(VerticalAlignment.CENTER); |
| | | return style; |
| | | } |
| | | |
| | | private static Object getColumnValue(Object rowObj, String source) { |
| | | if (rowObj == null || StringUtils.isBlank(source)) { |
| | | return null; |
| | | } |
| | | |
| | | if (rowObj instanceof Map) { |
| | | return getValueFromMap((Map<?, ?>) rowObj, source); |
| | | } |
| | | |
| | | String extendFieldKey = extractExtendFieldKey(source); |
| | | if (extendFieldKey != null) { |
| | | Object extendFields = getBeanValue(rowObj, "extendFields"); |
| | | if (extendFields instanceof Map) { |
| | | return ((Map<?, ?>) extendFields).get(extendFieldKey); |
| | | } |
| | | return null; |
| | | } |
| | | |
| | | return getBeanValue(rowObj, source); |
| | | } |
| | | |
| | | private static Object getValueFromMap(Map<?, ?> rowObj, String source) { |
| | | String extendFieldKey = extractExtendFieldKey(source); |
| | | if (extendFieldKey != null) { |
| | | Object extendFields = rowObj.get("extendFields"); |
| | | if (extendFields instanceof Map) { |
| | | return ((Map<?, ?>) extendFields).get(extendFieldKey); |
| | | } |
| | | return null; |
| | | } |
| | | return rowObj.get(source); |
| | | } |
| | | |
| | | private static String extractExtendFieldKey(String source) { |
| | | if (source == null || !source.startsWith("extendFields.[")) { |
| | | return null; |
| | | } |
| | | int startIndex = source.indexOf('['); |
| | | int endIndex = source.indexOf(']'); |
| | | if (startIndex < 0 || endIndex <= startIndex) { |
| | | return null; |
| | | } |
| | | return source.substring(startIndex + 1, endIndex); |
| | | } |
| | | |
| | | private static Object getBeanValue(Object rowObj, String source) { |
| | | Object value = invokeGetter(rowObj, source); |
| | | if (value != null) { |
| | | return value; |
| | | } |
| | | Field field = findField(rowObj.getClass(), source); |
| | | if (field == null) { |
| | | return null; |
| | | } |
| | | try { |
| | | field.setAccessible(true); |
| | | return field.get(rowObj); |
| | | } catch (IllegalAccessException ignore) { |
| | | return null; |
| | | } |
| | | } |
| | | |
| | | private static Object invokeGetter(Object target, String source) { |
| | | String suffix = Character.toUpperCase(source.charAt(0)) + source.substring(1); |
| | | String[] methodNames = new String[] { "get" + suffix, "is" + suffix }; |
| | | for (String methodName : methodNames) { |
| | | try { |
| | | Method method = target.getClass().getMethod(methodName); |
| | | return method.invoke(target); |
| | | } catch (Exception ignore) { |
| | | } |
| | | } |
| | | return null; |
| | | } |
| | | |
| | | private static Field findField(Class<?> clazz, String source) { |
| | | Class<?> current = clazz; |
| | | while (current != null && current != Object.class) { |
| | | try { |
| | | return current.getDeclaredField(source); |
| | | } catch (NoSuchFieldException ignore) { |
| | | current = current.getSuperclass(); |
| | | } |
| | | } |
| | | return null; |
| | | } |
| | | |
| | | /** |
| | | * 添加导入excel配置参数 |
| | | * 注:默认配置可满足当前需求 |
| | |
| | | return false; |
| | | } |
| | | |
| | | public static class ColumnMeta { |
| | | private String key; |
| | | private String source; |
| | | private String label; |
| | | private Boolean extendField; |
| | | |
| | | public String getKey() { |
| | | return key; |
| | | } |
| | | |
| | | public ColumnMeta setKey(String key) { |
| | | this.key = key; |
| | | return this; |
| | | } |
| | | |
| | | public String getSource() { |
| | | return source; |
| | | } |
| | | |
| | | public ColumnMeta setSource(String source) { |
| | | this.source = source; |
| | | return this; |
| | | } |
| | | |
| | | public String getLabel() { |
| | | return label; |
| | | } |
| | | |
| | | public ColumnMeta setLabel(String label) { |
| | | this.label = label; |
| | | return this; |
| | | } |
| | | |
| | | public Boolean getExtendField() { |
| | | return extendField; |
| | | } |
| | | |
| | | public ColumnMeta setExtendField(Boolean extendField) { |
| | | this.extendField = extendField; |
| | | return this; |
| | | } |
| | | } |
| | | |
| | | public static class ReportMeta { |
| | | private String title; |
| | | private String companyName; |
| | | private String printedBy; |
| | | private String reportDate; |
| | | private String reportDateValue; |
| | | |
| | | public String getTitle() { |
| | | return title; |
| | | } |
| | | |
| | | public ReportMeta setTitle(String title) { |
| | | this.title = title; |
| | | return this; |
| | | } |
| | | |
| | | public String getCompanyName() { |
| | | return companyName; |
| | | } |
| | | |
| | | public ReportMeta setCompanyName(String companyName) { |
| | | this.companyName = companyName; |
| | | return this; |
| | | } |
| | | |
| | | public String getReportDate() { |
| | | return reportDate; |
| | | } |
| | | |
| | | public ReportMeta setReportDate(String reportDate) { |
| | | this.reportDate = reportDate; |
| | | return this; |
| | | } |
| | | |
| | | public String getReportDateValue() { |
| | | return reportDateValue; |
| | | } |
| | | |
| | | public ReportMeta setReportDateValue(String reportDateValue) { |
| | | this.reportDateValue = reportDateValue; |
| | | return this; |
| | | } |
| | | |
| | | public String getPrintedBy() { |
| | | return printedBy; |
| | | } |
| | | |
| | | public ReportMeta setPrintedBy(String printedBy) { |
| | | this.printedBy = printedBy; |
| | | return this; |
| | | } |
| | | } |
| | | |
| | | |
| | | } |
| | | |