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 827f0ab..7454ca8 100644 --- a/src/main/java/com/gunshi/project/xyt/controller/WaterQualityController.java +++ b/src/main/java/com/gunshi/project/xyt/controller/WaterQualityController.java @@ -69,6 +69,8 @@ public class WaterQualityController { } @Operation(summary = "水质监测数据导出") + + @GetMapping("/export") public void export(WaterQualityPageSo waterQualityPageSo, HttpServletResponse response) { waterqualityService.export(waterQualityPageSo, response); 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 3df7786..58f5267 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 @@ -7,6 +7,7 @@ import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.plugins.pagination.Page; import com.gunshi.core.dateformat.DateFormatString; import com.gunshi.db.dao.BaseOrderDao; +import com.gunshi.db.dto.DateTimeRangeSo; import com.gunshi.project.xyt.mapper.WaterQualityMapper; import com.gunshi.project.xyt.model.StWaterQualityR; import com.gunshi.project.xyt.service.WaterQualityService; @@ -21,11 +22,14 @@ import org.springframework.transaction.annotation.Transactional; import org.springframework.web.multipart.MultipartFile; import java.io.IOException; +import java.io.OutputStream; import java.net.URLEncoder; 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. @@ -43,21 +47,28 @@ public class WaterQualityServiceImpl extends BaseOrderDao page(WaterQualityPageSo so) { + public Page page(WaterQualityPageSo pageQueryCriteria) { + + requireNonNull(pageQueryCriteria, "pageQueryCriteria should not be null"); + LambdaQueryWrapper query = Wrappers.lambdaQuery(); + DateTimeRangeSo timeSo = pageQueryCriteria.getTimeSo(); - if (so.getTimeSo() != null&& so.getTimeSo().hasStart()) { - query.ge(StWaterQualityR::getSpt, so.getTimeSo().getStart()); - } - if (so.getTimeSo() != null&& so.getTimeSo().hasEnd()) { - query.le(StWaterQualityR::getSpt,so.getTimeSo().getEnd()); + if (timeSo != null) { + if (timeSo.getStart() != null) { + query.ge(StWaterQualityR::getSpt, timeSo.getStart()); + } + if (timeSo.getEnd() != null) { + query.le(StWaterQualityR::getSpt, timeSo.getEnd()); + } } - return super.page(so.getPageSo().toPage(), query); + + return super.page(pageQueryCriteria.getPageSo().toPage(), query); } /** @@ -75,7 +86,7 @@ public class WaterQualityServiceImpl extends BaseOrderDao