package com.example.demo.utils;
import com.example.demo.entity.ExcelData;
import com.example.demo.entity.User;
import lombok.extern.slf4j.Slf4j;
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.*;
import javax.servlet.http.HttpServletResponse;
import java.io.BufferedOutputStream;
import java.io.FileInputStream;
import java.io.InputStream;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import static org.apache.poi.ss.usermodel.CellType.*;
/**
* 路径：com.example.demo.utils
* 类名：
* 功能：导入导出
* 备注：
* 创建人：typ
* 创建时间：2018/10/19 11:21
* 修改人：
* 修改备注：
* 修改时间：
*/
@Slf4j
public class ExcelUtil {
/**
* 方法名：exportExcel
* 功能：导出Excel
* 描述：
* 创建人：typ
* 创建时间：2018/10/19 16:00
* 修改人：
* 修改描述：
* 修改时间：
*/
public static void exportExcel(HttpServletResponse response, ExcelData data) {
log.info("导出解析开始，fileName:{}",data.getFileName());
try {
//实例化HSSFWorkbook
HSSFWorkbook workbook = new HSSFWorkbook();
//创建一个Excel表单，参数为sheet的名字
HSSFSheet sheet = workbook.createSheet("sheet");
//设置表头
setTitle(workbook, sheet, data.getHead());
//设置单元格并赋值
setData(sheet, data.getData());
//设置浏览器下载
setBrowser(response, workbook, data.getFileName());
log.info("导出解析成功!");
} catch (Exception e) {
log.info("导出解析失败!");
e.printStackTrace();
}
}
/**
* 方法名：setTitle
* 功能：设置表头
* 描述：
* 创建人：typ
* 创建时间：2018/10/19 10:20
* 修改人：
* 修改描述：
* 修改时间：
*/
private static void setTitle(HSSFWorkbook workbook, HSSFSheet sheet, String[] str) {
try {
HSSFRow row = sheet.createRow(0);
//设置列宽，setColumnWidth的第二个参数要乘以256，这个参数的单位是1/256个字符宽度
for (int i = 0; i <= str.length; i++) {
sheet.setColumnWidth(i, 15 * 256);
}
//设置为居中加粗,格式化时间格式
HSSFCellStyle style = workbook.createCellStyle();
HSSFFont font = workbook.createFont();
font.setBold(true);
style.setFont(font);
style.setDataFormat(HSSFDataFormat.getBuiltinFormat("m/d/yy h:mm"));
//创建表头名称
HSSFCell cell;
for (int j = 0; j < str.length; j++) {
cell = row.createCell(j);
cell.setCellValue(str[j]);
cell.setCellStyle(style);
}
} catch (Exception e) {
log.info("导出时设置表头失败！");
e.printStackTrace();
}
}
/**
* 方法名：setData
* 功能：表格赋值
* 描述：
* 创建人：typ
* 创建时间：2018/10/19 16:11
* 修改人：
* 修改描述：
* 修改时间：
*/
private static void setData(HSSFSheet sheet, List<String[]> data) {
try{
int rowNum = 1;
for (int i = 0; i < data.size(); i++) {
HSSFRow row = sheet.createRow(rowNum);
for (int j = 0; j < data.get(i).length; j++) {
row.createCell(j).setCellValue(data.get(i)[j]);
}
rowNum++;
}
log.info("表格赋值成功！");
}catch (Exception e){
log.info("表格赋值失败！");
e.printStackTrace();
}
}
/**
* 方法名：setBrowser
* 功能：使用浏览器下载
* 描述：
* 创建人：typ
* 创建时间：2018/10/19 16:20
* 修改人：
* 修改描述：
* 修改时间：
*/
private static void setBrowser(HttpServletResponse response, HSSFWorkbook workbook, String fileName) {
try {
//清空response
response.reset();
//设置response的Header
response.addHeader("Content-Disposition", "attachment;filename=" + fileName);
OutputStream os = new BufferedOutputStream(response.getOutputStream());
response.setContentType("application/vnd.ms-excel;charset=gb2312");
//将excel写入到输出流中
workbook.write(os);
os.flush();
os.close();
log.info("设置浏览器下载成功！");
} catch (Exception e) {
log.info("设置浏览器下载失败！");
e.printStackTrace();
}
}
/**
* 方法名：importExcel
* 功能：导入
* 描述：
* 创建人：typ
* 创建时间：2018/10/19 11:45
* 修改人：
* 修改描述：
* 修改时间：
*/
public static List<Object[]> importExcel(String fileName) {
log.info("导入解析开始，fileName:{}",fileName);
try {
List<Object[]> list = new ArrayList<>();
InputStream inputStream = new FileInputStream(fileName);
Workbook workbook = WorkbookFactory.create(inputStream);
Sheet sheet = workbook.getSheetAt(0);
//获取sheet的行数
int rows = sheet.getPhysicalNumberOfRows();
for (int i = 0; i < rows; i++) {
//过滤表头行
if (i == 0) {
continue;
}
//获取当前行的数据
Row row = sheet.getRow(i);
Object[] objects = new Object[row.getPhysicalNumberOfCells()];
int index = 0;
for (Cell cell : row) {
if (cell.getCellType().equals(NUMERIC)) {
objects[index] = (int) cell.getNumericCellValue();
}
if (cell.getCellType().equals(STRING)) {
objects[index] = cell.getStringCellValue();
}
if (cell.getCellType().equals(BOOLEAN)) {
objects[index] = cell.getBooleanCellValue();
}
if (cell.getCellType().equals(ERROR)) {
objects[index] = cell.getErrorCellValue();
}
index++;
}
list.add(objects);
}
log.info("导入文件解析成功！");
return list;
}catch (Exception e){
log.info("导入文件解析失败！");
e.printStackTrace();
}
return null;
}
//测试导入
public static void main(String[] args) {
try {
String fileName = "f:/test.xlsx";
List<Object[]> list = importExcel(fileName);
for (int i = 0; i < list.size(); i++) {
User user = new User();
user.setId((Integer) list.get(i)[0]);
user.setUsername((String) list.get(i)[1]);
user.setPassword((String) list.get(i)[2]);
user.setEnable((Integer) list.get(i)[3]);
System.out.println(user.toString());
}
} catch (Exception e) {
e.printStackTrace();
}
}
}
Springboot+Poi实现Excel的导入导出
5星 · 超过95%的资源 需积分: 45 70 浏览量
2018-10-22
22:04:40
上传
评论 29
收藏 61KB ZIP 举报 
springboot-excel.zip （17个子文件） 
springboot-excel 
.gitignore 268B .mvn wrapper 
maven-wrapper.jar 46KB maven-wrapper.properties 110B pom.xml 3KB mvnw 6KB src test java com example demo SpringbootExcelApplicationTests.java 621B main resources application.yml 625B java com example demo utils ExcelUtil.java 8KB SpringbootExcelApplication.java 326B controller ExcelController.java 2KB mapper UserMapper.java 777B service ExcelService.java 509B impl ExcelServiceImpl.java 3KB entity User.java 435B ExcelData.java 544B mvnw.cmd 5KB springboot-excel.iml 11KB
- 1
- 2
- 3
前往页