java實現數據的Excel導出, 自定義導出字段, 轉換字典值

第一版代碼:
基礎功能跳轉此文章java自定義Excel導出工具:

簡介

新增功能:

1. 添加自定義字段導出功能, 用戶可以選擇字段進行導出
2. 將字典類型數據進行轉換(如:0=女,1=男, 將0轉換為女, 1轉換為男)
3. 添加表頭格式
4. 隨機文件名稱, 防止多次導出時文件覆蓋問題

實現代碼

Excel注解

@Retention(RetentionPolicy.RUNTIME) @Target(ElementType.FIELD) public @interface Excel { /** * 導出到Excel中的名字 */ String name() default ""; /** * 日期格式, 如: yyyy-MM-dd */ String dateFormat() default ""; /** * 字典的key值 */ String dictKey() default ""; /** * 讀取內容轉表達式 (如: 0=男,1=女,2=未知) */ String dictExp() default ""; } 

Excel的導出工具類

/** * Excel的工具類 */ public class ExcelUtil<T> { /** * 工作薄 */ private Workbook wb; /** * 工作表 */ private Sheet sheet; /** * 需要導出的數據 */ private List<T> exportList; /** * 對象的class對象 */ private Class<T> clazz; /** * 被選中需要導出的字段名稱 */ private Map<String, Object> checkedFieldsName; /** * 被選中需要導出的字段對象 */ private List<Field> checkedFields; /** * 包含需要字典轉換的字段對象 */ private List<Field> fieldsContainDict; /** * 對象中的字典值 */ private Map<String, Map<String, String>> dicts; private ExcelUtil(){ } public ExcelUtil(Class<T> clazz){ this.clazz = clazz; } /** * * @param list * @param sheetName * @param fieldsName */ public void exportExcel(List<T> list, Map<String, Object> fieldsName, String sheetName){ // 初始化數據 init(list, sheetName, fieldsName); // 轉換字典值 try { convertDict(); } catch (IllegalAccessException e) { e.printStackTrace(); } // sheet第一行加入名稱數據 createTopRow(); // sheet其他行，添加目標數據 try { createOtherRow(); } catch (IllegalAccessException e) { e.printStackTrace(); } // 導出wb try(OutputStream outFile = new FileOutputStream(generateFileName())){ wb.write(outFile); } catch (IOException e) { e.printStackTrace(); } finally { try { wb.close(); } catch (IOException e) { e.printStackTrace(); } } } /** * 添加導出數據 */ private void createOtherRow() throws IllegalAccessException { for (int rowNum = 1; rowNum <= exportList.size(); rowNum++) { Row row = sheet.createRow(rowNum); T t = exportList.get(rowNum - 1); for (int colNum = 0; colNum < checkedFields.size(); colNum++) { Cell cell = row.createCell(colNum); Field field = checkedFields.get(colNum); field.setAccessible(true); // 單元格設置值 addCell(cell, field, t); } } } /** * 單元格中添加數據 * * @param cell 單元格 * @param field 字段 * @param t list中的一條數據 */ private void addCell(Cell cell, Field field, T t) throws IllegalAccessException { Class<?> fieldType = field.getType(); if (String.class == fieldType) { cell.setCellValue((String) field.get(t)); } else if ((Integer.TYPE == fieldType) || (Integer.class == fieldType)) { cell.setCellValue((Integer) field.get(t)); } else if ((Long.TYPE == fieldType) || (Long.class == fieldType)) { cell.setCellValue((Long) field.get(t)); } else if ((Double.TYPE == fieldType) || (Double.class == fieldType)) { cell.setCellValue((Double) field.get(t)); } else if ((Float.TYPE == fieldType) || (Float.class == fieldType)) { cell.setCellValue((Float) field.get(t)); } else if (Date.class == fieldType) { String dateFormat = field.getAnnotation(Excels.class).dateFormat(); cell.setCellValue(dateFormat((Date) field.get(t), dateFormat)); } } /** * 時間格式轉換 * @param date 日期 * @param dateFormat 日期格式 * @return */ private String dateFormat(Date date, String dateFormat) { if (dateFormat == null || "".equals(dateFormat)) { dateFormat = "yyyy-MM-dd HH:mm:ss"; } SimpleDateFormat df = new SimpleDateFormat(dateFormat); return df.format(date); } /** * sheet第一行加入名稱數據 */ private void createTopRow() { Row row = sheet.createRow(0); Map<String, CellStyle> styles = createStyles(wb); for (int index = 0; index < checkedFields.size(); index++) { Cell cell = row.createCell(index); cell.setCellValue(checkedFields.get(index).getAnnotation(Excels.class).name()); System.out.println(styles.get("header")); cell.setCellStyle(styles.get("header")); } } /** * 轉換字典值 * 將數據中字典值轉化為對應的值(注:字典值應為String格式) */ private void convertDict() throws IllegalAccessException { for (Field field : fieldsContainDict) { Excels annotation = field.getAnnotation(Excels.class); String dictKey = annotation.dictKey(); field.setAccessible(true); for (T t : exportList) { // 獲取字段值 String o = (String) field.get(t); field.set(t, dicts.get(dictKey).get(o)); } } } /** * 將數據導出Excel * * @param list 需要導出的數據 * @param sheetName 工作表名稱 */ public void exportExcel(List<T> list, String sheetName){ exportExcel(list, null, sheetName); } /** * 將數據導出Excel * * @param list 需要導出的數據 */ public void exportExcel(List<T> list) { exportExcel(list, null, "sheet"); } /** * 初始化 */ public void init(List<T> list ,String sheetName, Map<String, Object> fieldsName){ this.checkedFieldsName = fieldsName; this.exportList = list; // 初始化導出數據 initExportList(); // 初始化工作薄 initWorkbook(); // 初始化工作表 initSheet(sheetName); // 初始化checkedFields, fieldsContainDict initFields(); // 根據注解生成生成字典 generateObjDict(); } /** * 初始化導出數據 */ private void initExportList(){ // 防止導出過程中出現空指針 if(Objects.isNull(this.exportList)) { this.exportList = new ArrayList<>(); } } /** * 初始化工作簿 */ private void initWorkbook(){ this.wb = new SXSSFWorkbook(); } /** * 初始化工作表 */ private void initSheet(String sheetName){ this.sheet = wb.createSheet(sheetName); } /** * 初始化checkedFields, fieldsContainDict * fieldsContainDict含有字典表達式的字段 * checkedFields用戶選中的字段 * 1.如果checkedFieldsName沒有定義(未自定義導出字段),所有字段全部導出 * 2.如果checkedFieldsName進行了定義,根據定義字段進行導出 */ private void initFields(){ // 獲取對象所有字段對象 Field[] fields = clazz.getDeclaredFields(); // 過濾出checkedFields this.checkedFields = Arrays. asList(fields). stream(). filter(item -> { if(!Objects.isNull(this.checkedFieldsName)) { if (item.isAnnotationPresent(Excel.class)) { return checkedFieldsName.containsKey(item.getName()); } } else { return item.isAnnotationPresent(Excel.class); } return false; }) .collect(Collectors.toList()); // 過濾出fieldsContainDict this.fieldsContainDict = Arrays .asList(clazz.getDeclaredFields()) .stream() .filter(item -> !"".equals(item.getAnnotation(Excel.class).dictExp())) .collect(Collectors.toList( } /** * 通過掃描字段注解生成字典數據 */ private void generateObjDict(){ if(fieldsContainDict.size() == 0) { return; } if(dicts == null) { dicts = new HashMap<>(); // Map<String, List<Map<String, String>>> } for (Field field : fieldsContainDict) { String dictKey = field.getAnnotation(Excel.class).dictKey(); String exps = field.getAnnotation(Excel.class).dictExp(); String[] exp = exps.split(","); Map<String, String> keyV = new HashMap<>(); dicts.put(dictKey, keyV); for (String s : exp) { String[] out = s.split("="); keyV.put(out[0], out[1]); } System.out.println("字典值:"+ dicts); } } /** * 創建表格樣式 * * @param wb 工作薄對象 * @return 樣式列表 */ private Map<String, CellStyle> createStyles(Workbook wb) { Map<String, CellStyle> styles = new HashMap<String, CellStyle>(); // 數據格式 CellStyle style = wb.createCellStyle(); style.setAlignment(HorizontalAlignment.CENTER); style.setVerticalAlignment(VerticalAlignment.CENTER); style.setBorderRight(BorderStyle.THIN); style.setRightBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); style.setBorderLeft(BorderStyle.THIN); style.setLeftBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); style.setBorderTop(BorderStyle.THIN); style.setTopBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); style.setBorderBottom(BorderStyle.THIN); style.setBottomBorderColor(IndexedColors.GREY_50_PERCENT.getIndex()); Font dataFont = wb.createFont(); dataFont.setFontName("Arial"); dataFont.setFontHeightInPoints((short) 10); style.setFont(dataFont); styles.put("data", style); // 表頭格式 style = wb.createCellStyle(); style.cloneStyleFrom(styles.get("data")); style.setAlignment(HorizontalAlignment.CENTER); style.setVerticalAlignment(VerticalAlignment.CENTER); style.setFillForegroundColor(IndexedColors.GREY_50_PERCENT.getIndex()); style.setFillPattern(FillPatternType.SOLID_FOREGROUND); Font headerFont = wb.createFont(); headerFont.setFontName("Arial"); headerFont.setFontHeightInPoints((short) 10); headerFont.setBold(true); headerFont.setColor(IndexedColors.WHITE.getIndex()); style.setFont(headerFont); styles.put("header", style); return styles; } /** * 生成隨機名稱,防止文件復寫 * @return */ private String generateFileName(){ return "D:\\" + UUID.randomUUID().toString().replace("-", "") + ".xlsx"; } } 

使用方法

將對象加上工具類需要的注解:

@Data @NoArgsConstructor @AllArgsConstructor public class Student { @Excel(name = "姓名") private String name; @Excel(name = "年齡") private Integer age; @Excel(name = "出生日期", dateFormat = "yyyy-MM-dd") private Date birthday; @Excel(name = "性別", dictKey = "sex", dictExp = "1=男,2=女") private String sex; } 

測試字典轉換

public static void main(String[] args) { ArrayList<Student> data = new ArrayList<>(); Student student = new Student(); student.setName("tom"); student.setAge(19); student.setSex("1"); student.setBirthday(new Date()); data.add(student); ExcelUtil<Student> util = new ExcelUtil<>(Student.class); util.exportExcel(data, "人員信息表"); } 

輸出結果:

結果可以看出, 已將1轉換為男了

---

測試選擇字段導出:
若不自定義導出的字段, 工具將會把所有帶有Excel注解的字段進行導出, 如上方所示。
當輸入導出字段時，才能根據定義的字段進行導出，實現如下：

public static void main(String[] args) { ArrayList<Student> data = new ArrayList<>(); Student student = new Student(); student.setName("tom"); student.setAge(19); student.setSex("1"); student.setBirthday(new Date()); data.add(student); // 需要導出字段的名稱，放入map的key中即可(這里只導出姓名和性別) Map<String, Object> fieldsName = new HashMap<>(); fieldsName.put("name", null); fieldsName.put("sex", null); ExcelUtil<Student> util = new ExcelUtil<>(Student.class); // 將fieldsName放入方法中 util.exportExcel(data, fieldsName,"人員信息表"); }