package com.y9zb.web.util;
import com.y9zb.core.SplitSet;
import com.y9zb.core.TimeUtil;
import org.apache.poi.hssf.usermodel.*;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import javax.servlet.http.HttpServletResponse;
import java.io.OutputStream;
import java.io.UnsupportedEncodingException;
import java.lang.reflect.Field;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map.Entry;
/**
* 导出Excel
* @author Guang
*
*/
public class PoiExcelUtil {
private static final Logger logger =LoggerFactory.getLogger(PoiExcelUtil.class);
/**
* 导出Excel
* @param excelName 要导出的excel名称
* @param list 要导出的数据集合
* @param fieldMap 中英文字段对应Map,即要导出的excel表头
* @param response 使用response可以导出到浏览器
* @return
*/
public static <T> Boolean export(String excelName,List<T> list,LinkedHashMap, String> fieldMap,HttpServletResponse response){
// 设置默认文件名为当前时间:年月日时分秒
if (excelName==null || excelName=="") {
excelName = TimeUtil.getNoFormatToday()+TimeUtil.getNoFormatTime();
}
// 设置response头信息
response.reset();
response.setContentType("application/vnd.ms-excel"); // 改成输出excel文件
try {
response.setHeader("Content-disposition", "attachment; filename="
+new String((excelName+"_"+TimeUtil.getNoFormatToday()+TimeUtil.getNoFormatTime()).getBytes("gb2312"), "ISO-8859-1") + ".xls");
} catch (UnsupportedEncodingException e1) {
logger.info(e1.getMessage());
return false;
}
try {
//创建一个WorkBook,对应一个Excel文件
HSSFWorkbook wb=new HSSFWorkbook();
ListT>> lists= SplitSet.split(list, 30000);
for (int j = 0; j < lists.size(); j++) {
//在Workbook中,创建一个sheet,对应Excel中的工作薄(sheet)
HSSFSheet sheet=wb.createSheet(excelName+"_"+(j+1));
//给单子名称一个长度
sheet.setDefaultColumnWidth(18);
//创建单元格,并设置值表头 设置表头居中
HSSFCellStyle style=wb.createCellStyle();
//创建一个居中格式
style.setAlignment(HSSFCellStyle.ALIGN_CENTER);
// 填充工作表
fillSheet(sheet,lists.get(j),fieldMap,style);
}
//将文件输出
OutputStream ouputStream = response.getOutputStream();
wb.write(ouputStream);
ouputStream.flush();
ouputStream.close();
} catch (Exception e) {
logger.info("导出Excel失败!");
logger.error(e.getMessage());
return false;
}
return true;
}
/**
* 根据字段名获取字段对象
*
* @param fieldName
* 字段名
* @param clazz
* 包含该字段的类
* @return 字段
*/
public static Field getFieldByName(String fieldName, Class> clazz) {
logger.info("根据字段名获取字段对象:getFieldByName()");
// 拿到本类的所有字段
Field[] selfFields = clazz.getDeclaredFields();
// 如果本类中存在该字段,则返回
for (Field field : selfFields) {
//如果本类中存在该字段,则返回
if (field.getName().equals(fieldName)) {
return field;
}
}
// 否则,查看父类中是否存在此字段,如果有则返回
Class> superClazz = clazz.getSuperclass();
if (superClazz != null && superClazz != Object.class) {
//递归
return getFieldByName(fieldName, superClazz);
}
// 如果本类和父类都没有,则返回空
return null;
}
/**
* 根据字段名获取字段值
*
* @param fieldName 字段名
* @param o 对象
* @return 字段值
* @throws Exception 异常
*
*/
public static Object getFieldValueByName(String fieldName, Object o)
throws Exception {
logger.info("根据字段名获取字段值:getFieldValueByName()");
Object value = null;
//根据字段名得到字段对象
Field field = getFieldByName(fieldName, o.getClass());
//如果该字段存在,则取出该字段的值
if (field != null) {
field.setAccessible(true);//类中的成员变量为private,在类外边使用属性值,故必须进行此操作
value = field.get(o);//获取当前对象中当前Field的value
} else {
throw new Exception(o.getClass().getSimpleName() + "类不存在字段名 "
+ fieldName);
}
return value;
}
/**
* 根据带路径或不带路径的属性名获取属性值,即接受简单属性名,
* 如userName等,又接受带路径的属性名,如student.department.name等
*
* @param fieldNameSequence 带路径的属性名或简单属性名
* @param o 对象
* @return 属性值
* @throws Exception 异常
*
*/
public static Object getFieldValueByNameSequence(String fieldNameSequence,
Object o) throws Exception {
logger.info("根据带路径或不带路径的属性名获取属性值,即接受简单属性名:getFieldValueByNameSequence()");
Object value = null;
// 将fieldNameSequence进行拆分
String[] attributes = fieldNameSequence.split("\\.");
if (attributes.length == 1) {
value = getFieldValueByName(fieldNameSequence, o);
} else {
// 根据数组中第一个连接属性名获取连接属性对象,如student.department.name
Object fieldObj = getFieldValueByName(attributes[0], o);
//截取除第一个属性名之后的路径
String subFieldNameSequence = fieldNameSequence
.substring(fieldNameSequence.indexOf(".") + 1);
//递归得到最终的属性对象的值
value = getFieldValueByNameSequence(subFieldNameSequence, fieldObj);
}
return value;
}
/**
* 向工作表中填充数据
*
* @param sheet
* excel的工作表名称
* @param list
* 数据源
* @param fieldMap
* 中英文字段对应关系的Map
* @param style
* 表格中的格式
* @throws Exception
* 异常
*
*/
public static <T> void fillSheet(HSSFSheet sheet, List<T> list,
LinkedHashMap, String> fieldMap,HSSFCellStyle style) throws Exception {
logger.info("向工作表中填充数据:fillSheet()");
// 定义存放英文字段名和中文字段名的数组
String[] enFields = new String[fieldMap.size()];
String[] cnFields = new String[fieldMap.size()];
// 填充数组
int count = 0;
for (Entry, String> entry : fieldMap.entrySet()) {
enFields[count] = entry.getKey();
cnFields[count] = entry.getValue();
count++;
}
//在sheet中添加表头第0行,注意老版本poi对Excel的行数列数有限制short
HSSFRow row=sheet.createRow((int)0);
// 填充表头
for (int i = 0; i < cnFields.length; i++) {
HSSFCell cell=row.createCell(i);
cell.setCellValue(cnFields[i]);
cell.setCellStyle(style);
// sheet.autoSizeColumn(i);
}
// 填充内容
for (int index = 0; index < list.size(); index++) {
row = sheet.createRow(index + 1);
// 获取单个对象
T item = list.get(index);
for (int i = 0; i < enFields.length; i++) {
Object objValue = getFieldValueByNameSequence(enFields[i], item);
String fieldValue = objValue == null ? "" : objValue.toString();
row.createCell(i).setCellValue(fieldValue);
}
}
}
}
@RequestMapping(value = "query/excel/write")
public
@ResponseBody
HttpJsonResult queryExcel(HttpServletRequest request,
HttpServletResponse response, Map, Object> dataMap, String q_memberId, String q_memberName,
String q_memberPhone, String q_endTime, String q_startTime, String q_status) throws IOException {
HttpJsonResult jsonResult = new HttpJsonResult();
//queryMap--查询条件
Map, String> queryMap = new HashMap, String>();
queryMap.put("q_memberId", q_memberId);
queryMap.put("q_memberName", q_memberName);
queryMap.put("q_memberPhone", q_memberPhone);
queryMap.put("q_startTime", q_startTime);
queryMap.put("q_endTime", q_endTime);
queryMap.put("q_status", q_status);
queryMap.put("q_exportStatus", "0");
//count---共有多少条数据需要导出
Integer count = pickingCardService.count(queryMap);
//index---分批查询
int index = 0;
if (count % 10000 == 0) {
index = count / 10000;
} else {
index = count / 10000 + 1;
}
List pickingCardList = new ArrayList<>();
for (int i = 0; i < index; i++) {
PagerInfo pager = new PagerInfo();
pager.setPageIndex(i + 1);
pager.setPageSize(10000);
pager.setRowsCount(count);
ServiceResult> serviceResult = pickingCardService.page(queryMap, pager);
pickingCardList.addAll(serviceResult.getResult());
}
//定义导出excel的名字
String excelName = "提货券查询结果表";
// 获取需要转出的excle表头的map字段
LinkedHashMap, String> fieldMap = new LinkedHashMap, String>();
fieldMap.put("memberId", "用户ID");
fieldMap.put("memberName", "用户名");
fieldMap.put("memberPhone", "手机号");
fieldMap.put("createTimeString", "获得时间");
fieldMap.put("statusString", "使用状态");
for (int i = 0; i < pickingCardList.size(); i++) {
pickingCardList.get(i).setExportStatus(1);
pickingCardList.get(i).setCreateTimeString(TimeUtil.getDateTimeString(pickingCardList.get(i).getCreateTime()));
if (pickingCardList.get(i).getStatus() == 0) {
pickingCardList.get(i).setStatusString("已使用");
} else {
pickingCardList.get(i).setStatusString("可用");
}
}
//导出
Boolean is = PoiExcelUtil.export(excelName, pickingCardList, fieldMap, response);
if (is) {
jsonResult.setRows(true);
jsonResult.setMessage("导出成功");
pickingCardService.updateBatch(pickingCardList);
} else {
jsonResult.setRows(false);
jsonResult.setMessage("导出失败");
}
return jsonResult;
}