统计场景列表人数

lib/EmergencyService/zdxtEmergencyKnowledgeService/src/main/java/com/zdxt/knowledge/controller/SceneManageController.java

@@ -3,11 +3,16 @@ package com.zdxt.knowledge.controller;
 import com.zdxt.common.ZDResponse;
 import com.zdxt.common.controller.BaseController;
 import com.zdxt.common.util.TableDataInfo;
+import com.zdxt.knowledge.domain.QuestionFre;
 import com.zdxt.knowledge.domain.SceneImportDataModel;
 import com.zdxt.knowledge.domain.SceneManage;
+import com.zdxt.knowledge.domain.SceneRole;
+import com.zdxt.knowledge.service.IQuestionFreService;
 import com.zdxt.knowledge.service.ISceneManageService;
+import com.zdxt.knowledge.service.ISceneRoleService;
 import com.zdxt.knowledge.utils.ExcelUtil;
 import io.swagger.annotations.ApiOperation;
+import org.apache.commons.collections.CollectionUtils;
 import org.apache.shiro.authz.annotation.RequiresPermissions;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.core.io.Resource;
@@ -17,9 +22,11 @@ import org.springframework.ui.ModelMap;
 import org.springframework.web.bind.annotation.*;
 import org.springframework.web.multipart.MultipartFile;
 
+import java.util.Collection;
 import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
+import java.util.stream.Collectors;
 
 
 /**
@@ -36,6 +43,12 @@ public class SceneManageController extends BaseController {
     @Autowired
     private ISceneManageService sceneManageService;
 
+    @Autowired
+    private ISceneRoleService sceneRoleService;
+
+    @Autowired
+    private IQuestionFreService questionFreService;
+
     @RequiresPermissions("project:manage:view")
     @GetMapping()
     public String manage() {
@@ -63,6 +76,19 @@ public class SceneManageController extends BaseController {
     public TableDataInfo listApp(@RequestBody SceneManage sceneManage) {
         startPage();
         List<SceneManage> list = sceneManageService.selectSceneManageListAll(sceneManage);
+        if (CollectionUtils.isNotEmpty(list)) {
+            list = list.stream()
+                    .map(item -> {
+                        SceneRole sceneRole = new SceneRole();
+                        sceneRole.setSceneId(item.getGuid());
+                        item.setRoleTotal(sceneRoleService.selectSceneRoleList(sceneRole).size());
+                        QuestionFre questionFre = new QuestionFre();
+                        questionFre.setSceneId(item.getGuid());
+                        item.setUseTotal((long) questionFreService.selectQuestionFreList(questionFre).size());
+                        return item;
+                    })
+                    .collect(Collectors.toList());
+        }
         return getDataTable(list);
     }
 

lib/EmergencyService/zdxtEmergencyKnowledgeService/src/main/java/com/zdxt/knowledge/domain/QuestionFre.java

@@ -29,6 +29,11 @@ public class QuestionFre {
      */
     private String userId;
 
+    /**
+     * 场景id
+     */
+    private String sceneId;
+
     /**
      * 创建人
      */
@@ -118,4 +123,12 @@ public class QuestionFre {
     public void setUserId(String userId) {
         this.userId = userId;
     }
+
+    public String getSceneId() {
+        return sceneId;
+    }
+
+    public void setSceneId(String sceneId) {
+        this.sceneId = sceneId;
+    }
 }

lib/EmergencyService/zdxtEmergencyKnowledgeService/src/main/java/com/zdxt/knowledge/domain/SceneManage.java

