From b5f2aab02997ed9e718f20ea9b95304d580ed49d Mon Sep 17 00:00:00 2001
From: yinzuomei <576302893@qq.com>
Date: Tue, 23 Mar 2021 13:56:07 +0800
Subject: [PATCH] =?UTF-8?q?/data/report/project/categorydict=E6=8E=A5?=
 =?UTF-8?q?=E5=8F=A3=E8=B0=83=E6=95=B4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../controller/project/ProjectController.java         | 11 ++++-------
 .../datareport/service/project/ProjectService.java    |  4 ++--
 .../service/project/impl/ProjectServiceImpl.java      |  8 ++++----
 3 files changed, 10 insertions(+), 13 deletions(-)

diff --git a/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/controller/project/ProjectController.java b/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/controller/project/ProjectController.java
index 2bb49d5c25..2f704b6b16 100644
--- a/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/controller/project/ProjectController.java
+++ b/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/controller/project/ProjectController.java
@@ -16,10 +16,7 @@ import com.epmet.project.dto.form.DifficultyRankFormDTO;
 import com.epmet.project.dto.form.ProjectIncrTrendFormDTO;
 import com.epmet.project.dto.result.*;
 import org.springframework.beans.factory.annotation.Autowired;
-import org.springframework.web.bind.annotation.PostMapping;
-import org.springframework.web.bind.annotation.RequestBody;
-import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
+import org.springframework.web.bind.annotation.*;
 
 import java.util.List;
 
@@ -134,13 +131,13 @@ public class ProjectController {
      * 002、项目分类字典查询
      *      查询当前客户的父客户下，项目分类字典
      *
-     * @param tokenDto
+     * @param customerId
      * @return com.epmet.commons.tools.utils.Result<java.util.List<com.epmet.project.dto.result.ProjectCategoryDictResultDTO>>
      * @Author zhangyong
      * @Date 10:41 2021-03-22
      **/
     @PostMapping("categorydict")
-    public Result<List<ProjectCategoryDictResultDTO>> categoryDict(@LoginUser TokenDto tokenDto) {
-        return new Result<List<ProjectCategoryDictResultDTO>>().ok(projectService.getCategoryDict(tokenDto));
+    public Result<List<ProjectCategoryDictResultDTO>> categoryDict(@RequestHeader("CustomerId") String customerId) {
+        return new Result<List<ProjectCategoryDictResultDTO>>().ok(projectService.getCategoryDict(customerId));
     }
 }
diff --git a/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/ProjectService.java b/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/ProjectService.java
index bb2ffedd9c..917113fe12 100644
--- a/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/ProjectService.java
+++ b/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/ProjectService.java
@@ -74,10 +74,10 @@ public interface ProjectService {
     /**
      * 002、项目分类字典查询
      *
-     * @param tokenDto
+     * @param customerId
      * @return java.util.List<com.epmet.project.dto.result.ProjectCategoryDictResultDTO>
      * @Author zhangyong
      * @Date 14:26 2021-03-22
      **/
-    List<ProjectCategoryDictResultDTO> getCategoryDict(TokenDto tokenDto);
+    List<ProjectCategoryDictResultDTO> getCategoryDict(String customerId);
 }
diff --git a/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/impl/ProjectServiceImpl.java b/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/impl/ProjectServiceImpl.java
index 4da82573c9..4d2df70470 100644
--- a/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/impl/ProjectServiceImpl.java
+++ b/epmet-module/data-report/data-report-server/src/main/java/com/epmet/datareport/service/project/impl/ProjectServiceImpl.java
@@ -358,12 +358,12 @@ public class ProjectServiceImpl implements ProjectService {
     }
 
     @Override
-    public List<ProjectCategoryDictResultDTO> getCategoryDict(TokenDto tokenDto) {
+    public List<ProjectCategoryDictResultDTO> getCategoryDict(String customerId) {
         List<ProjectCategoryDictResultDTO> result = new ArrayList<>();
-        Result<String> parentCustomerId = operCrmOpenFeignClient.getExternalAndParentCustomerId(tokenDto.getCustomerId());
+        Result<String> parentCustomerId = operCrmOpenFeignClient.getExternalAndParentCustomerId(customerId);
         if (!parentCustomerId.success() && null == parentCustomerId.getData()) {
             logger.warn(String.format("调用%s服务查询外部客户的 父级客户ID 失败,入参%s", ServiceConstant.OPER_CRM_SERVER,
-                    JSON.toJSONString(tokenDto.getCustomerId())));
+                    JSON.toJSONString(customerId)));
             return result;
         }
         CustomerCategoryDTO argDto = new CustomerCategoryDTO();
@@ -373,7 +373,7 @@ public class ProjectServiceImpl implements ProjectService {
         Result<List<ProjectCategoryDictResultDTO>> parentCategoryDict = govIssueOpenFeignClient.getCustomerCategoryDict(argDto);
         if (!parentCategoryDict.success()) {
             logger.warn(String.format("调用%s服务查询当前客户的父客户下，项目分类字典失败,入参%s", ServiceConstant.GOV_ISSUE_SERVER,
-                    JSON.toJSONString(tokenDto.getCustomerId())));
+                    JSON.toJSONString(customerId)));
         } else {
             result = parentCategoryDict.getData();
             // 获取客户二级分类