From 7f9a1ccdad55ce31edd68cfb2c2bd8a4068481c7 Mon Sep 17 00:00:00 2001
From: chenjiantian <214228265@qq.com>
Date: 星期二, 14 十二月 2021 13:52:01 +0800
Subject: [PATCH] 灯杆模块

---
 ximon-admin/src/main/java/com/sandu/ximon/admin/service/ClientService.java |   44 ++++++++++++++++++++++++++++++++++----------
 1 files changed, 34 insertions(+), 10 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 9feb6f9..934a95a 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
@@ -4,52 +4,76 @@
 import com.sandu.common.execption.BusinessException;
 import com.sandu.common.service.impl.BaseServiceImpl;
 import com.sandu.ximon.admin.param.ClientPrarm;
-import com.sandu.ximon.dao.domain.Admin;
 import com.sandu.ximon.dao.domain.Client;
 import com.sandu.ximon.dao.mapper.ClientMapper;
 import lombok.AllArgsConstructor;
 import org.springframework.stereotype.Service;
 
+import java.util.List;
+
 @Service
 @AllArgsConstructor
 public class ClientService extends BaseServiceImpl<ClientMapper, Client> {
 
+    private final ClientMapper clientMapper;
+
     public boolean addClient(ClientPrarm clientPrarm) {
-        Client client=new Client();
+        Client client = new Client();
+        if (clientPrarm.getClientSuperior() != null) {
+            Client one = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getClientName, clientPrarm.getClientSuperior()));
+            if (one != null) {
+                client.setSuperiorId(one.getId());
+            } else {
+                throw new BusinessException("涓婄骇瀹㈡埛涓嶅瓨鍦�");
+            }
+        }
         client.setClientName(clientPrarm.getClientName());
         client.setLinkMan(clientPrarm.getLinkMan());
         client.setMobile(clientPrarm.getMobile());
         client.setClientSuperior(clientPrarm.getClientSuperior());
 
+
         return save(client);
     }
 
 
-    public boolean updateClient(Long id,ClientPrarm clientPrarm){
+    public boolean updateClient(Long id, ClientPrarm clientPrarm) {
         //Client one = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getId, id));
-        Client one=getById(id);
-        if(one==null){
+        Client one = getById(id);
+        if (one == null) {
             throw new BusinessException("璇ュ鎴蜂笉瀛樺湪");
         }
-        Client client=new Client();
+        Client client = new Client();
+        if (clientPrarm.getClientSuperior() != null) {
+            Client superior = getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getClientName, clientPrarm.getClientSuperior()));
+            if (one != null) {
+                client.setSuperiorId(superior.getId());
+            } else {
+                throw new BusinessException("涓婄骇瀹㈡埛涓嶅瓨鍦�");
+            }
+        }
         client.setId(id);
         client.setClientName(clientPrarm.getClientName());
         client.setLinkMan(clientPrarm.getLinkMan());
         client.setMobile(clientPrarm.getMobile());
         client.setClientSuperior(clientPrarm.getClientSuperior());
 
-       // update(client);
-      return updateById(client);
+        // update(client);
+        return updateById(client);
     }
 
-    public boolean deleteClient(Long id){
-        Client one=getById(id);
+    public boolean deleteClient(Long id) {
+        Client one = getById(id);
         if (one == null) {
             throw new BusinessException("璇ュ鎴蜂笉瀛樺湪");
         }
         return removeById(id);
     }
 
+    public List<Client>  clientList(Long userId) {
+        return clientMapper.clientList(userId);
+    }
+
     public Client findByPhone(String phone) {
         return getOne(Wrappers.lambdaQuery(Client.class).eq(Client::getMobile, phone).last("limit 1"));
     }

--
Gitblit v1.9.3