58 lines
2.0 KiB
Java
58 lines
2.0 KiB
Java
|
package com.gunshi.project.ss.service;
|
|||
|
|
|
|||
|
|
|
|||
|
|
import cn.hutool.poi.excel.ExcelUtil;
|
|||
|
|
import cn.hutool.poi.excel.ExcelWriter;
|
|||
|
|
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
|
|||
|
|
import com.gunshi.project.ss.mapper.StEvpoMapper;
|
|||
|
|
import com.gunshi.project.ss.model.StEvpo;
|
|||
|
|
import jakarta.servlet.http.HttpServletResponse;
|
|||
|
|
import lombok.extern.slf4j.Slf4j;
|
|||
|
|
import org.springframework.stereotype.Service;
|
|||
|
|
import org.springframework.transaction.annotation.Transactional;
|
|||
|
|
|
|||
|
|
import java.io.IOException;
|
|||
|
|
import java.net.URLEncoder;
|
|||
|
|
import java.util.List;
|
|||
|
|
|
|||
|
|
@Service
|
|||
|
|
@Slf4j
|
|||
|
|
@Transactional(rollbackFor = Exception.class)
|
|||
|
|
public class StEvpoService extends ServiceImpl<StEvpoMapper, StEvpo> {
|
|||
|
|
public void export(HttpServletResponse response) {
|
|||
|
|
List<StEvpo> list = this.lambdaQuery().orderByAsc(StEvpo::getMonth).list();
|
|||
|
|
list.stream().forEach(o ->{
|
|||
|
|
o.setEvaporation(o.getEvaporation().setScale(2));
|
|||
|
|
});
|
|||
|
|
// 通过工具类创建writer
|
|||
|
|
ExcelWriter writer = ExcelUtil.getWriter();
|
|||
|
|
|
|||
|
|
// 自定义标题别名（只包含需要导出的字段）
|
|||
|
|
writer.addHeaderAlias("month", "月份");
|
|||
|
|
writer.addHeaderAlias("evaporation", "日蒸发量(万m³)");
|
|||
|
|
|
|||
|
|
// 只写出设置了别名的字段，默认写出所有字段
|
|||
|
|
writer.setOnlyAlias(true);
|
|||
|
|
|
|||
|
|
// 一次性写出内容，使用默认样式
|
|||
|
|
writer.write(list, true);
|
|||
|
|
|
|||
|
|
// 设置响应内容类型
|
|||
|
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet;charset=utf-8");
|
|||
|
|
|
|||
|
|
try {
|
|||
|
|
// 设置响应头信息
|
|||
|
|
String fileName = URLEncoder.encode("蒸发量数据", "UTF-8");
|
|||
|
|
response.setHeader("Content-Disposition", "attachment;filename=" + fileName + ".xlsx");
|
|||
|
|
|
|||
|
|
// 将writer对象刷新到响应输出流中
|
|||
|
|
writer.flush(response.getOutputStream(), true);
|
|||
|
|
} catch (IOException e) {
|
|||
|
|
e.printStackTrace();
|
|||
|
|
} finally {
|
|||
|
|
// 关闭writer，释放内存
|
|||
|
|
writer.close();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|