From 489c190434088c96c10ec1ab179cf1e3564b36e6 Mon Sep 17 00:00:00 2001 From: "udf_admin@163.com" Date: Sun, 17 Dec 2017 16:12:01 +0800 Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0excel=E7=9B=B8=E5=85=B3?= =?UTF-8?q?=E5=B7=A5=E5=85=B7=E7=B1=BB?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- pom.xml | 1 + udf-starter-core/pom.xml | 12 + .../udf/core/utils/excel/ExportExcel.java | 363 ++++++++++++++++++ .../excel/ExportExcelHeaderMateData.java | 34 ++ .../udf/core/utils/excel/ImportExcel.java | 209 ++++++++++ .../java/org/itkk/udf/id/IdWorkerTest.java | 3 +- 6 files changed, 621 insertions(+), 1 deletion(-) create mode 100644 udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcel.java create mode 100644 udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcelHeaderMateData.java create mode 100644 udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ImportExcel.java diff --git a/pom.xml b/pom.xml index d5212fb..2adc954 100644 --- a/pom.xml +++ b/pom.xml @@ -28,6 +28,7 @@ 1.3.1 1.1 2.8.2 + 3.17 diff --git a/udf-starter-core/pom.xml b/udf-starter-core/pom.xml index 8a6baff..8a30036 100644 --- a/udf-starter-core/pom.xml +++ b/udf-starter-core/pom.xml @@ -97,6 +97,18 @@ commons-lang3 ${commons-lang3.version} + + + org.apache.poi + poi + ${poi.version} + + + org.apache.poi + poi-ooxml + ${poi.version} + + org.springframework.cloud diff --git a/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcel.java b/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcel.java new file mode 100644 index 0000000..945d401 --- /dev/null +++ b/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcel.java @@ -0,0 +1,363 @@ +/** + * ExportExcel.java + * Created at 2016-03-01 + * Created by Administrator + * Copyright (C) 2016 LLSFW, All rights reserved. + */ +package org.itkk.udf.core.utils.excel; + +import org.apache.commons.lang3.StringUtils; +import org.apache.poi.ss.usermodel.*; +import org.apache.poi.ss.util.CellRangeAddress; +import org.apache.poi.xssf.streaming.SXSSFSheet; +import org.apache.poi.xssf.streaming.SXSSFWorkbook; +import org.apache.poi.xssf.usermodel.XSSFClientAnchor; +import org.apache.poi.xssf.usermodel.XSSFRichTextString; +import org.itkk.udf.core.exception.SystemRuntimeException; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import javax.servlet.http.HttpServletResponse; +import java.io.FileOutputStream; +import java.io.IOException; +import java.io.OutputStream; +import java.io.UnsupportedEncodingException; +import java.net.URLEncoder; +import java.util.Date; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +/** + *

+ * ClassName: ExportExcel + *

+ *

+ * Description: 导出excel(2007+) + *

+ *

+ * Author: Administrator + *

+ *

+ * Date: 2016年3月23日 + *

+ */ +public class ExportExcel { + + /** + * 日志 + */ + private static Logger log = LoggerFactory.getLogger(ExportExcel.class); + + /** + * 字体 + */ + private static final String FONT_NAME = "Arial"; + + /** + *

+ * Field DATA: 样式名称 + *

+ */ + private static final String STYLE_DATA = "data"; + + /** + * 工作薄对象 + */ + private SXSSFWorkbook wb; + + /** + * 工作表对象 + */ + private SXSSFSheet sheet; + + /** + * 样式列表 + */ + private Map styles; + + /** + * 当前行号 + */ + private int rownum; + + /** + * 构造函数 + * + * @param title 表格标题，传“空值”，表示无标题 + * @param headerList 表头列表 + * @param rowAccessWindowSize 刷新行数 + * @throws SystemRuntimeException + */ + public ExportExcel(String title, List headerList, int rowAccessWindowSize) { + initialize(title, headerList, rowAccessWindowSize); + } + + /** + * 初始化函数 + * + * @param title 表格标题，传“空值”，表示无标题 + * @param headerList 表头列表 + * @param rowAccessWindowSize 刷新行数 + * @throws SystemRuntimeException + */ + private void initialize(String title, List headerList, int rowAccessWindowSize) { + final int num2 = 2; + final int num3 = 3; + final int num5 = 5; + final int num6 = 6; + final int num16 = 16; + final int num30 = 30; + final int num256 = 256; + this.wb = new SXSSFWorkbook(rowAccessWindowSize); //有没有被关闭的风险 + this.wb.setCompressTempFiles(true); // 缓冲区文件压缩 + this.sheet = this.wb.createSheet("Export"); + this.styles = createStyles(); + // 创建标题 + if (StringUtils.isNotBlank(title)) { + Row titleRow = this.sheet.createRow(this.rownum++); + titleRow.setHeightInPoints(num30); + Cell titleCell = titleRow.createCell(0); + titleCell.setCellStyle(this.styles.get("title")); + titleCell.setCellValue(title); + this.sheet.addMergedRegion(new CellRangeAddress(titleRow.getRowNum(), titleRow.getRowNum(), + titleRow.getRowNum(), headerList.size() - 1)); + } + // 创建表头 + if (headerList == null) { + throw new SystemRuntimeException("headerList not null!"); + } + Row headerRow = this.sheet.createRow(this.rownum++); + headerRow.setHeightInPoints(num16); + for (int i = 0; i < headerList.size(); i++) { + ExportExcelHeaderMateData headerItem = headerList.get(i); + Cell cell = headerRow.createCell(i); + cell.setCellStyle(this.styles.get("header")); + String[] ss = StringUtils.split(headerItem.getHeaderName(), "**", num2); + if (ss.length == num2) { + cell.setCellValue(ss[0]); + Comment comment = this.sheet.createDrawingPatriarch() + .createCellComment(new XSSFClientAnchor(0, 0, 0, 0, (short) num3, num3, (short) num5, num6)); + comment.setString(new XSSFRichTextString(ss[1])); + cell.setCellComment(comment); + } else { + cell.setCellValue(headerItem.getHeaderName()); + } + this.sheet.setColumnWidth(i, headerItem.getHeaderWidth() * num2 * num256); + } + log.debug("Initialize success."); + } + + /** + * 创建表格样式 + * + * @return 样式列表 + */ + private Map createStyles() { + final int num10 = 10; + final int num16 = 16; + + Map stylesMap = new HashMap<>(); + + // 标题样式 + CellStyle style = this.wb.createCellStyle(); + style.setAlignment(HorizontalAlignment.CENTER); + style.setVerticalAlignment(VerticalAlignment.CENTER); + Font titleFont = this.wb.createFont(); + titleFont.setFontName(FONT_NAME); + titleFont.setFontHeightInPoints((short) num16); + titleFont.setBold(true); + style.setFont(titleFont); + stylesMap.put("title", style); + + // 数据样式 + style = this.wb.createCellStyle(); + 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 = this.wb.createFont(); + dataFont.setFontName(FONT_NAME); + dataFont.setFontHeightInPoints((short) num10); + style.setFont(dataFont); + stylesMap.put(STYLE_DATA, style); + + // 数据样式1 + style = this.wb.createCellStyle(); + style.cloneStyleFrom(stylesMap.get(STYLE_DATA)); + style.setAlignment(HorizontalAlignment.LEFT); + stylesMap.put("data1", style); + + // 数据样式2 + style = this.wb.createCellStyle(); + style.cloneStyleFrom(stylesMap.get(STYLE_DATA)); + style.setAlignment(HorizontalAlignment.CENTER); + stylesMap.put("data2", style); + + // 数据样式3 + style = this.wb.createCellStyle(); + style.cloneStyleFrom(stylesMap.get(STYLE_DATA)); + style.setAlignment(HorizontalAlignment.RIGHT); + stylesMap.put("data3", style); + + // 表头样式 + style = this.wb.createCellStyle(); + style.cloneStyleFrom(stylesMap.get(STYLE_DATA)); + style.setWrapText(true); + style.setAlignment(HorizontalAlignment.CENTER); + style.setFillForegroundColor(IndexedColors.GREY_50_PERCENT.getIndex()); + style.setFillPattern(FillPatternType.SOLID_FOREGROUND); + Font headerFont = this.wb.createFont(); + headerFont.setFontName(FONT_NAME); + headerFont.setFontHeightInPoints((short) num10); + headerFont.setBold(true); + headerFont.setColor(IndexedColors.WHITE.getIndex()); + style.setFont(headerFont); + stylesMap.put("header", style); + + // 返回 + return stylesMap; + } + + /** + * 添加一行 + * + * @return 行对象 + */ + public Row addRow() { + return this.sheet.createRow(this.rownum++); + } + + /** + * 添加一个单元格 + * + * @param row 添加的行 + * @param column 添加列号 + * @param val 添加值 + * @return 单元格对象 + */ + public Cell addCell(Row row, int column, Object val) { + return this.addCell(row, column, val, 0); + } + + /** + *

+ * Description: 添加一个单元格 + *

+ * + * @param row 添加的行 + * @param column 添加列号 + * @param val 添加值 + * @param align 对齐方式（1：靠左；2：居中；3：靠右） + * @return 单元格 + */ + public Cell addCell(Row row, int column, Object val, int align) { + final int num3 = 3; + Cell cell = row.createCell(column); + CellStyle style = this.styles.get(STYLE_DATA + (align >= 1 && align <= num3 ? align : "")); + if (val == null) { + cell.setCellValue(""); + } else if (val instanceof String) { + cell.setCellValue((String) val); + } else if (val instanceof Integer) { + cell.setCellValue((Integer) val); + } else if (val instanceof Long) { + cell.setCellValue((Long) val); + } else if (val instanceof Double) { + cell.setCellValue((Double) val); + } else if (val instanceof Float) { + cell.setCellValue((Float) val); + } else if (val instanceof Date) { + DataFormat format = this.wb.createDataFormat(); + style.setDataFormat(format.getFormat("yyyy-MM-dd")); + cell.setCellValue((Date) val); + } else { + cell.setCellValue(val.toString()); + } + cell.setCellStyle(style); + return cell; + } + + /** + *

+ * Description: 合并单元格，行列位置都是从0开始 + *

+ * + * @param firstRow 起始行 + * @param lastRow 结束行 + * @param firstCol 其实列 + * @param lastCol 结束列 + */ + public void mergeCell(int firstRow, int lastRow, int firstCol, int lastCol) { + this.sheet.addMergedRegion(new CellRangeAddress(firstRow, lastRow, firstCol, lastCol)); + } + + /** + * 输出到客户端 + * + * @param response 响应对象 + * @param fileName 文件名 + * @throws IOException 异常 + */ + public void write(HttpServletResponse response, String fileName) throws IOException { + OutputStream os = null; + try { + response.reset(); + response.setContentType("application/octet-stream; charset=utf-8"); + response.setHeader("Content-Disposition", "attachment; filename=" + urlEncode(fileName)); + os = response.getOutputStream(); + this.wb.write(os); + } finally { + if (os != null) { + os.flush(); + os.close(); + } + } + } + + /** + * 输出到文件 + * + * @param name 输出文件名 + * @throws IOException 异常 + */ + public void writeFile(String name) throws IOException { + FileOutputStream os = null; + try { //NOSONAR + os = new FileOutputStream(name); + this.wb.write(os); + } finally { + if (os != null) { + os.flush(); + os.close(); + } + } + } + + /** + * 清理临时文件 + * + * @throws IOException 异常 + */ + public void dispose() throws IOException { + this.wb.dispose(); + this.wb.close(); + } + + /** + * URL 编码, Encode默认为UTF-8. + * + * @param part 附件名称 + * @return 编码后的附件名称 + * @throws UnsupportedEncodingException 异常 + */ + private String urlEncode(String part) throws UnsupportedEncodingException { + return URLEncoder.encode(part, "UTF-8"); + } + +} diff --git a/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcelHeaderMateData.java b/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcelHeaderMateData.java new file mode 100644 index 0000000..79c097f --- /dev/null +++ b/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcelHeaderMateData.java @@ -0,0 +1,34 @@ +package org.itkk.udf.core.utils.excel; + +import lombok.Data; + +/** + * Excel导出表头定义 + * + * @author Administrator + */ +@Data +public class ExportExcelHeaderMateData { + + /** + * 表头名称 + */ + private String headerName; + + /** + * 表头列宽(算法:最大字数) + */ + private int headerWidth; + + /** + * 构造函数 + * + * @param headerName 表头名称 + * @param headerWidth 表头列宽(算法:最大字数) + */ + public ExportExcelHeaderMateData(String headerName, int headerWidth) { + this.headerName = headerName; + this.headerWidth = headerWidth; + } + +} diff --git a/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ImportExcel.java b/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ImportExcel.java new file mode 100644 index 0000000..21612be --- /dev/null +++ b/udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ImportExcel.java @@ -0,0 +1,209 @@ +/** + * ImportExcel.java + * Created at 2016-03-01 + * Created by Administrator + * Copyright (C) 2016 LLSFW, All rights reserved. + */ + +package org.itkk.udf.core.utils.excel; + +import org.apache.commons.lang3.StringUtils; +import org.apache.poi.hssf.usermodel.HSSFWorkbook; +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.slf4j.Logger; +import org.slf4j.LoggerFactory; +import org.springframework.web.multipart.MultipartFile; + +import java.io.File; +import java.io.FileInputStream; +import java.io.IOException; +import java.io.InputStream; + +/** + * 导入excel(2003+) + * + * @author Administrator + */ +public class ImportExcel { + + /** + * 日志 + */ + private static final Logger LOG = LoggerFactory.getLogger(ImportExcel.class); + + /** + * 工作薄对象 + */ + private Workbook wb; + + /** + * 工作表对象 + */ + private Sheet sheet; + + /** + * 标题行号 + */ + private int headerNum; + + /** + * 构造函数 + * + * @param fileName 文件名称(完整路径) + * @param headerNum 标题行号，数据行号=标题行号+1 + * @throws IOException 异常 + */ + public ImportExcel(String fileName, int headerNum) throws IOException { + this(new File(fileName), headerNum); + } + + /** + * 构造函数 + * + * @param file 文件对象 + * @param headerNum 标题行号，数据行号=标题行号+1 + * @throws IOException 异常 + */ + public ImportExcel(File file, int headerNum) throws IOException { + this(file, headerNum, 0); + } + + /** + * 构造函数 + * + * @param fileName 文件名称(完整路径) + * @param headerNum 标题行号，数据行号=标题行号+1 + * @param sheetIndex 工作表编号 + * @throws IOException 异常 + */ + public ImportExcel(String fileName, int headerNum, int sheetIndex) throws IOException { + this(new File(fileName), headerNum, sheetIndex); + } + + /** + * 构造函数 + * + * @param file 导入文件对象 + * @param headerNum 标题行号，数据行号=标题行号+1 + * @param sheetIndex 工作表编号 + * @throws IOException 异常 + */ + public ImportExcel(File file, int headerNum, int sheetIndex) throws IOException { + this(file.getName(), new FileInputStream(file), headerNum, sheetIndex); + } + + /** + * 构造函数 + * + * @param multipartFile 上传文件对象 + * @param headerNum 标题行号，数据行号=标题行号+1 + * @param sheetIndex 工作表编号 + * @throws IOException 异常 + */ + public ImportExcel(MultipartFile multipartFile, int headerNum, int sheetIndex) throws IOException { + this(multipartFile.getOriginalFilename(), multipartFile.getInputStream(), headerNum, sheetIndex); + } + + /** + * 构造函数 + * + * @param fileName 文件名称(完整路径) + * @param is 输入流 + * @param headerNum 标题行号，数据行号=标题行号+1 + * @param sheetIndex 工作表编号 + * @throws IOException 异常 + */ + public ImportExcel(String fileName, InputStream is, int headerNum, int sheetIndex) throws IOException { + if (StringUtils.isBlank(fileName)) { + throw new IOException("导入文档为空!"); + } else if (fileName.toLowerCase().endsWith("xls")) { + this.wb = new HSSFWorkbook(is); + } else if (fileName.toLowerCase().endsWith("xlsx")) { + this.wb = new XSSFWorkbook(is); + } else { + throw new IOException("文档格式不正确!"); + } + if (this.wb.getNumberOfSheets() < sheetIndex) { + throw new IOException("文档中没有工作表!"); + } + this.sheet = this.wb.getSheetAt(sheetIndex); + this.headerNum = headerNum; + LOG.debug("Initialize success."); + } + + /** + * 获取行对象 + * + * @param rownum 行号 + * @return 行 + */ + public Row getRow(int rownum) { + return this.sheet.getRow(rownum); + } + + /** + * 获取数据行号 + * + * @return 行号 + */ + public int getDataRowNum() { + return this.headerNum + 1; + } + + /** + * 获取最后一个数据行号 + * + * @return 行号 + */ + public int getLastDataRowNum() { + return this.sheet.getLastRowNum() + this.headerNum; + } + + /** + * 获取最后一个列号 + * + * @return 列号 + */ + public int getLastCellNum() { + return this.getRow(this.headerNum).getLastCellNum(); + } + + /** + * 获取单元格值 + * + * @param row 获取的行 + * @param column 获取单元格列号 + * @return 单元格值 + */ + public Object getCellValue(Row row, int column) { + Object val = ""; + Cell cell = row.getCell(column); + if (cell != null) { + switch (cell.getCellTypeEnum()) { + case NUMERIC: + val = cell.getNumericCellValue(); + break; + case STRING: + val = cell.getStringCellValue(); + break; + case FORMULA: + val = cell.getCellFormula(); + break; + case BOOLEAN: + val = cell.getBooleanCellValue(); + break; + case ERROR: + val = cell.getErrorCellValue(); + break; + default: + val = null; + break; + } + } + return val; + } +} diff --git a/udf-starter-id/src/main/test/java/org/itkk/udf/id/IdWorkerTest.java b/udf-starter-id/src/main/test/java/org/itkk/udf/id/IdWorkerTest.java index 08ad482..2bd7f3e 100644 --- a/udf-starter-id/src/main/test/java/org/itkk/udf/id/IdWorkerTest.java +++ b/udf-starter-id/src/main/test/java/org/itkk/udf/id/IdWorkerTest.java @@ -1,7 +1,8 @@ -package org.itkk.udf.id; +package java.org.itkk.udf.id; import lombok.extern.slf4j.Slf4j; import org.itkk.udf.core.constant.IdWorkerConstant; +import org.itkk.udf.id.IdWorker; import org.junit.Test; import java.text.SimpleDateFormat;