93 lines
3.9 KiB
Java
93 lines
3.9 KiB
Java
package com.adc.da.common;
|
|
|
|
import org.apache.commons.beanutils.BeanUtils;
|
|
import org.apache.poi.hssf.usermodel.HSSFCellStyle;
|
|
import org.apache.poi.ss.usermodel.*;
|
|
import org.apache.poi.ss.util.CellRangeAddress;
|
|
import org.apache.poi.xssf.streaming.SXSSFWorkbook;
|
|
|
|
import java.util.*;
|
|
|
|
public class TableToExcel {
|
|
|
|
public static <T extends Object> Workbook createExcel(Map<String,String> columnMap, List<T> dataList, String sheelName, String fileName) {
|
|
|
|
if(sheelName==null)sheelName="sheet1";
|
|
|
|
int sheetRow=500;
|
|
sheetRow=sheetRow<0?200:sheetRow;
|
|
//第一步，创建一个webbook，对应一个excel文件，内存中sheetRow条记录后提交。
|
|
Workbook wb=new SXSSFWorkbook(sheetRow);
|
|
if(columnMap!=null&&dataList!=null) {
|
|
//获取需要生成的sheet数，以防数据量过大报错
|
|
int sheetCount=getSheetCount(sheetRow,dataList.size());
|
|
//循环生成sheet
|
|
for(int i=0;i<sheetCount;i++) {
|
|
//第二步，在webbook中添加一个sheet，对应Excel文件中的sheet
|
|
Sheet sheet=wb.createSheet(sheelName);
|
|
//第三步，在sheet中添加表头第0行，注意老版本poi对Excel的行数有限制short
|
|
Row titleRow=sheet.createRow(0);
|
|
|
|
Set<String> keys=columnMap.keySet();
|
|
//第一行（标题行）合并并设置标题
|
|
|
|
|
|
//创建表头行
|
|
Row headRow=sheet.createRow(0);
|
|
//生成表头行
|
|
int j=0;
|
|
for(String key:keys) {
|
|
Cell hCell=headRow.createCell(j++);
|
|
hCell.setCellValue(columnMap.get(key));
|
|
}
|
|
//从第二行开始填充数据
|
|
int c=1;
|
|
for(int k=i*sheetRow;k<dataList.size()&&k<(i+1)*sheetRow;k++) {
|
|
T t=dataList.get(k);
|
|
Row dataRow=sheet.createRow(c++);
|
|
//循环取值进行数据填充
|
|
int v=0;
|
|
for(String key:keys) {
|
|
try {
|
|
Object val= BeanUtils.getProperty(t, key);
|
|
if(val instanceof Long) {
|
|
dataRow.createCell(v++).setCellValue((Long)val);
|
|
}else if(val instanceof Double){
|
|
dataRow.createCell(v++).setCellValue((Double)val);
|
|
}else if(val instanceof Date) {
|
|
dataRow.createCell(v++).setCellValue((Date)val);
|
|
}else if(val instanceof Calendar) {
|
|
dataRow.createCell(v++).setCellValue((Calendar)val);
|
|
}else if(val instanceof Boolean) {
|
|
dataRow.createCell(v++).setCellValue((Boolean)val);
|
|
}else if(val==null) {
|
|
dataRow.createCell(v++).setCellValue("");
|
|
}else {
|
|
dataRow.createCell(v++).setCellValue((String)val);
|
|
}
|
|
} catch (Exception e) {
|
|
System.out.println("导出Excel-表格赋值异常");
|
|
}
|
|
}
|
|
}
|
|
//设置列宽
|
|
// setColumnWidth(sheet,columnMap);
|
|
}
|
|
}
|
|
return wb;
|
|
}
|
|
public static void setColumnWidth(Sheet sheet, Map<String,String> columnMap) {
|
|
if(sheet!=null&&columnMap!=null) {
|
|
List<String> list=new ArrayList<String>(columnMap.keySet());
|
|
for(int i=0;i<list.size();i++) {
|
|
sheet.autoSizeColumn(i,true);
|
|
}
|
|
}
|
|
}
|
|
public static int getSheetCount(int sheelRow,int total) {
|
|
if(total==0||total<sheelRow)return 1;
|
|
return total%sheelRow>0?total/sheelRow+1:total/sheelRow;
|
|
}
|
|
|
|
}
|