From 971f04a71aa3feca69255f5a4f21b68c0bb2004f Mon Sep 17 00:00:00 2001
From: Fuzy <admin@123>
Date: Wed, 24 Jul 2024 14:14:29 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E6=AD=A3?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../impl/WagesConversionServiceImpl.java      | 23 ++++++++++++++++---
 1 file changed, 20 insertions(+), 3 deletions(-)

diff --git a/ruoyi-wages/src/main/java/com/ruoyi/system/service/impl/WagesConversionServiceImpl.java b/ruoyi-wages/src/main/java/com/ruoyi/system/service/impl/WagesConversionServiceImpl.java
index 8b84c90..956053a 100644
--- a/ruoyi-wages/src/main/java/com/ruoyi/system/service/impl/WagesConversionServiceImpl.java
+++ b/ruoyi-wages/src/main/java/com/ruoyi/system/service/impl/WagesConversionServiceImpl.java
@@ -1,6 +1,9 @@
 package com.ruoyi.system.service.impl;
 
 import com.ruoyi.common.core.domain.AjaxResult;
+import com.ruoyi.common.core.domain.entity.SysRole;
+import com.ruoyi.common.core.domain.entity.SysUser;
+import com.ruoyi.common.utils.SecurityUtils;
 import com.ruoyi.system.mapper.WagesConversionMapper;
 import com.ruoyi.system.model.wages.dao.WagesConversionData;
 import com.ruoyi.system.model.wages.dto.WagesParamsDto;
@@ -8,9 +11,7 @@ import com.ruoyi.system.model.wages.vo.WagesConversionDataVO;
 import com.ruoyi.system.service.WagesConversionService;
 import org.springframework.stereotype.Service;
 
-import java.util.Collections;
-import java.util.List;
-import java.util.Optional;
+import java.util.*;
 import java.util.stream.Collectors;
 
 /**
@@ -23,6 +24,16 @@ public class WagesConversionServiceImpl implements WagesConversionService {
     private WagesConversionServiceImpl(WagesConversionMapper wagesConversionMapper) {
         this.wagesConversionMapper = wagesConversionMapper;
     }
+    private static Map<String, String> roleIdMap = new HashMap<>();
+    static {
+        roleIdMap.put("10", "010101");
+        roleIdMap.put("11", "011704");
+        roleIdMap.put("12", "011204");
+        roleIdMap.put("13", "011702");
+        roleIdMap.put("14", "011701");
+        roleIdMap.put("15", "011502");
+        roleIdMap.put("16", "010102");
+    }
 
     @Override
     public void save(WagesConversionDataVO wagesConversionDataVO) {
@@ -36,6 +47,12 @@ public class WagesConversionServiceImpl implements WagesConversionService {
         String pageSize = wagesParamsDto.getPageSize();
         String wageMonth = wagesParamsDto.getWageMonth();
         int n = (Integer.parseInt(pageNum) - 1) * Integer.parseInt(pageSize);
+        SysUser user = SecurityUtils.getLoginUser().getUser();
+        if (user.getRoles().size() != 0){
+            SysRole sysRole = user.getRoles().get(0);
+            String roleSort = sysRole.getRoleSort();
+            wagesParamsDto.setMineId(roleIdMap.getOrDefault(roleSort, "0"));
+        }
         return wagesConversionMapper.selectWagesConversionDataList(n, Integer.parseInt(pageSize), wageMonth, wagesParamsDto.getMineId());
     }
 
-- 
2.26.2