@@ -95,6 +95,12 @@ public class SceneManage extends PageDomain {
     @Excel(name = "使用人数")
     private Long useTotal;
 
+    /**
+     * 演练角色
+     */
+    @Excel(name = "演练角色")
+    private Integer roleTotal;
+
     /**
      * 创建时间
      */
@@ -275,4 +281,12 @@ public class SceneManage extends PageDomain {
     public void setUploadPath(String uploadPath) {
         this.uploadPath = uploadPath;
     }
+
+    public Integer getRoleTotal() {
+        return roleTotal;
+    }
+
+    public void setRoleTotal(Integer roleTotal) {
+        this.roleTotal = roleTotal;
+    }
 }

lib/EmergencyService/zdxtEmergencyKnowledgeService/src/main/resources/mapper/QuestionFreMapper.xml

@@ -8,6 +8,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
         <result property="guid"    column="guid"    />
         <result property="userId"    column="user_id"    />
         <result property="roleId"    column="role_id"    />
+        <result property="sceneId"    column="scene_id"    />
         <result property="creator"    column="creator"    />
         <result property="createTime"    column="create_time"    />
         <result property="updator"    column="updator"    />
@@ -15,7 +16,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
     </resultMap>
 
     <sql id="selectQuestionFreVo">
-        select guid, user_id, role_id, creator, create_time, updator, update_time from question_fre
+        select guid, user_id, role_id, scene_id, creator, create_time, updator, update_time from question_fre
     </sql>
 
     <select id="selectQuestionFreList" parameterType="QuestionFre" resultMap="QuestionFreResult">
@@ -23,6 +24,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
         <where>  
             <if test="userId != null  and userId != ''"> and user_id = #{userId}</if>
             <if test="roleId != null  and roleId != ''"> and role_id = #{roleId}</if>
+            <if test="sceneId != null  and sceneId != ''"> and scene_id = #{sceneId}</if>
             <if test="creator != null  and creator != ''"> and creator = #{creator}</if>
             <if test="updator != null  and updator != ''"> and updator = #{updator}</if>
         </where>
@@ -40,6 +42,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
             <if test="guid != null  and guid != ''">guid,</if>
             <if test="userId != null  and userId != ''">user_id,</if>
             <if test="roleId != null  and roleId != ''">role_id,</if>
+            <if test="sceneId != null  and sceneId != ''">scene_id,</if>
             <if test="creator != null  and creator != ''">creator,</if>
             <if test="createTime != null ">create_time,</if>
             <if test="updator != null  and updator != ''">updator,</if>
@@ -49,6 +52,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
             <if test="guid != null  and guid != ''">#{guid},</if>
             <if test="userId != null  and userId != ''">#{userId},</if>
             <if test="roleId != null  and roleId != ''">#{roleId},</if>
+            <if test="sceneId != null  and sceneId != ''">#{sceneId},</if>
             <if test="creator != null  and creator != ''">#{creator},</if>
             <if test="createTime != null ">#{createTime},</if>
             <if test="updator != null  and updator != ''">#{updator},</if>
@@ -61,6 +65,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
         <trim prefix="SET" suffixOverrides=",">
             <if test="userId != null  and userId != ''">user_id = #{userId},</if>
             <if test="roleId != null  and roleId != ''">role_id = #{roleId},</if>
+            <if test="sceneId != null  and sceneId != ''">scene_id = #{sceneId},</if>
             <if test="creator != null  and creator != ''">creator = #{creator},</if>
             <if test="createTime != null ">create_time = #{createTime},</if>
             <if test="updator != null  and updator != ''">updator = #{updator},</if>

lib/EmergencyService/zdxtEmergencyKnowledgeService/src/main/resources/mapper/SceneManageMapper.xml

@@ -143,7 +143,7 @@ PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN"
 <!--    </delete>-->
 
     <select id="selectSceneManageListAll" parameterType="SceneManage" resultMap="SceneManageResult">
-        select cu.guid, cu.name, cu.sort, cu.status, cu.is_open,  zd.upload_path from scene_manage as cu
+        select cu.guid, cu.name, cu.sort, cu.status, cu.is_open, cu.use_total, zd.upload_path from scene_manage as cu
         inner join zdxt_file as zd
         on cu.guid = zd.business_id
         <where>