package com.zy.asrs.controller;
|
|
import com.alibaba.excel.EasyExcel;
|
import com.alibaba.fastjson.JSON;
|
import com.baomidou.mybatisplus.mapper.EntityWrapper;
|
import com.baomidou.mybatisplus.plugins.Page;
|
import com.core.common.Cools;
|
import com.core.common.DateUtils;
|
import com.core.common.R;
|
import com.zy.asrs.entity.AgvWrkMastLog;
|
import com.zy.asrs.entity.ViewWorkCountInView;
|
import com.zy.asrs.mapper.ReportQueryMapper;
|
import com.zy.asrs.service.AgvWrkMastLogService;
|
import com.zy.common.web.BaseController;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.web.bind.annotation.*;
|
|
import javax.servlet.http.HttpServletResponse;
|
import java.io.IOException;
|
import java.net.URLEncoder;
|
import java.util.HashMap;
|
import java.util.List;
|
import java.util.Map;
|
|
@RestController
|
@RequestMapping("/report/download")
|
public class ReportDownloadController extends BaseController {
|
|
|
@Autowired
|
private ReportQueryMapper reportQueryMapper;
|
|
|
@Autowired
|
private AgvWrkMastLogService wrkMastLogService;
|
|
|
/**
|
* 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
|
* 日入库汇总查询
|
*
|
* @since 2.1.1
|
*/
|
@GetMapping("/in")
|
public void download1(HttpServletResponse response,
|
@RequestParam Map<String, Object> param) throws IOException {
|
try {
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
response.setCharacterEncoding("utf-8");
|
// 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
|
String fileName = URLEncoder.encode("日入库汇总", "UTF-8").replaceAll("\\+", "%20");
|
response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
|
// 这里需要设置不关闭流
|
EasyExcel.write(response.getOutputStream(), ViewWorkCountInView.class).autoCloseStream(Boolean.FALSE).sheet("sheet1")
|
.doWrite(getInData(param));
|
} catch (Exception e) {
|
// 重置response
|
response.reset();
|
response.setContentType("application/json");
|
response.setCharacterEncoding("utf-8");
|
Map<String, String> map = new HashMap<>();
|
map.put("status", "failure");
|
map.put("message", "下载文件失败" + e.getMessage());
|
response.getWriter().println(JSON.toJSONString(map));
|
}
|
}
|
|
|
private List<ViewWorkCountInView> getInData(Map<String, Object> param) {
|
String startTime = "1970.1.2";
|
String endTime = "2099.1.2";
|
if (!Cools.isEmpty(param.get("query_date"))) {
|
String queryDate = (String) param.get("query_date");
|
String[] split = queryDate.split(" - ");
|
startTime = split[0].split(" ")[0].replace("-", ".");
|
endTime = split[1].split(" ")[0].replace("-", ".");
|
}
|
List<ViewWorkCountInView> allCountIn = reportQueryMapper.selectWorkCountIn(null, null, (String) param.get("matnr"), startTime, endTime);
|
return allCountIn;
|
}
|
|
/**
|
* 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
|
* 日入库汇总查询
|
*
|
* @since 2.1.1
|
*/
|
@RequestMapping("/out")
|
public void download2(HttpServletResponse response,
|
@RequestParam Map<String, Object> param) throws IOException {
|
try {
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
response.setCharacterEncoding("utf-8");
|
// 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
|
String fileName = URLEncoder.encode("日出库汇总", "UTF-8").replaceAll("\\+", "%20");
|
response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
|
// 这里需要设置不关闭流
|
EasyExcel.write(response.getOutputStream(), ViewWorkCountInView.class).autoCloseStream(Boolean.FALSE).sheet("sheet1")
|
.doWrite(getOutData(param));
|
} catch (Exception e) {
|
// 重置response
|
response.reset();
|
response.setContentType("application/json");
|
response.setCharacterEncoding("utf-8");
|
Map<String, String> map = new HashMap<>();
|
map.put("status", "failure");
|
map.put("message", "下载文件失败" + e.getMessage());
|
response.getWriter().println(JSON.toJSONString(map));
|
}
|
}
|
|
|
private List<ViewWorkCountInView> getOutData(Map<String, Object> param) {
|
String startTime = "1970.1.2";
|
String endTime = "2099.1.2";
|
if (!Cools.isEmpty(param.get("query_date"))) {
|
String queryDate = (String) param.get("query_date");
|
String[] split = queryDate.split(" - ");
|
startTime = split[0].split(" ")[0].replace("-", ".");
|
endTime = split[1].split(" ")[0].replace("-", ".");
|
}
|
List<ViewWorkCountInView> allCountIn = reportQueryMapper.selectWorkCountOut(Integer.valueOf((String) param.get("pageNumber")), Integer.valueOf((String) param.get("pageSize")), (String) param.get("matnr"), startTime, endTime);
|
return allCountIn;
|
}
|
|
|
/**
|
* 文件下载并且失败的时候返回json（默认失败了会返回一个有部分数据的Excel）
|
* 日入库汇总查询
|
*
|
* @since 2.1.1
|
*/
|
@RequestMapping("/wrkMastLog")
|
public void wrkMastLog(HttpServletResponse response,
|
@RequestParam Map<String, Object> param) throws IOException {
|
try {
|
response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
|
response.setCharacterEncoding("utf-8");
|
// 这里URLEncoder.encode可以防止中文乱码 当然和easyexcel没有关系
|
String fileName = URLEncoder.encode("工作档维护日志", "UTF-8").replaceAll("\\+", "%20");
|
response.setHeader("Content-disposition", "attachment;filename*=utf-8''" + fileName + ".xlsx");
|
// 这里需要设置不关闭流
|
EasyExcel.write(response.getOutputStream(), ViewWorkCountInView.class).autoCloseStream(Boolean.FALSE).sheet("sheet1")
|
.doWrite(getWrkMastLogData(param));
|
} catch (Exception e) {
|
// 重置response
|
response.reset();
|
response.setContentType("application/json");
|
response.setCharacterEncoding("utf-8");
|
Map<String, String> map = new HashMap<>();
|
map.put("status", "failure");
|
map.put("message", "下载文件失败" + e.getMessage());
|
response.getWriter().println(JSON.toJSONString(map));
|
}
|
}
|
|
|
private List<AgvWrkMastLog> getWrkMastLogData(Map<String, Object> param) {
|
excludeTrash(param);
|
EntityWrapper<AgvWrkMastLog> wrapper = new EntityWrapper<>();
|
convert(param, wrapper);
|
return wrkMastLogService.selectList(wrapper);
|
}
|
|
private <T> void convert(Map<String, Object> map, EntityWrapper<T> wrapper) {
|
for (Map.Entry<String, Object> entry : map.entrySet()) {
|
String val = String.valueOf(entry.getValue());
|
if (val.contains(RANGE_TIME_LINK)) {
|
String[] dates = val.split(RANGE_TIME_LINK);
|
wrapper.ge(entry.getKey(), DateUtils.convert(dates[0]));
|
wrapper.le(entry.getKey(), DateUtils.convert(dates[1]));
|
} else {
|
if (entry.getKey().equals("manu_type")) {
|
wrapper.like(entry.getKey(), val);
|
} else {
|
wrapper.eq(entry.getKey(), val);
|
}
|
}
|
}
|
}
|
}
|
