控制层接口类
组长写的,拿来方便以后
package com.fx.bis.dx.app.plateform.controller.io;
import java.io.IOException;
import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletResponse;
import org.apache.commons.lang3.StringUtils;
import org.springframework.beans.BeansException;
import org.springframework.context.ApplicationContext;
import org.springframework.context.ApplicationContextAware;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import com.alibaba.fastjson.JSON;
import com.fx.bis.dx.app.plateform.constant.Module;
import com.fx.bis.dx.app.plateform.proxy.Excel;
import com.fx.bis.dx.app.plateform.proxy.ExcelProxy;
import com.fx.bis.dx.common.constant.Const;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import cn.hutool.core.io.IoUtil;
import cn.hutool.poi.excel.ExcelUtil;
import cn.hutool.poi.excel.ExcelWriter;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiParam;
/**
*Excels 操作控制器
*
*@author: wangyi
*
*/
@Controller
@Api(tags = "分布式文件上传控制器")
@RequestMapping(Const.API_AUTH_PREFIX + Module.API_IO_EXCEL)
public class ExcelsController implements ApplicationContextAware{
private static ApplicationContext applicationContext;
/**
* <p>
* columns:
* [
* {"prop":"gysName","width":200,"fixed":false,"label":"供应商名称"},
* {"prop":"gysLongName","width":300,"fixed":false,"label":"供应商全称"},
* {"prop":"status","width":100,"fixed":true,"label":"状态"}
* ]
*
* </p>
* @throws ClassNotFoundException
* @throws IllegalAccessException
* @throws InstantiationException
* @throws IOException
*/
@GetMapping("/dxExporter")
@ApiOperation(value = "导出Excels", produces = "application/octet-stream")
public void dxExporter(
@ApiParam(name = "actuator", value="导出执行器: ①面料导出:mlBaseExportImpl")@RequestParam(value = "actuator", required = true) String actuator,
@ApiParam(name = "condition", value="查询条件json")@RequestParam(value = "condition", required = false) String condition,
@ApiParam(name = "columns", value="导出列json,参数只能是prop,width,fixed,label, 其中pro,label必输,如[{\"prop\":\"mlNo\",\"label\":\"面料号\"},{\"prop\":\"brand\",\"label\":\"品牌\"}]") @RequestParam(value = "columns", required = true) String columns,
@ApiParam(name = "fileName", value="文件名称")@RequestParam(value = "fileName", required = true) String fileName,
@ApiParam(name = "title", value="标题合并单元格名称")@RequestParam(value = "title", required = false) String title,
@ApiParam(name = "xlsx", value="是否创建xlsx格式,默认xls格式")@RequestParam(value = "xlsx", required = true) boolean xlsx,
HttpServletResponse response
) {
if (StringUtils.isEmpty(actuator) || StringUtils.isEmpty(columns)) {
return;
}
ExcelWriter writer = null;
ServletOutputStream out = null;
try {
// 查询条件
Map<String, Object> conditionMap = JSON.parseObject(condition, Map.class);
// 自定义导出列
Map<String, String> columnsMap = Maps.newHashMap();
List<Map> parseArray = JSON.parseArray(columns, Map.class);
for (Map m : parseArray) {
Object K = m.get("prop");
Object V = m.get("label");
if (K == null || V == null) {
throw new RuntimeException("缺少prop和label参数");
}
columnsMap.put(K.toString(), V.toString());
}
// 代理执行获取封装数据
// Class<?> clazz = Class.forName(actuator);
// Excel proxy = new ExcelProxy(clazz.newInstance()).getProxy();
Object bean = getBean(actuator);
Excel proxy = new ExcelProxy(bean).getProxy();
List<Map<String, Object>> list = proxy.exportExcel(conditionMap);
// 数据过滤
List<Map<String, Object>> exList = Lists.newArrayList();
for (Map<String, Object> item : list) {
Map<String, Object> n = new LinkedHashMap<>();
for (Map.Entry<String, Object> entry : item.entrySet()) {
String key = entry.getKey();
Object value = entry.getValue();
if (columnsMap.containsKey(key)) {
n.put(columnsMap.get(key), value); // 属性名称作为K
}
}
exList.add(n);
}
// 倒序排序
// Collections.reverse(exList);
// 导出Excel
if (xlsx) {
writer = ExcelUtil.getWriter(true);
} else {
writer = ExcelUtil.getWriter();
}
// 合并标题
if (title != null && !title.equals("") && !title.equals("null")) {
writer.merge(columnsMap.size() -1 , title);
}
writer.write(exList, true); // 一次性写入
if (xlsx) {
fileName = fileName + ".xlsx";
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
response.setHeader("Content-Disposition","attachment;filename=" + fileName);
} else {
fileName = fileName + ".xls";
//response为HttpServletResponse对象
response.setContentType("application/vnd.ms-excel;charset=utf-8");
//是弹出下载对话框的文件名,不能为中文,中文请自行编码
response.setHeader("Content-Disposition","attachment;filename=" + fileName);
}
out = response.getOutputStream();
} catch (Exception e) {
e.printStackTrace();
} finally {
writer.flush(out, true);
// 关闭writer,释放内存
writer.close();
//此处记得关闭输出Servlet流
IoUtil.close(out);
}
}
@Override
public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
this.applicationContext = applicationContext;
}
public static ApplicationContext getApplicationContext() {
return applicationContext;
}
public static Object getBean(String name) {
return getApplicationContext().getBean(name);
}
}
另外两个相关的类
package com.fx.bis.dx.app.plateform.proxy;
import java.util.List;
import java.util.Map;
public interface Excel {
public List<Map<String, Object>> exportExcel(Map<String, Object> condition);
}
package com.fx.bis.dx.app.plateform.proxy;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
/**
* Excel Jdk Proxy
*
* @author: wangyi
*
*/
public class ExcelProxy implements InvocationHandler{
private Object target;
public ExcelProxy(Object target) {
this.target = target;
}
@SuppressWarnings("unchecked")
public <T> T getProxy() {
return (T) Proxy.newProxyInstance(target.getClass().getClassLoader(), target.getClass().getInterfaces(), this);
}
@Override
public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
return method.invoke(target, args);
}
}