【文件导出2】导出html文件数据

导出html文件数据

文章目录

• 导出html文件数据
• 前言
• 一、实现代码
• • 1.controller层
  • 2.接口层
  • 3.接口实现类
  • 4.FileUtil 工具类
• 二、文件导出效果
• 总结

前言

springBoot项目实现在线导出html文件数据的功能。

一、实现代码

1.controller层

@GetMapping("/record/_export")
@ApiOperation("查询记录导出")
public void recordExport(HttpServletResponse response, RecordSearchRequest query) throws IOException {
        ruleVerifyResultService.recordExport(response, query);
    }

RecordSearchRequest为请求查询条件，根据条件导出对应的记录。

2.接口层

public interface ResultService extends IService<RuleVerifyResultEntity> {
 /**
     * 记录导出
     * @param query
     * @return
     */
    void recordExport(HttpServletResponse response, RecordSearchRequest query) throws IOException;
}

3.接口实现类

@Override
public void queryRecordExport(HttpServletResponse response, RuleVerifyRecordSearchRequest query) throws IOException {
LocalDateTime startExecuteTime = null;
LocalDateTime endExecuteTime = null;
//设置导出文件名称
String fileName = "检验记录列表数据";
if (ObjectUtils.isNotEmpty(query.getExecuteStartTime())) {
startExecuteTime = query.getExecuteStartTime().atStartOfDay();
endExecuteTime = query.getExecuteStartTime().atTime(23, 59, 59);
}
//查询需要导出的数据，用list集合接收
List<RuleVerifyRecordEntity> recordList = ruleVerifyRecordMapper.selectList(Wrappers.<RuleVerifyRecordEntity>lambdaQuery().
// 关键字搜索
and(StringUtils.isNotEmpty(query.getKeywords()), wrapper -> 
wrapper.like(RuleVerifyRecordEntity::getTableName, query.getKeywords()).
or().like(RuleVerifyRecordEntity::getTableRemark, query.getKeywords()).
or().like(RuleVerifyRecordEntity::getDatasourceName, query.getKeywords())
)
orderByDesc(RuleVerifyRecordEntity::getExecuteStartTime)
);
//2.导出html数据
if (ObjectUtils.equals(ExportTypeEnum.HTML.getValue(), query.getExportType())) {
//构建文件信息
StringBuilder table = this.buildTableInfo(recordList);
InputStream inputStream = null;
FileInputStream fileInputStream = null;
ServletOutputStream outputStream = null;
try {
//string字符串转输入流
inputStream = FileUtil.getInputStream(table.toString());
//字节输入流转文件输入流
fileInputStream = FileUtil.convertToFileInputStream(inputStream);
//给浏览器设置响应头：Content-Disposition告诉浏览器以附件的形式打开这个文件
response.setHeader("Content-Disposition", "attachment;filename=" + URLEncoder.encode(fileName + ".html", "UTF-8"));
//给浏览器设置响应头：文件类型
response.setContentType("application/octet-stream;charset=utf-8");
//从response中获取输出流
outputStream = response.getOutputStream();
//进行流拷贝操作
FileUtil.copyStream(fileInputStream, outputStream);
outputStream.flush();
} catch (IOException e) {
log.error(e.getMessage());
e.printStackTrace();
} finally {
//关闭资源文件
try{
inputStream.close();
fileInputStream.close();
outputStream.close();
}catch(Exception e){
e.printStackTrace();
}
}
}
private StringBuilder buildTableInfo(List<RuleVerifyRecordEntity> recordList){
StringBuilder table = new StringBuilder();
// 添加table标签
table.append("
");// 生成表头
table.append("
");
table.append("");
table.append("");
table.append("");
table.append("");
table.append("");
table.append("
");
table.append("");
table.append("
");// 生成HTML表格// 生成表格数据for(RuleVerifyRecordEntity row : recordList){
table.append("
");
table.append("
");
table.append("
");
table.append("
");
table.append("
");
table.append("
");
table.append("
");
table.append("
");
table.append("
");}
table.append("

" + "表名称" + "
" + "规则异常" + "
" + "执行状态" + "
" + "调度信息" + "
" + "开始时间" + "
" + " " + "
" + "结束时间" + "


  
" + row.getTableName() + "
" + row.getRuleCount()) + "
" + (ObjectUtils.isNotEmpty(row.getExecuteStatus()) ? row.getExecuteStatus().toString() : null) + "
" + row.getSchedulerInfo() + " " + "
" + DateTimeUtil.toDateTimeStr(ObjectUtils.isNotEmpty(row.getExecuteStatus()) ? row.getExecuteStartTime() : LocalDateTime.now()) + "
" + " " + "
" + DateTimeUtil.toDateTimeStr(ObjectUtils.isNotEmpty(row.getExecuteEndTime()) ? row.getExecuteEndTime() : LocalDateTime.now()) + "
");
return table;
}
}

4.FileUtil 工具类

import javax.servlet.ServletOutputStream;
import java.io.*;
import java.nio.charset.StandardCharsets;
public class FileUtil {
//string字符串转输入流
public static InputStream getInputStream(String str) {
if (str != null && !"".equals(str.trim())) {
try {
ByteArrayInputStream inputStream = new ByteArrayInputStream(str.getBytes(StandardCharsets.UTF_8));
return inputStream;
} catch (Exception e) {
e.printStackTrace();
}
}
return null;
}
//字节输入流转文件输入流
public static FileInputStream convertToFileInputStream(InputStream inputStream) throws IOException {
File tempFile = File.createTempFile("temp", ".tmp");
tempFile.deleteOnExit();
try (FileOutputStream outputStream = new FileOutputStream(tempFile)) {
byte[] buffer = new byte[1024];
int bytesRead;
while ((bytesRead = inputStream.read(buffer)) != -1) {
outputStream.write(buffer, 0, bytesRead);
}
}
return new FileInputStream(tempFile);
}
//数据写入输出流
public static void copyStream(FileInputStream fileInputStream, ServletOutputStream outputStream) throws IOException {
int len = 0;
byte[] arr = new byte[1024];
while ((len = fileInputStream.read(arr)) != -1) {
outputStream.write(arr, 0, len);
}
}
}

二、文件导出效果

总结

本次实现了html文件的导出功能，下节接着实现xml文件的导出功能。