bug fix #6730

src/main/java/com/huiju/estateagents/center/taUser/mapper/TaUserMapper.java

     Integer validateWeight(@Param("buildingIds") List<String> buildingIds, @Param("buildingId") String buildingId, @Param("weight") String weight, @Param("userId")Integer userId);
 
     TaUser getAdminByAppID(@Param("appid") String appID);
+
+    List<String> getBuildingIdsOf(@Param("userId") Integer userId,@Param("buildingId") String buildingId);
 }

src/main/java/com/huiju/estateagents/controller/TsConsultantKpiController.java

 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
 import com.huiju.estateagents.base.BaseController;
 import com.huiju.estateagents.base.ResponseBean;
+import com.huiju.estateagents.center.taUser.entity.TaUser;
 import com.huiju.estateagents.common.DateUtils;
 import com.huiju.estateagents.common.ExcelUtils;
 import com.huiju.estateagents.common.StringUtils;
 import com.huiju.estateagents.excel.ConsultantKPIExport;
 import com.huiju.estateagents.excel.TaActivityDynamicEnlistExport;
 import com.huiju.estateagents.excel.handler.CustomCellWriteHandler;
+import io.swagger.models.auth.In;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import org.springframework.beans.factory.annotation.Autowired;
         }
 
         Integer orgId = getOrgId(request);
