引言
文件的下载是一个非常常见的功能，也有一些非常好的框架可以使用，这里我们就介绍一种比较常见的场景，下载Excel模版，导入功能通常会配有一个模版下载的功能，根据下载的模版，填充数据然后再上传。
需求
如图所示用户下载如下模版，模版中的地区是从数据库中读出来的。
分析
如果模版都是固定的内容，这种比较简单，我们可以直接把模版上传到我们的文件服务器，用户下载的时候可以直接在文件服务器上下载，或者是把文件放到我们的工程里面，直接读取文件然后以流的方式返回给前端，用户直接下载。
这里比较麻烦点的就是这个模版的内容不是固定的，需要在库里查询出来，写到excel然后在返给前端。同时这个文件又有一定的样式，一般有样式的文件，我们可以根据模版填充，而不是把说有的样式都用程序代码写，程序代码写这种样式会很麻烦。所以我们这里用的方式就是填充模版
准备模版
把准备好的模版放到我们工程里面，创建文件夹template 并把文件夹设置为resource,设置为resoucre后该文件夹下面的文件在mvn打包的时候回合我们的java类打在同一个路径下，方便我们获取文件
编写下拉选项的handler
public class SelectSheetWriteHandler implements SheetWriteHandler {
private String[] list;
private int firstRow;
private int lastRow;
private int firstCol;
private int lastCol;
public SelectSheetWriteHandler(String[] list, int firstRow, int lastRow, int firstCol, int lastCol) {
this.list = list;
this.firstRow = firstRow;
this.lastRow = lastRow;
this.firstCol = firstCol;
this.lastCol = lastCol;
}
@Override
public void beforeSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
}
@Override
public void afterSheetCreate(WriteWorkbookHolder writeWorkbookHolder, WriteSheetHolder writeSheetHolder) {
// 区间设置 第一列第一行和第二行的数据。由于第一行是头，所以第一、二行的数据实际上是第二三行
CellRangeAddressList cellRangeAddressList = new CellRangeAddressList(firstRow, lastRow, firstCol, lastCol);
DataValidationHelper helper = writeSheetHolder.getSheet().getDataValidationHelper();
DataValidationConstraint constraint = helper.createExplicitListConstraint(list);
DataValidation dataValidation = helper.createValidation(constraint, cellRangeAddressList);
dataValidation.setShowErrorBox(true);
writeSheetHolder.getSheet().addValidationData(dataValidation);
}
}
下载的方法
public void export(HttpServletResponse response) throws IOException {
//模板文件
InputStream templateFile = TenantInsuranceFundServiceImpl.class.getResourceAsStream("/五险一金导入模板.xlsx");
//导出后的文件名
String fileName = "五险一金导入模板";
List<BaseDictionaryVO> baseDictionaryVOS = dictionaryService.queryDictionaryDetail(DictionaryEnum.AREA.getCode(), DictionaryEnum.AREA.getName());
Set<String> areaNameList = new HashSet<>();
if(!CollectionUtils.isEmpty(baseDictionaryVOS)){
areaNameList = baseDictionaryVOS.stream().map(BaseDictionaryVO::getName).collect(Collectors.toSet());
}
SelectSheetWriteHandler selectSheetWriteHandler = new SelectSheetWriteHandler(areaNameList.toArray(new String[areaNameList.size()]), 2, 5000, 0, 0);
//写入
ExcelWriter excelWriter = null;
try {
//流输出
excelWriter = EasyExcel.write(response.getOutputStream())
.withTemplate(templateFile)
.registerWriteHandler(selectSheetWriteHandler)
.build();
WriteSheet writeSheet = EasyExcel.writerSheet().build();
// 直接写入list数据 这里没有数据需要填写直接一个空集合
excelWriter.fill(Arrays.asList(), writeSheet);
//浏览器下载操作
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
response.setCharacterEncoding("utf-8");
// 这里URLEncoder.encode可以防止中文乱码
fileName = URLEncoder.encode(fileName, "UTF-8").replaceAll("\\+", "%20");
response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
excelWriter.finish();
} catch (Exception e) {
// 重置response
response.reset();
response.setContentType("application/json");
response.setCharacterEncoding("utf-8");
Map<String, String> map = new HashMap<String, String>();
map.put("code", "1");
map.put("message", "下载模板失败" + e.getMessage());
response.getWriter().println(JSON.toJSONString(map));
} finally {
// 千万别忘记关闭流
if (excelWriter != null) {
excelWriter.finish();
}
}
}
