运管开发

shuili-system/src/main/java/com/kms/yg/sz/controller/BsSgcSzFxzzController.java

@@ -3,10 +3,14 @@ package com.kms.yg.sz.controller;
 import java.util.Arrays;
 import java.util.List;
 
+import cn.hutool.core.collection.CollectionUtil;
 import com.kms.system.service.SysDeptService;
 import com.kms.system.service.SysXzqhService;
 import com.kms.yg.sz.domain.BsSgcSzFxzz;
+import com.kms.yg.sz.domain.BsSgcSzFxzzcy;
+import com.kms.yg.sz.domain.dto.BsSgcSzFxzzDto;
 import com.kms.yg.sz.service.BsSgcSzFxzzService;
+import com.kms.yg.sz.service.BsSgcSzFxzzcyService;
 import com.shuili.common.core.controller.BaseController;
 import com.shuili.common.core.domain.SearchParam;
 import com.shuili.common.utils.poi.ExcelUtil;
@@ -15,6 +19,7 @@ import com.kms.common.utils.BaseEntityUtils;
 
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
+import org.springframework.beans.BeanUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
@@ -48,6 +53,9 @@ public class BsSgcSzFxzzController extends BaseController
     @Autowired
     private BsSgcSzFxzzService bsSgcSzFxzzService;
 
+    @Autowired
+    private BsSgcSzFxzzcyService bsSgcSzFxzzcyService;
+
     @Autowired
     private SysXzqhService sysXzqhService;
 
@@ -93,10 +101,22 @@ public class BsSgcSzFxzzController extends BaseController
     @Log(title = "防汛组织新增", businessType = BusinessType.INSERT)
     @PostMapping
     @ApiOperation("防汛组织新增")
-    public AjaxResult add(@RequestBody BsSgcSzFxzz bsSgcSzFxzz)
+    public AjaxResult add(@RequestBody BsSgcSzFxzzDto dto)
     {
+
+        BsSgcSzFxzz bsSgcSzFxzz = new BsSgcSzFxzz();
+        BeanUtils.copyProperties(dto,bsSgcSzFxzz);
         BaseEntityUtils.preInsert(bsSgcSzFxzz);
-        return toAjax(bsSgcSzFxzzService.save(bsSgcSzFxzz));
+        bsSgcSzFxzzService.save(bsSgcSzFxzz);
+        String id = bsSgcSzFxzz.getId();
+        List<BsSgcSzFxzzcy> cy = dto.getCy();
+        if (CollectionUtil.isNotEmpty(cy)) {
+            cy.stream().forEach(x->x.setOrganizationId(id));
+            return toAjax(bsSgcSzFxzzcyService.saveBatch(cy));
+        }else {
+            return AjaxResult.success();
+        }
+
     }
 
     /**

shuili-system/src/main/java/com/kms/yg/sz/domain/dto/BsSgcSzFxzzDto.java

@@ -0,0 +1,17 @@
+package com.kms.yg.sz.domain.dto;
+
+import com.kms.yg.sz.domain.BsSgcSzFxzz;
+import com.kms.yg.sz.domain.BsSgcSzFxzzcy;
+import lombok.Data;
+
+import java.io.Serializable;
+import java.util.List;
+
+@Data
+public class BsSgcSzFxzzDto extends BsSgcSzFxzz implements Serializable {
+
+
+    private List<BsSgcSzFxzzcy> cy;
+
+
+}