package com.ay.voipwork.until.excel;
import com.alibaba.fastjson.JSONArray;
import com.alibaba.fastjson.JSONObject;
import com.ay.voipwork.Exception.ExportException;
import org.apache.poi.hssf.usermodel.HSSFWorkbook;
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.streaming.SXSSFWorkbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import java.io.FileOutputStream;
import java.io.IOException;
import java.net.URLEncoder;
import java.util.Map;
import java.util.Set;
import java.util.function.Consumer;
//使用 直接 ExcelUntil.createExcel( response, EsConstant.headers, array );即可
public class ExcelUntil {
private static Logger log = LoggerFactory.getLogger( ExcelUntil.class );
private static final String EXTENSION_XLS = "xls";
private static final String EXTENSION_XLSX = "xlsx";
//headers 是表格头 jsonArray 是要导出的数据 这部分我的逻辑是将查询结果放在内存中,然后一次
//性导出
public static void createExcel(HttpServletResponse response, String[] heads, JSONArray jsonArray) {
try {
response.setContentType( "application/vnd.ms-excel" );
response.setHeader( "Content-Disposition", "attachment;filename=" + URLEncoder.encode( "phone.xlsx", "utf-8" ) );
Workbook workbook = getWorkbook("phone.xlsx");// 创建一个Excel文件
Sheet sheet = workbook.createSheet();// 创建一个Excel的Sheet
Row titleRow = sheet.createRow( 0 );//创建表头
for (int i = 0; i < heads.length; i++) {//表头赋值
titleRow.createCell( i ).setCellValue( heads[i] );
}
if (jsonArray.size() > 0) {//单元格赋值
for (int i = 0; i < jsonArray.size(); i++) {
Row row = sheet.createRow( i + 1 );
JSONObject json = jsonArray.getJSONObject( i ); // 遍历 jsonarray
for (int j = 0; j < heads.length; j++) {
row.createCell( j ).setCellValue( String.valueOf(json.get( heads[j] ) ));//赋值
}
}
}
ServletOutputStream outputStream = response.getOutputStream();
workbook.write( outputStream );
outputStream.flush();
outputStream.close();
workbook.close();
} catch (Exception e) {
e.printStackTrace();
throw new ExportException( "导出异常" );
}
}
/***
* <pre>
* 取得Workbook对象(xls和xlsx对象不同,不过都是Workbook的实现类)
* xls:HSSFWorkbook
* xlsx:XSSFWorkbook
* @param filename
* @return
* @throws IOException
* </pre>
*/
public static Workbook getWorkbook(String filename) {
Workbook workbook = null;
if (filename.endsWith( EXTENSION_XLS )) {
workbook = new HSSFWorkbook();
} else if (filename.endsWith( EXTENSION_XLSX )) {
XSSFWorkbook xssfWorkbook = new XSSFWorkbook();
workbook = new SXSSFWorkbook(xssfWorkbook, 1000);
}
return workbook;
}
}
总结: 这个小工具会直接在浏览器生成一个excel,保存到本地即可。
选哟导入excel处理的mvn依赖
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>4.1.1</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>4.1.1</version>
</dependency>