+        Integer userId = getUserId(request);
         try {
             // 汇总明细
-            IPage<TsConsultantKpi> result = iTsConsultantKpiService.stsKPIDaily(pageNum, pageSize, orgId, buildingId, startDate, endDate, StringUtils.humpToLine(asc), StringUtils.humpToLine(desc));
+            IPage<TsConsultantKpi> result = iTsConsultantKpiService.stsKPIDaily(pageNum, pageSize, orgId, userId, buildingId, startDate, endDate, StringUtils.humpToLine(asc), StringUtils.humpToLine(desc));
             // 汇总
-            TsConsultantKpi total = iTsConsultantKpiService.stsKPITotalByOrg(orgId, buildingId, startDate, endDate);
+            TsConsultantKpi total = iTsConsultantKpiService.stsKPITotalByOrg(orgId, userId, buildingId, startDate, endDate);
             return ResponseBean.success(new HashMap<String, Object>(){{
                 put("paged", result);
                 put("total", total);
         }
 
         Integer orgId = getOrgId(request);
+        Integer userId = getUserId(request);
         try {
 
             // 汇总明细
-            List<ConsultantKPIExport> result = iTsConsultantKpiService.stsKPIDailyExport(orgId, buildingId, startDate, endDate);
+            List<ConsultantKPIExport> result = iTsConsultantKpiService.stsKPIDailyExport(orgId, userId, buildingId, startDate, endDate);
             ExcelUtils.flush(response, ConsultantKPIExport.class, result, "置业顾问KPI统计" + DateUtils.today());
         } catch (Exception e){
 //            e.printStackTrace();

src/main/java/com/huiju/estateagents/mapper/TsConsultantKpiMapper.java

 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.huiju.estateagents.entity.TaPerson;
-import com.huiju.estateagents.entity.TaShareCount;
 import com.huiju.estateagents.entity.TsConsultantKpi;
 import com.baomidou.mybatisplus.core.mapper.BaseMapper;
 import com.huiju.estateagents.excel.ConsultantKPIExport;
 @Mapper
 public interface TsConsultantKpiMapper extends BaseMapper<TsConsultantKpi> {
 
+    /**
+     * 置业KPI 汇总
+     * @param page
+     * @param orgId
+     * @param buildingIds
+     * @param startDate
+     * @param endDate
+     * @param asc
+     * @param desc
+     * @return
+     */
     IPage<TsConsultantKpi> stsKPIDaily(IPage<TsConsultantKpi> page,
                                        @Param("orgId") Integer orgId,
-                                       @Param("buildingId") String buildingId,
+                                       @Param("buildingIds") List<String> buildingIds,
                                        @Param("startDate") String startDate,
                                        @Param("endDate") String endDate,
                                        @Param("asc") String asc,
                                        @Param("desc") String desc);
 
-    TsConsultantKpi stsKPITotalByOrg(@Param("orgId") Integer orgId, @Param("buildingId") String buildingId, @Param("startDate") String startDate, @Param("endDate") String endDate);
+    TsConsultantKpi stsKPITotalByOrg(@Param("orgId") Integer orgId,
+                                     @Param("buildingIds") List<String> buildingIds,
+                                     @Param("startDate") String startDate,
+                                     @Param("endDate") String endDate);
 
-    List<ConsultantKPIExport> stsKPIDailyExport(@Param("orgId") Integer orgId, @Param("buildingId") String buildingId, @Param("startDate") String startDate, @Param("endDate") String endDate);
+    List<ConsultantKPIExport> stsKPIDailyExport(@Param("orgId") Integer orgId,
+                                                @Param("buildingIds") List<String> buildingIds,
+                                                @Param("startDate") String startDate,
+                                                @Param("endDate") String endDate);
 
     /**
      * 统计当前时间段内的, 已经固化的置业数据的所有客户
      * @param orgId
-     * @param buildingId
+     * @param buildingIds
      * @param startDate
      * @param endDate
      * @return
      */
-    Integer stsAllCustomersByOrg(@Param("orgId") Integer orgId, @Param("buildingId") String buildingId, @Param("startDate") String startDate, @Param("endDate") String endDate);
+    Integer stsAllCustomersByOrg(@Param("orgId") Integer orgId,
+                                 @Param("buildingIds") List<String> buildingIds,
+                                 @Param("startDate") String startDate,
+                                 @Param("endDate") String endDate);
 
 
     /**

src/main/java/com/huiju/estateagents/service/ITsConsultantKpiService.java

  */
 public interface ITsConsultantKpiService extends IService<TsConsultantKpi> {
 
-    IPage<TsConsultantKpi> stsKPIDaily(Integer pageNum, Integer pageSize, Integer orgId, String buildingId, String startDate, String endDate, String asc, String desc);
+    IPage<TsConsultantKpi> stsKPIDaily(Integer pageNum, Integer pageSize, Integer orgId, Integer userId, String buildingId, String startDate, String endDate, String asc, String desc);
 
-    TsConsultantKpi stsKPITotalByOrg(Integer orgId, String buildingId, String startDate, String endDate);
+    TsConsultantKpi stsKPITotalByOrg(Integer orgId, Integer userId, String buildingId, String startDate, String endDate);
 
-    List<ConsultantKPIExport> stsKPIDailyExport(Integer orgId, String buildingId, String startDate, String endDate);
+    List<ConsultantKPIExport> stsKPIDailyExport(Integer orgId, Integer userId, String buildingId, String startDate, String endDate);
 
     IPage<TaRecommendCustomer> getCustomerListOfConsultant(IPage<TaRecommendCustomer> page, Integer orgId, String userId, String buildingId, LocalDateTime startDate, LocalDateTime endDate);
 

src/main/java/com/huiju/estateagents/service/impl/TsConsultantKpiServiceImpl.java

 
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
+import com.huiju.estateagents.center.taUser.entity.TaUser;
+import com.huiju.estateagents.center.taUser.mapper.TaUserMapper;
 import com.huiju.estateagents.common.StringUtils;
 import com.huiju.estateagents.entity.*;
 import com.huiju.estateagents.excel.ConsultantKPIExport;
 import org.springframework.stereotype.Service;
 
 import java.time.LocalDateTime;
+import java.util.ArrayList;
 import java.util.List;
 
 /**
     @Autowired
     TsConsultantKpiMapper tsConsultantKpiMapper;
 
+    @Autowired
+    TaUserMapper taUserMapper;
+
     @Autowired
     TaRecommendCustomerMapper taRecommendCustomerMapper;
 
+    /**
+     * 获取用户授权楼盘
+     * 如果 buildingId 参数存在, 则校验参数是否在授权范围内
+     *
+     * @param userId
+     * @param buildingId
+     * @return List 长度 0 代表未授权楼盘, null 代表管理员, 不校验楼盘授权
+     */
+    public List<String> getBuildingListOf(Integer userId, String buildingId) {
+        List<String> unAuthrized = new ArrayList<String>(){{
+            add("*");   // 构造SQL  building_id in ('*') , 目的是为了阻止检索到数据
+        }};
+
+        TaUser taUser = taUserMapper.selectById(userId);
+        if (taUser == null) {
+            return unAuthrized;
+        }
+
+        if (taUser.getIsAdmin()) {
+            return null;
+        }
+
+        List<String> buildings = taUserMapper.getBuildingIdsOf(userId, buildingId);
+        return null == buildings || buildings.size() == 0 ? unAuthrized : buildings;
+    }
+
     @Override
-    public IPage<TsConsultantKpi> stsKPIDaily(Integer pageNum, Integer pageSize, Integer orgId, String buildingId, String startDate, String endDate, String asc, String desc) {
+    public IPage<TsConsultantKpi> stsKPIDaily(Integer pageNum, Integer pageSize, Integer orgId, Integer userId, String buildingId, String startDate, String endDate, String asc, String desc) {
         if (StringUtils.isEmpty(asc) && StringUtils.isEmpty(desc)) {
             desc = "new_persons";
         }
 
+        List<String> buildingIds =  getBuildingListOf(userId, buildingId);
         IPage<TsConsultantKpi> page = new Page<>(pageNum, pageSize);
-        return tsConsultantKpiMapper.stsKPIDaily(page, orgId, buildingId, startDate, endDate, asc, desc);
+        return tsConsultantKpiMapper.stsKPIDaily(page, orgId, buildingIds, startDate, endDate, asc, desc);
     }
 
     @Override
-    public TsConsultantKpi stsKPITotalByOrg(Integer orgId, String buildingId, String startDate, String endDate) {
+    public TsConsultantKpi stsKPITotalByOrg(Integer orgId, Integer userId, String buildingId, String startDate, String endDate) {
+        List<String> buildingIds =  getBuildingListOf(userId, buildingId);
+
         // 合计数据, 但是不包含 客户总计 列
-        TsConsultantKpi result = tsConsultantKpiMapper.stsKPITotalByOrg(orgId, buildingId, startDate, endDate);
+        TsConsultantKpi result = tsConsultantKpiMapper.stsKPITotalByOrg(orgId, buildingIds, startDate, endDate);
 
         // 统计当前时间段内的已经固化的置业的客户
-        Integer totalPersons = tsConsultantKpiMapper.stsAllCustomersByOrg(orgId, buildingId, startDate, endDate);
+        Integer totalPersons = tsConsultantKpiMapper.stsAllCustomersByOrg(orgId, buildingIds, startDate, endDate);
         result.setTotalPersons(totalPersons);
 
         return result;
     }
 
     @Override
-    public List<ConsultantKPIExport> stsKPIDailyExport(Integer orgId, String buildingId, String startDate, String endDate) {
-        return tsConsultantKpiMapper.stsKPIDailyExport(orgId, buildingId, startDate, endDate);
+    public List<ConsultantKPIExport> stsKPIDailyExport(Integer orgId, Integer userId, String buildingId, String startDate, String endDate) {
+        List<String> buildingIds =  getBuildingListOf(userId, buildingId);
+        return tsConsultantKpiMapper.stsKPIDailyExport(orgId, buildingIds, startDate, endDate);
     }
 
     @Override

src/main/resources/mapper/TaUserMapper.xml

         AND m.is_admin = 1
     </select>
 
+    <select id="getBuildingIdsOf" resultType="java.lang.String">
+        select building_id from ta_person_building
+        where user_id = #{userId}
+        <if test="buildingId != null and buildingId != ''">
+            and building_id != #{buildingId}
+        </if>
+    </select>
+
 </mapper>

src/main/resources/mapper/TsConsultantKpiMapper.xml

             WHERE
                 t.org_id = #{orgId}
                 AND t.statis_date BETWEEN #{startDate} AND #{endDate}
-            <if test="buildingId != null and buildingId != ''">
-                AND t.building_id = #{buildingId}
+            <if test="buildingIds != null">
+                AND t.building_id in
+                <foreach item="buildingId" index="index" collection="buildingIds"
+                         open="(" separator="," close=")">
+                    #{buildingId}
+                </foreach>
             </if>
             GROUP BY
                 t.user_id,
         WHERE
             t.org_id = #{orgId}
             AND t.statis_date BETWEEN #{startDate} AND #{endDate}
-        <if test="buildingId != null and buildingId != ''">
-            AND t.building_id = #{buildingId}
+        <if test="buildingIds != null">
+            AND t.building_id in
+            <foreach item="buildingId" index="index" collection="buildingIds"
+                     open="(" separator="," close=")">
+                #{buildingId}
+            </foreach>
         </if>
     </select>
 
         WHERE
             t.org_id = #{orgId}
         AND t.statis_date BETWEEN #{startDate} AND #{endDate}
-        <if test="buildingId != null and buildingId != ''">
-            AND t.building_id = #{buildingId}
-        </if>
+    <if test="buildingIds != null">
+        AND t.building_id in
+        <foreach item="buildingId" index="index" collection="buildingIds"
+                 open="(" separator="," close=")">
+            #{buildingId}
+        </foreach>
+    </if>
         GROUP BY
             t.user_id,
             t.building_id
             p.org_id = #{orgId}
             AND p.`status` > 0
             AND t.statis_date BETWEEN #{startDate} AND #{endDate}
-        <if test="buildingId != null and buildingId != ''">
-            AND t.building_id = #{buildingId}
+        <if test="buildingIds != null">
+            AND t.building_id in
+            <foreach item="buildingId" index="index" collection="buildingIds"
+                     open="(" separator="," close=")">
+                #{buildingId}
+            </foreach>
         </if>
     </select>