From 1f653021b7c278f2e505101ce0364e2160be51d1 Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期二, 08 十一月 2022 13:37:32 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/system/service/impl/RoleServiceImpl.java |   35 ++++++++++++++++++++++++++++++++++-
 1 files changed, 34 insertions(+), 1 deletions(-)

diff --git a/src/main/java/com/zy/crm/system/service/impl/RoleServiceImpl.java b/src/main/java/com/zy/crm/system/service/impl/RoleServiceImpl.java
index da93c1f..02b3021 100644
--- a/src/main/java/com/zy/crm/system/service/impl/RoleServiceImpl.java
+++ b/src/main/java/com/zy/crm/system/service/impl/RoleServiceImpl.java
@@ -1,12 +1,45 @@
 package com.zy.crm.system.service.impl;
 
 import com.baomidou.mybatisplus.service.impl.ServiceImpl;
-import org.springframework.stereotype.Service;
+import com.core.common.Cools;
 import com.zy.crm.system.entity.Role;
+import com.zy.crm.system.entity.User;
 import com.zy.crm.system.mapper.RoleMapper;
 import com.zy.crm.system.service.RoleService;
+import com.zy.crm.system.service.UserService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Service;
+
+import java.util.ArrayList;
+import java.util.Collections;
+import java.util.List;
 
 @Service("roleService")
 public class RoleServiceImpl extends ServiceImpl<RoleMapper, Role> implements RoleService {
 
+    @Autowired
+    private UserService userService;
+
+    @Override
+    public List<User> getUserByRoleCode(Long hostId, Long deptId, Long userId) {
+        Role role = this.selectById(userService.selectById(userId).getRoleId());
+        if (Cools.isEmpty(role) || Cools.isEmpty(role.getCode())) {
+            return new ArrayList<>();
+        }
+        switch (role.getCode()) {
+            case "boss":
+            case "manager":
+                return userService.getUserByDept(hostId, deptId);
+            case "salesman":
+                User user = userService.selectById(userId);
+                if (deptId.equals(user.getDeptId())) {
+                    return Collections.singletonList(user);
+                } else {
+                    return new ArrayList<>();
+                }
+            default:
+                return new ArrayList<>();
+        }
+    }
+
 }

--
Gitblit v1.9.1