From 8e2d20f3e4fbf754610fdf517ef34e86be353332 Mon Sep 17 00:00:00 2001
From: huangzj <826100833@qq.com>
Date: Fri, 4 Aug 2023 11:43:44 +0800
Subject: [PATCH] =?UTF-8?q?=E6=8F=90=E4=BA=A4?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/CsEngineeringServiceImpl.java        | 20 ++++++++++++++++++-
 1 file changed, 19 insertions(+), 1 deletion(-)

diff --git a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java
index b12d137..fbc7b64 100644
--- a/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java
+++ b/cs-device/cs-device-boot/src/main/java/com/njcn/csdevice/service/impl/CsEngineeringServiceImpl.java
@@ -1,6 +1,7 @@
 package com.njcn.csdevice.service.impl;
 
 import cn.hutool.core.collection.CollectionUtil;
+import com.alibaba.fastjson.JSONArray;
 import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
 import com.baomidou.mybatisplus.core.metadata.IPage;
 import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
@@ -17,17 +18,21 @@ import com.njcn.csdevice.pojo.param.CsEngineeringQueryParm;
 import com.njcn.csdevice.pojo.po.CsEngineeringPO;
 import com.njcn.csdevice.pojo.po.CsEngineeringUserPO;
 import com.njcn.csdevice.pojo.po.CsLedger;
+import com.njcn.csdevice.pojo.po.CsMarketData;
 import com.njcn.csdevice.pojo.vo.CsEngineeringVO;
 import com.njcn.csdevice.service.CsEngineeringService;
+import com.njcn.csdevice.service.CsMarketDataService;
 import com.njcn.redis.utils.RedisUtil;
 import com.njcn.system.api.AreaFeignClient;
 import com.njcn.system.pojo.po.Area;
+import com.njcn.user.enums.AppRoleEnum;
 import com.njcn.web.utils.RequestUtil;
 import lombok.RequiredArgsConstructor;
 import org.apache.commons.lang.StringUtils;
 import org.springframework.beans.BeanUtils;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
+import org.springframework.util.CollectionUtils;
 
 import java.util.ArrayList;
 import java.util.List;
@@ -54,7 +59,7 @@ public class CsEngineeringServiceImpl extends ServiceImpl<CsEngineeringMapper, C
     private final CsEngineeringUserMapper  csEngineeringUserMapper;
     private final RoleEngineerDevServiceImpl roleEngineerDevService;
 
-
+    private final CsMarketDataService csMarketDataService;
 
 
 
@@ -81,6 +86,19 @@ public class CsEngineeringServiceImpl extends ServiceImpl<CsEngineeringMapper, C
         po.setUserId(RequestUtil.getUserIndex());
 
         csEngineeringUserMapper.insert(po);
+        //工程人员新增工程，默认把工程加入关注列表
+        String role = RequestUtil.getUserRole();
+        List<String> strings = JSONArray.parseArray(role, String.class);
+        if(CollectionUtils.isEmpty(strings)){
+            throw new BusinessException(AlgorithmResponseEnum.UNKNOW_ROLE);
+        }
+        role=strings.get(0);
+        if(Objects.equals(role, AppRoleEnum.ENGINEERING_USER.getCode())){
+            CsMarketData csMarketData = new CsMarketData();
+            csMarketData.setEngineerId(csEngineeringPO.getId());
+            csMarketData.setUserId(RequestUtil.getUserIndex());
+            csMarketDataService.save(csMarketData);
+        }
         return save;
     }