Alawn/udf
1
0
Fork 0
mirror of https://gitee.com/wangkang/udf.git synced 2026-10-27 11:33:45 +00:00
Code Issues Releases Wiki Activity

添加excel相关工具类

Browse Source
This commit is contained in:
udf_admin@163.com
2017-12-17 16:12:01 +08:00
parent e33684e2a4
commit 489c190434
6 changed files with 621 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pom.xml
+1
View File
@@ -28,6 +28,7 @@
<mybatis-spring-boot-starter.version>1.3.1</mybatis-spring-boot-starter.version>
<jsqlparser.version>1.1</jsqlparser.version>
<aliyun-sdk-oss.version>2.8.2</aliyun-sdk-oss.version>
<poi.version>3.17</poi.version>
</properties>
<!--子模块 -->
<modules>
udf-starter-core/pom.xml
+12
View File
@@ -97,6 +97,18 @@
<artifactId>commons-lang3</artifactId>
<version>${commons-lang3.version}</version>
</dependency>
<!-- POI -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>${poi.version}</version>
</dependency>
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>${poi.version}</version>
</dependency>
<!-- spring-cloud-starter-config 分布式配置-客户端 -->
<dependency>
<groupId>org.springframework.cloud</groupId>
udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcel.java
+363
View File
@@ -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;
/**
* <p>
* ClassName: ExportExcel
* </p>
* <p>
* Description: 导出excel(2007+)
* </p>
* <p>
* Author: Administrator
* </p>
* <p>
* Date: 2016年3月23日
* </p>
*/
public class ExportExcel {
/**
* 日志
*/
private static Logger log = LoggerFactory.getLogger(ExportExcel.class);
/**
* 字体
*/
private static final String FONT_NAME = "Arial";
/**
* <p>
* Field DATA: 样式名称
* </p>
*/
private static final String STYLE_DATA = "data";
/**
* 工作薄对象
*/
private SXSSFWorkbook wb;
/**
* 工作表对象
*/
private SXSSFSheet sheet;
/**
* 样式列表
*/
private Map<String, CellStyle> styles;
/**
* 当前行号
*/
private int rownum;
/**
* 构造函数
*
* @param title 表格标题，传“空值”，表示无标题
* @param headerList 表头列表
* @param rowAccessWindowSize 刷新行数
* @throws SystemRuntimeException
*/
public ExportExcel(String title, List<ExportExcelHeaderMateData> headerList, int rowAccessWindowSize) {
initialize(title, headerList, rowAccessWindowSize);
}
/**
* 初始化函数
*
* @param title 表格标题，传“空值”，表示无标题
* @param headerList 表头列表
* @param rowAccessWindowSize 刷新行数
* @throws SystemRuntimeException
*/
private void initialize(String title, List<ExportExcelHeaderMateData> 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<String, CellStyle> createStyles() {
final int num10 = 10;
final int num16 = 16;
Map<String, CellStyle> 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);
}
/**
* <p>
* Description: 添加一个单元格
* </p>
*
* @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;
}
/**
* <p>
* Description: 合并单元格，行列位置都是从0开始
* </p>
*
* @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");
}
}
udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ExportExcelHeaderMateData.java
+34
View File
@@ -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;
}
}
udf-starter-core/src/main/java/org/itkk/udf/core/utils/excel/ImportExcel.java
+209
View File
@@ -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;
}
}
udf-starter-id/src/main/test/java/org/itkk/udf/id/IdWorkerTest.java
+2 -1
View File
@@ -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;