概览

dsk-module/dsk-biz-api/src/main/java/com/dsk/search/service/MarketAnalysisService.java

@@ -4,9 +4,13 @@ import cn.hutool.core.bean.BeanUtil;
 import com.alibaba.fastjson2.JSONObject;
 import com.dsk.common.core.domain.AjaxResult;
 import com.dsk.common.utils.DskOpenApiUtil;
+import com.dsk.jsk.service.EnterpriseService;
+import org.apache.commons.collections4.MapUtils;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
+import org.springframework.util.ObjectUtils;
 
+import java.util.List;
 import java.util.Map;
 
 /**
@@ -20,6 +24,9 @@ public class MarketAnalysisService {
     @Autowired
     private DskOpenApiUtil dskOpenApiUtil;
 
+    @Autowired
+    private EnterpriseService enterpriseService;
+
     public AjaxResult areaGroupByProvince() {
         Map<String, Object> map = dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/areaGroupByProvince", new JSONObject());
         return BeanUtil.toBean(map, AjaxResult.class);
@@ -117,11 +124,27 @@ public class MarketAnalysisService {
 
     public AjaxResult combineRecentlyBid(JSONObject object) {
         Map<String, Object> map = dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/combine/recentlyBid", object);
+        if (!ObjectUtils.isEmpty(map.get("data"))) {
+            List<Map<String, Object>> list = (List<Map<String, Object>>) map.get("data");
+            list.parallelStream().forEach(res -> {
+                Integer companyId = MapUtils.getInteger(res, "companyId");
+                String companyName = MapUtils.getString(res, "companyName");
+                res.put("uipId", enterpriseService.getUipIdByCompanyNameOrCompanyId(companyName, companyId));
+            });
+        }
         return BeanUtil.toBean(map, AjaxResult.class);
     }
 
     public AjaxResult combineBidByYear(JSONObject object) {
         Map<String, Object> map = dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/combine/bidByYear", object);
+        if (!ObjectUtils.isEmpty(map.get("data"))) {
+            List<Map<String, Object>> list = (List<Map<String, Object>>) map.get("data");
+            list.parallelStream().forEach(res -> {
+                Integer companyId = MapUtils.getInteger(res, "tendereeId");
+                String companyName = MapUtils.getString(res, "tenderee");
+                res.put("uipId", enterpriseService.getUipIdByCompanyNameOrCompanyId(companyName, companyId));
+            });
+        }
         return BeanUtil.toBean(map, AjaxResult.class);
     }
 
@@ -132,6 +155,14 @@ public class MarketAnalysisService {
 
     public AjaxResult combineGroupByType(JSONObject object) {
         Map<String, Object> map = dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/combine/groupByType", object);
+        if (!ObjectUtils.isEmpty(map.get("data"))) {
+            List<Map<String, Object>> list = (List<Map<String, Object>>) map.get("data");
+            list.parallelStream().forEach(res -> {
+                Integer companyId = MapUtils.getInteger(res, "companyId");
+                String companyName = MapUtils.getString(res, "companyName");
+                res.put("uipId", enterpriseService.getUipIdByCompanyNameOrCompanyId(companyName, companyId));
+            });
+        }
         return BeanUtil.toBean(map, AjaxResult.class);
     }
 
@@ -142,6 +173,14 @@ public class MarketAnalysisService {
 
     public AjaxResult combinePeojectTop(JSONObject object) {
         Map<String, Object> map = dskOpenApiUtil.requestBody("/nationzj/marketAnalysis/combine/peojectTop", object);
+        if (!ObjectUtils.isEmpty(map.get("data"))) {
+            List<Map<String, Object>> list = (List<Map<String, Object>>) map.get("data");
+            list.parallelStream().forEach(res -> {
+                Integer companyId = MapUtils.getInteger(res, "projectUnitId");
+                String companyName = MapUtils.getString(res, "projectUnit");
+                res.put("uipId", enterpriseService.getUipIdByCompanyNameOrCompanyId(companyName, companyId));
+            });
+        }
         return BeanUtil.toBean(map, AjaxResult.class);
     }
 }