修改dto

neim-biz/src/main/java/com/jiayue/biz/controller/InverterInfoController.java

@@ -73,21 +73,21 @@ public class InverterInfoController extends BaseController {
 
     //新增逆变器
     @PostMapping("/addInverterInfo")
-    public void addInverterInfo(InverterInfo inverterInfo, String stationId){
-        inverterInfoService.addInverterInfo(inverterInfo,stationId);
+    public void addInverterInfo(@RequestBody InverterInfoDto inverterInfoDto){
+        inverterInfoService.addInverterInfo(inverterInfoDto);
 
     }
 
     //修改逆变器
     @PutMapping("/updateInverterInfo")
-    public void updateInverterInfo(InverterInfo inverterInfo,String stationId){
-        inverterInfoService.updateInverterInfo(inverterInfo,stationId);
+    public void updateInverterInfo(@RequestBody InverterInfoDto inverterInfoDto){
+        inverterInfoService.updateInverterInfo(inverterInfoDto);
 
     }
     //删除逆变器
     @DeleteMapping("/deleteInverterInfo")
-    public void deleteInverterInfo(InverterInfo inverterInfo,String stationId){
-        inverterInfoService.deleteInverterInfo(inverterInfo,stationId);
+    public void deleteInverterInfo(@RequestBody InverterInfoDto inverterInfoDto){
+        inverterInfoService.deleteInverterInfo(inverterInfoDto);
     }
 
 

neim-biz/src/main/java/com/jiayue/biz/dto/InverterInfoDto.java

@@ -1,10 +1,14 @@
 package com.jiayue.biz.dto;
 
+import com.baomidou.mybatisplus.annotation.IdType;
+import com.baomidou.mybatisplus.annotation.TableId;
 import com.jiayue.biz.domain.InverterInfo;
+import com.jiayue.common.annotation.Excel;
 import lombok.AllArgsConstructor;
 import lombok.Data;
 import lombok.NoArgsConstructor;
 
+import java.math.BigDecimal;
 import java.util.List;
 
 @Data
@@ -16,7 +20,42 @@ public class InverterInfoDto {
 
     private String stationName;
 
-    private List<InverterInfo> inverterInfo;
+    /**
+     * id
+     */
+    private String InverterInfoId;
+
+    /**
+     * 名称
+     */
+    private String name;
+
+    /**
+     * 型号
+     */
+    private String modelNumber;
+    /**
+     * 经度
+     */
+    private BigDecimal longitude;
+
+    /**
+     * 纬度
+     */
+    private BigDecimal latitude;
+    /**
+     * 额定功率（KW）
+     */
+    private BigDecimal capacity;
+    /**
+     * 光伏组件型号
+     */
+    private String batteryModel;
+
+    /**
+     * 光伏组件数量（个）
+     */
+    private Integer batteryNumber;
 
 
 

neim-biz/src/main/java/com/jiayue/biz/service/InverterInfoService.java

@@ -14,12 +14,12 @@ public interface InverterInfoService {
     List<InverterInfoDto> selectInverterInfo();
 
     //新增逆变器
-    void addInverterInfo(InverterInfo inverterInfo, String stationId);
+    void addInverterInfo(InverterInfoDto inverterInfoDto);
 
     //修改逆变器
-    void updateInverterInfo(InverterInfo inverterInfo,String stationId);
+    void updateInverterInfo(InverterInfoDto inverterInfoDto);
     //删除逆变器
-    void deleteInverterInfo(InverterInfo inverterInfo,String stationId);
+    void deleteInverterInfo(InverterInfoDto inverterInfoDto);
 
 
 }

neim-biz/src/main/java/com/jiayue/biz/service/impl/InverterInfoServiceImpl.java

@@ -23,16 +23,20 @@ public class InverterInfoServiceImpl implements InverterInfoService {
 
 
     //查询逆变器
-    public List<InverterInfoDto> selectInverterInfo(){
+    public List<InverterInfoDto> selectInverterInfo() {
         List<StationInfo> stationInfos = stationInfoService.selectStationInfo();
         ArrayList<InverterInfoDto> list = new ArrayList<>();
         for (StationInfo stationInfo : stationInfos) {
-            if(stationInfo.getStationBasicInfo().getStationType().equals("光")){
-                InverterInfoDto inverterInfoDto = new InverterInfoDto();
-                inverterInfoDto.setId(stationInfo.getId());
-                inverterInfoDto.setStationName(stationInfo.getStationBasicInfo().getStationName());
-                inverterInfoDto.setInverterInfo(stationInfo.getInverterInfos());
-                list.add(inverterInfoDto);
+            if (stationInfo.getStationBasicInfo().getStationType().equals("光") && stationInfo.getInverterInfos() != null) {
+                for (InverterInfo inverterInfo : stationInfo.getInverterInfos()) {
+                    InverterInfoDto inverterInfoDto = new InverterInfoDto();
+                    inverterInfoDto.setId(stationInfo.getId());
+                    inverterInfoDto.setStationName(stationInfo.getStationBasicInfo().getStationName());
+                    this.getInverterInfoDto(inverterInfo, inverterInfoDto);
+                    list.add(inverterInfoDto);
+
+                }
+
             }
 
         }
@@ -41,20 +45,22 @@ public class InverterInfoServiceImpl implements InverterInfoService {
     }
 
     //新增逆变器
-    public void addInverterInfo(InverterInfo inverterInfo,String stationId){
-        inverterInfo.setId(UUID.randomUUID().toString());
-        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(stationId);
+    public void addInverterInfo(InverterInfoDto inverterInfoDto) {
+        inverterInfoDto.setId(UUID.randomUUID().toString());
+        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(inverterInfoDto.getId());
+        InverterInfo inverterInfo = new InverterInfo();
+        this.getInverterInfo(inverterInfo, inverterInfoDto);
         oneStationInfo.getInverterInfos().add(inverterInfo);
         stationInfoService.saveStation(oneStationInfo);
 
     }
 
     //修改逆变器
-    public void updateInverterInfo(InverterInfo inverterInfo,String stationId){
-        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(stationId);
+    public void updateInverterInfo(InverterInfoDto inverterInfoDto) {
+        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(inverterInfoDto.getId());
         List<InverterInfo> inverterInfos = oneStationInfo.getInverterInfos().stream().map(s -> {
-            if (s.getId().equals(inverterInfo.getId())) {
-                s = inverterInfo;
+            if (s.getId().equals(inverterInfoDto.getInverterInfoId())) {
+                this.getInverterInfo(s,inverterInfoDto);
             }
             return s;
         }).collect(Collectors.toList());
@@ -62,13 +68,47 @@ public class InverterInfoServiceImpl implements InverterInfoService {
         stationInfoService.saveStation(oneStationInfo);
 
     }
+
     //删除逆变器
-    public void deleteInverterInfo(InverterInfo inverterInfo,String stationId){
-        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(stationId);
-        List<InverterInfo> collect = oneStationInfo.getInverterInfos().stream().filter(s -> s.getId().equals(inverterInfo.getId())).collect(Collectors.toList());
+    public void deleteInverterInfo(InverterInfoDto inverterInfoDto) {
+        StationInfo oneStationInfo = stationInfoService.getOneStationInfo(inverterInfoDto.getId());
+        List<InverterInfo> collect = oneStationInfo.getInverterInfos().stream().filter(s -> s.getId().equals(inverterInfoDto.getInverterInfoId())).collect(Collectors.toList());
         oneStationInfo.getInverterInfos().remove(collect.get(0));
         stationInfoService.saveStation(oneStationInfo);
     }
 
+    /**
+     * 实体转换dto
+     *
+     * @param inverterInfo
+     * @param inverterInfoDto
+     */
+    public void getInverterInfoDto(InverterInfo inverterInfo, InverterInfoDto inverterInfoDto) {
+        inverterInfoDto.setInverterInfoId(inverterInfo.getId());
+        inverterInfoDto.setName(inverterInfo.getName());
+        inverterInfoDto.setLatitude(inverterInfo.getLatitude());
+        inverterInfoDto.setBatteryModel(inverterInfo.getBatteryModel());
+        inverterInfoDto.setLongitude(inverterInfo.getLongitude());
+        inverterInfoDto.setCapacity(inverterInfo.getCapacity());
+        inverterInfoDto.setBatteryNumber(inverterInfo.getBatteryNumber());
+        inverterInfoDto.setModelNumber(inverterInfo.getModelNumber());
+    }
+
+    /**
+     * dto转换实体
+     *
+     * @param inverterInfo
+     * @param inverterInfoDto
+     */
+    public void getInverterInfo(InverterInfo inverterInfo, InverterInfoDto inverterInfoDto) {
+        inverterInfo.setId(inverterInfoDto.getInverterInfoId());
+        inverterInfo.setName(inverterInfoDto.getName());
+        inverterInfo.setLatitude(inverterInfoDto.getLatitude());
+        inverterInfo.setBatteryModel(inverterInfoDto.getBatteryModel());
+        inverterInfo.setLongitude(inverterInfoDto.getLongitude());
+        inverterInfo.setCapacity(inverterInfoDto.getCapacity());
+        inverterInfo.setBatteryNumber(inverterInfoDto.getBatteryNumber());
+        inverterInfo.setModelNumber(inverterInfoDto.getModelNumber());
+    }
 
 }