From dda268997ca8f8a364f7c19b45d7a43a50a98efe Mon Sep 17 00:00:00 2001
From: liuhaonan <31457034@qq.com>
Date: 星期二, 25 十月 2022 18:00:11 +0800
Subject: [PATCH] changes

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/service/ClientService.java |  169 ++++++++++++++++++++++++++++++++++++++++++++++++++++++--
 1 files changed, 162 insertions(+), 7 deletions(-)

diff --git a/ximon-admin/src/main/java/com/sandu/ximon/admin/service/ClientService.java b/ximon-admin/src/main/java/com/sandu/ximon/admin/service/ClientService.java
index 3e9aecd..8d6d8b7 100644
--- a/ximon-admin/src/main/java/com/sandu/ximon/admin/service/ClientService.java
+++ b/ximon-admin/src/main/java/com/sandu/ximon/admin/service/ClientService.java
@@ -1,12 +1,18 @@
 package com.sandu.ximon.admin.service;
 
+import cn.hutool.core.bean.BeanUtil;
 import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
 import com.baomidou.mybatisplus.core.toolkit.Wrappers;
 import com.sandu.common.enums.RoleLevelStatus;
 import com.sandu.common.execption.BusinessException;
-import com.sandu.common.object.BaseConditionVO;
+import com.sandu.common.redis.RedisService;
+import com.sandu.common.security.LoginUserInfo;
+import com.sandu.common.security.token.TokenProvider;
 import com.sandu.common.service.impl.BaseServiceImpl;
+import com.sandu.common.util.IpUtil;
 import com.sandu.common.util.SpringContextHolder;
+import com.sandu.ximon.admin.dto.ClientDto;
+import com.sandu.ximon.admin.dto.ClientDtoNode;
 import com.sandu.ximon.admin.param.AddClientPrarm;
 import com.sandu.ximon.admin.param.PwdParam;
 import com.sandu.ximon.admin.param.UpdateClientPrarm;
@@ -16,29 +22,38 @@
 import com.sandu.ximon.dao.bo.MenuNode;
 import com.sandu.ximon.dao.domain.*;
 import com.sandu.ximon.dao.enums.AdministratorEnums;
+import com.sandu.ximon.dao.enums.OrderByEnums;
 import com.sandu.ximon.dao.mapper.ClientMapper;
 import lombok.AllArgsConstructor;
+import org.springframework.beans.BeanUtils;
+import org.springframework.security.core.GrantedAuthority;
 import org.springframework.security.crypto.password.PasswordEncoder;
 import org.springframework.stereotype.Service;
 import org.springframework.transaction.annotation.Transactional;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.List;
+import java.util.stream.Collectors;
 
 @Service
 @AllArgsConstructor
 public class ClientService extends BaseServiceImpl<ClientMapper, Client> {
 
-    private final ClientMapper clientMapper;
     private final PasswordEncoder passwordEncoder;
     private final ClientRoleRelationService clientRoleRelationService;
     private final RoleService roleService;
+    private final TokenProvider tokenProvider;
 
 
     public boolean addClient(AddClientPrarm addClientPrarm) {
 
         if (getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getClientName, addClientPrarm.getClientName().trim())) != null) {
-            throw new BusinessException("璇ョ敤鎴峰悕宸插瓨鍦紒");
+            throw new BusinessException("璇ョ敤鎴峰悕宸插湪瀹㈡埛涓瓨鍦紒");
+        }
+        Admin admin = SpringContextHolder.getBean(AdminService.class).findByUserName(addClientPrarm.getClientName());
+        if (admin != null) {
+            throw new BusinessException("褰撳墠璐﹀彿" + addClientPrarm.getClientName() + "宸茬粡鍦ㄧ鐞嗗憳涓瓨鍦�");
         }
         Long userId = SecurityUtils.getUserId();
         boolean clientId = findClientId();
