diff --git a/src/main/java/com/gunshi/project/xyt/controller/WaterQualityController.java b/src/main/java/com/gunshi/project/xyt/controller/WaterQualityController.java index dbc8f15..916c25a 100644 --- a/src/main/java/com/gunshi/project/xyt/controller/WaterQualityController.java +++ b/src/main/java/com/gunshi/project/xyt/controller/WaterQualityController.java @@ -78,9 +78,9 @@ public class WaterQualityController { @Operation(summary = "水质监测数据导入") @PostMapping("/import") - public R importExcel(@RequestParam("file") MultipartFile file) { + public R importExcel(@RequestParam("file") MultipartFile file) { if (file.isEmpty()) { - return R.error(400, "请选择上传文件"); + return R.error(400, "请选择上传文件",false); } try { @@ -89,7 +89,7 @@ public class WaterQualityController { } catch (IOException e) { - return R.error(500, "文件上传失败: " + e.getMessage()); + return R.error(500, "文件上传失败: " + e.getMessage(),false); } } diff --git a/src/main/java/com/gunshi/project/xyt/service/WaterQualityService.java b/src/main/java/com/gunshi/project/xyt/service/WaterQualityService.java index 9934eb3..d725d6f 100644 --- a/src/main/java/com/gunshi/project/xyt/service/WaterQualityService.java +++ b/src/main/java/com/gunshi/project/xyt/service/WaterQualityService.java @@ -22,13 +22,13 @@ public interface WaterQualityService { void export(WaterQualityPageSo waterQualityPageSo, HttpServletResponse response); - String importExcel(MultipartFile file) throws IOException; + Boolean importExcel(MultipartFile file) throws IOException; - String add(StWaterQualityR stWaterQualityR); + Boolean add(StWaterQualityR stWaterQualityR); - String addList(List stWaterQualityRList); + Boolean addList(List stWaterQualityRList); - String update(StWaterQualityR StWaterQualityR); +// String update(StWaterQualityR StWaterQualityR); - String delete(String id); +// String delete(String id); } diff --git a/src/main/java/com/gunshi/project/xyt/service/impl/WaterQualityServiceImpl.java b/src/main/java/com/gunshi/project/xyt/service/impl/WaterQualityServiceImpl.java index f59e470..af7c931 100644 --- a/src/main/java/com/gunshi/project/xyt/service/impl/WaterQualityServiceImpl.java +++ b/src/main/java/com/gunshi/project/xyt/service/impl/WaterQualityServiceImpl.java @@ -28,8 +28,6 @@ import java.nio.charset.StandardCharsets; import java.util.ArrayList; import java.util.List; -import static java.util.Objects.requireNonNull; - /** * Description: * Created by xusan on 2023/1/23. @@ -67,6 +65,7 @@ public class WaterQualityServiceImpl extends BaseOrderDao data = new ArrayList<>(); EasyExcel.read(file.getInputStream(), StWaterQualityR.class, new PageReadListener(data::addAll)).sheet().doRead(); - for (StWaterQualityR o : data) { - o.setId(null); - } - return this.addList(data); + } @@ -142,41 +138,60 @@ public class WaterQualityServiceImpl extends BaseOrderDao queryWrapper = Wrappers.lambdaQuery(); + + queryWrapper.eq(StWaterQualityR::getStcd,o.getStcd()) + .eq(StWaterQualityR::getSpt,o.getSpt()); + + if (this.count(queryWrapper) > 0) { + + log.info("站码和采样时间重复 {}", String.format("%s", o)); + + throw new IllegalArgumentException("站码和采样时间必须唯一"); + + } + + return this.save(o); + } @Override - public String addList(List stWaterQualityRList) { + public Boolean addList(List stWaterQualityRList) { try { for (StWaterQualityR o : stWaterQualityRList) { this.add(o); } }catch (Exception e) { - return "失败"; + return false; } - return "成功"; + return true; // return this.saveBatch(stWaterQualityRList) ? "成功":"失败"; + } - /** - * 修改水质 - * - * @param o 水质 - */ - @Override - public String update(StWaterQualityR o) { - return this.updateById(o) ? "成功":"失败"; - } - - /** - * 删除水质 - * - * @param id id - */ - @Override - public String delete(String id) { - return this.removeById(id) ? "成功":"失败"; - } +// /** +// * 修改水质 +// * +// * @param o 水质 +// */ +// @Override +// public String update(StWaterQualityR o) { +// return this.updateById(o) ? "成功":"失败"; +// } +// +// /** +// * 删除水质 +// * +// * @param id id +// */ +// @Override +// public String delete(String id) { +// return this.removeById(id) ? "成功":"失败"; +// } }