调整

src/main/java/com/gxwebsoft/tower/controller/TowerProjectController.java

@@ -1,20 +1,18 @@
 package com.gxwebsoft.tower.controller;
 
+import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
+import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
+import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.gxwebsoft.common.core.annotation.OperationLog;
 import com.gxwebsoft.common.core.web.ApiResult;
 import com.gxwebsoft.common.core.web.BaseController;
 import com.gxwebsoft.common.core.web.BatchParam;
 import com.gxwebsoft.common.core.web.PageResult;
 import com.gxwebsoft.common.system.entity.User;
-import com.gxwebsoft.tower.entity.TowerProject;
-import com.gxwebsoft.tower.entity.TowerProjectOrder;
-import com.gxwebsoft.tower.entity.TowerProjectPlace;
-import com.gxwebsoft.tower.entity.TowerProjectUser;
+import com.gxwebsoft.common.system.service.UserService;
+import com.gxwebsoft.tower.entity.*;
 import com.gxwebsoft.tower.param.TowerProjectParam;
-import com.gxwebsoft.tower.service.TowerProjectOrderService;
-import com.gxwebsoft.tower.service.TowerProjectPlaceService;
-import com.gxwebsoft.tower.service.TowerProjectService;
-import com.gxwebsoft.tower.service.TowerProjectUserService;
+import com.gxwebsoft.tower.service.*;
 import io.swagger.annotations.Api;
 import io.swagger.annotations.ApiOperation;
 import org.springframework.security.access.prepost.PreAuthorize;
@@ -22,8 +20,7 @@ import org.springframework.transaction.annotation.Transactional;
 import org.springframework.web.bind.annotation.*;
 
 import javax.annotation.Resource;
