From 68d2e92f824d72b98aaf68cd5a936945c1912bed Mon Sep 17 00:00:00 2001
From: luxiaotao1123 <t1341870251@63.com>
Date: 星期一, 21 十一月 2022 15:14:45 +0800
Subject: [PATCH] #

---
 src/main/java/com/zy/crm/system/service/impl/RoleServiceImpl.java |   42 +++++++++++++++++++++++++++++++++++++++++-
 1 files changed, 41 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..2b6b684 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,52 @@
 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.common.model.enums.RoleType;
 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<>();
+        }
+    }
+
+    @Override
+    public Boolean judgeLeader(Long roleId) {
+        Role role = this.selectById(roleId);
+        return (role.getCode().equals(RoleType.boss.toString()) || role.getCode().equals(RoleType.manager.toString()));
+    }
+
 }

--
Gitblit v1.9.1