From 64efb660b2c119c00432434c0f651f8996483f18 Mon Sep 17 00:00:00 2001
From: fuliqi <fuliqi@qq.com>
Date: 星期五, 21 二月 2025 15:51:13 +0800
Subject: [PATCH] OSD大华数据格式+数据中心重点标签统计数

---
 ycl-server/src/main/java/com/ycl/platform/service/impl/ContractResultServiceImpl.java |   51 +++++++++++++++++++++++++++++++++++++++------------
 1 files changed, 39 insertions(+), 12 deletions(-)

diff --git a/ycl-server/src/main/java/com/ycl/platform/service/impl/ContractResultServiceImpl.java b/ycl-server/src/main/java/com/ycl/platform/service/impl/ContractResultServiceImpl.java
index 4a77bd9..80320a1 100644
--- a/ycl-server/src/main/java/com/ycl/platform/service/impl/ContractResultServiceImpl.java
+++ b/ycl-server/src/main/java/com/ycl/platform/service/impl/ContractResultServiceImpl.java
@@ -1,6 +1,5 @@
 package com.ycl.platform.service.impl;
 
-import com.baomidou.mybatisplus.extension.conditions.query.LambdaQueryChainWrapper;
 import com.baomidou.mybatisplus.extension.conditions.update.LambdaUpdateChainWrapper;
 import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
 import com.ycl.platform.domain.entity.ContractResult;
@@ -9,15 +8,20 @@
 import com.ycl.platform.mapper.ContractResultMapper;
 import com.ycl.platform.mapper.ContractResultRecordMapper;
 import com.ycl.platform.service.IContractResultService;
-import com.ycl.platform.service.ITContractService;
-import com.ycl.platform.service.YwUnitService;
+import com.ycl.system.entity.BaseEntity;
+import com.ycl.system.entity.SysRole;
+import com.ycl.system.entity.SysUser;
 import com.ycl.utils.SecurityUtils;
+import enumeration.general.PublishType;
+import lombok.RequiredArgsConstructor;
 import org.springframework.beans.factory.annotation.Autowired;
 import org.springframework.stereotype.Service;
 import utils.DateUtils;
 
 import java.util.Arrays;
 import java.util.List;
+import java.util.Map;
+import java.util.Objects;
 
 /**
  * 鑰冩牳缁撴灉Service涓氬姟灞傚鐞�
@@ -26,11 +30,11 @@
  * @date 2024-04-01
  */
 @Service
+@RequiredArgsConstructor
 public class ContractResultServiceImpl extends ServiceImpl<ContractResultMapper, ContractResult> implements IContractResultService {
-    @Autowired
-    private ContractResultMapper checkResultMapper;
-    @Autowired
-    private ContractResultRecordMapper contractResultRecordMapper;
+
+    private final ContractResultMapper checkResultMapper;
+    private final ContractResultRecordMapper contractResultRecordMapper;
 
 
     /**
@@ -52,8 +56,31 @@
      */
     @Override
     public List<ContractResultVO> selectCheckResultList(ContractResultVO contractResult) {
+        //閬垮厤鏌ヨ鏉′欢澶辨晥
+        if (contractResult.getUnitId() == null) {
+            Integer unitId = SecurityUtils.getUnitId();
+            Long uid = null;
+            if (Objects.nonNull(unitId)) {
+                uid = Long.valueOf(unitId);
+            }
+            contractResult.setUnitId(uid);
+        }
+        roleControl(contractResult);
         return checkResultMapper.selectCheckResultList(contractResult);
     }
+
+    private void roleControl(ContractResultVO contractResult) {
+        List<SysRole> roles = SecurityUtils.getLoginUser().getUser().getRoles();
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        if (!user.isAdmin()) {
+            for (SysRole role : roles) {
+                if (role.getPermissions().contains("check:contract:role:publish")) {
+                    contractResult.setPublish(1);
+                }
+            }
+        }
+    }
+
 
     /**
      * 鏂板鑰冩牳缁撴灉
@@ -108,17 +135,17 @@
     }
 
     @Override
-    public List<ContractResultRecord> selectCheckResultRecordList(Long resultId) {
-        return new LambdaQueryChainWrapper<>(contractResultRecordMapper)
-                .eq(ContractResultRecord::getResultId, resultId)
-                .list();
+    public List<ContractResultRecord> selectCheckResultRecordList(Long resultId, Long contractId) {
+        return checkResultMapper.selectCheckResultRecordList(resultId, contractId);
     }
 
     @Override
     public Boolean publish(Long id) {
+        ContractResult contractResult = checkResultMapper.selectById(id);
+
         return new LambdaUpdateChainWrapper<>(checkResultMapper)
                 .eq(ContractResult::getId, id)
-                .set(ContractResult::getPublish, 1)
+                .set(ContractResult::getPublish, contractResult.getPublish() == 0 ? 1 : 0)
                 .set(ContractResult::getPublishId, SecurityUtils.getLoginUser().getUserId())
                 .update();
     }

--
Gitblit v1.8.0