项目统计

oa-back/ruoyi-admin/src/main/java/com/ruoyi/web/controller/oa/CmcProjectController.java

@@ -1,13 +1,14 @@
 package com.ruoyi.web.controller.oa;
 
 import java.io.IOException;
-import java.util.ArrayList;
-import java.util.Arrays;
-import java.util.Date;
-import java.util.List;
+import java.util.*;
+import java.util.stream.Collectors;
 import javax.servlet.http.HttpServletResponse;
 
+import com.alibaba.fastjson2.JSONArray;
+import com.alibaba.fastjson2.JSONObject;
 import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.utils.DateUtils;
 import com.ruoyi.common.utils.StringUtils;
 import com.ruoyi.oa.domain.CmcManageDept;
 import com.ruoyi.oa.service.ICmcManageDeptService;
@@ -114,7 +115,7 @@ public class CmcProjectController extends BaseController
     }
 
     /**
-     * 获取cmc项目详细信息
+     * 项目自动编号
      */
     @GetMapping("/projectNumber")
     public AjaxResult getProjectNumberByYear(CmcProject cmcProject)
@@ -125,6 +126,108 @@ public class CmcProjectController extends BaseController
         return success(projectNumber);
     }
 
+    /**
+     * 获取cmc项目统计信息
+     */
+    @GetMapping("/statistic")
+    public AjaxResult getProjectStatistic(CmcProject cmcProject)
+    {
+        JSONObject jsonObject = new JSONObject();
+        JSONArray yearArray = new JSONArray();
+        JSONObject yearObject = new JSONObject();
+        if (cmcProject.getProjectNumber() == null) {
+            //每年项目数量
+            for (int i = 2019; i <= Calendar.getInstance().get(Calendar.YEAR); i++) {
+                cmcProject.setProjectNumber(String.valueOf(i));
+                List<CmcProject> yearList = cmcProjectService.selectCmcProjectList(cmcProject);
+                yearObject.put(String.valueOf(i), yearList.size());
+            }
+            //输入年份院内外项目数量
+            JSONArray sourceArray = new JSONArray();
+            JSONObject sourceObject = new JSONObject();
+            cmcProject.setProjectNumber(null);
+            cmcProject.setProjectSource("1");
+            sourceObject.put("市场", cmcProjectService.selectCmcProjectList(cmcProject).size());
+            cmcProject.setProjectSource("0");
+            sourceObject.put("院内", cmcProjectService.selectCmcProjectList(cmcProject).size());
+            sourceArray.add(sourceObject);
+            jsonObject.put("source", sourceArray);
+            //输入年份各类型项目数量
+            cmcProject.setProjectSource(null);
+            Map<Object, List<CmcProject>> typeGroup = cmcProjectService.selectCmcProjectList(cmcProject).stream().collect(Collectors.groupingBy(item -> Optional.ofNullable(item.getProjectType())));
+            Set<Map.Entry<Object, List<CmcProject>>> typeEntries = typeGroup.entrySet();
+            JSONArray typeArray = new JSONArray();
+            JSONObject typeObject = new JSONObject();
+            typeEntries.forEach(item -> {
+                if (item.getKey() != Optional.empty()) {
+                    String str = item.getKey().toString();
+                    typeObject.put(str.substring(str.indexOf("[") + 1, str.indexOf("]")), item.getValue().size());
+                }
+            });
+            typeArray.add(typeObject);
+            jsonObject.put("type", typeArray);
+            //输入年份各承担部门项目数量
+            JSONArray deptArray = new JSONArray();
+            JSONObject deptObject = new JSONObject();
+            for (int i = 107; i <= 115; i++) {
+                cmcProject.setUndertakingDept(String.valueOf(i));
+                List<CmcProject> deptList = new ArrayList<>();
+                if (i != 113)
+                    deptList = cmcProjectService.selectCmcProjectList(cmcProject);
+                else
+                    deptList = cmcProjectService.selectCmcInvestProjectList(cmcProject);
+                deptObject.put(deptService.selectDeptById((long)i).getDeptName(), deptList.size());
+            }
+            deptArray.add(deptObject);
+            jsonObject.put("dept", deptArray);
+        }
+        else {
+            //输入年份项目数量
+            yearObject.put(cmcProject.getProjectNumber(), cmcProjectService.selectCmcProjectList(cmcProject).size());
+            //输入年份院内外项目数量
+            JSONArray sourceArray = new JSONArray();
+            JSONObject sourceObject = new JSONObject();
+            CmcProject project = new CmcProject();
+            project.setProjectNumber(cmcProject.getProjectNumber() + "W");
+            sourceObject.put("市场", cmcProjectService.selectCmcProjectList(project).size());
+            project.setProjectNumber(cmcProject.getProjectNumber() + "C");
+            sourceObject.put("院内", cmcProjectService.selectCmcProjectList(project).size());
+            sourceArray.add(sourceObject);
+            jsonObject.put("source", sourceArray);
+            //输入年份各类型项目数量
+            JSONArray typeArray = new JSONArray();
+            JSONObject typeObject = new JSONObject();
+            Map<Object, List<CmcProject>> typeGroup = cmcProjectService.selectCmcProjectList(cmcProject).stream().collect(Collectors.groupingBy(item -> Optional.ofNullable(item.getProjectType())));
+            Set<Map.Entry<Object, List<CmcProject>>> typeEntries = typeGroup.entrySet();
+            typeEntries.forEach(item -> {
+                if (item.getKey() != Optional.empty()) {
+                    String str = item.getKey().toString();
+                    typeObject.put(str.substring(str.indexOf("[") + 1, str.indexOf("]")), item.getValue().size());
+                }
+            });
+            typeArray.add(typeObject);
+            jsonObject.put("type", typeArray);
+            //输入年份各承担部门项目数量
+            JSONArray deptArray = new JSONArray();
+            JSONObject deptObject = new JSONObject();
+            project.setProjectNumber(cmcProject.getProjectNumber());
+            for (int i = 107; i <= 115; i++) {
+                project.setUndertakingDept(String.valueOf(i));
+                List<CmcProject> deptList = new ArrayList<>();
+                if (i != 113)
+                    deptList = cmcProjectService.selectCmcProjectList(project);
+                else
+                    deptList = cmcProjectService.selectCmcInvestProjectList(project);
+                deptObject.put(deptService.selectDeptById((long)i).getDeptName(), deptList.size());
+            }
+            deptArray.add(deptObject);
+            jsonObject.put("dept", deptArray);
+        }
+        yearArray.add(yearObject);
+        jsonObject.put("year", yearArray);
+        return success(jsonObject);
+    }
+
     /**
      * 新增cmc项目
      */
@@ -230,4 +333,6 @@ public class CmcProjectController extends BaseController
         }
         return userList;
     }
+
+
 }