java 后台权限管理系统

package cc.mrbird.common.util; import java.io.FileOutputStream; import java.io.IOException; import java.io.OutputStream; import java.lang.reflect.Field; import java.nio.charset.Charset; import java.util.ArrayList; import java.util.HashMap; import java.util.List; import java.util.Map; import javax.servlet.http.HttpServletResponse; import org.apache.commons.beanutils.BeanUtils; import org.apache.commons.lang3.StringUtils; import org.apache.poi.hssf.util.HSSFColor; import org.apache.poi.ss.usermodel.CellStyle; import org.apache.poi.ss.usermodel.Font; import org.apache.poi.xssf.streaming.SXSSFCell; import org.apache.poi.xssf.streaming.SXSSFRow; import org.apache.poi.xssf.streaming.SXSSFSheet; import org.apache.poi.xssf.streaming.SXSSFWorkbook; import com.csvreader.CsvWriter; import cc.mrbird.common.annotation.ExportConfig; import cc.mrbird.common.handler.ExportHandler; import cc.mrbird.common.util.poi.convert.ExportConvert; import cc.mrbird.common.util.poi.pojo.ExportItem; import org.slf4j.Logger; import org.slf4j.LoggerFactory; public class ExcelUtils { private Logger log = LoggerFactory.getLogger(this.getClass()); private Class<?> mClass = null; private HttpServletResponse mResponse = null; // 分Sheet机制：每个Sheet最多多少条数据 private Integer mMaxSheetRecords = 10000; // 缓存数据格式器实例,避免多次使用反射进行实例化 private Map<String, ExportConvert> mConvertInstanceCache = new HashMap<>(); protected ExcelUtils() { } private ExcelUtils(Class<?> clazz) { this(clazz, null); } private ExcelUtils(Class<?> clazz, HttpServletResponse response) { this.mResponse = response; this.mClass = clazz; } /** * 用于生成本地文件 * * @param clazz 实体Class对象 * @return ExcelUtils */ static ExcelUtils builder(Class<?> clazz) { return new ExcelUtils(clazz); } /** * 用于浏览器导出 * * @param clazz 实体Class对象 * @param response 原生HttpServletResponse对象 * @return ExcelUtils */ public static ExcelUtils export(Class<?> clazz, HttpServletResponse response) { return new ExcelUtils(clazz, response); } /** * 分Sheet机制：每个Sheet最多多少条数据(默认10000) * * @param size 数据条数 * @return this */ public ExcelUtils setMaxSheetRecords(Integer size) { this.mMaxSheetRecords = size; return this; } /** * 导出Excel(此方式需依赖浏览器实现文件下载,故应先使用export()构造器) * * @param data 数据集合 * @param sheetName 工作表名字 * @return true-操作成功,false-操作失败 */ public boolean toExcel(List<?> data, String sheetName) { requiredexportParams(); try { return toExcel(data, sheetName, mResponse.getOutputStream()); } catch (IOException e) { log.error(e.getMessage()); } return false; } /** * 针对转换方法的默认实现(提供默认样式和文件命名规则) * * @param data 数据集合 * @param sheetName 工作表名字 * @param out 输出流 * @return true-操作成功,false-操作失败 */ boolean toExcel(List<?> data, String sheetName, OutputStream out) { return toExcel(data, sheetName, new ExportHandler() { @Override public CellStyle headCellStyle(SXSSFWorkbook wb) { CellStyle cellStyle = wb.createCellStyle(); Font font = wb.createFont(); cellStyle.setFillForegroundColor((short) 12); cellStyle.setFillPattern(CellStyle.SOLID_FOREGROUND);// 填充模式 cellStyle.setBorderTop(CellStyle.BORDER_THIN);// 上边框为细边框 cellStyle.setBorderRight(CellStyle.BORDER_THIN);// 右边框为细边框 cellStyle.setBorderBottom(CellStyle.BORDER_THIN);// 下边框为细边框 cellStyle.setBorderLeft(CellStyle.BORDER_THIN);// 左边框为细边框 cellStyle.setAlignment(CellStyle.ALIGN_LEFT);// 对齐 cellStyle.setFillForegroundColor(HSSFColor.GREEN.index); cellStyle.setFillBackgroundColor(HSSFColor.GREEN.index); font.setBoldweight(Font.BOLDWEIGHT_NORMAL); // font.setFontHeightInPoints((short) 12);// 字体大小 font.setColor(HSSFColor.WHITE.index); // 应用标题字体到标题样式 cellStyle.setFont(font); return cellStyle; } @Override public String exportFileName(String sheetName) { return String.format("导出-%s-%s", sheetName, System.currentTimeMillis()); } }, out); } private boolean toExcel(List<?> data, String sheetName, ExportHandler handler, OutputStream out) { requiredbuilderParams(); if (data == null || data.isEmpty()) { return false; } // 导出列查询。 ExportConfig currentExportConfig; ExportItem currentExportItem; List<ExportItem> exportItems = new ArrayList<>(); for (Field field : mClass.getDeclaredFields()) { currentExportConfig = field.getAnnotation(ExportConfig.class); if (currentExportConfig != null) { currentExportItem = new ExportItem().setField(field.getName()) .setDisplay("field".equals(currentExportConfig.value()) ? field.getName() : currentExportConfig.value()) .setWidth(currentExportConfig.width()).setConvert(currentExportConfig.convert()) .setColor(currentExportConfig.color()).setReplace(currentExportConfig.replace()); exportItems.add(currentExportItem); } } // 创建新的工作薄。 SXSSFWorkbook wb = POIUtils.newSXSSFWorkbook(); double sheetNo = Math.ceil((double) data.size() / mMaxSheetRecords);// 取出一共有多少个sheet. // =====多sheet生成填充数据===== int index = 0; while (index <= (sheetNo == 0.0 ? sheetNo : sheetNo - 1)) { SXSSFSheet sheet = POIUtils.newSXSSFSheet(wb, sheetName + (index == 0 ? "" : "_" + index)); // 创建表头 SXSSFRow headerRow = POIUtils.newSXSSFRow(sheet, 0); for (int i = 0; i < exportItems.size(); i++) { SXSSFCell cell = POIUtils.newSXSSFCell(headerRow, i); POIUtils.setColumnWidth(sheet, i, exportItems.get(i).getWidth(), exportItems.get(i).getDisplay()); cell.setCellValue(exportItems.get(i).getDisplay()); CellStyle style = handler.headCellStyle(wb); if (style != null) { cell.setCellStyle(style); } } SXSSFRow bodyRow; String cellValue; SXSSFCell cell; CellStyle style = wb.createCellStyle(); Font font = wb.createFont(); style.setFont(font); // 产生数据行 if (!data.isEmpty()) { int startNo = index * mMaxSheetRecords; int endNo = Math.min(startNo + mMaxSheetRecords, data.size()); int i = startNo; while (i < endNo) { bodyRow = POIUtils.newSXSSFRow(sheet, i + 1 - startNo); for (int j = 0; j < exportItems.size(); j++) { // 处理单元格值 cellValue = exportItems.get(j).getReplace(); if ("".equals(cellValue)) { try { cellValue = BeanUtils.getProperty(data.get(i), exportItems.get(j).getField());