From 705c498e382a8a9af2ddcc6961710ecfd7ccbe10 Mon Sep 17 00:00:00 2001 From: chenxiwang <1520264117@qq.com> Date: Wed, 24 Jul 2024 09:02:59 +0800 Subject: [PATCH] =?UTF-8?q?=E7=94=9F=E6=80=81=E6=B5=81=E9=87=8F=E7=9B=91?= =?UTF-8?q?=E6=8E=A7=E6=8E=A5=E5=8F=A3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../xyt/controller/StWaterRController.java | 64 +++++++++++++------ .../controller/StWaterRRealController.java | 17 ++++- .../xyt/mapper/StWaterRRealMapper.java | 5 +- .../xyt/service/StWaterRRealService.java | 13 ++-- .../resources/mapper/StWaterRRealMapper.xml | 8 +++ 5 files changed, 82 insertions(+), 25 deletions(-) diff --git a/src/main/java/com/gunshi/project/xyt/controller/StWaterRController.java b/src/main/java/com/gunshi/project/xyt/controller/StWaterRController.java index 34e85b9..cb2a23e 100644 --- a/src/main/java/com/gunshi/project/xyt/controller/StWaterRController.java +++ b/src/main/java/com/gunshi/project/xyt/controller/StWaterRController.java @@ -1,22 +1,11 @@ package com.gunshi.project.xyt.controller; -import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; -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.xyt.entity.vo.StWaterRVo; -import com.gunshi.project.xyt.model.StWaterR; -import com.gunshi.project.xyt.service.StWaterRService; -import com.gunshi.project.xyt.util.ConvertUtil; -import com.gunshi.project.xyt.util.DateUtil; -import com.gunshi.project.xyt.util.ExcelUtil; -import com.gunshi.project.xyt.validate.markers.Insert; -import com.gunshi.project.xyt.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 java.io.Serializable; +import java.util.HashMap; +import java.util.List; +import java.util.Map; +import java.util.stream.Collectors; + import org.springframework.beans.factory.annotation.Autowired; import org.springframework.validation.annotation.Validated; import org.springframework.web.bind.annotation.CrossOrigin; @@ -25,10 +14,30 @@ 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.RequestParam; import org.springframework.web.bind.annotation.RestController; -import java.io.Serializable; -import java.util.List; +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.xyt.entity.vo.StWaterRVo; +import com.gunshi.project.xyt.model.ResMonthEcoFlow; +import com.gunshi.project.xyt.model.StWaterR; +import com.gunshi.project.xyt.service.ResMonthEcoFlowService; +import com.gunshi.project.xyt.service.StWaterRService; +import com.gunshi.project.xyt.util.ConvertUtil; +import com.gunshi.project.xyt.util.DateUtil; +import com.gunshi.project.xyt.util.ExcelUtil; +import com.gunshi.project.xyt.validate.markers.Insert; +import com.gunshi.project.xyt.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; /** * 描述: 供水量表 @@ -43,6 +52,9 @@ public class StWaterRController { @Autowired private StWaterRService service; + @Autowired + private ResMonthEcoFlowService resMonthEcoFlowService; + @Operation(summary = "新增") @PostMapping("/insert") @@ -77,6 +89,20 @@ public class StWaterRController { return R.ok(service.list(wrapper)); } + @Operation(summary = "获取月核定流量") + @GetMapping("/getResMonthEcoFlow") + public R> getResMonthEcoFlow(@Schema(name = "startTime", description = "开始时间") @RequestParam("startTime") String startTime, @Schema(name = "endTime", description = "开始时间") @RequestParam("endTime") String endTime) { + Map resMap = new HashMap<>(); + QueryWrapper wrapper = new QueryWrapper() + .ge(StringUtils.isNotEmpty(startTime), "month", Integer.valueOf(startTime.substring(5, 7))) + .le(StringUtils.isNotEmpty(endTime), "month", Integer.valueOf(endTime.substring(5, 7))); + List list = resMonthEcoFlowService.list(wrapper); + if(CollectionUtils.isNotEmpty(list)){ + resMap = list.stream().collect(Collectors.toMap(ResMonthEcoFlow::getMonth, ResMonthEcoFlow::getValue)); + } + return R.ok(resMap); + } + @Operation(summary = "分页") @PostMapping("/page") public R> page(@RequestBody @Validated StWaterR stWaterR) { diff --git a/src/main/java/com/gunshi/project/xyt/controller/StWaterRRealController.java b/src/main/java/com/gunshi/project/xyt/controller/StWaterRRealController.java index b2967d5..6aa0d6f 100644 --- a/src/main/java/com/gunshi/project/xyt/controller/StWaterRRealController.java +++ b/src/main/java/com/gunshi/project/xyt/controller/StWaterRRealController.java @@ -15,11 +15,13 @@ import org.springframework.web.bind.annotation.RequestMapping; import org.springframework.web.bind.annotation.RestController; import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.baomidou.mybatisplus.core.toolkit.CollectionUtils; import com.baomidou.mybatisplus.core.toolkit.ObjectUtils; import com.gunshi.core.result.R; import com.gunshi.project.xyt.model.ResMonthEcoFlow; import com.gunshi.project.xyt.model.StWaterRReal; import com.gunshi.project.xyt.service.ResMonthEcoFlowService; +import com.gunshi.project.xyt.service.StStbprpBService; import com.gunshi.project.xyt.service.StWaterRRealService; import com.gunshi.project.xyt.validate.markers.Insert; import com.gunshi.project.xyt.validate.markers.Update; @@ -37,6 +39,9 @@ import io.swagger.v3.oas.annotations.tags.Tag; @RequestMapping(value="/stWaterRReal") public class StWaterRRealController { + @Autowired + private StStbprpBService stStbprpBService; + @Autowired private StWaterRRealService service; @@ -81,7 +86,17 @@ public class StWaterRRealController { @Operation(summary = "列表") @PostMapping("/list") public R> list() { - return R.ok(service.lambdaQuery().list()); + List list = service.listRelated(); + if(CollectionUtils.isNotEmpty(list)){ + // 获取当月核定流量 + ResMonthEcoFlow resMonthEcoFlow = resMonthEcoFlowService.getOne(new QueryWrapper().eq("month", Calendar.getInstance().get(Calendar.MONTH) + 1)); + if(ObjectUtils.isNotEmpty(resMonthEcoFlow)){ + for(StWaterRReal real : list){ + real.setResMonthEcoFlow(resMonthEcoFlow); + } + } + } + return R.ok(list); } // @Operation(summary = "分页") diff --git a/src/main/java/com/gunshi/project/xyt/mapper/StWaterRRealMapper.java b/src/main/java/com/gunshi/project/xyt/mapper/StWaterRRealMapper.java index 2206fff..dc2ddab 100644 --- a/src/main/java/com/gunshi/project/xyt/mapper/StWaterRRealMapper.java +++ b/src/main/java/com/gunshi/project/xyt/mapper/StWaterRRealMapper.java @@ -4,6 +4,8 @@ import com.baomidou.mybatisplus.core.mapper.BaseMapper; import com.gunshi.project.xyt.model.StWaterRReal; import org.apache.ibatis.annotations.Mapper; +import java.util.List; + /** * 描述: 供水量实时表 * author: xusan @@ -12,4 +14,5 @@ import org.apache.ibatis.annotations.Mapper; @Mapper public interface StWaterRRealMapper extends BaseMapper { -} \ No newline at end of file + List listRelated(); +} diff --git a/src/main/java/com/gunshi/project/xyt/service/StWaterRRealService.java b/src/main/java/com/gunshi/project/xyt/service/StWaterRRealService.java index 84f1aaf..96c63a5 100644 --- a/src/main/java/com/gunshi/project/xyt/service/StWaterRRealService.java +++ b/src/main/java/com/gunshi/project/xyt/service/StWaterRRealService.java @@ -1,13 +1,15 @@ package com.gunshi.project.xyt.service; +import java.util.List; + +import org.springframework.stereotype.Service; +import org.springframework.transaction.annotation.Transactional; + import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; import com.gunshi.project.xyt.mapper.StWaterRRealMapper; import com.gunshi.project.xyt.model.StWaterRReal; -import lombok.extern.slf4j.Slf4j; -import org.springframework.stereotype.Service; -import org.springframework.transaction.annotation.Transactional; -import java.util.Date; +import lombok.extern.slf4j.Slf4j; /** * 描述: 供水量实时表 @@ -20,6 +22,9 @@ import java.util.Date; public class StWaterRRealService extends ServiceImpl { + public List listRelated() { + return baseMapper.listRelated(); + } } diff --git a/src/main/resources/mapper/StWaterRRealMapper.xml b/src/main/resources/mapper/StWaterRRealMapper.xml index 0e6a3e7..9562afd 100644 --- a/src/main/resources/mapper/StWaterRRealMapper.xml +++ b/src/main/resources/mapper/StWaterRRealMapper.xml @@ -2,4 +2,12 @@ +