|
|
@@ -0,0 +1,502 @@
|
|
|
+package cn.hobbystocks.auc.common.utils;
|
|
|
+
|
|
|
+import lombok.extern.slf4j.Slf4j;
|
|
|
+import org.apache.poi.ss.usermodel.*;
|
|
|
+import org.apache.poi.ss.util.CellRangeAddressList;
|
|
|
+import org.apache.poi.xssf.streaming.SXSSFWorkbook;
|
|
|
+import org.apache.poi.xssf.usermodel.XSSFDataValidation;
|
|
|
+
|
|
|
+import javax.servlet.http.HttpServletResponse;
|
|
|
+import java.io.File;
|
|
|
+import java.io.FileOutputStream;
|
|
|
+import java.io.IOException;
|
|
|
+import java.io.OutputStream;
|
|
|
+import java.lang.reflect.Field;
|
|
|
+import java.lang.reflect.Method;
|
|
|
+import java.util.*;
|
|
|
+import java.util.stream.Collectors;
|
|
|
+
|
|
|
+@Slf4j
|
|
|
+public class ExcelUtils<T> {
|
|
|
+
|
|
|
+ /**
|
|
|
+ * Excel sheet最大行数,默认65536
|
|
|
+ */
|
|
|
+ public static final int sheetSize = 65536;
|
|
|
+ /**
|
|
|
+ * 文件保存路径
|
|
|
+ */
|
|
|
+ private String filepath;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 工作表名称
|
|
|
+ */
|
|
|
+ private String sheetName;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 导出类型(EXPORT:导出数据;IMPORT:导入模板)
|
|
|
+ */
|
|
|
+ private Excel.Type type;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 工作薄对象
|
|
|
+ */
|
|
|
+ private Workbook wb;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 工作表对象
|
|
|
+ */
|
|
|
+ private Sheet sheet;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 样式列表
|
|
|
+ */
|
|
|
+ private Map<String, CellStyle> styles;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 导入导出数据列表
|
|
|
+ */
|
|
|
+ private List<T> list;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 注解列表
|
|
|
+ */
|
|
|
+ private List<Object[]> fields;
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 实体对象
|
|
|
+ */
|
|
|
+ public Class<T> clazz;
|
|
|
+
|
|
|
+ public ExcelUtils(Class<T> clazz) {
|
|
|
+ this.clazz = clazz;
|
|
|
+ }
|
|
|
+
|
|
|
+ public void init(List<T> list, String sheetName, Excel.Type type) {
|
|
|
+ if (list == null) {
|
|
|
+ list = new ArrayList<T>();
|
|
|
+ }
|
|
|
+ this.list = list;
|
|
|
+ this.sheetName = sheetName;
|
|
|
+ this.type = type;
|
|
|
+ createExcelField();
|
|
|
+ createWorkbook();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 导出
|
|
|
+ */
|
|
|
+ public void exportExcelHttp(List<T> list, HttpServletResponse response, String sheetName) {
|
|
|
+ this.init(list, sheetName, Excel.Type.EXPORT);
|
|
|
+ OutputStream out = null;
|
|
|
+ try {
|
|
|
+ // 取出一共有多少个sheet.
|
|
|
+ double sheetNo = Math.ceil(list.size() / sheetSize);
|
|
|
+ for (int index = 0; index <= sheetNo; index++) {
|
|
|
+ createSheet(sheetNo, index);
|
|
|
+ // 产生一行
|
|
|
+ Row row = sheet.createRow(0);
|
|
|
+ int column = 0;
|
|
|
+ // 写入各个字段的列头名称
|
|
|
+ for (Object[] os : fields) {
|
|
|
+ Excel excel = (Excel) os[1];
|
|
|
+ this.createCell(excel, row, column++);
|
|
|
+ }
|
|
|
+ if (Excel.Type.EXPORT.equals(type)) {
|
|
|
+ fillExcelData(index, row);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ String filename = encodingFilename(sheetName);
|
|
|
+ out = new FileOutputStream(getAbsoluteFile(filename));
|
|
|
+// wb.write(out);
|
|
|
+ //导出数据
|
|
|
+ //设置Http响应头告诉浏览器下载这个附件
|
|
|
+ response.setHeader("Content-Disposition", "attachment;Filename=" + new String(sheetName.getBytes("gb2312"), "ISO8859-1") + ".xlsx");
|
|
|
+ out = response.getOutputStream();
|
|
|
+ wb.write(out);
|
|
|
+ } catch (Exception e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ log.error("导出Excel异常{}", e.getMessage());
|
|
|
+ throw new RuntimeException("导出Excel失败,请联系系统管理员!");
|
|
|
+ } finally {
|
|
|
+ if (wb != null) {
|
|
|
+ try {
|
|
|
+ wb.close();
|
|
|
+ } catch (IOException e1) {
|
|
|
+ e1.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ if (out != null) {
|
|
|
+ try {
|
|
|
+ out.close();
|
|
|
+ } catch (IOException e1) {
|
|
|
+ e1.printStackTrace();
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取下载路径
|
|
|
+ *
|
|
|
+ * @param filename 文件名称
|
|
|
+ */
|
|
|
+ public String getAbsoluteFile(String filename)
|
|
|
+ {
|
|
|
+ String downloadPath = Global.getDownloadPath() + filename;
|
|
|
+ File desc = new File(downloadPath);
|
|
|
+ if (!desc.getParentFile().exists())
|
|
|
+ {
|
|
|
+ desc.getParentFile().mkdirs();
|
|
|
+ }
|
|
|
+ return downloadPath;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 得到所有定义字段
|
|
|
+ */
|
|
|
+ private void createExcelField() {
|
|
|
+ this.fields = new ArrayList<Object[]>();
|
|
|
+ List<Field> tempFields = new ArrayList<>();
|
|
|
+ tempFields.addAll(Arrays.asList(clazz.getSuperclass().getDeclaredFields()));
|
|
|
+ tempFields.addAll(Arrays.asList(clazz.getDeclaredFields()));
|
|
|
+ for (Field field : tempFields) {
|
|
|
+ // 单注解
|
|
|
+ if (field.isAnnotationPresent(Excel.class)) {
|
|
|
+ putToField(field, field.getAnnotation(Excel.class));
|
|
|
+ }
|
|
|
+
|
|
|
+ // 多注解
|
|
|
+ if (field.isAnnotationPresent(Excels.class)) {
|
|
|
+ Excels attrs = field.getAnnotation(Excels.class);
|
|
|
+ Excel[] excels = attrs.value();
|
|
|
+ for (Excel excel : excels) {
|
|
|
+ putToField(field, excel);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ this.fields = this.fields.stream().sorted(Comparator.comparing(objects -> ((Excel) objects[1]).sort())).collect(Collectors.toList());
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建一个工作簿
|
|
|
+ */
|
|
|
+ public void createWorkbook() {
|
|
|
+ this.wb = new SXSSFWorkbook(500);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建工作表
|
|
|
+ *
|
|
|
+ * @param sheetNo sheet数量
|
|
|
+ * @param index 序号
|
|
|
+ */
|
|
|
+ public void createSheet(double sheetNo, int index) {
|
|
|
+ try {
|
|
|
+ this.sheet = wb.createSheet();
|
|
|
+ } catch (Exception e) {
|
|
|
+ e.printStackTrace();
|
|
|
+ }
|
|
|
+ this.styles = createStyles(wb);
|
|
|
+ // 设置工作表的名称.
|
|
|
+ if (sheetNo == 0) {
|
|
|
+ wb.setSheetName(index, sheetName);
|
|
|
+ } else {
|
|
|
+ wb.setSheetName(index, sheetName + index);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建表格样式
|
|
|
+ *
|
|
|
+ * @param wb 工作薄对象
|
|
|
+ * @return 样式列表
|
|
|
+ */
|
|
|
+ private Map<String, CellStyle> createStyles(Workbook wb) {
|
|
|
+ // 写入各条记录,每条记录对应excel表中的一行
|
|
|
+ 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;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建单元格
|
|
|
+ */
|
|
|
+ public Cell createCell(Excel attr, Row row, int column) {
|
|
|
+ // 创建列
|
|
|
+ Cell cell = row.createCell(column);
|
|
|
+ // 写入列信息
|
|
|
+ cell.setCellValue(attr.name());
|
|
|
+ setDataValidation(attr, row, column);
|
|
|
+ cell.setCellStyle(styles.get("header"));
|
|
|
+ return cell;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 创建表格样式
|
|
|
+ */
|
|
|
+ public void setDataValidation(Excel attr, Row row, int column) {
|
|
|
+ if (attr.name().indexOf("注:") >= 0) {
|
|
|
+ sheet.setColumnWidth(column, 6000);
|
|
|
+ } else {
|
|
|
+ // 设置列宽
|
|
|
+ sheet.setColumnWidth(column, (int) ((attr.width() + 0.72) * 256));
|
|
|
+ row.setHeight((short) (attr.height() * 20));
|
|
|
+ }
|
|
|
+ // 如果设置了提示信息则鼠标放上去提示.
|
|
|
+ if (StringUtils.isNotEmpty(attr.prompt())) {
|
|
|
+ // 这里默认设了2-101列提示.
|
|
|
+ setXSSFPrompt(sheet, "", attr.prompt(), 1, 100, column, column);
|
|
|
+ }
|
|
|
+ // 如果设置了combo属性则本列只能选择不能输入
|
|
|
+ if (attr.combo().length > 0) {
|
|
|
+ // 这里默认设了2-101列只能选择不能输入.
|
|
|
+ setXSSFValidation(sheet, attr.combo(), 1, 100, column, column);
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 设置 POI XSSFSheet 单元格提示
|
|
|
+ *
|
|
|
+ * @param sheet 表单
|
|
|
+ * @param promptTitle 提示标题
|
|
|
+ * @param promptContent 提示内容
|
|
|
+ * @param firstRow 开始行
|
|
|
+ * @param endRow 结束行
|
|
|
+ * @param firstCol 开始列
|
|
|
+ * @param endCol 结束列
|
|
|
+ */
|
|
|
+ public void setXSSFPrompt(Sheet sheet, String promptTitle, String promptContent, int firstRow, int endRow,
|
|
|
+ int firstCol, int endCol) {
|
|
|
+ DataValidationHelper helper = sheet.getDataValidationHelper();
|
|
|
+ DataValidationConstraint constraint = helper.createCustomConstraint("DD1");
|
|
|
+ CellRangeAddressList regions = new CellRangeAddressList(firstRow, endRow, firstCol, endCol);
|
|
|
+ DataValidation dataValidation = helper.createValidation(constraint, regions);
|
|
|
+ dataValidation.createPromptBox(promptTitle, promptContent);
|
|
|
+ dataValidation.setShowPromptBox(true);
|
|
|
+ sheet.addValidationData(dataValidation);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 设置某些列的值只能输入预制的数据,显示下拉框.
|
|
|
+ *
|
|
|
+ * @param sheet 要设置的sheet.
|
|
|
+ * @param textlist 下拉框显示的内容
|
|
|
+ * @param firstRow 开始行
|
|
|
+ * @param endRow 结束行
|
|
|
+ * @param firstCol 开始列
|
|
|
+ * @param endCol 结束列
|
|
|
+ * @return 设置好的sheet.
|
|
|
+ */
|
|
|
+ public void setXSSFValidation(Sheet sheet, String[] textlist, int firstRow, int endRow, int firstCol, int endCol) {
|
|
|
+ DataValidationHelper helper = sheet.getDataValidationHelper();
|
|
|
+ // 加载下拉列表内容
|
|
|
+ DataValidationConstraint constraint = helper.createExplicitListConstraint(textlist);
|
|
|
+ // 设置数据有效性加载在哪个单元格上,四个参数分别是:起始行、终止行、起始列、终止列
|
|
|
+ CellRangeAddressList regions = new CellRangeAddressList(firstRow, endRow, firstCol, endCol);
|
|
|
+ // 数据有效性对象
|
|
|
+ DataValidation dataValidation = helper.createValidation(constraint, regions);
|
|
|
+ // 处理Excel兼容性问题
|
|
|
+ if (dataValidation instanceof XSSFDataValidation) {
|
|
|
+ dataValidation.setSuppressDropDownArrow(true);
|
|
|
+ dataValidation.setShowErrorBox(true);
|
|
|
+ } else {
|
|
|
+ dataValidation.setSuppressDropDownArrow(false);
|
|
|
+ }
|
|
|
+
|
|
|
+ sheet.addValidationData(dataValidation);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 放到字段集合中
|
|
|
+ */
|
|
|
+ private void putToField(Field field, Excel attr) {
|
|
|
+ if (attr != null && (attr.type() == Excel.Type.ALL || attr.type() == type)) {
|
|
|
+ this.fields.add(new Object[]{field, attr});
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 填充excel数据
|
|
|
+ *
|
|
|
+ * @param index 序号
|
|
|
+ * @param row 单元格行
|
|
|
+ */
|
|
|
+ public void fillExcelData(int index, Row row) {
|
|
|
+ int startNo = index * sheetSize;
|
|
|
+ int endNo = Math.min(startNo + sheetSize, list.size());
|
|
|
+ for (int i = startNo; i < endNo; i++) {
|
|
|
+ row = sheet.createRow(i + 1 - startNo);
|
|
|
+ // 得到导出对象.
|
|
|
+ T vo = (T) list.get(i);
|
|
|
+ int column = 0;
|
|
|
+ for (Object[] os : fields) {
|
|
|
+ Field field = (Field) os[0];
|
|
|
+ Excel excel = (Excel) os[1];
|
|
|
+ // 设置实体类私有属性可访问
|
|
|
+ field.setAccessible(true);
|
|
|
+ this.addCell(excel, row, vo, field, column++);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 添加单元格
|
|
|
+ */
|
|
|
+ public Cell addCell(Excel attr, Row row, T vo, Field field, int column) {
|
|
|
+ Cell cell = null;
|
|
|
+ try {
|
|
|
+ // 设置行高
|
|
|
+ row.setHeight((short) (attr.height() * 20));
|
|
|
+ // 根据Excel中设置情况决定是否导出,有些情况需要保持为空,希望用户填写这一列.
|
|
|
+ if (attr.isExport()) {
|
|
|
+ // 创建cell
|
|
|
+ cell = row.createCell(column);
|
|
|
+ if (null != styles) {
|
|
|
+ cell.setCellStyle(styles.get("data"));
|
|
|
+ }
|
|
|
+ // 用于读取对象中的属性
|
|
|
+// Object value = getTargetValue(vo, field, attr);
|
|
|
+ Object value = getValue(vo, field.getName());
|
|
|
+ String dateFormat = attr.dateFormat();
|
|
|
+ String dictType = attr.dictType();
|
|
|
+ String readConverterExp = attr.readConverterExp();
|
|
|
+ String readFormatDataMethod = attr.readFormatDataMethod();
|
|
|
+ String desensitize = attr.desensitize();
|
|
|
+ String desensitizeType = attr.desensitizeType();
|
|
|
+ if (StringUtils.isNotEmpty(dateFormat) && StringUtils.isNotNull(value)) {
|
|
|
+ if (StringUtils.isNotEmpty(String.valueOf(value))) {
|
|
|
+ cell.setCellValue(DateUtils.parseDateToStr(dateFormat, (Date) value));
|
|
|
+ }
|
|
|
+
|
|
|
+ } else if (StringUtils.isNotEmpty(readConverterExp) && StringUtils.isNotNull(value)) {
|
|
|
+ cell.setCellValue(convertByExp(String.valueOf(value), readConverterExp));
|
|
|
+ } else if (StringUtils.isNotEmpty(dictType) && StringUtils.isNotNull(value)) {
|
|
|
+ //cell.setCellValue(convertDictByExp(Convert.toStr(value), dictType, attr.separator()));
|
|
|
+ } else if (StringUtils.isNotEmpty(readFormatDataMethod) && "get".equals(readFormatDataMethod)) {
|
|
|
+ Object val = getValue(vo, field.getName());
|
|
|
+ cell.setCellValue(String.valueOf(val));
|
|
|
+ } else {
|
|
|
+ // 设置列类型
|
|
|
+ setCellVo(value, attr, cell);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ log.error("导出Excel失败{}", e);
|
|
|
+ }
|
|
|
+ return cell;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 以类的属性的get方法方法形式获取值
|
|
|
+ *
|
|
|
+ * @param o
|
|
|
+ * @param name
|
|
|
+ * @return value
|
|
|
+ * @throws Exception
|
|
|
+ */
|
|
|
+ private Object getValue(Object o, String name) throws Exception {
|
|
|
+ if (StringUtils.isNotEmpty(name)) {
|
|
|
+ Class<?> clazz = o.getClass();
|
|
|
+ String methodName = "get" + name.substring(0, 1).toUpperCase() + name.substring(1);
|
|
|
+ Method method = clazz.getMethod(methodName);
|
|
|
+ o = method.invoke(o);
|
|
|
+ }
|
|
|
+ return null == o ? "" : o;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 解析字典值
|
|
|
+ *
|
|
|
+ * @param dictValue 字典值
|
|
|
+ * @param dictType 字典类型
|
|
|
+ * @param separator 分隔符
|
|
|
+ * @return 字典标签
|
|
|
+ */
|
|
|
+ /* public static String convertDictByExp(String dictValue, String dictType, String separator) throws Exception {
|
|
|
+ return DictUtils.getDictLabel(dictType, dictValue, separator);
|
|
|
+ }*/
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 编码文件名
|
|
|
+ */
|
|
|
+ public String encodingFilename(String filename) {
|
|
|
+ if (StringUtils.isEmpty(filename)) {
|
|
|
+ String path = UUID.randomUUID().toString() + "_" + filename + ".xlsx";
|
|
|
+ filename = StringUtils.isNotEmpty(filepath) ? filepath + "//" + path : path;
|
|
|
+ } else {
|
|
|
+ filename = StringUtils.isNotEmpty(filepath) ? filepath + "//" + filename + ".xlsx" : filename + ".xlsx";
|
|
|
+ }
|
|
|
+ return filename;
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 设置单元格信息
|
|
|
+ *
|
|
|
+ * @param value 单元格值
|
|
|
+ * @param attr 注解相关
|
|
|
+ * @param cell 单元格信息
|
|
|
+ */
|
|
|
+ public void setCellVo(Object value, Excel attr, Cell cell) {
|
|
|
+ if (Excel.ColumnType.STRING == attr.cellType()) {
|
|
|
+ cell.setCellType(CellType.NUMERIC);
|
|
|
+ cell.setCellValue(StringUtils.isNull(value) ? attr.defaultValue() : value + attr.suffix());
|
|
|
+ } else if (Excel.ColumnType.NUMERIC == attr.cellType()) {
|
|
|
+ cell.setCellType(CellType.NUMERIC);
|
|
|
+ cell.setCellValue(Integer.parseInt(value + ""));
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 解析导出值 0=男,1=女,2=未知
|
|
|
+ *
|
|
|
+ * @param propertyValue 参数值
|
|
|
+ * @param converterExp 翻译注解
|
|
|
+ * @return 解析后值
|
|
|
+ * @throws Exception
|
|
|
+ */
|
|
|
+ public static String convertByExp(String propertyValue, String converterExp) throws Exception {
|
|
|
+ try {
|
|
|
+ String[] convertSource = converterExp.split(",");
|
|
|
+ for (String item : convertSource) {
|
|
|
+ String[] itemArray = item.split("=");
|
|
|
+ if (itemArray[0].equals(propertyValue)) {
|
|
|
+ return itemArray[1];
|
|
|
+ }
|
|
|
+ }
|
|
|
+ } catch (Exception e) {
|
|
|
+ throw e;
|
|
|
+ }
|
|
|
+ return propertyValue;
|
|
|
+ }
|
|
|
+}
|