-import java.util.List;
-import java.util.Set;
+import java.util.*;
 
 /**
  * 项目管理控制器
@@ -43,13 +40,51 @@ public class TowerProjectController extends BaseController {
     private TowerProjectUserService towerProjectUserService;
     @Resource
     private TowerProjectPlaceService towerProjectPlaceService;
+    @Resource
+    private TowerEquipmentService equipmentService;
+    @Resource
+    private UserService userService;
+
+    @PreAuthorize("hasAuthority('tower:towerEquipment:list')")
+    @OperationLog
+    @ApiOperation("项目数据")
+    @GetMapping("/data")
+    public ApiResult<Map<String, Integer>> data() {
+        Map<String, Integer> data = new HashMap<>();
+        Integer totalNum = towerProjectService.count();
+        Integer startedNum = towerProjectService.count(
+                new LambdaQueryWrapper<TowerProject>()
+                        .eq(TowerProject::getProjectStatus, "建设中")
+        );
+        Integer notStartNum = towerProjectService.count(
+                new LambdaQueryWrapper<TowerProject>()
+                        .eq(TowerProject::getProjectStatus, "未开工")
+        );
+        data.put("totalNum", totalNum);
+        data.put("startedNum", startedNum);
+        data.put("notStartNum", notStartNum);
+        return success(data);
+    }
 
     @PreAuthorize("hasAuthority('tower:towerProject:list')")
     @ApiOperation("分页查询项目管理")
     @GetMapping("/page")
     public ApiResult<PageResult<TowerProject>> page(TowerProjectParam param) {
         // 使用关联查询
-        return success(towerProjectService.pageRel(param));
+        PageResult<TowerProject> list = towerProjectService.pageRel(param);
+        for (TowerProject project : list.getList()) {
+            project.setEquipmentNum(towerProjectOrderService.count(
+                    new LambdaQueryWrapper<TowerProjectOrder>()
+                            .eq(TowerProjectOrder::getProjectId, project.getProjectId())
+            ));
+            project.setOrders(
+                    new HashSet<>(towerProjectOrderService.list(
+                            new LambdaQueryWrapper<TowerProjectOrder>().eq(TowerProjectOrder::getProjectId, project.getProjectId())
+                    ))
+            );
+            if (!project.getDirector().equals("0")) project.setDirectorUser(userService.getById(project.getDirector()));
+        }
+        return success(list);
     }
 
     @PreAuthorize("hasAuthority('tower:towerProject:list')")
@@ -68,6 +103,18 @@ public class TowerProjectController extends BaseController {
         return success(towerProjectService.getByIdRel(id));
     }
 
+    @ApiOperation("添加项目管理")
+    @PostMapping("/new")
+    public ApiResult<?> newProject(@RequestBody TowerProject towerProject) {
+        // 记录当前登录用户id
+        User loginUser = getLoginUser();
+        if (loginUser != null) {
+            towerProject.setUserId(loginUser.getUserId());
+        }
+        towerProjectService.save(towerProject);
+        return success(towerProject);
+    }
+
     @PreAuthorize("hasAuthority('tower:towerProject:save')")
     @Transactional(rollbackFor = {Exception.class})
     @ApiOperation("添加项目管理")
@@ -76,42 +123,42 @@ public class TowerProjectController extends BaseController {
         // 记录当前登录用户id
         User loginUser = getLoginUser();
         if (loginUser != null) {
-          towerProject.setUserId(loginUser.getUserId());
+            towerProject.setUserId(loginUser.getUserId());
         }
-      System.out.println("towerProject = " + towerProject.getPlaces2());
+        System.out.println("towerProject = " + towerProject.getPlaces2());
         if (towerProjectService.save(towerProject)) {
-          final Integer projectId = towerProject.getProjectId();
+            final Integer projectId = towerProject.getProjectId();
             // 批量添加项目成员
             if (!towerProject.getUsers().isEmpty()) {
-              final Set<TowerProjectUser> users = towerProject.getUsers();
-              users.forEach(d->{
-                d.setProjectId(projectId);
-              });
-              towerProjectUserService.saveBatch(users);
+                final Set<TowerProjectUser> users = towerProject.getUsers();
+                users.forEach(d -> {
+                    d.setProjectId(projectId);
+                });
+                towerProjectUserService.saveBatch(users);
             }
             // 批量添加项目订单
             if (!towerProject.getOrders().isEmpty()) {
-              final Set<TowerProjectOrder> orders = towerProject.getOrders();
-              orders.forEach(d->{
-                d.setProjectId(projectId);
-              });
-              towerProjectOrderService.saveBatch(orders);
+                final Set<TowerProjectOrder> orders = towerProject.getOrders();
+                orders.forEach(d -> {
+                    d.setProjectId(projectId);
+                });
+                towerProjectOrderService.saveBatch(orders);
             }
             // 批量添加检查部位配置
             if (!towerProject.getPlaces1().isEmpty()) {
-              final Set<TowerProjectPlace> places1 = towerProject.getPlaces1();
-              places1.forEach(d->{
-                d.setProjectId(projectId);
-              });
-              towerProjectPlaceService.saveBatch(places1);
+                final Set<TowerProjectPlace> places1 = towerProject.getPlaces1();
+                places1.forEach(d -> {
+                    d.setProjectId(projectId);
+                });
+                towerProjectPlaceService.saveBatch(places1);
             }
             // 批量添加保养部位配置
             if (!towerProject.getPlaces2().isEmpty()) {
-              final Set<TowerProjectPlace> places2 = towerProject.getPlaces2();
-              places2.forEach(d->{
-                d.setProjectId(projectId);
-              });
-              towerProjectPlaceService.saveBatch(places2);
+                final Set<TowerProjectPlace> places2 = towerProject.getPlaces2();
+                places2.forEach(d -> {
+                    d.setProjectId(projectId);
+                });
+                towerProjectPlaceService.saveBatch(places2);
             }
             return success("添加成功");
         }
@@ -127,7 +174,25 @@ public class TowerProjectController extends BaseController {
             // 批量修改项目成员
             towerProjectUserService.saveOrUpdateBatch(towerProject.getUsers());
             // 批量修改项目订单
-            towerProjectOrderService.saveOrUpdateBatch(towerProject.getOrders());
+            Set<TowerProjectOrder> projectOrders = towerProject.getOrders();
+            if (towerProject.getProjectStatus().equals("建设中")) {
+                for (TowerProjectOrder order : projectOrders) {
+                    equipmentService.update(
+                            new LambdaUpdateWrapper<TowerEquipment>()
+                                    .eq(TowerEquipment::getFilingNo, order.getFilingNo())
+                                    .set(TowerEquipment::getStatus, 4)
+                    );
+                }
+            } else if (towerProject.getProjectStatus().equals("完工")) {
+                for (TowerProjectOrder order : projectOrders) {
+                    equipmentService.update(
+                            new LambdaUpdateWrapper<TowerEquipment>()
+                                    .eq(TowerEquipment::getFilingNo, order.getFilingNo())
+                                    .set(TowerEquipment::getStatus, 0)
+                    );
+                }
+            }
+            towerProjectOrderService.saveOrUpdateBatch(projectOrders);
             // 批量修改检查/保养部位配置
             towerProjectPlaceService.saveOrUpdateBatch(towerProject.getPlaces1());
             towerProjectPlaceService.saveOrUpdateBatch(towerProject.getPlaces2());