Excel表格导出工具类
import org.apache.poi.hssf.usermodel.*;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.springframework.http.HttpStatus;
import org.springframework.http.ResponseEntity;
import org.springframework.web.multipart.MultipartFile;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
public class ExcelUtil {
public static Workbook getWorkbook(MultipartFile file) throws IOException {
String fileName = file.getOriginalFilename();
if (!fileName.matches("^.+\\.(?i)(xls)$") && !fileName.matches("^.+\\.(?i)(xlsx)$")) {
return null;
}
boolean isExcel2003 = true;
if (fileName.matches("^.+\\.(?i)(xlsx)$")) {
isExcel2003 = false;
}
InputStream input = file.getInputStream();
Workbook work = null;
if (isExcel2003) {
work = new HSSFWorkbook(input);
} else {
work = new XSSFWorkbook(input);
}
return work;
/**
*
* Sheet sheet = work.getSheetAt(0);//得到sheet
* int colLength = row.getLastCellNum();//得到总列数
* for(int i=2;i<=sheet.getLastRowNum();i++){//遍历行
* Cell cell = row.getCell(j);//当前单元格对象,根据行和列确定单元格
* String value = (cell == null) == true ? null : cell.getStringCellValue().trim();//得到字符串
* Date data = cell.getDateCellValue();//得到时间
* }
*/
}
/**
* Excel表格导出
* @param response HttpServletResponse对象
* @param excelData Excel表格的数据,封装为List<List<String>>
* @param sheetName sheet的名字
* @param fileName 导出Excel的文件名
* @param columnWidth Excel表格的宽度,建议为15
* @throws IOException 抛IO异常
*/
public static void exportExcel(HttpServletResponse response,
List<List<String>> excelData,
String sheetName,
String fileName,
int columnWidth) throws IOException {
//声明一个工作簿
HSSFWorkbook workbook = new HSSFWorkbook();
//生成一个表格,设置表格名称
HSSFSheet sheet = workbook.createSheet(sheetName);
//设置表格列宽度
sheet.setDefaultColumnWidth(columnWidth);
//写入List<List<String>>中的数据
int rowIndex = 0;
for(List<String> data : excelData){
//创建一个row行,然后自增1
HSSFRow row = sheet.createRow(rowIndex++);
//遍历添加本行数据
for (int i = 0; i < data.size(); i++) {
//创建一个单元格
HSSFCell cell = row.createCell(i);
//创建一个内容对象
HSSFRichTextString text = new HSSFRichTextString(data.get(i));
//将内容对象的文字内容写入到单元格中
cell.setCellValue(text);
}
}
//准备将Excel的输出流通过response输出到页面下载
//八进制输出流
response.setContentType("application/octet-stream");
//设置导出Excel的名称
response.setHeader("Content-disposition", "attachment;filename=" + fileName);
try {
//刷新缓冲
response.flushBuffer();
//workbook将Excel写入到response的输出流中,供页面下载该Excel文件
workbook.write(response.getOutputStream());
} catch (IOException e) {
e.printStackTrace();
}
// //关闭workbook
// workbook.close();
}
}
控制层
PageMerchantDTO 封装了请求参数,分页和一些其他的参数,去掉也可以,直接全查
@ApiOperation("导出")
@GetMapping("/auth/export")
public void export(PageMerchantDTO merchantDTO, HttpServletResponse response) throws IOException {
//返回结果集
List<List<String>> result = merchantService.export(merchantDTO);
//获取时间的工具自己生产就行了
String date = DateUtil.getCurrentDate(DateUtil.DATE_FORMAT_YYYY_MM_DD);
String sheetName = "XXX列表";
String fileName = "XXXX列表"+date+".xls";
ExcelUtil.exportExcel(response, result, sheetName, fileName, 15);
}
接口
List<List<String>> export(PageMerchantDTO merchantDTO);
service层
@Override
public List<List<String>> export(PageMerchantDTO merchantDTO) {
List<List<String>> strList=new ArrayList<>();
//头部也是一个list
List<String> head = new ArrayList<>();
head.add("名称");
head.add("项目");
head.add("联系人");
head.add("入驻时间");
head.add("到期时间");
head.add("状态");
//放到第一行显示,下面的数据以此类推
strList.add(head);
//这个方法就是从数据库查询出来的一个分页list 我这是调用了一个分页的数据,根据项目本身的业务逻辑写就可以了,不用管我这里的方法
PageResult page = findPage(merchantDTO);
if(page.getContent().size()!=0){
//强转了对象,可以忽略
List<Merchant> contentList = (List<Merchant>) page.getContent();
for (int i = 0; i <contentList.size() ; i++) {
//给表格进行赋值
List<String> data = new ArrayList<>();
data.add(contentList.get(i).getXXX());
data.add(contentList.get(i).getXXX());
data.add(contentList.get(i).getXXX());
data.add(contentList.get(i).getXXX());
data.add(contentList.get(i).getXXX());
data.add(contentList.get(i).getXXX());
strList.add(data);
}
}
return strList;
}