@@ -242,6 +257,27 @@
         if (removeById(id)) {
             //鍒犺鑹插叧绯昏〃淇℃伅
             flag = clientRoleRelationService.removeById(one);
+            List<Pole> poles = SpringContextHolder.getBean(PoleService.class).list(Wrappers.lambdaQuery(Pole.class).eq(Pole::getClientId, id));
+            poles.forEach(
+                    pole -> {
+                        pole.setClientId(-1L);
+                        pole.setUserId(-1L);
+                    }
+            );
+
+            //鍒犻櫎鐢ㄦ埛鍚庨渶瑕佽嚜鍔ㄨВ缁戣鐢ㄦ埛鐨勮澶�
+
+            //鐏潌id闆嗗悎
+            List<Long> collect = poles.stream().map(p -> p.getId()).collect(Collectors.toList());
+            List<Long> collect1 = poles.stream().map(Pole::getId).collect(Collectors.toList());
+            //鍒犻櫎缁戝畾鍏崇郴
+            SpringContextHolder.getBean(PoleBindingService.class)
+                    .remove(Wrappers.lambdaQuery(PoleBinding.class)
+                            .in(PoleBinding::getPoleId, collect));
+            //鏇存敼鐏潌褰掑睘
+            flag = SpringContextHolder.getBean(PoleService.class).updateBatchById(poles);
+
+
             //鍒犻櫎澶辫触鍥炴粴鏁版嵁
             if (!flag) {
                 throw new BusinessException("鍒犻櫎鐢ㄦ埛澶辫触锛�");
@@ -260,8 +296,94 @@
         return flag;
     }
 
-    public List<Client> clientList(Long userId, BaseConditionVO baseConditionVO) {
-        return clientMapper.clientList(userId, baseConditionVO.getPageNo(), baseConditionVO.getPageSize());
+    /**
+     * 鐢ㄦ埛鍒楄〃
+     *
+     * @param keyword
+     * @param order
+     * @param seq
+     * @return
+     */
+    public List<ClientDtoNode> listLikeClient(String keyword, Integer order, Integer seq) {
+
+
+        LambdaQueryWrapper<Client> wrapper = Wrappers.lambdaQuery(Client.class);
+        if (AdministratorEnums.NORMAL.getCode().equals(SecurityUtils.getAdministratorIdentity())) {
+            wrapper.eq(Client::getSuperiorId, SecurityUtils.getUserId()).or(w -> w.eq(
+                    Client::getId, SecurityUtils.getUserId()
+            ));
+        }
+        if (keyword != null && !keyword.isEmpty()) {
+            wrapper.like(Client::getClientName, keyword)
+                    .or(w1 -> w1.like(Client::getMobile, keyword))
+                    .or(w2 -> w2.like(Client::getLinkMan, keyword));
+        }
+        //鎺掑簭瀛楁
+        String orderByResult = "id";
+        //姝e簭銆佸�掑彊
+        String orderBySeq = OrderByEnums.ASC.getCode();
+        if (order != null) {
+            switch (order) {
+                case 1:
+                    orderByResult = OrderByEnums.CLIENT_CREATE_TIME.getCode();
+                    break;
+                default:
+            }
+        }
+        if (seq != null) {
+            switch (seq) {
+                case 1:
+                    orderBySeq = OrderByEnums.ASC.getCode();
+                    break;
+                case 2:
+                    orderBySeq = OrderByEnums.DESC.getCode();
+                    break;
+                default:
+                    break;
+            }
+        }
+        //鎺掑簭鏂瑰紡
+        String orderBy = orderByResult + " " + orderBySeq;
+
+//        if (baseConditionVO != null) {
+//            PageHelper.startPage(baseConditionVO.getPageNo(), baseConditionVO.getPageSize(), orderBy);
+//        }
+
+        List<Client> clientList = list(wrapper);
+
+        List<ClientDto> clientDtoList = new ArrayList<>();
+        for (Client demo : clientList) {
+            ClientDto clientDto = new ClientDto();
+            BeanUtil.copyProperties(demo, clientDto);
+            ClientRoleRelation one = clientRoleRelationService.getOne(Wrappers.lambdaQuery(ClientRoleRelation.class)
+                    .eq(ClientRoleRelation::getClientId, demo.getId()));
+
+            if (one != null) {
+                clientDto.setRoleId(one.getRoleId());
+            }
+            clientDtoList.add(clientDto);
+        }
+
+        return clientDtoList.stream()
+                .filter(bean -> null == bean.getSuperiorId())
+                .map(bean -> covertClientDtoNode(bean, clientDtoList)).collect(Collectors.toList());
+    }
+
+    /**
+     * 鏁版嵁鏁寸悊鎴愭爲鐘跺浘
+     *
+     * @param clientDto
+     * @param clientDtoList
+     * @return
+     */
+    private ClientDtoNode covertClientDtoNode(ClientDto clientDto, List<ClientDto> clientDtoList) {
+        ClientDtoNode node = new ClientDtoNode();
+        BeanUtils.copyProperties(clientDto, node);
+        List<ClientDtoNode> children = clientDtoList.stream()
+                .filter(subClientDto -> (subClientDto.getSuperiorId() != null && subClientDto.getSuperiorId().equals(clientDto.getId())))
+                .map(subClientDto -> covertClientDtoNode(subClientDto, clientDtoList)).collect(Collectors.toList());
+        node.setChildren(children);
+        return node;
     }
 
     public Client findByPhone(String phone) {
@@ -288,7 +410,7 @@
 
     /**
      * \
-     * 鍏朵粬绫荤敤鏉ユ煡鎵惧鎴穒d浣跨敤  濡傛灉娌℃湁涓婄骇瀹㈡埛 杩欒繑鍥炵敤鎴稩D
+     * 鍏朵粬绫荤敤鏉ユ煡鎵惧鎴穒d浣跨敤  濡傛灉娌℃湁涓婄骇瀹㈡埛 杩斿洖-1
      *
      * @param
      * @return
@@ -298,7 +420,7 @@
         if (one != null && one.getSuperiorId() != null) {
             return one.getSuperiorId();
         } else {
-            return userId;
+            return -1L;
         }
 
     }
@@ -378,4 +500,37 @@
         List<MenuNode> resultList = SpringContextHolder.getBean(MenuService.class).getUserPermissionListById(menuIds);
         return resultList;
     }
+
+    /**
+     * 鐢熸垚渚挎嵎鐧诲綍KEY
+     */
+    public String creatKeyByLogin(String keyValue) {
+        if (keyValue == null) {
+            throw new BusinessException("key涓嶈兘涓虹┖锛�");
+        }
+
+        Object object = SpringContextHolder.getBean(RedisService.class).get(keyValue);
+
+        if (object != null) {
+            return object.toString();
+        } else {
+            Client client = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getUserKey, keyValue));
+            if (client == null) {
+                throw new BusinessException("鐢ㄦ埛涓嶅瓨鍦ㄦ垨key杩囨湡,璇疯仈绯荤鐞嗗憳閲嶆柊鑾峰彇鏈�鏂発ey锛�");
+            }
+
+            Collection<GrantedAuthority> grantedAuthorities = roleService.mapToGrantedAuthorities(client.getId(), false);
+
+            LoginUserInfo loginUserInfo = new LoginUserInfo();
+            loginUserInfo.setUserId(client.getId());
+            loginUserInfo.setAccount(client.getClientName());
+            loginUserInfo.setIp(IpUtil.getRealIp());
+            loginUserInfo.setAdministratorType(AdministratorEnums.NORMAL.getCode());
+            loginUserInfo.setPermission(grantedAuthorities.stream().map(GrantedAuthority::getAuthority).collect(Collectors.joining(",")));
+
+            // 鐢熸垚浠ょ墝
+            String token = tokenProvider.createToken(loginUserInfo);
+            return token;
+        }
+    }
 }

--
Gitblit v1.9.3