219 lines
9.8 KiB
Java
219 lines
9.8 KiB
Java
package com.gunshi.project.hsz.controller;
|
||
|
||
import com.alibaba.fastjson2.JSON;
|
||
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
|
||
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
|
||
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
|
||
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
||
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
|
||
import com.gunshi.core.result.R;
|
||
import com.gunshi.project.hsz.entity.vo.OsmoticWaterRVo;
|
||
import com.gunshi.project.hsz.listener.OsmoticWaterRImportListener;
|
||
import com.gunshi.project.hsz.model.OsmoticWaterR;
|
||
import com.gunshi.project.hsz.model.OsmoticWaterRule;
|
||
import com.gunshi.project.hsz.service.OsmoticWaterRService;
|
||
import com.gunshi.project.hsz.service.OsmoticWaterRuleService;
|
||
import com.gunshi.project.hsz.util.ConvertUtil;
|
||
import com.gunshi.project.hsz.util.DateUtil;
|
||
import com.gunshi.project.hsz.util.ExcelUtil;
|
||
import com.gunshi.project.hsz.util.ResultJson;
|
||
import com.gunshi.project.hsz.util.excel.ExcelResult;
|
||
import com.gunshi.project.hsz.common.validate.markers.Insert;
|
||
import com.gunshi.project.hsz.common.validate.markers.Update;
|
||
import io.swagger.v3.oas.annotations.Operation;
|
||
import io.swagger.v3.oas.annotations.media.Schema;
|
||
import io.swagger.v3.oas.annotations.tags.Tag;
|
||
import jakarta.servlet.http.HttpServletResponse;
|
||
import org.springframework.beans.factory.annotation.Autowired;
|
||
import org.springframework.http.MediaType;
|
||
import org.springframework.validation.annotation.Validated;
|
||
import org.springframework.web.bind.annotation.CrossOrigin;
|
||
import org.springframework.web.bind.annotation.GetMapping;
|
||
import org.springframework.web.bind.annotation.PathVariable;
|
||
import org.springframework.web.bind.annotation.PostMapping;
|
||
import org.springframework.web.bind.annotation.RequestBody;
|
||
import org.springframework.web.bind.annotation.RequestMapping;
|
||
import org.springframework.web.bind.annotation.RequestPart;
|
||
import org.springframework.web.bind.annotation.RestController;
|
||
import org.springframework.web.multipart.MultipartFile;
|
||
|
||
import java.io.File;
|
||
import java.io.FileNotFoundException;
|
||
import java.io.IOException;
|
||
import java.io.InputStream;
|
||
import java.io.OutputStream;
|
||
import java.io.Serializable;
|
||
import java.net.URLEncoder;
|
||
import java.util.ArrayList;
|
||
import java.util.HashMap;
|
||
import java.util.List;
|
||
import java.util.Map;
|
||
|
||
/**
|
||
* 描述: 水质采样记录表
|
||
* author: xusan
|
||
* date: 2024-07-08 17:40:37
|
||
*/
|
||
@Tag(name = "水质采样记录表")
|
||
@RestController
|
||
@RequestMapping(value="/osmoticWaterR")
|
||
public class OsmoticWaterRController {
|
||
|
||
@Autowired
|
||
private OsmoticWaterRService service;
|
||
|
||
@Autowired
|
||
private OsmoticWaterRuleService osmoticWaterRuleService;
|
||
|
||
@Operation(summary = "新增")
|
||
@PostMapping("/insert")
|
||
public R<OsmoticWaterR> insert(@Validated(Insert.class) @RequestBody OsmoticWaterR dto) {
|
||
boolean result = service.save(dto);
|
||
return R.ok(result ? dto : null);
|
||
}
|
||
|
||
@Operation(summary = "修改")
|
||
@PostMapping("/update")
|
||
public R<OsmoticWaterR> update(@Validated(Update.class) @RequestBody OsmoticWaterR dto) {
|
||
boolean result = service.updateById(dto);
|
||
return R.ok(result ? dto : null);
|
||
}
|
||
|
||
@Operation(summary = "删除")
|
||
@GetMapping("/del/{id}")
|
||
public R<Boolean> del(@Schema(name = "id") @PathVariable("id") Serializable id) {
|
||
return R.ok(service.removeById(id));
|
||
}
|
||
|
||
@Operation(summary = "获取最新一条数据")
|
||
@PostMapping("/getLastData")
|
||
public R<Map<String, Object>> getLastData() {
|
||
Map<String, Object> resMap = new HashMap<>();
|
||
QueryWrapper qw = new QueryWrapper<>();
|
||
qw.orderBy(true, false, "tm").last(" limit 1");
|
||
OsmoticWaterR osmoticWaterR = service.getOne(qw);
|
||
if(ObjectUtils.isNotEmpty(osmoticWaterR)){
|
||
resMap = JSON.parseObject(JSON.toJSONString(osmoticWaterR), Map.class);
|
||
// 获取标准值
|
||
List<OsmoticWaterRule> list = osmoticWaterRuleService.list();
|
||
if(CollectionUtils.isNotEmpty(list)){
|
||
for (OsmoticWaterRule rule : list){
|
||
String code = rule.getCode();
|
||
String codeStandard = code + "Standard";
|
||
if(resMap.containsKey(code)){
|
||
if("~".equals(rule.getCondition())){
|
||
resMap.put(codeStandard, "(" + rule.getOne() + rule.getCondition() + rule.getTwo() + ")");
|
||
} else {
|
||
String codeLevel = resMap.get(code + "Level").toString();
|
||
if(codeLevel.contains("Ⅰ")){
|
||
resMap.put(codeStandard, "(" + rule.getCondition() + rule.getOne() + ")");
|
||
} else if (codeLevel.contains("Ⅱ")){
|
||
resMap.put(codeStandard, "(" + rule.getCondition() + rule.getTwo() + ")");
|
||
} else if (codeLevel.contains("Ⅲ")){
|
||
resMap.put(codeStandard, "(" + rule.getCondition() + rule.getThree() + ")");
|
||
} else if (codeLevel.contains("Ⅳ")){
|
||
resMap.put(codeStandard, "(" + rule.getCondition() + rule.getFour() + ")");
|
||
} else if (codeLevel.contains("Ⅴ")){
|
||
resMap.put(codeStandard, "(" + rule.getCondition() + rule.getFive() + ")");
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
return R.ok(resMap);
|
||
}
|
||
|
||
@Operation(summary = "列表")
|
||
@PostMapping("/list")
|
||
public R<List<OsmoticWaterR>> list(@RequestBody @Validated OsmoticWaterR osmoticWaterR) {
|
||
QueryWrapper<OsmoticWaterR> wrapper = new QueryWrapper<OsmoticWaterR>()
|
||
.ge(ObjectUtils.isNotNull(osmoticWaterR.getStartTime()), "tm", osmoticWaterR.getStartTime())
|
||
.le(ObjectUtils.isNotNull(osmoticWaterR.getEndTime()), "tm", osmoticWaterR.getEndTime());
|
||
if(StringUtils.isNotBlank(osmoticWaterR.getOrderField())){
|
||
wrapper.orderBy(true, ObjectUtils.isEmpty(osmoticWaterR.getIsAsc()) ? false : osmoticWaterR.getIsAsc(), osmoticWaterR.getOrderField());
|
||
}
|
||
return R.ok(service.list(wrapper));
|
||
}
|
||
|
||
@Operation(summary = "分页")
|
||
@PostMapping("/page")
|
||
public R<Page<OsmoticWaterR>> page(@RequestBody @Validated OsmoticWaterR osmoticWaterR) {
|
||
QueryWrapper<OsmoticWaterR> wrapper = new QueryWrapper<>();
|
||
wrapper.ge(ObjectUtils.isNotNull(osmoticWaterR.getStartTime()), "tm", osmoticWaterR.getStartTime())
|
||
.le(ObjectUtils.isNotNull(osmoticWaterR.getEndTime()), "tm", osmoticWaterR.getEndTime());
|
||
if(StringUtils.isNotBlank(osmoticWaterR.getOrderField())){
|
||
wrapper.orderBy(true, ObjectUtils.isEmpty(osmoticWaterR.getIsAsc()) ? false : osmoticWaterR.getIsAsc(), osmoticWaterR.getOrderField());
|
||
}
|
||
return R.ok(service.page(osmoticWaterR.getPageSo().toPage(), wrapper));
|
||
}
|
||
|
||
/**
|
||
* 获取导入模板
|
||
*/
|
||
@Operation(summary = "获取导入模板")
|
||
@PostMapping("/importTemplate")
|
||
public void importTemplate(HttpServletResponse response) {
|
||
ExcelUtil.exportExcel(new ArrayList<>(), "水质整编表", OsmoticWaterRVo.class, response, "水质整编表");
|
||
}
|
||
|
||
/**
|
||
* 导入数据
|
||
*
|
||
* @param file 导入文件
|
||
*
|
||
*/
|
||
@Operation(summary = "导入数据")
|
||
@PostMapping(value = "/importData", consumes = MediaType.MULTIPART_FORM_DATA_VALUE)
|
||
public ResultJson<String> importData(@RequestPart("file") MultipartFile file) throws Exception {
|
||
ExcelResult<OsmoticWaterRVo> result = ExcelUtil.importExcel(file.getInputStream(), OsmoticWaterRVo.class, new OsmoticWaterRImportListener());
|
||
return ResultJson.ok(result.getAnalysis());
|
||
}
|
||
|
||
@Operation(summary = "导出")
|
||
@PostMapping("/exportOsmoticWaterRDataExcel")
|
||
@CrossOrigin
|
||
public void exportOsmoticWaterRDataExcel(@RequestBody @Validated OsmoticWaterR osmoticWaterR, HttpServletResponse response) {
|
||
String filename = "水质整编表";
|
||
if (ObjectUtils.isNotEmpty(osmoticWaterR.getStartTime())) {
|
||
filename.concat("_").concat(DateUtil.convertDateToString(osmoticWaterR.getStartTime()));
|
||
}
|
||
if (ObjectUtils.isNotEmpty(osmoticWaterR.getStartTime())) {
|
||
filename.concat("_").concat(DateUtil.convertDateToString(osmoticWaterR.getEndTime()));
|
||
}
|
||
List<OsmoticWaterRVo> vos = ConvertUtil.entityToVoList(this.list(osmoticWaterR).getData(), OsmoticWaterRVo.class);
|
||
ExcelUtil.exportExcel(vos, filename, OsmoticWaterRVo.class, response, "水质整编表");
|
||
}
|
||
|
||
|
||
@Operation(summary = "文件下载")
|
||
@PostMapping("/downloadFile")
|
||
@CrossOrigin
|
||
public void downloadFile(HttpServletResponse response) throws IOException {
|
||
String filePath = "doc/2002年6月1日_GB3838-2002地表水环境质量标准.pdf";
|
||
File file = new File(filePath);
|
||
if (!file.exists()) {
|
||
new FileNotFoundException("文件不存在！");
|
||
}
|
||
response.setCharacterEncoding("utf-8");
|
||
response.setContentType("application/octet-stream; charset=UTF-8");
|
||
response.setHeader("content-disposition", "attachment;filename*=utf-8'zh_cn'" + URLEncoder.encode("GB3838-2002地表水环境质量标准.pdf", "UTF-8"));
|
||
InputStream inputStream = null;
|
||
OutputStream outputStream = null;
|
||
try {
|
||
inputStream = this.getClass().getClassLoader().getResourceAsStream(filePath);
|
||
outputStream = response.getOutputStream();
|
||
byte[] buffer = new byte[1024];
|
||
int len = 0;
|
||
while ((len = inputStream.read(buffer)) != -1) {
|
||
outputStream.write(buffer, 0, len);
|
||
}
|
||
} catch (Exception e) {
|
||
e.printStackTrace();
|
||
} finally {
|
||
inputStream.close();
|
||
outputStream.close();
|
||
}
|
||
}
|
||
}
|